Improved Agent Speed/Prompts (#118)

Author: whitead

paperqa/agent.py

@@ -82,6 +82,7 @@ class AnswerTool(BaseTool):
     description = "Ask a researcher to propose an answer using evidence from papers. The input is the question to be answered."
     docs: Docs = None
     answer: Answer = None
+    return_direct = True
 
     def __init__(self, docs, answer):
         # call the parent class constructor
@@ -129,7 +130,7 @@ def _run(self, query: str) -> str:
             )
 
         papers = paperscraper.search_papers(
-            query, limit=20, verbose=False, pdir=self.docs.index_path
+            query, limit=5, _limit=20, verbose=False, pdir=self.docs.index_path
         )
         for path, data in papers.items():
             try:

paperqa/qaprompts.py

@@ -48,6 +48,7 @@
     template="We want to answer the following question: {question} \n"
     "Provide three keyword searches (one search per line) "
     "that will find papers to help answer the question. Do not use boolean operators. "
+    "Provide some broad and some specific searches. "
     "Recent years are 2021, 2022, 2023.\n\n"
     "1.",
 )

paperqa/version.py

@@ -1 +1 @@
-__version__ = "1.6.3"
+__version__ = "1.8.0"

tests/test_paperqa.py

@@ -72,7 +72,7 @@ async def test_token_count(self):
             get_callbacks=lambda x: [TestHandler()],
         )
         assert answer.tokens > 100
-        assert answer.cost > 0.01
+        assert answer.cost > 0.001
 
 
 def test_evidence():