Added support for pydantic v2 (#193)

paperqa/contrib/zotero.py

@@ -4,7 +4,10 @@
 from pathlib import Path
 from typing import List, Optional, Union, cast
 
-from pydantic import BaseModel
+try:
+    from pydantic.v1 import BaseModel
+except ImportError:
+    from pydantic import BaseModel
 
 try:
     from pyzotero import zotero

paperqa/docs.py

@@ -15,7 +15,11 @@
 from langchain.memory import ConversationTokenBufferMemory
 from langchain.memory.chat_memory import BaseChatMemory
 from langchain.vectorstores import FAISS, VectorStore
-from pydantic import BaseModel, validator
+
+try:
+    from pydantic.v1 import BaseModel, validator
+except ImportError:
+    from pydantic import BaseModel, validator
 
 from .chains import get_score, make_chain
 from .paths import PAPERQA_DIR

paperqa/types.py

@@ -7,7 +7,12 @@
     CallbackManagerForChainRun,
 )
 from langchain.prompts import PromptTemplate
-from pydantic import BaseModel, validator
+
+try:
+    from pydantic.v1 import BaseModel, validator
+except ImportError:
+    from pydantic import BaseModel, validator
+
 
 from .prompts import (
     citation_prompt,

paperqa/version.py

@@ -1 +1 @@
-__version__ = "3.9.1"
+__version__ = "3.10.0"