Changed types to use factories (#164)

Future-House · Jul 18, 2023 · 497f870 · 497f870
1 parent 9db08fa
commit 497f870
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/paperqa/types.py b/paperqa/types.py
@@ -7,7 +7,7 @@
     CallbackManagerForChainRun,
 )
 from langchain.prompts import PromptTemplate
-from pydantic import BaseModel, validator
+from pydantic import BaseModel, Field, validator
 
 from .prompts import citation_prompt, qa_prompt, select_paper_prompt, summary_prompt
 
@@ -31,10 +31,10 @@ class Text(BaseModel):
 
 
 class PromptCollection(BaseModel):
-    summary: PromptTemplate = summary_prompt
-    qa: PromptTemplate = qa_prompt
-    select: PromptTemplate = select_paper_prompt
-    cite: PromptTemplate = citation_prompt
+    summary: PromptTemplate = Field(default_factory=lambda: summary_prompt)
+    qa: PromptTemplate = Field(default_factory=lambda: qa_prompt)
+    select: PromptTemplate = Field(default_factory=lambda: select_paper_prompt)
+    cite: PromptTemplate = Field(default_factory=lambda: citation_prompt)
     pre: Optional[PromptTemplate] = None
     post: Optional[PromptTemplate] = None
 

diff --git a/paperqa/version.py b/paperqa/version.py
@@ -1 +1 @@
-__version__ = "3.3.1"
+__version__ = "3.3.2"