Change logic on arxiv reader for article id inputs (#1044)

* Chagne logic on arxiv reader for article ID inputs * v1.4.15
Josh-XT · Oct 10, 2023 · b78ae02 · b78ae02
1 parent 1c4eba9
commit b78ae02
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/agixt/readers/arxiv.py b/agixt/readers/arxiv.py
@@ -27,11 +27,15 @@ async def write_arxiv_articles_to_memory(
             results = arxiv.Search(
                 id_list=articles, query=query, max_results=max_articles
             )
-        if article_ids and not query:  # Comma separated list of article IDs
+        elif article_ids != None and article_ids != "":
+            # Comma separated list of article IDs
             articles = article_ids.split(",") if "," in article_ids else [article_ids]
+            articles = [article.strip() for article in articles]
             results = arxiv.Search(id_list=articles)
-        if query and not article_ids:  # Search query
+        elif query != None and query != "":  # Search query
             results = arxiv.Search(query=query, max_results=max_articles)
+        else:
+            return False
         if results:
             base_path = os.path.join(os.getcwd(), "WORKSPACE")
             for result in results.results():

diff --git a/agixt/version b/agixt/version
@@ -1 +1 @@
-v1.4.14
+v1.4.15