Use a submodule for pydantic v1 compat (#9371)

<!-- Thank you for contributing to LangChain!

Replace this entire comment with:
  - Description: a description of the change, 
  - Issue: the issue # it fixes (if applicable),
  - Dependencies: any dependencies required for this change,
- Tag maintainer: for a quicker response, tag the relevant maintainer
(see below),
- Twitter handle: we announce bigger features on Twitter. If your PR
gets announced and you'd like a mention, we'll gladly shout you out!

Please make sure your PR is passing linting and testing before
submitting. Run `make format`, `make lint` and `make test` to check this
locally.

See contribution guidelines for more information on how to write/run
tests, lint, etc:

https://github.com/hwchase17/langchain/blob/master/.github/CONTRIBUTING.md

If you're adding a new integration, please include:
1. a test for the integration, preferably unit tests that do not rely on
network access,
2. an example notebook showing its use. These live is docs/extras
directory.

If no one reviews your PR within a few days, please @-mention one of
@baskaryan, @eyurtsev, @hwchase17, @rlancemartin.
 -->

libs/experimental/langchain_experimental/__init__.py

@@ -1,24 +0,0 @@
-import importlib
-import sys
-
-## Create namespaces for pydantic v1 and v2.
-# This code must stay at the top of the file before other modules may
-# attempt to import pydantic since it adds pydantic_v1 and pydantic_v2 to sys.modules.
-#
-# This hack is done for the following reasons:
-# * Langchain will attempt to remain compatible with both pydantic v1 and v2 since
-#   both dependencies and dependents may be stuck on either version of v1 or v2.
-# * Creating namespaces for pydantic v1 and v2 should allow us to write code that
-#   unambiguously uses either v1 or v2 API.
-# * This change is easier to roll out and roll back.
-
-try:
-    pydantic_v1 = importlib.import_module("pydantic.v1")
-except ImportError:
-    pydantic_v1 = importlib.import_module("pydantic")
-
-if "pydantic_v1" not in sys.modules:
-    # Use a conditional because langchain experimental
-    # will use the same strategy to add pydantic_v1 to sys.modules
-    # and may run prior to langchain core package.
-    sys.modules["pydantic_v1"] = pydantic_v1

libs/experimental/langchain_experimental/autonomous_agents/autogpt/agent.py

@@ -13,7 +13,6 @@ from langchain.schema.messages import AIMessage, HumanMessage, SystemMessage
 from langchain.tools.base import BaseTool
 from langchain.tools.human.tool import HumanInputRun
 from langchain.vectorstores.base import VectorStoreRetriever
-from pydantic_v1 import ValidationError
 
 from langchain_experimental.autonomous_agents.autogpt.output_parser import (
     AutoGPTOutputParser,
@@ -23,6 +22,7 @@ from langchain_experimental.autonomous_agents.autogpt.prompt import AutoGPTPromp
 from langchain_experimental.autonomous_agents.autogpt.prompt_generator import (
     FINISH_NAME,
 )
+from langchain_experimental.pydantic_v1 import ValidationError
 
 
 class AutoGPT:

libs/experimental/langchain_experimental/autonomous_agents/autogpt/memory.py

@@ -2,7 +2,8 @@ from typing import Any, Dict, List
 
 from langchain.memory.chat_memory import BaseChatMemory, get_prompt_input_key
 from langchain.vectorstores.base import VectorStoreRetriever
-from pydantic_v1 import Field
+
+from langchain_experimental.pydantic_v1 import Field
 
 
 class AutoGPTMemory(BaseChatMemory):

libs/experimental/langchain_experimental/autonomous_agents/autogpt/prompt.py

@@ -7,9 +7,9 @@ from langchain.prompts.chat import (
 from langchain.schema.messages import BaseMessage, HumanMessage, SystemMessage
 from langchain.tools.base import BaseTool
 from langchain.vectorstores.base import VectorStoreRetriever
-from pydantic_v1 import BaseModel
 
 from langchain_experimental.autonomous_agents.autogpt.prompt_generator import get_prompt
+from langchain_experimental.pydantic_v1 import BaseModel
 
 
 class AutoGPTPrompt(BaseChatPromptTemplate, BaseModel):

libs/experimental/langchain_experimental/autonomous_agents/baby_agi/baby_agi.py

@@ -6,7 +6,6 @@ from langchain.callbacks.manager import CallbackManagerForChainRun
 from langchain.chains.base import Chain
 from langchain.schema.language_model import BaseLanguageModel
 from langchain.vectorstores.base import VectorStore
-from pydantic_v1 import BaseModel, Field
 
 from langchain_experimental.autonomous_agents.baby_agi.task_creation import (
     TaskCreationChain,
@@ -17,6 +16,7 @@ from langchain_experimental.autonomous_agents.baby_agi.task_execution import (
 from langchain_experimental.autonomous_agents.baby_agi.task_prioritization import (
     TaskPrioritizationChain,
 )
+from langchain_experimental.pydantic_v1 import BaseModel, Field
 
 
 class BabyAGI(Chain, BaseModel):

libs/experimental/langchain_experimental/autonomous_agents/hugginggpt/task_planner.py

@@ -13,7 +13,8 @@ from langchain.prompts.chat import (
     SystemMessagePromptTemplate,
 )
 from langchain.tools.base import BaseTool
-from pydantic_v1 import BaseModel
+
+from langchain_experimental.pydantic_v1 import BaseModel
 
 DEMONSTRATIONS = [
     {

libs/experimental/langchain_experimental/cpal/models.py

@@ -6,9 +6,15 @@ from typing import Any, Optional, Union
 import duckdb
 import pandas as pd
 from langchain.graphs.networkx_graph import NetworkxEntityGraph
-from pydantic_v1 import BaseModel, Field, PrivateAttr, root_validator, validator
 
 from langchain_experimental.cpal.constants import Constant
+from langchain_experimental.pydantic_v1 import (
+    BaseModel,
+    Field,
+    PrivateAttr,
+    root_validator,
+    validator,
+)
 
 
 class NarrativeModel(BaseModel):

libs/experimental/langchain_experimental/generative_agents/generative_agent.py

@@ -5,9 +5,9 @@ from typing import Any, Dict, List, Optional, Tuple
 from langchain.chains import LLMChain
 from langchain.prompts import PromptTemplate
 from langchain.schema.language_model import BaseLanguageModel
-from pydantic_v1 import BaseModel, Field
 
 from langchain_experimental.generative_agents.memory import GenerativeAgentMemory
+from langchain_experimental.pydantic_v1 import BaseModel, Field
 
 
 class GenerativeAgent(BaseModel):

libs/experimental/langchain_experimental/llms/anthropic_functions.py

@@ -19,7 +19,8 @@ from langchain.schema.messages import (
     BaseMessage,
     SystemMessage,
 )
-from pydantic_v1 import root_validator
+
+from langchain_experimental.pydantic_v1 import root_validator
 
 prompt = """In addition to responding, you can use tools. \
 You have access to the following tools.

libs/experimental/langchain_experimental/llms/jsonformer_decoder.py

@@ -6,7 +6,8 @@ from typing import TYPE_CHECKING, Any, List, Optional, cast
 
 from langchain.callbacks.manager import CallbackManagerForLLMRun
 from langchain.llms.huggingface_pipeline import HuggingFacePipeline
-from pydantic_v1 import Field, root_validator
+
+from langchain_experimental.pydantic_v1 import Field, root_validator
 
 if TYPE_CHECKING:
     import jsonformer

libs/experimental/langchain_experimental/llms/rellm_decoder.py

@@ -6,7 +6,8 @@ from typing import TYPE_CHECKING, Any, List, Optional, cast
 from langchain.callbacks.manager import CallbackManagerForLLMRun
 from langchain.llms.huggingface_pipeline import HuggingFacePipeline
 from langchain.llms.utils import enforce_stop_tokens
-from pydantic_v1 import Field, root_validator
+
+from langchain_experimental.pydantic_v1 import Field, root_validator
 
 if TYPE_CHECKING:
     import rellm

libs/experimental/langchain_experimental/pal_chain/base.py

@@ -15,10 +15,10 @@ from langchain.chains.base import Chain
 from langchain.chains.llm import LLMChain
 from langchain.schema.language_model import BaseLanguageModel
 from langchain.utilities import PythonREPL
-from pydantic_v1 import Extra, Field
 
 from langchain_experimental.pal_chain.colored_object_prompt import COLORED_OBJECT_PROMPT
 from langchain_experimental.pal_chain.math_prompt import MATH_PROMPT
+from langchain_experimental.pydantic_v1 import Extra, Field
 
 COMMAND_EXECUTION_FUNCTIONS = ["system", "exec", "execfile", "eval"]
 

libs/experimental/langchain_experimental/plan_and_execute/agent_executor.py

@@ -5,7 +5,6 @@ from langchain.callbacks.manager import (
     CallbackManagerForChainRun,
 )
 from langchain.chains.base import Chain
-from pydantic_v1 import Field
 
 from langchain_experimental.plan_and_execute.executors.base import BaseExecutor
 from langchain_experimental.plan_and_execute.planners.base import BasePlanner
@@ -13,6 +12,7 @@ from langchain_experimental.plan_and_execute.schema import (
     BaseStepContainer,
     ListStepContainer,
 )
+from langchain_experimental.pydantic_v1 import Field
 
 
 class PlanAndExecute(Chain):

libs/experimental/langchain_experimental/plan_and_execute/executors/base.py

@@ -3,9 +3,9 @@ from typing import Any
 
 from langchain.callbacks.manager import Callbacks
 from langchain.chains.base import Chain
-from pydantic_v1 import BaseModel
 
 from langchain_experimental.plan_and_execute.schema import StepResponse
+from langchain_experimental.pydantic_v1 import BaseModel
 
 
 class BaseExecutor(BaseModel):

libs/experimental/langchain_experimental/plan_and_execute/planners/base.py

@@ -3,9 +3,9 @@ from typing import Any, List, Optional
 
 from langchain.callbacks.manager import Callbacks
 from langchain.chains.llm import LLMChain
-from pydantic_v1 import BaseModel
 
 from langchain_experimental.plan_and_execute.schema import Plan, PlanOutputParser
+from langchain_experimental.pydantic_v1 import BaseModel
 
 
 class BasePlanner(BaseModel):

libs/experimental/langchain_experimental/plan_and_execute/schema.py

@@ -2,7 +2,8 @@ from abc import abstractmethod
 from typing import List, Tuple
 
 from langchain.schema import BaseOutputParser
-from pydantic_v1 import BaseModel, Field
+
+from langchain_experimental.pydantic_v1 import BaseModel, Field
 
 
 class Step(BaseModel):

libs/experimental/langchain_experimental/pydantic_v1/__init__.py

@@ -0,0 +1,23 @@
+from importlib import metadata
+
+## Create namespaces for pydantic v1 and v2.
+# This code must stay at the top of the file before other modules may
+# attempt to import pydantic since it adds pydantic_v1 and pydantic_v2 to sys.modules.
+#
+# This hack is done for the following reasons:
+# * Langchain will attempt to remain compatible with both pydantic v1 and v2 since
+#   both dependencies and dependents may be stuck on either version of v1 or v2.
+# * Creating namespaces for pydantic v1 and v2 should allow us to write code that
+#   unambiguously uses either v1 or v2 API.
+# * This change is easier to roll out and roll back.
+
+try:
+    from pydantic.v1 import *  # noqa: F403
+except ImportError:
+    from pydantic import *  # noqa: F403
+
+
+try:
+    _PYDANTIC_MAJOR_VERSION: int = int(metadata.version("pydantic").split(".")[0])
+except metadata.PackageNotFoundError:
+    _PYDANTIC_MAJOR_VERSION = 0

libs/experimental/langchain_experimental/pydantic_v1/dataclasses.py

@@ -0,0 +1,4 @@
+try:
+    from pydantic.v1.dataclasses import *  # noqa: F403
+except ImportError:
+    from pydantic.dataclasses import *  # noqa: F403

libs/experimental/langchain_experimental/pydantic_v1/main.py

@@ -0,0 +1,4 @@
+try:
+    from pydantic.v1.main import *  # noqa: F403
+except ImportError:
+    from pydantic.main import *  # noqa: F403

libs/experimental/langchain_experimental/smart_llm/base.py

@@ -13,7 +13,8 @@ from langchain.prompts.chat import (
     HumanMessagePromptTemplate,
 )
 from langchain.schema import LLMResult, PromptValue
-from pydantic_v1 import Extra, root_validator
+
+from langchain_experimental.pydantic_v1 import Extra, root_validator
 
 
 class SmartLLMChain(Chain):

libs/experimental/langchain_experimental/sql/base.py

@@ -13,7 +13,8 @@ from langchain.schema import BasePromptTemplate
 from langchain.schema.language_model import BaseLanguageModel
 from langchain.tools.sql_database.prompt import QUERY_CHECKER
 from langchain.utilities.sql_database import SQLDatabase
-from pydantic_v1 import Extra, Field, root_validator
+
+from langchain_experimental.pydantic_v1 import Extra, Field, root_validator
 
 INTERMEDIATE_STEPS_KEY = "intermediate_steps"
 

libs/experimental/langchain_experimental/tot/base.py

@@ -20,8 +20,8 @@ from langchain.callbacks.manager import (
     CallbackManagerForChainRun,
 )
 from langchain.chains.base import Chain
-from pydantic_v1 import Extra
 
+from langchain_experimental.pydantic_v1 import Extra
 from langchain_experimental.tot.checker import ToTChecker
 from langchain_experimental.tot.controller import ToTController
 from langchain_experimental.tot.memory import ToTDFSMemory

libs/experimental/langchain_experimental/tot/thought.py

@@ -3,7 +3,7 @@ from __future__ import annotations
 from enum import Enum
 from typing import Set
 
-from pydantic_v1 import BaseModel, Field
+from langchain_experimental.pydantic_v1 import BaseModel, Field
 
 
 class ThoughtValidity(Enum):

libs/experimental/langchain_experimental/tot/thought_generation.py

@@ -11,8 +11,8 @@ from typing import Any, Dict, List, Tuple
 
 from langchain.chains.llm import LLMChain
 from langchain.prompts.base import BasePromptTemplate
-from pydantic_v1 import Field
 
+from langchain_experimental.pydantic_v1 import Field
 from langchain_experimental.tot.prompts import COT_PROMPT, PROPOSE_PROMPT
 
 

libs/experimental/tests/unit_tests/fake_llm.py

@@ -3,7 +3,8 @@ from typing import Any, Dict, List, Mapping, Optional, cast
 
 from langchain.callbacks.manager import CallbackManagerForLLMRun
 from langchain.llms.base import LLM
-from pydantic_v1 import validator
+
+from langchain_experimental.pydantic_v1 import validator
 
 
 class FakeLLM(LLM):

libs/langchain/langchain/__init__.py

@@ -1,38 +1,8 @@
 # ruff: noqa: E402
 """Main entrypoint into package."""
-import importlib
-import sys
 from importlib import metadata
 from typing import Optional
 
-## Create namespaces for pydantic v1 and v2.
-# This code must stay at the top of the file before other modules may
-# attempt to import pydantic since it adds pydantic_v1 and pydantic_v2 to sys.modules.
-#
-# This hack is done for the following reasons:
-# * Langchain will attempt to remain compatible with both pydantic v1 and v2 since
-#   both dependencies and dependents may be stuck on either version of v1 or v2.
-# * Creating namespaces for pydantic v1 and v2 should allow us to write code that
-#   unambiguously uses either v1 or v2 API.
-# * This change is easier to roll out and roll back.
-
-try:
-    pydantic_v1 = importlib.import_module("pydantic.v1")
-except ImportError:
-    pydantic_v1 = importlib.import_module("pydantic")
-
-if "pydantic_v1" not in sys.modules:
-    # Use a conditional because langchain experimental
-    # will use the same strategy to add pydantic_v1 to sys.modules
-    # and may run prior to langchain core package.
-    sys.modules["pydantic_v1"] = pydantic_v1
-
-try:
-    _PYDANTIC_MAJOR_VERSION: int = int(metadata.version("pydantic").split(".")[0])
-except metadata.PackageNotFoundError:
-    _PYDANTIC_MAJOR_VERSION = 0
-
-
 from langchain.agents import MRKLChain, ReActChain, SelfAskWithSearchChain
 from langchain.cache import BaseCache
 from langchain.chains import (

libs/langchain/langchain/agents/agent.py

@@ -10,7 +10,6 @@ from pathlib import Path
 from typing import Any, Callable, Dict, List, Optional, Sequence, Tuple, Union
 
 import yaml
-from pydantic_v1 import BaseModel, root_validator
 
 from langchain.agents.agent_iterator import AgentExecutorIterator
 from langchain.agents.agent_types import AgentType
@@ -27,6 +26,7 @@ from langchain.chains.base import Chain
 from langchain.chains.llm import LLMChain
 from langchain.prompts.few_shot import FewShotPromptTemplate
 from langchain.prompts.prompt import PromptTemplate
+from langchain.pydantic_v1 import BaseModel, root_validator
 from langchain.schema import (
     AgentAction,
     AgentFinish,

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

@@ -2,9 +2,8 @@ from __future__ import annotations
 
 from typing import TYPE_CHECKING, List
 
-from pydantic_v1 import Field
-
 from langchain.agents.agent_toolkits.base import BaseToolkit
+from langchain.pydantic_v1 import Field
 from langchain.tools import BaseTool
 from langchain.tools.amadeus.closest_airport import AmadeusClosestAirport
 from langchain.tools.amadeus.flight_search import AmadeusFlightSearch

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

@@ -2,8 +2,7 @@
 from abc import ABC, abstractmethod
 from typing import List
 
-from pydantic_v1 import BaseModel
-
+from langchain.pydantic_v1 import BaseModel
 from langchain.tools import BaseTool
 
 

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

@@ -2,9 +2,8 @@ from __future__ import annotations
 
 from typing import List, Optional
 
-from pydantic_v1 import root_validator
-
 from langchain.agents.agent_toolkits.base import BaseToolkit
+from langchain.pydantic_v1 import root_validator
 from langchain.tools import BaseTool
 from langchain.tools.file_management.copy import CopyFileTool
 from langchain.tools.file_management.delete import DeleteFileTool

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

@@ -2,9 +2,8 @@ from __future__ import annotations
 
 from typing import TYPE_CHECKING, List
 
-from pydantic_v1 import Field
-
 from langchain.agents.agent_toolkits.base import BaseToolkit
+from langchain.pydantic_v1 import Field
 from langchain.tools import BaseTool
 from langchain.tools.gmail.create_draft import GmailCreateDraft
 from langchain.tools.gmail.get_message import GmailGetMessage

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

@@ -2,10 +2,9 @@ from __future__ import annotations
 
 from typing import Any, List, Optional, Sequence
 
-from pydantic_v1 import Field
-
 from langchain.agents.agent_toolkits.base import BaseToolkit
 from langchain.agents.agent_toolkits.nla.tool import NLATool
+from langchain.pydantic_v1 import Field
 from langchain.schema.language_model import BaseLanguageModel
 from langchain.tools.base import BaseTool
 from langchain.tools.openapi.utils.openapi_utils import OpenAPISpec

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

@@ -2,9 +2,8 @@ from __future__ import annotations
 
 from typing import TYPE_CHECKING, List
 
-from pydantic_v1 import Field
-
 from langchain.agents.agent_toolkits.base import BaseToolkit
+from langchain.pydantic_v1 import Field
 from langchain.tools import BaseTool
 from langchain.tools.office365.create_draft_message import O365CreateDraftMessage
 from langchain.tools.office365.events_search import O365SearchEvents

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

@@ -5,7 +5,6 @@ from functools import partial
 from typing import Any, Callable, Dict, List, Optional
 
 import yaml
-from pydantic_v1 import Field
 
 from langchain.agents.agent import AgentExecutor
 from langchain.agents.agent_toolkits.openapi.planner_prompt import (
@@ -33,6 +32,7 @@ from langchain.chains.llm import LLMChain
 from langchain.llms.openai import OpenAI
 from langchain.memory import ReadOnlySharedMemory
 from langchain.prompts import PromptTemplate
+from langchain.pydantic_v1 import Field
 from langchain.schema import BasePromptTemplate
 from langchain.schema.language_model import BaseLanguageModel
 from langchain.tools.base import BaseTool

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

@@ -3,9 +3,8 @@ from __future__ import annotations
 
 from typing import TYPE_CHECKING, List, Optional, Type, cast
 
-from pydantic_v1 import Extra, root_validator
-
 from langchain.agents.agent_toolkits.base import BaseToolkit
+from langchain.pydantic_v1 import Extra, root_validator
 from langchain.tools.base import BaseTool
 from langchain.tools.playwright.base import (
     BaseBrowserTool,

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

@@ -1,8 +1,6 @@
 """Toolkit for interacting with a Power BI dataset."""
 from typing import List, Optional, Union
 
-from pydantic_v1 import Field
-
 from langchain.agents.agent_toolkits.base import BaseToolkit
 from langchain.callbacks.base import BaseCallbackManager
 from langchain.chains.llm import LLMChain
@@ -13,6 +11,7 @@ from langchain.prompts.chat import (
     HumanMessagePromptTemplate,
     SystemMessagePromptTemplate,
 )
+from langchain.pydantic_v1 import Field
 from langchain.schema.language_model import BaseLanguageModel
 from langchain.tools import BaseTool
 from langchain.tools.powerbi.prompt import (

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

@@ -1,9 +1,8 @@
 """Toolkit for interacting with Spark SQL."""
 from typing import List
 
-from pydantic_v1 import Field
-
 from langchain.agents.agent_toolkits.base import BaseToolkit
+from langchain.pydantic_v1 import Field
 from langchain.schema.language_model import BaseLanguageModel
 from langchain.tools import BaseTool
 from langchain.tools.spark_sql.tool import (

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

@@ -1,9 +1,8 @@
 """Toolkit for interacting with an SQL database."""
 from typing import List
 
-from pydantic_v1 import Field
-
 from langchain.agents.agent_toolkits.base import BaseToolkit
+from langchain.pydantic_v1 import Field
 from langchain.schema.language_model import BaseLanguageModel
 from langchain.tools import BaseTool
 from langchain.tools.sql_database.tool import (

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

@@ -1,10 +1,9 @@
 """Toolkit for interacting with a vector store."""
 from typing import List
 
-from pydantic_v1 import BaseModel, Field
-
 from langchain.agents.agent_toolkits.base import BaseToolkit
 from langchain.llms.openai import OpenAI
+from langchain.pydantic_v1 import BaseModel, Field
 from langchain.schema.language_model import BaseLanguageModel
 from langchain.tools import BaseTool
 from langchain.tools.vectorstore.tool import (

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

@@ -1,7 +1,5 @@
 from typing import Any, List, Optional, Sequence, Tuple
 
-from pydantic_v1 import Field
-
 from langchain.agents.agent import Agent, AgentOutputParser
 from langchain.agents.chat.output_parser import ChatOutputParser
 from langchain.agents.chat.prompt import (
@@ -18,6 +16,7 @@ from langchain.prompts.chat import (
     HumanMessagePromptTemplate,
     SystemMessagePromptTemplate,
 )
+from langchain.pydantic_v1 import Field
 from langchain.schema import AgentAction, BasePromptTemplate
 from langchain.schema.language_model import BaseLanguageModel
 from langchain.tools.base import BaseTool

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

@@ -3,8 +3,6 @@ from __future__ import annotations
 
 from typing import Any, List, Optional, Sequence
 
-from pydantic_v1 import Field
-
 from langchain.agents.agent import Agent, AgentOutputParser
 from langchain.agents.agent_types import AgentType
 from langchain.agents.conversational.output_parser import ConvoOutputParser
@@ -13,6 +11,7 @@ from langchain.agents.utils import validate_tools_single_input
 from langchain.callbacks.base import BaseCallbackManager
 from langchain.chains import LLMChain
 from langchain.prompts import PromptTemplate
+from langchain.pydantic_v1 import Field
 from langchain.schema.language_model import BaseLanguageModel
 from langchain.tools.base import BaseTool
 

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

@@ -3,8 +3,6 @@ from __future__ import annotations
 
 from typing import Any, List, Optional, Sequence, Tuple
 
-from pydantic_v1 import Field
-
 from langchain.agents.agent import Agent, AgentOutputParser
 from langchain.agents.conversational_chat.output_parser import ConvoOutputParser
 from langchain.agents.conversational_chat.prompt import (
@@ -21,6 +19,7 @@ from langchain.prompts.chat import (
     MessagesPlaceholder,
     SystemMessagePromptTemplate,
 )
+from langchain.pydantic_v1 import Field
 from langchain.schema import AgentAction, BaseOutputParser, BasePromptTemplate
 from langchain.schema.language_model import BaseLanguageModel
 from langchain.schema.messages import AIMessage, BaseMessage, HumanMessage

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

@@ -3,8 +3,6 @@ from __future__ import annotations
 
 from typing import Any, Callable, List, NamedTuple, Optional, Sequence
 
-from pydantic_v1 import Field
-
 from langchain.agents.agent import Agent, AgentExecutor, AgentOutputParser
 from langchain.agents.agent_types import AgentType
 from langchain.agents.mrkl.output_parser import MRKLOutputParser
@@ -14,6 +12,7 @@ from langchain.agents.utils import validate_tools_single_input
 from langchain.callbacks.base import BaseCallbackManager
 from langchain.chains import LLMChain
 from langchain.prompts import PromptTemplate
+from langchain.pydantic_v1 import Field
 from langchain.schema.language_model import BaseLanguageModel
 from langchain.tools.base import BaseTool
 

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

@@ -4,8 +4,6 @@ from dataclasses import dataclass
 from json import JSONDecodeError
 from typing import Any, List, Optional, Sequence, Tuple, Union
 
-from pydantic_v1 import root_validator
-
 from langchain.agents import BaseSingleActionAgent
 from langchain.callbacks.base import BaseCallbackManager
 from langchain.callbacks.manager import Callbacks
@@ -16,6 +14,7 @@ from langchain.prompts.chat import (
     HumanMessagePromptTemplate,
     MessagesPlaceholder,
 )
+from langchain.pydantic_v1 import root_validator
 from langchain.schema import (
     AgentAction,
     AgentFinish,

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

@@ -4,8 +4,6 @@ from dataclasses import dataclass
 from json import JSONDecodeError
 from typing import Any, List, Optional, Sequence, Tuple, Union
 
-from pydantic_v1 import root_validator
-
 from langchain.agents import BaseMultiActionAgent
 from langchain.callbacks.base import BaseCallbackManager
 from langchain.callbacks.manager import Callbacks
@@ -16,6 +14,7 @@ from langchain.prompts.chat import (
     HumanMessagePromptTemplate,
     MessagesPlaceholder,
 )
+from langchain.pydantic_v1 import root_validator
 from langchain.schema import (
     AgentAction,
     AgentFinish,

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

@@ -1,8 +1,6 @@
 """Chain that implements the ReAct paper from https://arxiv.org/pdf/2210.03629.pdf."""
 from typing import Any, List, Optional, Sequence
 
-from pydantic_v1 import Field
-
 from langchain.agents.agent import Agent, AgentExecutor, AgentOutputParser
 from langchain.agents.agent_types import AgentType
 from langchain.agents.react.output_parser import ReActOutputParser
@@ -12,6 +10,7 @@ from langchain.agents.tools import Tool
 from langchain.agents.utils import validate_tools_single_input
 from langchain.docstore.base import Docstore
 from langchain.docstore.document import Document
+from langchain.pydantic_v1 import Field
 from langchain.schema import BasePromptTemplate
 from langchain.schema.language_model import BaseLanguageModel
 from langchain.tools.base import BaseTool

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

@@ -1,14 +1,13 @@
 """Chain that does self-ask with search."""
 from typing import Any, Sequence, Union
 
-from pydantic_v1 import Field
-
 from langchain.agents.agent import Agent, AgentExecutor, AgentOutputParser
 from langchain.agents.agent_types import AgentType
 from langchain.agents.self_ask_with_search.output_parser import SelfAskOutputParser
 from langchain.agents.self_ask_with_search.prompt import PROMPT
 from langchain.agents.tools import Tool
 from langchain.agents.utils import validate_tools_single_input
+from langchain.pydantic_v1 import Field
 from langchain.schema import BasePromptTemplate
 from langchain.schema.language_model import BaseLanguageModel
 from langchain.tools.base import BaseTool

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

@@ -1,8 +1,6 @@
 import re
 from typing import Any, List, Optional, Sequence, Tuple
 
-from pydantic_v1 import Field
-
 from langchain.agents.agent import Agent, AgentOutputParser
 from langchain.agents.structured_chat.output_parser import (
     StructuredChatOutputParserWithRetries,
@@ -15,6 +13,7 @@ from langchain.prompts.chat import (
     HumanMessagePromptTemplate,
     SystemMessagePromptTemplate,
 )
+from langchain.pydantic_v1 import Field
 from langchain.schema import AgentAction, BasePromptTemplate
 from langchain.schema.language_model import BaseLanguageModel
 from langchain.tools import BaseTool

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

@@ -5,11 +5,10 @@ import logging
 import re
 from typing import Optional, Union
 
-from pydantic_v1 import Field
-
 from langchain.agents.agent import AgentOutputParser
 from langchain.agents.structured_chat.prompt import FORMAT_INSTRUCTIONS
 from langchain.output_parsers import OutputFixingParser
+from langchain.pydantic_v1 import Field
 from langchain.schema import AgentAction, AgentFinish, OutputParserException
 from langchain.schema.language_model import BaseLanguageModel
 

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

@@ -8,8 +8,8 @@ from uuid import UUID
 
 from langsmith.schemas import RunBase as BaseRunV2
 from langsmith.schemas import RunTypeEnum as RunTypeEnumDep
-from pydantic_v1 import BaseModel, Field, root_validator
 
+from langchain.pydantic_v1 import BaseModel, Field, root_validator
 from langchain.schema import LLMResult
 
 

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

@@ -3,8 +3,6 @@ from __future__ import annotations
 
 from typing import Any, Dict, List, Optional
 
-from pydantic_v1 import Field, root_validator
-
 from langchain.callbacks.manager import (
     AsyncCallbackManagerForChainRun,
     CallbackManagerForChainRun,
@@ -12,6 +10,7 @@ from langchain.callbacks.manager import (
 from langchain.chains.api.prompt import API_RESPONSE_PROMPT, API_URL_PROMPT
 from langchain.chains.base import Chain
 from langchain.chains.llm import LLMChain
+from langchain.pydantic_v1 import Field, root_validator
 from langchain.schema import BasePromptTemplate
 from langchain.schema.language_model import BaseLanguageModel
 from langchain.utilities.requests import TextRequestsWrapper

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

@@ -4,7 +4,6 @@ from __future__ import annotations
 import json
 from typing import Any, Dict, List, NamedTuple, Optional, cast
 
-from pydantic_v1 import BaseModel, Field
 from requests import Response
 
 from langchain.callbacks.manager import CallbackManagerForChainRun, Callbacks
@@ -12,6 +11,7 @@ from langchain.chains.api.openapi.requests_chain import APIRequesterChain
 from langchain.chains.api.openapi.response_chain import APIResponderChain
 from langchain.chains.base import Chain
 from langchain.chains.llm import LLMChain
+from langchain.pydantic_v1 import BaseModel, Field
 from langchain.schema.language_model import BaseLanguageModel
 from langchain.tools.openapi.utils.api_models import APIOperation
 from langchain.utilities.requests import Requests

libs/langchain/langchain/chains/base.py

@@ -10,7 +10,6 @@ from pathlib import Path
 from typing import Any, Dict, List, Optional, Union
 
 import yaml
-from pydantic_v1 import Field, root_validator, validator
 
 import langchain
 from langchain.callbacks.base import BaseCallbackManager
@@ -23,6 +22,7 @@ from langchain.callbacks.manager import (
 )
 from langchain.load.dump import dumpd
 from langchain.load.serializable import Serializable
+from langchain.pydantic_v1 import Field, root_validator, validator
 from langchain.schema import RUN_KEY, BaseMemory, RunInfo
 from langchain.schema.runnable import Runnable, RunnableConfig
 

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

@@ -3,14 +3,13 @@
 from abc import ABC, abstractmethod
 from typing import Any, Dict, List, Optional, Tuple
 
-from pydantic_v1 import Field
-
 from langchain.callbacks.manager import (
     AsyncCallbackManagerForChainRun,
     CallbackManagerForChainRun,
 )
 from langchain.chains.base import Chain
 from langchain.docstore.document import Document
+from langchain.pydantic_v1 import Field
 from langchain.text_splitter import RecursiveCharacterTextSplitter, TextSplitter
 
 

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

@@ -4,13 +4,12 @@ from __future__ import annotations
 
 from typing import Any, Dict, List, Optional, Tuple
 
-from pydantic_v1 import Extra, root_validator
-
 from langchain.callbacks.manager import Callbacks
 from langchain.chains.combine_documents.base import BaseCombineDocumentsChain
 from langchain.chains.combine_documents.reduce import ReduceDocumentsChain
 from langchain.chains.llm import LLMChain
 from langchain.docstore.document import Document
+from langchain.pydantic_v1 import Extra, root_validator
 
 
 class MapReduceDocumentsChain(BaseCombineDocumentsChain):

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

@@ -4,13 +4,12 @@ from __future__ import annotations
 
 from typing import Any, Dict, List, Optional, Sequence, Tuple, Union, cast
 
-from pydantic_v1 import Extra, root_validator
-
 from langchain.callbacks.manager import Callbacks
 from langchain.chains.combine_documents.base import BaseCombineDocumentsChain
 from langchain.chains.llm import LLMChain
 from langchain.docstore.document import Document
 from langchain.output_parsers.regex import RegexParser
+from langchain.pydantic_v1 import Extra, root_validator
 
 
 class MapRerankDocumentsChain(BaseCombineDocumentsChain):

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

@@ -4,11 +4,10 @@ from __future__ import annotations
 
 from typing import Any, Callable, List, Optional, Protocol, Tuple
 
-from pydantic_v1 import Extra
-
 from langchain.callbacks.manager import Callbacks
 from langchain.chains.combine_documents.base import BaseCombineDocumentsChain
 from langchain.docstore.document import Document
+from langchain.pydantic_v1 import Extra
 
 
 class CombineDocsProtocol(Protocol):

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

@@ -4,8 +4,6 @@ from __future__ import annotations
 
 from typing import Any, Dict, List, Tuple
 
-from pydantic_v1 import Extra, Field, root_validator
-
 from langchain.callbacks.manager import Callbacks
 from langchain.chains.combine_documents.base import (
     BaseCombineDocumentsChain,
@@ -13,6 +11,7 @@ from langchain.chains.combine_documents.base import (
 from langchain.chains.llm import LLMChain
 from langchain.docstore.document import Document
 from langchain.prompts.prompt import PromptTemplate
+from langchain.pydantic_v1 import Extra, Field, root_validator
 from langchain.schema import BasePromptTemplate, format_document
 
 

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

@@ -2,8 +2,6 @@
 
 from typing import Any, Dict, List, Optional, Tuple
 
-from pydantic_v1 import Extra, Field, root_validator
-
 from langchain.callbacks.manager import Callbacks
 from langchain.chains.combine_documents.base import (
     BaseCombineDocumentsChain,
@@ -11,6 +9,7 @@ from langchain.chains.combine_documents.base import (
 from langchain.chains.llm import LLMChain
 from langchain.docstore.document import Document
 from langchain.prompts.prompt import PromptTemplate
+from langchain.pydantic_v1 import Extra, Field, root_validator
 from langchain.schema import BasePromptTemplate, format_document
 
 

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

@@ -1,5 +1,5 @@
 """Models for the Constitutional AI chain."""
-from pydantic_v1 import BaseModel
+from langchain.pydantic_v1 import BaseModel
 
 
 class ConstitutionalPrinciple(BaseModel):

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

@@ -1,11 +1,10 @@
 """Chain that carries on a conversation and calls an LLM."""
 from typing import Dict, List
 
-from pydantic_v1 import Extra, Field, root_validator
-
 from langchain.chains.conversation.prompt import PROMPT
 from langchain.chains.llm import LLMChain
 from langchain.memory.buffer import ConversationBufferMemory
+from langchain.pydantic_v1 import Extra, Field, root_validator
 from langchain.schema import BaseMemory, BasePromptTemplate
 
 

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

@@ -7,8 +7,6 @@ from abc import abstractmethod
 from pathlib import Path
 from typing import Any, Callable, Dict, List, Optional, Tuple, Union
 
-from pydantic_v1 import Extra, Field, root_validator
-
 from langchain.callbacks.manager import (
     AsyncCallbackManagerForChainRun,
     CallbackManagerForChainRun,
@@ -20,6 +18,7 @@ from langchain.chains.combine_documents.stuff import StuffDocumentsChain
 from langchain.chains.conversational_retrieval.prompts import CONDENSE_QUESTION_PROMPT
 from langchain.chains.llm import LLMChain
 from langchain.chains.question_answering import load_qa_chain
+from langchain.pydantic_v1 import Extra, Field, root_validator
 from langchain.schema import BasePromptTemplate, BaseRetriever, Document
 from langchain.schema.language_model import BaseLanguageModel
 from langchain.schema.messages import BaseMessage

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

@@ -3,13 +3,12 @@ from __future__ import annotations
 
 from typing import TYPE_CHECKING, Any, Dict, List, Optional
 
-from pydantic_v1 import Extra, root_validator
-
 from langchain.callbacks.manager import CallbackManagerForChainRun
 from langchain.chains.base import Chain
 from langchain.chains.elasticsearch_database.prompts import ANSWER_PROMPT, DSL_PROMPT
 from langchain.chains.llm import LLMChain
 from langchain.output_parsers.json import SimpleJsonOutputParser
+from langchain.pydantic_v1 import Extra, root_validator
 from langchain.schema import BaseLLMOutputParser, BasePromptTemplate
 from langchain.schema.language_model import BaseLanguageModel
 

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

@@ -5,7 +5,6 @@ from abc import abstractmethod
 from typing import Any, Dict, List, Optional, Sequence, Tuple
 
 import numpy as np
-from pydantic_v1 import Field
 
 from langchain.callbacks.manager import (
     CallbackManagerForChainRun,
@@ -18,6 +17,7 @@ from langchain.chains.flare.prompts import (
 )
 from langchain.chains.llm import LLMChain
 from langchain.llms import OpenAI
+from langchain.pydantic_v1 import Field
 from langchain.schema import BasePromptTemplate, BaseRetriever, Generation
 from langchain.schema.language_model import BaseLanguageModel
 

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

@@ -4,8 +4,6 @@ from __future__ import annotations
 import re
 from typing import Any, Dict, List, Optional
 
-from pydantic_v1 import Field
-
 from langchain.base_language import BaseLanguageModel
 from langchain.callbacks.manager import CallbackManagerForChainRun
 from langchain.chains.base import Chain
@@ -16,6 +14,7 @@ from langchain.chains.graph_qa.prompts import (
 )
 from langchain.chains.llm import LLMChain
 from langchain.graphs.arangodb_graph import ArangoGraph
+from langchain.pydantic_v1 import Field
 from langchain.schema import BasePromptTemplate
 
 

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

@@ -3,13 +3,12 @@ from __future__ import annotations
 
 from typing import Any, Dict, List, Optional
 
-from pydantic_v1 import Field
-
 from langchain.callbacks.manager import CallbackManagerForChainRun
 from langchain.chains.base import Chain
 from langchain.chains.graph_qa.prompts import ENTITY_EXTRACTION_PROMPT, GRAPH_QA_PROMPT
 from langchain.chains.llm import LLMChain
 from langchain.graphs.networkx_graph import NetworkxEntityGraph, get_entities
+from langchain.pydantic_v1 import Field
 from langchain.schema import BasePromptTemplate
 from langchain.schema.language_model import BaseLanguageModel
 

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

@@ -4,13 +4,12 @@ from __future__ import annotations
 import re
 from typing import Any, Dict, List, Optional
 
-from pydantic_v1 import Field
-
 from langchain.callbacks.manager import CallbackManagerForChainRun
 from langchain.chains.base import Chain
 from langchain.chains.graph_qa.prompts import CYPHER_GENERATION_PROMPT, CYPHER_QA_PROMPT
 from langchain.chains.llm import LLMChain
 from langchain.graphs.neo4j_graph import Neo4jGraph
+from langchain.pydantic_v1 import Field
 from langchain.schema import BasePromptTemplate
 from langchain.schema.language_model import BaseLanguageModel
 

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

@@ -3,8 +3,6 @@ from __future__ import annotations
 
 from typing import Any, Dict, List, Optional
 
-from pydantic_v1 import Field
-
 from langchain.callbacks.manager import CallbackManagerForChainRun
 from langchain.chains.base import Chain
 from langchain.chains.graph_qa.prompts import (
@@ -13,6 +11,7 @@ from langchain.chains.graph_qa.prompts import (
 )
 from langchain.chains.llm import LLMChain
 from langchain.graphs.hugegraph import HugeGraph
+from langchain.pydantic_v1 import Field
 from langchain.schema import BasePromptTemplate
 from langchain.schema.language_model import BaseLanguageModel
 

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

@@ -3,13 +3,12 @@ from __future__ import annotations
 
 from typing import Any, Dict, List, Optional
 
-from pydantic_v1 import Field
-
 from langchain.callbacks.manager import CallbackManagerForChainRun
 from langchain.chains.base import Chain
 from langchain.chains.graph_qa.prompts import CYPHER_QA_PROMPT, KUZU_GENERATION_PROMPT
 from langchain.chains.llm import LLMChain
 from langchain.graphs.kuzu_graph import KuzuGraph
+from langchain.pydantic_v1 import Field
 from langchain.schema import BasePromptTemplate
 from langchain.schema.language_model import BaseLanguageModel
 

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

@@ -3,13 +3,12 @@ from __future__ import annotations
 
 from typing import Any, Dict, List, Optional
 
-from pydantic_v1 import Field
-
 from langchain.callbacks.manager import CallbackManagerForChainRun
 from langchain.chains.base import Chain
 from langchain.chains.graph_qa.prompts import CYPHER_QA_PROMPT, NGQL_GENERATION_PROMPT
 from langchain.chains.llm import LLMChain
 from langchain.graphs.nebula_graph import NebulaGraph
+from langchain.pydantic_v1 import Field
 from langchain.schema import BasePromptTemplate
 from langchain.schema.language_model import BaseLanguageModel
 

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

@@ -3,8 +3,6 @@ from __future__ import annotations
 import re
 from typing import Any, Dict, List, Optional
 
-from pydantic_v1 import Field
-
 from langchain.base_language import BaseLanguageModel
 from langchain.callbacks.manager import CallbackManagerForChainRun
 from langchain.chains.base import Chain
@@ -15,6 +13,7 @@ from langchain.chains.graph_qa.prompts import (
 from langchain.chains.llm import LLMChain
 from langchain.graphs import NeptuneGraph
 from langchain.prompts.base import BasePromptTemplate
+from langchain.pydantic_v1 import Field
 
 INTERMEDIATE_STEPS_KEY = "intermediate_steps"
 

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

@@ -5,8 +5,6 @@ from __future__ import annotations
 
 from typing import Any, Dict, List, Optional
 
-from pydantic_v1 import Field
-
 from langchain.callbacks.manager import CallbackManagerForChainRun
 from langchain.chains.base import Chain
 from langchain.chains.graph_qa.prompts import (
@@ -18,6 +16,7 @@ from langchain.chains.graph_qa.prompts import (
 from langchain.chains.llm import LLMChain
 from langchain.graphs.rdf_graph import RdfGraph
 from langchain.prompts.base import BasePromptTemplate
+from langchain.pydantic_v1 import Field
 from langchain.schema.language_model import BaseLanguageModel
 
 

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

@@ -7,13 +7,13 @@ from __future__ import annotations
 from typing import Any, Dict, List, Optional
 
 import numpy as np
-from pydantic_v1 import Extra
 
 from langchain.callbacks.manager import CallbackManagerForChainRun
 from langchain.chains.base import Chain
 from langchain.chains.hyde.prompts import PROMPT_MAP
 from langchain.chains.llm import LLMChain
 from langchain.embeddings.base import Embeddings
+from langchain.pydantic_v1 import Extra
 from langchain.schema.language_model import BaseLanguageModel
 
 

libs/langchain/langchain/chains/llm.py

@@ -4,8 +4,6 @@ from __future__ import annotations
 import warnings
 from typing import Any, Dict, List, Optional, Sequence, Tuple, Union
 
-from pydantic_v1 import Extra, Field
-
 from langchain.callbacks.manager import (
     AsyncCallbackManager,
     AsyncCallbackManagerForChainRun,
@@ -16,6 +14,7 @@ from langchain.callbacks.manager import (
 from langchain.chains.base import Chain
 from langchain.load.dump import dumpd
 from langchain.prompts.prompt import PromptTemplate
+from langchain.pydantic_v1 import Extra, Field
 from langchain.schema import (
     BaseLLMOutputParser,
     BasePromptTemplate,

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

@@ -5,12 +5,11 @@ import logging
 import warnings
 from typing import Any, Dict, List, Optional
 
-from pydantic_v1 import Extra, Field, root_validator
-
 from langchain.callbacks.manager import CallbackManagerForChainRun
 from langchain.chains.base import Chain
 from langchain.chains.llm import LLMChain
 from langchain.chains.llm_bash.prompt import PROMPT
+from langchain.pydantic_v1 import Extra, Field, root_validator
 from langchain.schema import BasePromptTemplate, OutputParserException
 from langchain.schema.language_model import BaseLanguageModel
 from langchain.utilities.bash import BashProcess

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

@@ -4,8 +4,6 @@ from __future__ import annotations
 import warnings
 from typing import Any, Dict, List, Optional
 
-from pydantic_v1 import Extra, root_validator
-
 from langchain.callbacks.manager import CallbackManagerForChainRun
 from langchain.chains.base import Chain
 from langchain.chains.llm import LLMChain
@@ -17,6 +15,7 @@ from langchain.chains.llm_checker.prompt import (
 )
 from langchain.chains.sequential import SequentialChain
 from langchain.prompts import PromptTemplate
+from langchain.pydantic_v1 import Extra, root_validator
 from langchain.schema.language_model import BaseLanguageModel
 
 

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

@@ -7,7 +7,6 @@ import warnings
 from typing import Any, Dict, List, Optional
 
 import numexpr
-from pydantic_v1 import Extra, root_validator
 
 from langchain.callbacks.manager import (
     AsyncCallbackManagerForChainRun,
@@ -16,6 +15,7 @@ from langchain.callbacks.manager import (
 from langchain.chains.base import Chain
 from langchain.chains.llm import LLMChain
 from langchain.chains.llm_math.prompt import PROMPT
+from langchain.pydantic_v1 import Extra, root_validator
 from langchain.schema import BasePromptTemplate
 from langchain.schema.language_model import BaseLanguageModel
 

libs/langchain/langchain/chains/llm_requests.py

@@ -3,11 +3,10 @@ from __future__ import annotations
 
 from typing import Any, Dict, List, Optional
 
-from pydantic_v1 import Extra, Field, root_validator
-
 from langchain.callbacks.manager import CallbackManagerForChainRun
 from langchain.chains import LLMChain
 from langchain.chains.base import Chain
+from langchain.pydantic_v1 import Extra, Field, root_validator
 from langchain.utilities.requests import TextRequestsWrapper
 
 DEFAULT_HEADERS = {

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

@@ -6,13 +6,12 @@ import warnings
 from pathlib import Path
 from typing import Any, Dict, List, Optional
 
-from pydantic_v1 import Extra, root_validator
-
 from langchain.callbacks.manager import CallbackManagerForChainRun
 from langchain.chains.base import Chain
 from langchain.chains.llm import LLMChain
 from langchain.chains.sequential import SequentialChain
 from langchain.prompts.prompt import PromptTemplate
+from langchain.pydantic_v1 import Extra, root_validator
 from langchain.schema.language_model import BaseLanguageModel
 
 PROMPTS_DIR = Path(__file__).parent / "prompts"

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

@@ -4,8 +4,6 @@ from __future__ import annotations
 import re
 from typing import Any, Dict, List, Optional
 
-from pydantic_v1 import Extra
-
 from langchain.base_language import BaseLanguageModel
 from langchain.callbacks.manager import (
     AsyncCallbackManagerForChainRun,
@@ -15,6 +13,7 @@ from langchain.chains.base import Chain
 from langchain.chains.llm import LLMChain
 from langchain.chains.llm_symbolic_math.prompt import PROMPT
 from langchain.prompts.base import BasePromptTemplate
+from langchain.pydantic_v1 import Extra
 
 
 class LLMSymbolicMathChain(Chain):

libs/langchain/langchain/chains/mapreduce.py

@@ -7,8 +7,6 @@ from __future__ import annotations
 
 from typing import Any, Dict, List, Mapping, Optional
 
-from pydantic_v1 import Extra
-
 from langchain.callbacks.manager import CallbackManagerForChainRun, Callbacks
 from langchain.chains import ReduceDocumentsChain
 from langchain.chains.base import Chain
@@ -17,6 +15,7 @@ from langchain.chains.combine_documents.map_reduce import MapReduceDocumentsChai
 from langchain.chains.combine_documents.stuff import StuffDocumentsChain
 from langchain.chains.llm import LLMChain
 from langchain.docstore.document import Document
+from langchain.pydantic_v1 import Extra
 from langchain.schema import BasePromptTemplate
 from langchain.schema.language_model import BaseLanguageModel
 from langchain.text_splitter import TextSplitter

libs/langchain/langchain/chains/moderation.py

@@ -1,10 +1,9 @@
 """Pass input through a moderation endpoint."""
 from typing import Any, Dict, List, Optional
 
-from pydantic_v1 import root_validator
-
 from langchain.callbacks.manager import CallbackManagerForChainRun
 from langchain.chains.base import Chain
+from langchain.pydantic_v1 import root_validator
 from langchain.utils import get_from_dict_or_env
 
 

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

@@ -4,13 +4,12 @@ from __future__ import annotations
 import warnings
 from typing import Any, Dict, List, Optional
 
-from pydantic_v1 import Extra, root_validator
-
 from langchain.callbacks.manager import CallbackManagerForChainRun
 from langchain.chains.base import Chain
 from langchain.chains.llm import LLMChain
 from langchain.chains.natbot.prompt import PROMPT
 from langchain.llms.openai import OpenAI
+from langchain.pydantic_v1 import Extra, root_validator
 from langchain.schema.language_model import BaseLanguageModel
 
 

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

@@ -12,8 +12,6 @@ from typing import (
     Union,
 )
 
-from pydantic_v1 import BaseModel
-
 from langchain.base_language import BaseLanguageModel
 from langchain.chains import LLMChain
 from langchain.output_parsers.openai_functions import (
@@ -22,6 +20,7 @@ from langchain.output_parsers.openai_functions import (
     PydanticOutputFunctionsParser,
 )
 from langchain.prompts import BasePromptTemplate
+from langchain.pydantic_v1 import BaseModel
 from langchain.schema import BaseLLMOutputParser
 
 PYTHON_TO_JSON_TYPES = {

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

@@ -1,13 +1,12 @@
 from typing import Iterator, List
 
-from pydantic_v1 import BaseModel, Field
-
 from langchain.chains.llm import LLMChain
 from langchain.chains.openai_functions.utils import get_llm_kwargs
 from langchain.output_parsers.openai_functions import (
     PydanticOutputFunctionsParser,
 )
 from langchain.prompts.chat import ChatPromptTemplate, HumanMessagePromptTemplate
+from langchain.pydantic_v1 import BaseModel, Field
 from langchain.schema.language_model import BaseLanguageModel
 from langchain.schema.messages import HumanMessage, SystemMessage
 

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

@@ -1,7 +1,5 @@
 from typing import Any, List
 
-from pydantic_v1 import BaseModel
-
 from langchain.chains.base import Chain
 from langchain.chains.llm import LLMChain
 from langchain.chains.openai_functions.utils import (
@@ -14,6 +12,7 @@ from langchain.output_parsers.openai_functions import (
     PydanticAttrOutputFunctionsParser,
 )
 from langchain.prompts import ChatPromptTemplate
+from langchain.pydantic_v1 import BaseModel
 from langchain.schema.language_model import BaseLanguageModel
 
 

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

@@ -1,7 +1,5 @@
 from typing import Any, List, Optional, Type, Union
 
-from pydantic_v1 import BaseModel, Field
-
 from langchain.chains.llm import LLMChain
 from langchain.chains.openai_functions.utils import get_llm_kwargs
 from langchain.output_parsers.openai_functions import (
@@ -10,6 +8,7 @@ from langchain.output_parsers.openai_functions import (
 )
 from langchain.prompts import PromptTemplate
 from langchain.prompts.chat import ChatPromptTemplate, HumanMessagePromptTemplate
+from langchain.pydantic_v1 import BaseModel, Field
 from langchain.schema import BaseLLMOutputParser
 from langchain.schema.language_model import BaseLanguageModel
 from langchain.schema.messages import HumanMessage, SystemMessage

libs/langchain/langchain/chains/prompt_selector.py

@@ -1,10 +1,9 @@
 from abc import ABC, abstractmethod
 from typing import Callable, List, Tuple
 
-from pydantic_v1 import BaseModel, Field
-
 from langchain.chat_models.base import BaseChatModel
 from langchain.llms.base import BaseLLM
+from langchain.pydantic_v1 import BaseModel, Field
 from langchain.schema import BasePromptTemplate
 from langchain.schema.language_model import BaseLanguageModel
 

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

@@ -3,12 +3,11 @@ from __future__ import annotations
 import json
 from typing import Any, Dict, List, Optional
 
-from pydantic_v1 import Field
-
 from langchain.callbacks.manager import CallbackManagerForChainRun
 from langchain.chains.base import Chain
 from langchain.chains.llm import LLMChain
 from langchain.chains.qa_generation.prompt import PROMPT_SELECTOR
+from langchain.pydantic_v1 import Field
 from langchain.schema import BasePromptTemplate
 from langchain.schema.language_model import BaseLanguageModel
 from langchain.text_splitter import RecursiveCharacterTextSplitter, TextSplitter

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

@@ -7,8 +7,6 @@ import re
 from abc import ABC, abstractmethod
 from typing import Any, Dict, List, Optional, Tuple
 
-from pydantic_v1 import Extra, root_validator
-
 from langchain.callbacks.manager import (
     AsyncCallbackManagerForChainRun,
     CallbackManagerForChainRun,
@@ -26,6 +24,7 @@ from langchain.chains.qa_with_sources.map_reduce_prompt import (
     QUESTION_PROMPT,
 )
 from langchain.docstore.document import Document
+from langchain.pydantic_v1 import Extra, root_validator
 from langchain.schema import BasePromptTemplate
 from langchain.schema.language_model import BaseLanguageModel
 

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

@@ -2,8 +2,6 @@
 
 from typing import Any, Dict, List
 
-from pydantic_v1 import Field
-
 from langchain.callbacks.manager import (
     AsyncCallbackManagerForChainRun,
     CallbackManagerForChainRun,
@@ -11,6 +9,7 @@ from langchain.callbacks.manager import (
 from langchain.chains.combine_documents.stuff import StuffDocumentsChain
 from langchain.chains.qa_with_sources.base import BaseQAWithSourcesChain
 from langchain.docstore.document import Document
+from langchain.pydantic_v1 import Field
 from langchain.schema import BaseRetriever
 
 

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

@@ -3,8 +3,6 @@
 import warnings
 from typing import Any, Dict, List
 
-from pydantic_v1 import Field, root_validator
-
 from langchain.callbacks.manager import (
     AsyncCallbackManagerForChainRun,
     CallbackManagerForChainRun,
@@ -12,6 +10,7 @@ from langchain.callbacks.manager import (
 from langchain.chains.combine_documents.stuff import StuffDocumentsChain
 from langchain.chains.qa_with_sources.base import BaseQAWithSourcesChain
 from langchain.docstore.document import Document
+from langchain.pydantic_v1 import Field, root_validator
 from langchain.vectorstores.base import VectorStore
 
 

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

@@ -5,7 +5,7 @@ from abc import ABC, abstractmethod
 from enum import Enum
 from typing import Any, List, Optional, Sequence, Union
 
-from pydantic_v1 import BaseModel
+from langchain.pydantic_v1 import BaseModel
 
 
 class Visitor(ABC):

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

@@ -1,4 +1,4 @@
-from pydantic_v1 import BaseModel
+from langchain.pydantic_v1 import BaseModel
 
 
 class AttributeInfo(BaseModel):

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

@@ -6,8 +6,6 @@ import warnings
 from abc import abstractmethod
 from typing import Any, Dict, List, Optional
 
-from pydantic_v1 import Extra, Field, root_validator
-
 from langchain.callbacks.manager import (
     AsyncCallbackManagerForChainRun,
     CallbackManagerForChainRun,
@@ -20,6 +18,7 @@ from langchain.chains.llm import LLMChain
 from langchain.chains.question_answering import load_qa_chain
 from langchain.chains.question_answering.stuff_prompt import PROMPT_SELECTOR
 from langchain.prompts import PromptTemplate
+from langchain.pydantic_v1 import Extra, Field, root_validator
 from langchain.schema import BaseRetriever, Document
 from langchain.schema.language_model import BaseLanguageModel
 from langchain.vectorstores.base import VectorStore

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

@@ -4,14 +4,13 @@ from __future__ import annotations
 from abc import ABC
 from typing import Any, Dict, List, Mapping, NamedTuple, Optional
 
-from pydantic_v1 import Extra
-
 from langchain.callbacks.manager import (
     AsyncCallbackManagerForChainRun,
     CallbackManagerForChainRun,
     Callbacks,
 )
 from langchain.chains.base import Chain
+from langchain.pydantic_v1 import Extra
 
 
 class Route(NamedTuple):

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

@@ -2,12 +2,11 @@ from __future__ import annotations
 
 from typing import Any, Dict, List, Optional, Sequence, Tuple, Type
 
-from pydantic_v1 import Extra
-
 from langchain.callbacks.manager import CallbackManagerForChainRun
 from langchain.chains.router.base import RouterChain
 from langchain.docstore.document import Document
 from langchain.embeddings.base import Embeddings
+from langchain.pydantic_v1 import Extra
 from langchain.vectorstores.base import VectorStore
 
 

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

@@ -3,8 +3,6 @@ from __future__ import annotations
 
 from typing import Any, Dict, List, Optional, Type, cast
 
-from pydantic_v1 import root_validator
-
 from langchain.callbacks.manager import (
     AsyncCallbackManagerForChainRun,
     CallbackManagerForChainRun,
@@ -12,6 +10,7 @@ from langchain.callbacks.manager import (
 from langchain.chains import LLMChain
 from langchain.chains.router.base import RouterChain
 from langchain.output_parsers.json import parse_and_check_json_markdown
+from langchain.pydantic_v1 import root_validator
 from langchain.schema import BaseOutputParser, BasePromptTemplate, OutputParserException
 from langchain.schema.language_model import BaseLanguageModel
 

libs/langchain/langchain/chains/sequential.py

@@ -1,13 +1,12 @@
 """Chain pipeline where the outputs of one step feed directly into next."""
 from typing import Any, Dict, List, Optional
 
-from pydantic_v1 import Extra, root_validator
-
 from langchain.callbacks.manager import (
     AsyncCallbackManagerForChainRun,
     CallbackManagerForChainRun,
 )
 from langchain.chains.base import Chain
+from langchain.pydantic_v1 import Extra, root_validator
 from langchain.utils.input import get_color_mapping
 
 

libs/langchain/langchain/chains/transform.py

@@ -3,13 +3,12 @@ import functools
 import logging
 from typing import Any, Awaitable, Callable, Dict, List, Optional
 
-from pydantic_v1 import Field
-
 from langchain.callbacks.manager import (
     AsyncCallbackManagerForChainRun,
     CallbackManagerForChainRun,
 )
 from langchain.chains.base import Chain
+from langchain.pydantic_v1 import Field
 
 logger = logging.getLogger(__name__)