Made model name customizable

Author: whitead

README.md

@@ -101,9 +101,9 @@ print(answer.formatted_answer)
 
 ## FAQ
 
-### How is this different from gpt-index?
+### How is this different from LlamaIndex?
 
-It's not that different! This is similar to the tree response method in GPT-index. I just have included some prompts I find useful, readers that give page numbers/line numbers, and am focused on one tasks - answering technical questions with cited sources.
+It's not that different! This is similar to the tree response method in LlamaIndex. I just have included some prompts I find useful, readers that give page numbers/line numbers, and am focused on one tasks - answering technical questions with cited sources.
 
 ### How is this different from LangChain?
 

paperqa/docs.py

@@ -64,6 +64,7 @@ def __init__(
         summary_llm: Optional[LLM] = None,
         name: str = "default",
         index_path: Optional[Path] = None,
+        model_name: str = 'gpt-3.5-turbo'
     ) -> None:
         """Initialize the collection of documents.
 
@@ -75,13 +76,14 @@ def __init__(
             summary_llm: The language model to use for summarizing documents. If None, llm is used.
             name: The name of the collection.
             index_path: The path to the index file IF pickled. If None, defaults to using name in $HOME/.paperqa/name
+            model_name: The name of the model to use assuming OpenAI. Default - gpt-3.5-turbo
         """
         self.docs = dict()
         self.chunk_size_limit = chunk_size_limit
         self.keys = set()
         self._faiss_index = None
         if llm is None:
-            llm = OpenAIChat(temperature=0.1, max_tokens=512, prefix_messages=chat_pref)
+            llm = OpenAIChat(temperature=0.1, max_tokens=512, prefix_messages=chat_pref, model_name=model_name)
         if summary_llm is None:
             summary_llm = llm
         self.update_llm(llm, summary_llm)

paperqa/version.py

@@ -1 +1 @@
-__version__ = "0.0.21"
+__version__ = "0.0.30"

tests/test_paperqa.py

@@ -94,14 +94,14 @@ def test_docs_pickle():
             list(
                 docs.get_evidence(
                     paperqa.Answer("What date is flag day in Canada?"),
-                    k=1,
+                    k=3,
                     max_sources=1,
                 )
             )[-1].context,
             list(
                 docs2.get_evidence(
                     paperqa.Answer("What date is flag day in Canada?"),
-                    k=1,
+                    k=3,
                     max_sources=1,
                 )
             )[-1].context,