new model

agent_utils.py

@@ -10,7 +10,7 @@ import pandas as pd
 
 def wiki_ret(question: str) -> str:
     """ Retrieve docs from wikipedia """
-
+    print("wiki")
     # Search
     search_docs = WikipediaLoader(query=question, 
                                 load_max_docs=2).load()
@@ -29,6 +29,7 @@ def wiki_ret(question: str) -> str:
 
 def arxiv_ret(query: str, max_results: int = 3) -> str:
     """Search arXiv for a given query and return a summary of top results."""
+    print("arxiv")
     search = arxiv.Search(
         query=query,
         max_results=max_results,
@@ -48,6 +49,7 @@ def arxiv_ret(query: str, max_results: int = 3) -> str:
 
 def tavily_ret(query: str, max_results: int = 3) -> str:
     """Use Tavily to retrieve web-based information about a topic."""
+    print("tavily")
     api_key = os.getenv("TAVILY_API_KEY")
     if not api_key:
         return "Tavily API key not found."
@@ -71,7 +73,7 @@ def handle_file_tool(input: dict) -> str:
     """Reads the file attached to a question"""
     question = input.get("question", "")
     file_url = input.get("file", "")
-    print(file_url)
+    print("file_url: ", file_url)
 
     if not file_url:
         return "No file provided."