Updated fetch function in app.py

app.py

@@ -51,7 +51,7 @@ from smolagents import CodeAgent, HfApiModel, tool
 
 @tool
 def fetch_latest_arxiv_papers(keywords: list, num_results: int = 5) -> list:
-    """Fetches the latest research papers from arXiv based on **strict keyword presence**.
+    """Fetches the latest research papers from arXiv based on provided keywords.
 
     Args:
         keywords: A list of keywords to search for relevant papers.
@@ -68,9 +68,9 @@ def fetch_latest_arxiv_papers(keywords: list, num_results: int = 5) -> list:
     try:
         print(f"DEBUG: Searching arXiv papers with keywords: {keywords}")
 
-        # Force strict keyword search using "title" and "abstract" only
-        query = "+AND+".join([f"(ti:\"{kw}\"+OR+abs:\"{kw}\")" for kw in keywords])  
-        query_encoded = urllib.parse.quote(query)  # Encode query for URL
+        # Format query using "AND" to enforce strict keyword presence
+        query = "+AND+".join([f"ti:{kw}+OR+abs:{kw}" for kw in keywords])  
+        query_encoded = urllib.parse.quote(query)  # Encode spaces and special characters
 
         url = f"http://export.arxiv.org/api/query?search_query={query_encoded}&start=0&max_results=20&sortBy=submittedDate&sortOrder=descending"
 
@@ -83,8 +83,8 @@ def fetch_latest_arxiv_papers(keywords: list, num_results: int = 5) -> list:
             title = entry.title.lower()
             abstract = entry.summary.lower()
 
-            # Ensure ALL keywords appear in **either** title or abstract
-            if all(kw.lower() in title or kw.lower() in abstract for kw in keywords):
+            # ✅ Ensure at least one keyword appears in the title or abstract
+            if any(kw.lower() in title or kw.lower() in abstract for kw in keywords):
                 papers.append({
                     "title": entry.title,
                     "authors": ", ".join(author.name for author in entry.authors),
@@ -93,11 +93,7 @@ def fetch_latest_arxiv_papers(keywords: list, num_results: int = 5) -> list:
                     "link": entry.link
                 })
 
-        # If no relevant papers found, return "No results found."
-        if not papers:
-            return [{"error": "No results found. Try different keywords."}]
-
-        # Prioritize papers where keywords appear in the **title**
+        # ✅ Sort papers: First prioritize keyword in title, then abstract
         papers.sort(key=lambda x: sum(kw.lower() in x["title"].lower() for kw in keywords), reverse=True)
 
         return papers[:num_results]  # Return top-matching papers