Add test file

agent.py

@@ -109,7 +109,7 @@ def create_assistant_tools(cfg):
             [ask_transcripts]
     )
 
-def initialize_agent(_cfg, update_func):
+def initialize_agent(_cfg, update_func=None):
     financial_bot_instructions = """
     - You are a helpful financial assistant, with expertise in financial reporting, in conversation with a user. 
     - Respond in a compact format by using appropriate units of measure (e.g., K for thousands, M for millions, B for billions). 

test_agent.py

@@ -0,0 +1,33 @@
+import unittest
+import os
+
+from omegaconf import OmegaConf
+from vectara_agent.agent import Agent
+
+from agent import initialize_agent
+
+from dotenv import load_dotenv
+load_dotenv(override=True)
+
+class TestAgentResponses(unittest.TestCase):
+
+    def test_responses(self):
+
+        cfg = OmegaConf.create({
+                'customer_id': str(os.environ['VECTARA_CUSTOMER_ID']),
+                'corpus_id': str(os.environ['VECTARA_CORPUS_ID']),
+                'api_key': str(os.environ['VECTARA_API_KEY']),
+                'examples': os.environ.get('QUERY_EXAMPLES', None)
+            })
+
+        agent = initialize_agent(_cfg=cfg)
+        self.assertIsInstance(agent, Agent)
+        
+        # Basic number questions
+        self.assertIn('274.52B', agent.chat('Was was the revenue for Apple in 2020? Just provide the number.'))
+        self.assertIn('amazon', agent.chat('Which company had the highest revenue in 2023? Just provide the name.').lower())
+        self.assertIn('amazon', agent.chat('Which company had the lowest profits in 2022?').lower())
+
+
+if __name__ == "__main__":
+    unittest.main()