Pass config variable to SearchAgent

nulib · Dec 17, 2024 · eea458f · eea458f
1 parent 7e4cae7
commit eea458f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/chat/src/handlers/chat.py b/chat/src/handlers/chat.py
@@ -29,7 +29,7 @@ def handler(event, context):
 
     metrics = MetricsHandler()
     callbacks = [AgentHandler(config.socket, config.ref), metrics]
-    search_agent = SearchAgent(model=chat_model(event), streaming=True)
+    search_agent = SearchAgent(model=chat_model(config), streaming=True)
     try:
         search_agent.invoke(config.question, config.ref, forget=config.forget, callbacks=callbacks)
         log_metrics(context, metrics, config)