Update app.py

app.py

@@ -44,6 +44,48 @@ def get_current_time_in_timezone(timezone: str) -> str:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
 
+
+@tool
+
+
+def get_latest_ai_agent_arxiv_papers(num_papers=5):
+    """
+    Searches arXiv for the latest research papers on AI agents and retrieves
+    the specified number of results.
+
+    Args:
+        num_papers: The number of papers to retrieve (default: 5).
+
+    Returns:
+        A list of dictionaries, where each dictionary represents a paper and
+        contains its title, authors, and URL. Returns an empty list if any errors occur.
+        Prints informative messages to the console during the search.
+    """
+    import arxiv
+    
+    try:
+        search = arxiv.Search(
+            query="AI agents",  # Your search term
+            max_results=num_papers,
+            sort_by=arxiv.SortCriterion.SubmittedDate # Sort by submission date for latest
+        )
+
+        papers = []
+        for result in search.results():
+            paper_info = {
+                "title": result.title,
+                "authors": ", ".join([author.name for author in result.authors]), # Nicer author format
+                "url": result.pdf_url,
+                "date": result.published.date(), # Get the date part of the datetime object
+                "abstract": result.summary
+            }
+            papers.append(paper_info)
+        return papers
+
+    except Exception as e:
+        print(f"An error occurred during the arXiv search: {e}")
+        return []
+        
 final_answer = FinalAnswerTool()
 
 # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
@@ -66,7 +108,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer,find_product,greet], ## add your tools here (don't remove final answer)
+    tools=[final_answer,find_product,greet,get_latest_ai_agent_arxiv_papers], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,