MWBadra/Context-Aware-Conversational-Agent
0
1�
2 �4�i� � �$ � S SK r S SKrS SKrS SKJr S SKJr S SKJr S SK J3r4 S SK J r S SK
Jr S SKJr \R" " \R$ S S59 \R& " \5 r " S S \ R, 5 r\S
:X a \ R0 " 5 gg)� N)� load_dotenv)�ChatGroq)�ContextAwareAgentManager)�ContextPresenceJudgeTool)�
WebSearchTool)�ContextRelevanceCheckerTool)�ContextSplitterToolz)%(asctime)s - %(levelname)s - %(message)s)�level�formatc �: � \ rS rSrSr\S 5 rS rS rS r Sr6g) �TestContextAwareAgent� zv7Integration Tests for the Context-Aware Agent.8Verifies that the LLM, Tools, and Agent Workflow integrate correctly.9c �` � [ R S5 [ 5 [ R " S5 U l [ R " S5 U l U R10 ( a U R ( d [ S5 e[ R S5 [ SSS9U l [ U R S 9[ 5 [ U R S 9[ U R S 9/U l [ U R U R S119U l U R R# 5 U l g ) z�12This function is run only once before all tests begin.13 14It aims to prepare the environment, models, and tools to save processing time.15 z"Setting up the test environment...� GROQ_API_KEY�TAVILY_API_KEYz<Missing API Keys in .env file. Cannot run integration tests.z)Initializing LLM and Tools for testing...r zllama-3.3-70b-versatile)� temperature�16model_name)�llm)r �toolsN)�logger�infor �os�getenv�groq_key�17tavily_key�18ValueErrorr r r r r r r r �
agent_manager� build_agent�agent_executor)�clss �jD:\books and slides\Cellula26\Cellula\Final_Project\Context-Aware-Conversational-Agent\tests\test_agent.py�19setUpClass� TestContextAwareAgent.setUpClass s� � � � � �8�9��
��y�y� �0�� ����#3�4����|�|�3�>�>� � [�\� \�� � �?�@��q�5N�O���
%���� 1� �O� '�C�G�G� 4� �C�G�G� ,� 20�� � 5����� � � R��� � .� .� :� :� <��� c � � Sn[ R SU 35 U R R USS.5 nUR SS5 nU R US5 U R
U[ S5 U R [ U5 S:� S 5 [ R S21US S S
35 g ! [ a nU R SU 35 S nAg S nAff = f)zY22Test Case 1: User provides context.23The agent should identify the context and NOT fail.24z>Here is my Python code: `print(x)`. It gives a NameError. Why?z.Running Test 1 (Context Provided) with query: � ��input� chat_history�output�#The agent returned a None response.�$The agent response must be a string.�25 �#The response is suspiciously short.z& Test 1 Passed. Agent Answer snippet: N�d �...z1 Agent crashed on context-provided query. Error: � r r r �invoke�get�assertIsNotNone�assertIsInstance�str�26assertTrue�len� Exception�fail��self�query�result�response�es r! �test_context_provided_flow�0TestContextAwareAgent.test_context_provided_flow3 s� � �27 Q��� � �D�U�G�L�M�
O� �(�(�/�/�%�QS�0T�U�F� �z�z�(�B�/�H� � � ��+P� Q� � !� !�(�C�1W� X� �O�O�C��M�B� .�0U� V� �K�K�@��$�3� �@P�PS�T� U��� O� �I�I�I�!��M� N� N�� O�� �BB0 �028C�:C�Cc � � Sn[ R SU 35 U R R USS.5 nUR SS5 nU R US5 U R
U[ S5 U R [ U5 S:� S 5 [ R S29US S S
35 g ! [ a nU R SU 35 S nAg S nAff = f)zi30Test Case 2: User asks a general question.31The agent MUST trigger the WebSearchTool to find the answer.32z:What is the latest version of LangChain released recently?z-Running Test 2 (Context Missing) with query: r& r' r* r+ r, r- r. z%Test 2 Passed. Agent Answer snippet: Nr/ r0 z/Agent crashed on context-missing query. Error: r1 r; s r! �test_context_missing_flow�/TestContextAwareAgent.test_context_missing_flowJ s� � �33 M��� � �C�E�7�K�L� M� �(�(�/�/�%�QS�0T�U�F� �z�z�(�B�/�H�
� � ��+P� Q� � !� !�(�C�1W� X� �O�O�C��M�B� .�0U� V� �K�K�?���#��?O�s�S� T��� M� �I�I�G��s�K� L� L�� M�rC c � � Sn[ R SU 35 U R R USS.5 nUR SS5 nU R US5 U R
SUS5 U R S UR 5 S345 [ R S US S
S35 g ! [ a nU R SU 35 S nAg S nAff = f)z�35Test Case 3: User provides irrelevant context.36The agent MUST ignore the context using ContextRelevanceChecker and answer the core question.37z`Here is a recipe for chicken: 1 cup rice, 2 chicken breasts. How do I fix a NameError in Python?z0Running Test 3 (Irrelevant Context) with query: r& r' r* r+ � NameErrorz1The agent failed to address the Python NameError.�chickenzFThe agent hallucinated and talked about the irrelevant chicken recipe.z%Test 3 Passed. Agent Answer snippet: Nr/ r0 z2Agent crashed on irrelevant-context query. Error: ) r r r r2 r3 r4 �assertIn� assertNotIn�lowerr9 r: r; s r! � test_irrelevant_context_flow�2TestContextAwareAgent.test_irrelevant_context_flowb s� � �38 s��� � �F�u�g�N�O� P��(�(�/�/�%�QS�0T�U�F� �z�z�(�B�/�H� � � ��+P� Q� �M�M�+�x�1d� e� � � �Y����(8� ;C�
D� �K�K�?���#��?O�s�S� T��� P� �I�I�J�1�#�N� O� O�� P�s �BB/ �/39C�9C�C� N) �__name__�40__module__� __qualname__�__firstlineno__�__doc__� classmethodr" rA rE rM �__static_attributes__rO r$ r! r
r
s- � ��41 �=� �=�8O�.M�0Pr$ r
�__main__)�unittestr �logging�dotenvr �langchain_groqr �agent.agent_runnerr � tools.context_presence_judger �tools.web_search_toolr � tools.context_relevance_checkerr �tools.context_splitterr � basicConfig�INFO� getLoggerrP r �TestCaser
�mainrO r$ r! �<module>rf sw �� � � � � #� 7� A� /� G� 6� � � �'�,�,�/Z� [� � � �8� $��fP�H� -� -� fP�P �z�� �M�M�O� r$