Upload 2 files

agent.py

@@ -90,7 +90,7 @@ def web_search(query: str) -> str:
     
     Args:
         query: The search query."""
-    search_docs = TavilySearchResults(max_results=3).invoke(query=query)
+    search_docs = TavilySearch(max_results=3).invoke(query=query)
     formatted_search_docs = "\n\n---\n\n".join(
         [
             f'<Document source="{doc.metadata["source"]}" page="{doc.metadata.get("page", "")}"/>\n{doc.page_content}\n</Document>'

app.py

@@ -24,13 +24,13 @@ class BasicAgent:
         self.graph = build_graph()
 
     def __call__(self, question: str) -> str:
-        print(f"Agent received question (first 50 chars): {question[:50]}...")
+        print(f"Agent received question : {question}...")
         # Wrap the question in a HumanMessage from langchain_core
         messages = [HumanMessage(content=question)]
         messages = self.graph.invoke({"messages": messages})
         answer = messages['messages'][-1].content
         print(f"Raw answer from agent: {answer}")
-        prefix = "Answer: "
+        prefix = "FINAL ANSWER: "
         if answer.startswith(prefix):
             # Remove the prefix only if it exists
             return answer[len(prefix):].strip()