Change default model in app

BlueBrain · Sep 4, 2024 · d7449b0 · d7449b0
1 parent 548b9ac
commit d7449b0
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/scholarag/app/config.py b/src/scholarag/app/config.py
@@ -84,7 +84,7 @@ class SettingsOpenAI(BaseModel):
     """OpenAI settings."""
 
     token: SecretStr | None = None
-    model: str = "gpt-3.5-turbo"
+    model: str = "gpt-4o-mini"
     temperature: float = 0
     max_tokens: int | None = None
 

diff --git a/src/scholarag/generative_question_answering.py b/src/scholarag/generative_question_answering.py
@@ -104,7 +104,7 @@ def run(
         Returns
         -------
         generated_text, finish_reason
-            Answers to the question (in theory), reason for the LLM to stop
+            Answers to the question (in theory), reason for the LLM to stop.
         """
         # Put the documents in the prompt with the correct formats
         docs = self._process_retrieved_contexts(contexts)
@@ -158,12 +158,12 @@ async def arun(
         contexts
             Contexts to use to answer the question.
         system_prompt
-            System prompt for the LLM. Leave None for default
+            System prompt for the LLM. Leave None for default.
 
         Returns
         -------
         generated_text, finish_reason
-            Answers to the question (in theory), reason for the LLM to stop
+            Answers to the question (in theory), reason for the LLM to stop.
         """
         # Put the documents in the prompt with the correct formats.
         docs = self._process_retrieved_contexts(contexts)