x

libs/langchain/langchain/agents/agent.py

@@ -10,7 +10,11 @@ from pathlib import Path
 from typing import Any, Callable, Dict, List, Optional, Sequence, Tuple, Union
 
 import yaml
-from pydantic import BaseModel, root_validator
+
+try:
+    from pydantic.v1 import BaseModel, root_validator
+except ImportError:
+    from pydantic import BaseModel, root_validator
 
 from langchain.agents.agent_iterator import AgentExecutorIterator
 from langchain.agents.agent_types import AgentType

libs/langchain/langchain/agents/agent_toolkits/amadeus/toolkit.py

@@ -2,7 +2,10 @@ from __future__ import annotations
 
 from typing import TYPE_CHECKING, List
 
-from pydantic import Field
+try:
+    from pydantic.v1 import Field
+except ImportError:
+    from pydantic import Field
 
 from langchain.agents.agent_toolkits.base import BaseToolkit
 from langchain.tools import BaseTool

libs/langchain/langchain/agents/agent_toolkits/azure_cognitive_services.py

@@ -19,7 +19,7 @@ class AzureCognitiveServicesToolkit(BaseToolkit):
     def get_tools(self) -> List[BaseTool]:
         """Get the tools in the toolkit."""
 
-        tools = [
+        tools: List[BaseTool] = [
             AzureCogsFormRecognizerTool(),
             AzureCogsSpeech2TextTool(),
             AzureCogsText2SpeechTool(),

libs/langchain/langchain/agents/agent_toolkits/base.py

@@ -2,7 +2,10 @@
 from abc import ABC, abstractmethod
 from typing import List
 
-from pydantic import BaseModel
+try:
+    from pydantic.v1 import BaseModel
+except ImportError:
+    from pydantic import BaseModel
 
 from langchain.tools import BaseTool
 

libs/langchain/langchain/agents/agent_toolkits/file_management/toolkit.py

@@ -1,8 +1,11 @@
 from __future__ import annotations
 
-from typing import List, Optional
+from typing import List, Optional, Type, cast
 
-from pydantic import root_validator
+try:
+    from pydantic.v1 import root_validator
+except ImportError:
+    from pydantic import root_validator
 
 from langchain.agents.agent_toolkits.base import BaseToolkit
 from langchain.tools import BaseTool
@@ -15,7 +18,7 @@ from langchain.tools.file_management.read import ReadFileTool
 from langchain.tools.file_management.write import WriteFileTool
 
 _FILE_TOOLS = {
-    tool_cls.__fields__["name"].default: tool_cls
+    cast(Type[BaseTool], tool_cls).__fields__["name"].default: tool_cls
     for tool_cls in [
         CopyFileTool,
         DeleteFileTool,

libs/langchain/langchain/agents/agent_toolkits/gmail/toolkit.py

@@ -2,7 +2,10 @@ from __future__ import annotations
 
 from typing import TYPE_CHECKING, List
 
-from pydantic import Field
+try:
+    from pydantic.v1 import Field
+except ImportError:
+    from pydantic import Field
 
 from langchain.agents.agent_toolkits.base import BaseToolkit
 from langchain.tools import BaseTool

libs/langchain/langchain/agents/agent_toolkits/nla/toolkit.py

@@ -2,7 +2,10 @@ from __future__ import annotations
 
 from typing import Any, List, Optional, Sequence
 
-from pydantic import Field
+try:
+    from pydantic.v1 import Field
+except ImportError:
+    from pydantic import Field
 
 from langchain.agents.agent_toolkits.base import BaseToolkit
 from langchain.agents.agent_toolkits.nla.tool import NLATool

libs/langchain/langchain/agents/agent_toolkits/office365/toolkit.py

@@ -2,7 +2,10 @@ from __future__ import annotations
 
 from typing import TYPE_CHECKING, List
 
-from pydantic import Field
+try:
+    from pydantic.v1 import Field
+except ImportError:
+    from pydantic import Field
 
 from langchain.agents.agent_toolkits.base import BaseToolkit
 from langchain.tools import BaseTool

libs/langchain/langchain/agents/agent_toolkits/openapi/planner.py

@@ -5,7 +5,11 @@ from functools import partial
 from typing import Any, Callable, Dict, List, Optional
 
 import yaml
-from pydantic import Field
+
+try:
+    from pydantic.v1 import Field
+except ImportError:
+    from pydantic import Field
 
 from langchain.agents.agent import AgentExecutor
 from langchain.agents.agent_toolkits.openapi.planner_prompt import (

libs/langchain/langchain/agents/agent_toolkits/playwright/toolkit.py

@@ -3,7 +3,10 @@ from __future__ import annotations
 
 from typing import TYPE_CHECKING, List, Optional, Type, cast
 
-from pydantic import Extra, root_validator
+try:
+    from pydantic.v1 import Extra, root_validator
+except ImportError:
+    from pydantic import Extra, root_validator
 
 from langchain.agents.agent_toolkits.base import BaseToolkit
 from langchain.tools.base import BaseTool

libs/langchain/langchain/agents/agent_toolkits/powerbi/toolkit.py

@@ -1,7 +1,10 @@
 """Toolkit for interacting with a Power BI dataset."""
 from typing import List, Optional, Union
 
-from pydantic import Field
+try:
+    from pydantic.v1 import Field
+except ImportError:
+    from pydantic import Field
 
 from langchain.agents.agent_toolkits.base import BaseToolkit
 from langchain.callbacks.base import BaseCallbackManager

libs/langchain/langchain/agents/agent_toolkits/spark_sql/toolkit.py

@@ -1,7 +1,10 @@
 """Toolkit for interacting with Spark SQL."""
 from typing import List
 
-from pydantic import Field
+try:
+    from pydantic.v1 import Field
+except ImportError:
+    from pydantic import Field
 
 from langchain.agents.agent_toolkits.base import BaseToolkit
 from langchain.schema.language_model import BaseLanguageModel

libs/langchain/langchain/agents/agent_toolkits/sql/toolkit.py

@@ -1,7 +1,10 @@
 """Toolkit for interacting with an SQL database."""
 from typing import List
 
-from pydantic import Field
+try:
+    from pydantic.v1 import Field
+except ImportError:
+    from pydantic import Field
 
 from langchain.agents.agent_toolkits.base import BaseToolkit
 from langchain.schema.language_model import BaseLanguageModel

libs/langchain/langchain/agents/agent_toolkits/vectorstore/toolkit.py

@@ -1,7 +1,10 @@
 """Toolkit for interacting with a vector store."""
 from typing import List
 
-from pydantic import BaseModel, Field
+try:
+    from pydantic.v1 import BaseModel, Field
+except ImportError:
+    from pydantic import BaseModel, Field
 
 from langchain.agents.agent_toolkits.base import BaseToolkit
 from langchain.llms.openai import OpenAI

libs/langchain/langchain/agents/chat/base.py

@@ -1,6 +1,9 @@
 from typing import Any, List, Optional, Sequence, Tuple
 
-from pydantic import Field
+try:
+    from pydantic.v1 import Field
+except ImportError:
+    from pydantic import Field
 
 from langchain.agents.agent import Agent, AgentOutputParser
 from langchain.agents.chat.output_parser import ChatOutputParser

libs/langchain/langchain/agents/conversational/base.py

@@ -3,7 +3,10 @@ from __future__ import annotations
 
 from typing import Any, List, Optional, Sequence
 
-from pydantic import Field
+try:
+    from pydantic.v1 import Field
+except ImportError:
+    from pydantic import Field
 
 from langchain.agents.agent import Agent, AgentOutputParser
 from langchain.agents.agent_types import AgentType

libs/langchain/langchain/agents/conversational_chat/base.py

@@ -3,7 +3,10 @@ from __future__ import annotations
 
 from typing import Any, List, Optional, Sequence, Tuple
 
-from pydantic import Field
+try:
+    from pydantic.v1 import Field
+except ImportError:
+    from pydantic import Field
 
 from langchain.agents.agent import Agent, AgentOutputParser
 from langchain.agents.conversational_chat.output_parser import ConvoOutputParser

libs/langchain/langchain/agents/mrkl/base.py

@@ -3,7 +3,10 @@ from __future__ import annotations
 
 from typing import Any, Callable, List, NamedTuple, Optional, Sequence
 
-from pydantic import Field
+try:
+    from pydantic.v1 import Field
+except ImportError:
+    from pydantic import Field
 
 from langchain.agents.agent import Agent, AgentExecutor, AgentOutputParser
 from langchain.agents.agent_types import AgentType

libs/langchain/langchain/agents/openai_functions_agent/base.py

@@ -4,7 +4,10 @@ from dataclasses import dataclass
 from json import JSONDecodeError
 from typing import Any, List, Optional, Sequence, Tuple, Union
 
-from pydantic import root_validator
+try:
+    from pydantic.v1 import root_validator
+except ImportError:
+    from pydantic import root_validator
 
 from langchain.agents import BaseSingleActionAgent
 from langchain.callbacks.base import BaseCallbackManager

libs/langchain/langchain/agents/openai_functions_multi_agent/base.py

@@ -4,7 +4,10 @@ from dataclasses import dataclass
 from json import JSONDecodeError
 from typing import Any, List, Optional, Sequence, Tuple, Union
 
-from pydantic import root_validator
+try:
+    from pydantic.v1 import root_validator
+except ImportError:
+    from pydantic import root_validator
 
 from langchain.agents import BaseMultiActionAgent
 from langchain.callbacks.base import BaseCallbackManager

libs/langchain/langchain/agents/react/base.py

@@ -1,7 +1,10 @@
 """Chain that implements the ReAct paper from https://arxiv.org/pdf/2210.03629.pdf."""
 from typing import Any, List, Optional, Sequence
 
-from pydantic import Field
+try:
+    from pydantic.v1 import Field
+except ImportError:
+    from pydantic import Field
 
 from langchain.agents.agent import Agent, AgentExecutor, AgentOutputParser
 from langchain.agents.agent_types import AgentType

libs/langchain/langchain/agents/self_ask_with_search/base.py

@@ -1,7 +1,10 @@
 """Chain that does self-ask with search."""
 from typing import Any, Sequence, Union
 
-from pydantic import Field
+try:
+    from pydantic.v1 import Field
+except ImportError:
+    from pydantic import Field
 
 from langchain.agents.agent import Agent, AgentExecutor, AgentOutputParser
 from langchain.agents.agent_types import AgentType

libs/langchain/langchain/agents/structured_chat/base.py

@@ -1,7 +1,10 @@
 import re
 from typing import Any, List, Optional, Sequence, Tuple
 
-from pydantic import Field
+try:
+    from pydantic.v1 import Field
+except ImportError:
+    from pydantic import Field
 
 from langchain.agents.agent import Agent, AgentOutputParser
 from langchain.agents.structured_chat.output_parser import (

libs/langchain/langchain/agents/structured_chat/output_parser.py

@@ -5,7 +5,10 @@ import logging
 import re
 from typing import Optional, Union
 
-from pydantic import Field
+try:
+    from pydantic.v1 import Field
+except ImportError:
+    from pydantic import Field
 
 from langchain.agents.agent import AgentOutputParser
 from langchain.agents.structured_chat.prompt import FORMAT_INSTRUCTIONS

libs/langchain/langchain/callbacks/tracers/schemas.py

@@ -8,7 +8,11 @@ from uuid import UUID
 
 from langsmith.schemas import RunBase as BaseRunV2
 from langsmith.schemas import RunTypeEnum as RunTypeEnumDep
-from pydantic import BaseModel, Field, root_validator
+
+try:
+    from pydantic.v1 import BaseModel, Field, root_validator
+except ImportError:
+    from pydantic import BaseModel, Field, root_validator
 
 from langchain.schema import LLMResult
 

libs/langchain/langchain/chains/api/base.py

@@ -3,7 +3,10 @@ from __future__ import annotations
 
 from typing import Any, Dict, List, Optional
 
-from pydantic import Field, root_validator
+try:
+    from pydantic.v1 import Field, root_validator
+except ImportError:
+    from pydantic import Field, root_validator
 
 from langchain.callbacks.manager import (
     AsyncCallbackManagerForChainRun,

libs/langchain/langchain/chains/api/openapi/chain.py

@@ -4,7 +4,10 @@ from __future__ import annotations
 import json
 from typing import Any, Dict, List, NamedTuple, Optional, cast
 
-from pydantic import BaseModel, Field
+try:
+    from pydantic.v1 import BaseModel, Field
+except ImportError:
+    from pydantic import BaseModel, Field
 from requests import Response
 
 from langchain.callbacks.manager import CallbackManagerForChainRun, Callbacks

libs/langchain/langchain/chains/base.py

@@ -10,7 +10,11 @@ from pathlib import Path
 from typing import Any, Dict, List, Optional, Union
 
 import yaml
-from pydantic import Field, root_validator, validator
+
+try:
+    from pydantic.v1 import Field, root_validator, validator
+except ImportError:
+    from pydantic import Field, root_validator, validator
 
 import langchain
 from langchain.callbacks.base import BaseCallbackManager

libs/langchain/langchain/chains/combine_documents/base.py

@@ -3,7 +3,10 @@
 from abc import ABC, abstractmethod
 from typing import Any, Dict, List, Optional, Tuple
 
-from pydantic import Field
+try:
+    from pydantic.v1 import Field
+except ImportError:
+    from pydantic import Field
 
 from langchain.callbacks.manager import (
     AsyncCallbackManagerForChainRun,

libs/langchain/langchain/chains/combine_documents/map_reduce.py

@@ -4,7 +4,10 @@ from __future__ import annotations
 
 from typing import Any, Dict, List, Optional, Tuple
 
-from pydantic import Extra, root_validator
+try:
+    from pydantic.v1 import Extra, root_validator
+except ImportError:
+    from pydantic import Extra, root_validator
 
 from langchain.callbacks.manager import Callbacks
 from langchain.chains.combine_documents.base import BaseCombineDocumentsChain

libs/langchain/langchain/chains/combine_documents/map_rerank.py

@@ -4,7 +4,10 @@ from __future__ import annotations
 
 from typing import Any, Dict, List, Optional, Sequence, Tuple, Union, cast
 
-from pydantic import Extra, root_validator
+try:
+    from pydantic.v1 import Extra, root_validator
+except ImportError:
+    from pydantic import Extra, root_validator
 
 from langchain.callbacks.manager import Callbacks
 from langchain.chains.combine_documents.base import BaseCombineDocumentsChain

libs/langchain/langchain/chains/combine_documents/reduce.py

@@ -4,7 +4,10 @@ from __future__ import annotations
 
 from typing import Any, Callable, List, Optional, Protocol, Tuple
 
-from pydantic import Extra
+try:
+    from pydantic.v1 import Extra
+except ImportError:
+    from pydantic import Extra
 
 from langchain.callbacks.manager import Callbacks
 from langchain.chains.combine_documents.base import BaseCombineDocumentsChain

libs/langchain/langchain/chains/combine_documents/refine.py

@@ -4,7 +4,10 @@ from __future__ import annotations
 
 from typing import Any, Dict, List, Tuple
 
-from pydantic import Extra, Field, root_validator
+try:
+    from pydantic.v1 import Extra, Field, root_validator
+except ImportError:
+    from pydantic import Extra, Field, root_validator
 
 from langchain.callbacks.manager import Callbacks
 from langchain.chains.combine_documents.base import (

libs/langchain/langchain/chains/combine_documents/stuff.py

@@ -2,7 +2,10 @@
 
 from typing import Any, Dict, List, Optional, Tuple
 
-from pydantic import Extra, Field, root_validator
+try:
+    from pydantic.v1 import Extra, Field, root_validator
+except ImportError:
+    from pydantic import Extra, Field, root_validator
 
 from langchain.callbacks.manager import Callbacks
 from langchain.chains.combine_documents.base import (

libs/langchain/langchain/chains/constitutional_ai/models.py

@@ -1,5 +1,8 @@
 """Models for the Constitutional AI chain."""
-from pydantic import BaseModel
+try:
+    from pydantic.v1 import BaseModel
+except ImportError:
+    from pydantic import BaseModel
 
 
 class ConstitutionalPrinciple(BaseModel):

libs/langchain/langchain/chains/conversation/base.py

@@ -1,7 +1,10 @@
 """Chain that carries on a conversation and calls an LLM."""
 from typing import Dict, List
 
-from pydantic import Extra, Field, root_validator
+try:
+    from pydantic.v1 import Extra, Field, root_validator
+except ImportError:
+    from pydantic import Extra, Field, root_validator
 
 from langchain.chains.conversation.prompt import PROMPT
 from langchain.chains.llm import LLMChain

libs/langchain/langchain/chains/conversational_retrieval/base.py

@@ -7,7 +7,10 @@ from abc import abstractmethod
 from pathlib import Path
 from typing import Any, Callable, Dict, List, Optional, Tuple, Union
 
-from pydantic import Extra, Field, root_validator
+try:
+    from pydantic.v1 import Extra, Field, root_validator
+except ImportError:
+    from pydantic import Extra, Field, root_validator
 
 from langchain.callbacks.manager import (
     AsyncCallbackManagerForChainRun,

libs/langchain/langchain/chains/elasticsearch_database/base.py

@@ -3,7 +3,10 @@ from __future__ import annotations
 
 from typing import TYPE_CHECKING, Any, Dict, List, Optional
 
-from pydantic import Extra, root_validator
+try:
+    from pydantic.v1 import Extra, root_validator
+except ImportError:
+    from pydantic import Extra, root_validator
 
 from langchain.callbacks.manager import CallbackManagerForChainRun
 from langchain.chains.base import Chain

libs/langchain/langchain/chains/flare/base.py

@@ -5,7 +5,11 @@ from abc import abstractmethod
 from typing import Any, Dict, List, Optional, Sequence, Tuple
 
 import numpy as np
-from pydantic import Field
+
+try:
+    from pydantic.v1 import Field
+except ImportError:
+    from pydantic import Field
 
 from langchain.callbacks.manager import (
     CallbackManagerForChainRun,

libs/langchain/langchain/chains/graph_qa/arangodb.py

@@ -4,7 +4,10 @@ from __future__ import annotations
 import re
 from typing import Any, Dict, List, Optional
 
-from pydantic import Field
+try:
+    from pydantic.v1 import Field
+except ImportError:
+    from pydantic import Field
 
 from langchain.base_language import BaseLanguageModel
 from langchain.callbacks.manager import CallbackManagerForChainRun

libs/langchain/langchain/chains/graph_qa/base.py

@@ -3,7 +3,10 @@ from __future__ import annotations
 
 from typing import Any, Dict, List, Optional
 
-from pydantic import Field
+try:
+    from pydantic.v1 import Field
+except ImportError:
+    from pydantic import Field
 
 from langchain.callbacks.manager import CallbackManagerForChainRun
 from langchain.chains.base import Chain

libs/langchain/langchain/chains/graph_qa/cypher.py

@@ -4,7 +4,10 @@ from __future__ import annotations
 import re
 from typing import Any, Dict, List, Optional
 
-from pydantic import Field
+try:
+    from pydantic.v1 import Field
+except ImportError:
+    from pydantic import Field
 
 from langchain.callbacks.manager import CallbackManagerForChainRun
 from langchain.chains.base import Chain

libs/langchain/langchain/chains/graph_qa/hugegraph.py

@@ -3,7 +3,10 @@ from __future__ import annotations
 
 from typing import Any, Dict, List, Optional
 
-from pydantic import Field
+try:
+    from pydantic.v1 import Field
+except ImportError:
+    from pydantic import Field
 
 from langchain.callbacks.manager import CallbackManagerForChainRun
 from langchain.chains.base import Chain

libs/langchain/langchain/chains/graph_qa/kuzu.py

@@ -3,7 +3,10 @@ from __future__ import annotations
 
 from typing import Any, Dict, List, Optional
 
-from pydantic import Field
+try:
+    from pydantic.v1 import Field
+except ImportError:
+    from pydantic import Field
 
 from langchain.callbacks.manager import CallbackManagerForChainRun
 from langchain.chains.base import Chain

libs/langchain/langchain/chains/graph_qa/nebulagraph.py

@@ -3,7 +3,10 @@ from __future__ import annotations
 
 from typing import Any, Dict, List, Optional
 
-from pydantic import Field
+try:
+    from pydantic.v1 import Field
+except ImportError:
+    from pydantic import Field
 
 from langchain.callbacks.manager import CallbackManagerForChainRun
 from langchain.chains.base import Chain

libs/langchain/langchain/chains/graph_qa/neptune_cypher.py

@@ -3,7 +3,10 @@ from __future__ import annotations
 import re
 from typing import Any, Dict, List, Optional
 
-from pydantic import Field
+try:
+    from pydantic.v1 import Field
+except ImportError:
+    from pydantic import Field
 
 from langchain.base_language import BaseLanguageModel
 from langchain.callbacks.manager import CallbackManagerForChainRun

libs/langchain/langchain/chains/graph_qa/sparql.py

@@ -5,7 +5,10 @@ from __future__ import annotations
 
 from typing import Any, Dict, List, Optional
 
-from pydantic import Field
+try:
+    from pydantic.v1 import Field
+except ImportError:
+    from pydantic import Field
 
 from langchain.callbacks.manager import CallbackManagerForChainRun
 from langchain.chains.base import Chain

libs/langchain/langchain/chains/hyde/base.py

@@ -7,7 +7,11 @@ from __future__ import annotations
 from typing import Any, Dict, List, Optional
 
 import numpy as np
-from pydantic import Extra
+
+try:
+    from pydantic.v1 import Extra
+except ImportError:
+    from pydantic import Extra
 
 from langchain.callbacks.manager import CallbackManagerForChainRun
 from langchain.chains.base import Chain

libs/langchain/langchain/chains/llm.py

@@ -4,7 +4,10 @@ from __future__ import annotations
 import warnings
 from typing import Any, Dict, List, Optional, Sequence, Tuple, Union
 
-from pydantic import Extra, Field
+try:
+    from pydantic.v1 import Extra, Field
+except ImportError:
+    from pydantic import Extra, Field
 
 from langchain.callbacks.manager import (
     AsyncCallbackManager,

libs/langchain/langchain/chains/llm_bash/base.py

@@ -5,7 +5,10 @@ import logging
 import warnings
 from typing import Any, Dict, List, Optional
 
-from pydantic import Extra, Field, root_validator
+try:
+    from pydantic.v1 import Extra, Field, root_validator
+except ImportError:
+    from pydantic import Extra, Field, root_validator
 
 from langchain.callbacks.manager import CallbackManagerForChainRun
 from langchain.chains.base import Chain

libs/langchain/langchain/chains/llm_checker/base.py

@@ -4,7 +4,10 @@ from __future__ import annotations
 import warnings
 from typing import Any, Dict, List, Optional
 
-from pydantic import Extra, root_validator
+try:
+    from pydantic.v1 import Extra, root_validator
+except ImportError:
+    from pydantic import Extra, root_validator
 
 from langchain.callbacks.manager import CallbackManagerForChainRun
 from langchain.chains.base import Chain

libs/langchain/langchain/chains/llm_math/base.py

@@ -7,7 +7,11 @@ import warnings
 from typing import Any, Dict, List, Optional
 
 import numexpr
-from pydantic import Extra, root_validator
+
+try:
+    from pydantic.v1 import Extra, root_validator
+except ImportError:
+    from pydantic import Extra, root_validator
 
 from langchain.callbacks.manager import (
     AsyncCallbackManagerForChainRun,

libs/langchain/langchain/chains/llm_requests.py

@@ -3,7 +3,10 @@ from __future__ import annotations
 
 from typing import Any, Dict, List, Optional
 
-from pydantic import Extra, Field, root_validator
+try:
+    from pydantic.v1 import Extra, Field, root_validator
+except ImportError:
+    from pydantic import Extra, Field, root_validator
 
 from langchain.callbacks.manager import CallbackManagerForChainRun
 from langchain.chains import LLMChain

libs/langchain/langchain/chains/llm_summarization_checker/base.py

@@ -6,7 +6,10 @@ import warnings
 from pathlib import Path
 from typing import Any, Dict, List, Optional
 
-from pydantic import Extra, root_validator
+try:
+    from pydantic.v1 import Extra, root_validator
+except ImportError:
+    from pydantic import Extra, root_validator
 
 from langchain.callbacks.manager import CallbackManagerForChainRun
 from langchain.chains.base import Chain

libs/langchain/langchain/chains/llm_symbolic_math/base.py

@@ -4,7 +4,10 @@ from __future__ import annotations
 import re
 from typing import Any, Dict, List, Optional
 
-from pydantic import Extra
+try:
+    from pydantic.v1 import Extra
+except ImportError:
+    from pydantic import Extra
 
 from langchain.base_language import BaseLanguageModel
 from langchain.callbacks.manager import (

libs/langchain/langchain/chains/mapreduce.py

@@ -7,7 +7,10 @@ from __future__ import annotations
 
 from typing import Any, Dict, List, Mapping, Optional
 
-from pydantic import Extra
+try:
+    from pydantic.v1 import Extra
+except ImportError:
+    from pydantic import Extra
 
 from langchain.callbacks.manager import CallbackManagerForChainRun, Callbacks
 from langchain.chains import ReduceDocumentsChain

libs/langchain/langchain/chains/moderation.py

@@ -1,7 +1,10 @@
 """Pass input through a moderation endpoint."""
 from typing import Any, Dict, List, Optional
 
-from pydantic import root_validator
+try:
+    from pydantic.v1 import root_validator
+except ImportError:
+    from pydantic import root_validator
 
 from langchain.callbacks.manager import CallbackManagerForChainRun
 from langchain.chains.base import Chain

libs/langchain/langchain/chains/natbot/base.py

@@ -4,7 +4,10 @@ from __future__ import annotations
 import warnings
 from typing import Any, Dict, List, Optional
 
-from pydantic import Extra, root_validator
+try:
+    from pydantic.v1 import Extra, root_validator
+except ImportError:
+    from pydantic import Extra, root_validator
 
 from langchain.callbacks.manager import CallbackManagerForChainRun
 from langchain.chains.base import Chain

libs/langchain/langchain/chains/openai_functions/base.py

@@ -10,9 +10,13 @@ from typing import (
     Tuple,
     Type,
     Union,
+    cast,
 )
 
-from pydantic import BaseModel
+try:
+    from pydantic.v1 import BaseModel
+except ImportError:
+    from pydantic import BaseModel
 
 from langchain.base_language import BaseLanguageModel
 from langchain.chains import LLMChain
@@ -87,7 +91,7 @@ def _get_python_function_arguments(function: Callable, arg_descriptions: dict) -
         if arg == "return":
             continue
         if isinstance(arg_type, type) and issubclass(arg_type, BaseModel):
-            properties[arg] = arg_type.schema()
+            properties[arg] = cast(Type[BaseModel], arg_type).schema()
         elif arg_type.__name__ in PYTHON_TO_JSON_TYPES:
             properties[arg] = {"type": PYTHON_TO_JSON_TYPES[arg_type.__name__]}
         if arg in arg_descriptions:
@@ -147,7 +151,7 @@ def convert_to_openai_function(
     if isinstance(function, dict):
         return function
     elif isinstance(function, type) and issubclass(function, BaseModel):
-        schema = function.schema()
+        schema = cast(Type[BaseModel], function).schema()
         return {
             "name": schema["title"],
             "description": schema["description"],

libs/langchain/langchain/chains/openai_functions/citation_fuzzy_match.py

@@ -1,6 +1,9 @@
 from typing import Iterator, List
 
-from pydantic import BaseModel, Field
+try:
+    from pydantic.v1 import BaseModel, Field
+except ImportError:
+    from pydantic import BaseModel, Field
 
 from langchain.chains.llm import LLMChain
 from langchain.chains.openai_functions.utils import get_llm_kwargs

libs/langchain/langchain/chains/openai_functions/extraction.py

@@ -1,6 +1,9 @@
 from typing import Any, List
 
-from pydantic import BaseModel
+try:
+    from pydantic.v1 import BaseModel
+except ImportError:
+    from pydantic import BaseModel
 
 from langchain.chains.base import Chain
 from langchain.chains.llm import LLMChain

libs/langchain/langchain/chains/openai_functions/qa_with_structure.py

@@ -1,6 +1,9 @@
 from typing import Any, List, Optional, Type, Union
 
-from pydantic import BaseModel, Field
+try:
+    from pydantic.v1 import BaseModel, Field
+except ImportError:
+    from pydantic import BaseModel, Field
 
 from langchain.chains.llm import LLMChain
 from langchain.chains.openai_functions.utils import get_llm_kwargs

libs/langchain/langchain/chains/prompt_selector.py

@@ -1,7 +1,10 @@
 from abc import ABC, abstractmethod
 from typing import Callable, List, Tuple
 
-from pydantic import BaseModel, Field
+try:
+    from pydantic.v1 import BaseModel, Field
+except ImportError:
+    from pydantic import BaseModel, Field
 
 from langchain.chat_models.base import BaseChatModel
 from langchain.llms.base import BaseLLM

libs/langchain/langchain/chains/qa_generation/base.py

@@ -3,7 +3,10 @@ from __future__ import annotations
 import json
 from typing import Any, Dict, List, Optional
 
-from pydantic import Field
+try:
+    from pydantic.v1 import Field
+except ImportError:
+    from pydantic import Field
 
 from langchain.callbacks.manager import CallbackManagerForChainRun
 from langchain.chains.base import Chain

libs/langchain/langchain/chains/qa_with_sources/base.py

@@ -7,7 +7,10 @@ import re
 from abc import ABC, abstractmethod
 from typing import Any, Dict, List, Optional
 
-from pydantic import Extra, root_validator
+try:
+    from pydantic.v1 import Extra, root_validator
+except ImportError:
+    from pydantic import Extra, root_validator
 
 from langchain.callbacks.manager import (
     AsyncCallbackManagerForChainRun,

libs/langchain/langchain/chains/qa_with_sources/retrieval.py

@@ -2,7 +2,10 @@
 
 from typing import Any, Dict, List
 
-from pydantic import Field
+try:
+    from pydantic.v1 import Field
+except ImportError:
+    from pydantic import Field
 
 from langchain.callbacks.manager import (
     AsyncCallbackManagerForChainRun,

libs/langchain/langchain/chains/qa_with_sources/vector_db.py

@@ -3,7 +3,10 @@
 import warnings
 from typing import Any, Dict, List
 
-from pydantic import Field, root_validator
+try:
+    from pydantic.v1 import Field, root_validator
+except ImportError:
+    from pydantic import Field, root_validator
 
 from langchain.callbacks.manager import (
     AsyncCallbackManagerForChainRun,

libs/langchain/langchain/chains/query_constructor/ir.py

@@ -5,7 +5,10 @@ from abc import ABC, abstractmethod
 from enum import Enum
 from typing import Any, List, Optional, Sequence, Union
 
-from pydantic import BaseModel
+try:
+    from pydantic.v1 import BaseModel
+except ImportError:
+    from pydantic import BaseModel
 
 
 class Visitor(ABC):

libs/langchain/langchain/chains/query_constructor/schema.py

@@ -1,4 +1,7 @@
-from pydantic import BaseModel
+try:
+    from pydantic.v1 import BaseModel
+except ImportError:
+    from pydantic import BaseModel
 
 
 class AttributeInfo(BaseModel):

libs/langchain/langchain/chains/retrieval_qa/base.py

@@ -6,7 +6,10 @@ import warnings
 from abc import abstractmethod
 from typing import Any, Dict, List, Optional
 
-from pydantic import Extra, Field, root_validator
+try:
+    from pydantic.v1 import Extra, Field, root_validator
+except ImportError:
+    from pydantic import Extra, Field, root_validator
 
 from langchain.callbacks.manager import (
     AsyncCallbackManagerForChainRun,

libs/langchain/langchain/chains/router/base.py

@@ -4,7 +4,10 @@ from __future__ import annotations
 from abc import ABC
 from typing import Any, Dict, List, Mapping, NamedTuple, Optional
 
-from pydantic import Extra
+try:
+    from pydantic.v1 import Extra
+except ImportError:
+    from pydantic import Extra
 
 from langchain.callbacks.manager import (
     AsyncCallbackManagerForChainRun,

libs/langchain/langchain/chains/router/embedding_router.py

@@ -2,7 +2,10 @@ from __future__ import annotations
 
 from typing import Any, Dict, List, Optional, Sequence, Tuple, Type
 
-from pydantic import Extra
+try:
+    from pydantic.v1 import Extra
+except ImportError:
+    from pydantic import Extra
 
 from langchain.callbacks.manager import CallbackManagerForChainRun
 from langchain.chains.router.base import RouterChain

libs/langchain/langchain/chains/router/llm_router.py

@@ -3,7 +3,10 @@ from __future__ import annotations
 
 from typing import Any, Dict, List, Optional, Type, cast
 
-from pydantic import root_validator
+try:
+    from pydantic.v1 import root_validator
+except ImportError:
+    from pydantic import root_validator
 
 from langchain.callbacks.manager import (
     AsyncCallbackManagerForChainRun,

libs/langchain/langchain/chains/sequential.py

@@ -1,7 +1,10 @@
 """Chain pipeline where the outputs of one step feed directly into next."""
 from typing import Any, Dict, List, Optional
 
-from pydantic import Extra, root_validator
+try:
+    from pydantic.v1 import Extra, root_validator
+except ImportError:
+    from pydantic import Extra, root_validator
 
 from langchain.callbacks.manager import (
     AsyncCallbackManagerForChainRun,

libs/langchain/langchain/chat_models/anyscale.py

@@ -7,7 +7,11 @@ import sys
 from typing import TYPE_CHECKING, Optional, Set
 
 import requests
-from pydantic import Field, root_validator
+
+try:
+    from pydantic.v1 import Field, root_validator
+except ImportError:
+    from pydantic import Field, root_validator
 
 from langchain.chat_models.openai import (
     ChatOpenAI,

libs/langchain/langchain/chat_models/azure_openai.py

@@ -4,7 +4,10 @@ from __future__ import annotations
 import logging
 from typing import Any, Dict, Mapping
 
-from pydantic import root_validator
+try:
+    from pydantic.v1 import root_validator
+except ImportError:
+    from pydantic import root_validator
 
 from langchain.chat_models.openai import ChatOpenAI
 from langchain.schema import ChatResult

libs/langchain/langchain/chat_models/azureml_endpoint.py

@@ -1,7 +1,10 @@
 import json
 from typing import Any, Dict, List, Optional
 
-from pydantic import validator
+try:
+    from pydantic.v1 import validator
+except ImportError:
+    from pydantic import validator
 
 from langchain.callbacks.manager import CallbackManagerForLLMRun
 from langchain.chat_models.base import SimpleChatModel

libs/langchain/langchain/chat_models/base.py

@@ -14,7 +14,10 @@ from typing import (
     cast,
 )
 
-from pydantic import Field, root_validator
+try:
+    from pydantic.v1 import Field, root_validator
+except ImportError:
+    from pydantic import Field, root_validator
 
 import langchain
 from langchain.callbacks.base import BaseCallbackManager

libs/langchain/langchain/chat_models/google_palm.py

@@ -4,7 +4,10 @@ from __future__ import annotations
 import logging
 from typing import TYPE_CHECKING, Any, Callable, Dict, List, Optional
 
-from pydantic import BaseModel, root_validator
+try:
+    from pydantic.v1 import BaseModel, root_validator
+except ImportError:
+    from pydantic import BaseModel, root_validator
 from tenacity import (
     before_sleep_log,
     retry,

libs/langchain/langchain/chat_models/human.py

@@ -5,7 +5,11 @@ from io import StringIO
 from typing import Any, Callable, Dict, List, Mapping, Optional
 
 import yaml
-from pydantic import Field
+
+try:
+    from pydantic.v1 import Field
+except ImportError:
+    from pydantic import Field
 
 from langchain.callbacks.manager import (
     AsyncCallbackManagerForLLMRun,

libs/langchain/langchain/chat_models/jinachat.py

@@ -15,7 +15,10 @@ from typing import (
     Union,
 )
 
-from pydantic import Field, root_validator
+try:
+    from pydantic.v1 import Field, root_validator
+except ImportError:
+    from pydantic import Field, root_validator
 from tenacity import (
     before_sleep_log,
     retry,

libs/langchain/langchain/chat_models/mlflow_ai_gateway.py

@@ -3,7 +3,10 @@ import logging
 from functools import partial
 from typing import Any, Dict, List, Mapping, Optional
 
-from pydantic import BaseModel, Extra
+try:
+    from pydantic.v1 import BaseModel, Extra
+except ImportError:
+    from pydantic import BaseModel, Extra
 
 from langchain.callbacks.manager import (
     AsyncCallbackManagerForLLMRun,
@@ -26,7 +29,7 @@ from langchain.schema.messages import (
 logger = logging.getLogger(__name__)
 
 
-class ChatParams(BaseModel, extra=Extra.allow):
+class ChatParams(BaseModel):
     """Parameters for the MLflow AI Gateway LLM."""
 
     temperature: float = 0.0
@@ -35,6 +38,11 @@ class ChatParams(BaseModel, extra=Extra.allow):
     stop: Optional[List[str]] = None
     max_tokens: Optional[int] = None
 
+    class Config:
+        """Configuration for this pydantic object."""
+
+        extra = Extra.forbid
+
 
 class ChatMLflowAIGateway(BaseChatModel):
     """

libs/langchain/langchain/chat_models/openai.py

@@ -17,7 +17,10 @@ from typing import (
     Union,
 )
 
-from pydantic import Field, root_validator
+try:
+    from pydantic.v1 import Field, root_validator
+except ImportError:
+    from pydantic import Field, root_validator
 
 from langchain.callbacks.manager import (
     AsyncCallbackManagerForLLMRun,

libs/langchain/langchain/chat_models/vertexai.py

@@ -2,7 +2,10 @@
 from dataclasses import dataclass, field
 from typing import TYPE_CHECKING, Any, Dict, List, Optional
 
-from pydantic import root_validator
+try:
+    from pydantic.v1 import root_validator
+except ImportError:
+    from pydantic import root_validator
 
 from langchain.callbacks.manager import CallbackManagerForLLMRun
 from langchain.chat_models.base import BaseChatModel

libs/langchain/langchain/document_loaders/apify_dataset.py

@@ -1,6 +1,9 @@
 from typing import Any, Callable, Dict, List
 
-from pydantic import BaseModel, root_validator
+try:
+    from pydantic.v1 import BaseModel, root_validator
+except ImportError:
+    from pydantic import BaseModel, root_validator
 
 from langchain.docstore.document import Document
 from langchain.document_loaders.base import BaseLoader

libs/langchain/langchain/document_loaders/blob_loaders/schema.py

@@ -13,7 +13,10 @@ from io import BufferedReader, BytesIO
 from pathlib import PurePath
 from typing import Any, Generator, Iterable, Mapping, Optional, Union
 
-from pydantic import BaseModel, root_validator
+try:
+    from pydantic.v1 import BaseModel, root_validator
+except ImportError:
+    from pydantic import BaseModel, root_validator
 
 PathLike = Union[str, PurePath]
 

libs/langchain/langchain/document_loaders/docugami.py

@@ -8,7 +8,11 @@ from pathlib import Path
 from typing import Any, Dict, List, Mapping, Optional, Sequence, Union
 
 import requests
-from pydantic import BaseModel, root_validator
+
+try:
+    from pydantic.v1 import BaseModel, root_validator
+except ImportError:
+    from pydantic import BaseModel, root_validator
 
 from langchain.docstore.document import Document
 from langchain.document_loaders.base import BaseLoader

libs/langchain/langchain/document_loaders/dropbox.py

@@ -13,7 +13,10 @@ import tempfile
 from pathlib import Path
 from typing import Any, Dict, List, Optional
 
-from pydantic import BaseModel, root_validator
+try:
+    from pydantic.v1 import BaseModel, root_validator
+except ImportError:
+    from pydantic import BaseModel, root_validator
 
 from langchain.docstore.document import Document
 from langchain.document_loaders.base import BaseLoader

libs/langchain/langchain/document_loaders/embaas.py

@@ -3,7 +3,11 @@ import warnings
 from typing import Any, Dict, Iterator, List, Optional
 
 import requests
-from pydantic import BaseModel, root_validator, validator
+
+try:
+    from pydantic.v1 import BaseModel, root_validator, validator
+except ImportError:
+    from pydantic import BaseModel, root_validator, validator
 from typing_extensions import NotRequired, TypedDict
 
 from langchain.docstore.document import Document

libs/langchain/langchain/document_loaders/github.py

@@ -3,7 +3,11 @@ from datetime import datetime
 from typing import Dict, Iterator, List, Literal, Optional, Union
 
 import requests
-from pydantic import BaseModel, root_validator, validator
+
+try:
+    from pydantic.v1 import BaseModel, root_validator, validator
+except ImportError:
+    from pydantic import BaseModel, root_validator, validator
 
 from langchain.docstore.document import Document
 from langchain.document_loaders.base import BaseLoader

libs/langchain/langchain/document_loaders/googledrive.py

@@ -13,7 +13,10 @@ import os
 from pathlib import Path
 from typing import Any, Dict, List, Optional, Sequence, Union
 
-from pydantic import BaseModel, root_validator, validator
+try:
+    from pydantic.v1 import BaseModel, root_validator, validator
+except ImportError:
+    from pydantic import BaseModel, root_validator, validator
 
 from langchain.docstore.document import Document
 from langchain.document_loaders.base import BaseLoader

libs/langchain/langchain/document_loaders/onedrive.py

@@ -8,7 +8,10 @@ from enum import Enum
 from pathlib import Path
 from typing import TYPE_CHECKING, Dict, List, Optional, Type, Union
 
-from pydantic import BaseModel, BaseSettings, Field, FilePath, SecretStr
+try:
+    from pydantic.v1 import BaseModel, BaseSettings, Field, FilePath, SecretStr
+except ImportError:
+    from pydantic import BaseModel, BaseSettings, Field, FilePath, SecretStr
 
 from langchain.docstore.document import Document
 from langchain.document_loaders.base import BaseLoader

libs/langchain/langchain/document_loaders/onedrive_file.py

@@ -3,7 +3,10 @@ from __future__ import annotations
 import tempfile
 from typing import TYPE_CHECKING, List
 
-from pydantic import BaseModel, Field
+try:
+    from pydantic.v1 import BaseModel, Field
+except ImportError:
+    from pydantic import BaseModel, Field
 
 from langchain.docstore.document import Document
 from langchain.document_loaders.base import BaseLoader

libs/langchain/langchain/document_loaders/youtube.py

@@ -6,8 +6,14 @@ from pathlib import Path
 from typing import Any, Dict, List, Optional, Sequence, Union
 from urllib.parse import parse_qs, urlparse
 
-from pydantic import root_validator
-from pydantic.dataclasses import dataclass
+try:
+    from pydantic.v1 import root_validator
+except ImportError:
+    from pydantic import root_validator
+try:
+    from pydantic.v1.dataclasses import dataclass
+except ImportError:
+    from pydantic.dataclasses import dataclass
 
 from langchain.docstore.document import Document
 from langchain.document_loaders.base import BaseLoader

libs/langchain/langchain/document_transformers/embeddings_redundant_filter.py

@@ -2,7 +2,11 @@
 from typing import Any, Callable, List, Sequence
 
 import numpy as np
-from pydantic import BaseModel, Field
+
+try:
+    from pydantic.v1 import BaseModel, Field
+except ImportError:
+    from pydantic import BaseModel, Field
 
 from langchain.embeddings.base import Embeddings
 from langchain.schema import BaseDocumentTransformer, Document

libs/langchain/langchain/document_transformers/long_context_reorder.py

@@ -1,7 +1,10 @@
 """Reorder documents"""
 from typing import Any, List, Sequence
 
-from pydantic import BaseModel
+try:
+    from pydantic.v1 import BaseModel
+except ImportError:
+    from pydantic import BaseModel
 
 from langchain.schema import BaseDocumentTransformer, Document
 

libs/langchain/langchain/document_transformers/openai_functions.py

@@ -1,7 +1,10 @@
 """Document transformers that use OpenAI Functions models"""
 from typing import Any, Dict, Optional, Sequence, Type, Union
 
-from pydantic import BaseModel
+try:
+    from pydantic.v1 import BaseModel
+except ImportError:
+    from pydantic import BaseModel
 
 from langchain.chains.llm import LLMChain
 from langchain.chains.openai_functions import create_tagging_chain

libs/langchain/langchain/embeddings/aleph_alpha.py

@@ -1,6 +1,9 @@
 from typing import Any, Dict, List, Optional
 
-from pydantic import BaseModel, root_validator
+try:
+    from pydantic.v1 import BaseModel, root_validator
+except ImportError:
+    from pydantic import BaseModel, root_validator
 
 from langchain.embeddings.base import Embeddings
 from langchain.utils import get_from_dict_or_env

libs/langchain/langchain/embeddings/awa.py

@@ -1,6 +1,9 @@
 from typing import Any, Dict, List
 
-from pydantic import BaseModel, root_validator
+try:
+    from pydantic.v1 import BaseModel, root_validator
+except ImportError:
+    from pydantic import BaseModel, root_validator
 
 from langchain.embeddings.base import Embeddings
 

libs/langchain/langchain/embeddings/bedrock.py

@@ -2,7 +2,10 @@ import json
 import os
 from typing import Any, Dict, List, Optional
 
-from pydantic import BaseModel, Extra, root_validator
+try:
+    from pydantic.v1 import BaseModel, Extra, root_validator
+except ImportError:
+    from pydantic import BaseModel, Extra, root_validator
 
 from langchain.embeddings.base import Embeddings