langchain 0.2.16__py3-none-any.whl → 0.3.0__py3-none-any.whl

langchain/chains/combine_documents/reduce.py

@@ -6,6 +6,7 @@ from typing import Any, Callable, List, Optional, Protocol, Tuple
 
 from langchain_core.callbacks import Callbacks
 from langchain_core.documents import Document
+from pydantic import ConfigDict
 
 from langchain.chains.combine_documents.base import BaseCombineDocumentsChain
 
@@ -204,9 +205,10 @@ class ReduceDocumentsChain(BaseCombineDocumentsChain):
     If None, it will keep trying to collapse documents to fit token_max.
     Otherwise, after it reaches the max number, it will throw an error"""
 
-    class Config:
-        arbitrary_types_allowed = True
-        extra = "forbid"
+    model_config = ConfigDict(
+        arbitrary_types_allowed=True,
+        extra="forbid",
+    )
 
     @property
     def _collapse_chain(self) -> BaseCombineDocumentsChain:

langchain/chains/combine_documents/refine.py

@@ -8,7 +8,7 @@ from langchain_core.callbacks import Callbacks
 from langchain_core.documents import Document
 from langchain_core.prompts import BasePromptTemplate, format_document
 from langchain_core.prompts.prompt import PromptTemplate
-from langchain_core.pydantic_v1 import Field, root_validator
+from pydantic import ConfigDict, Field, model_validator
 
 from langchain.chains.combine_documents.base import (
     BaseCombineDocumentsChain,
@@ -98,20 +98,23 @@ class RefineDocumentsChain(BaseCombineDocumentsChain):
             _output_keys = _output_keys + ["intermediate_steps"]
         return _output_keys
 
-    class Config:
-        arbitrary_types_allowed = True
-        extra = "forbid"
+    model_config = ConfigDict(
+        arbitrary_types_allowed=True,
+        extra="forbid",
+    )
 
-    @root_validator(pre=True)
-    def get_return_intermediate_steps(cls, values: Dict) -> Dict:
+    @model_validator(mode="before")
+    @classmethod
+    def get_return_intermediate_steps(cls, values: Dict) -> Any:
         """For backwards compatibility."""
         if "return_refine_steps" in values:
             values["return_intermediate_steps"] = values["return_refine_steps"]
             del values["return_refine_steps"]
         return values
 
-    @root_validator(pre=True)
-    def get_default_document_variable_name(cls, values: Dict) -> Dict:
+    @model_validator(mode="before")
+    @classmethod
+    def get_default_document_variable_name(cls, values: Dict) -> Any:
         """Get default document variable name, if not provided."""
         if "initial_llm_chain" not in values:
             raise ValueError("initial_llm_chain must be provided")

langchain/chains/combine_documents/stuff.py

@@ -8,8 +8,8 @@ from langchain_core.documents import Document
 from langchain_core.language_models import LanguageModelLike
 from langchain_core.output_parsers import BaseOutputParser, StrOutputParser
 from langchain_core.prompts import BasePromptTemplate, format_document
-from langchain_core.pydantic_v1 import Field, root_validator
 from langchain_core.runnables import Runnable, RunnablePassthrough
+from pydantic import ConfigDict, Field, model_validator
 
 from langchain.chains.combine_documents.base import (
     DEFAULT_DOCUMENT_PROMPT,
@@ -156,12 +156,14 @@ class StuffDocumentsChain(BaseCombineDocumentsChain):
     document_separator: str = "\n\n"
     """The string with which to join the formatted documents"""
 
-    class Config:
-        arbitrary_types_allowed = True
-        extra = "forbid"
+    model_config = ConfigDict(
+        arbitrary_types_allowed=True,
+        extra="forbid",
+    )
 
-    @root_validator(pre=True)
-    def get_default_document_variable_name(cls, values: Dict) -> Dict:
+    @model_validator(mode="before")
+    @classmethod
+    def get_default_document_variable_name(cls, values: Dict) -> Any:
         """Get default document variable name, if not provided.
 
         If only one variable is present in the llm_chain.prompt,

langchain/chains/constitutional_ai/models.py

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

langchain/chains/conversation/base.py

@@ -1,11 +1,12 @@
 """Chain that carries on a conversation and calls an LLM."""
 
-from typing import Dict, List
+from typing import List
 
 from langchain_core._api import deprecated
 from langchain_core.memory import BaseMemory
 from langchain_core.prompts import BasePromptTemplate
-from langchain_core.pydantic_v1 import Field, root_validator
+from pydantic import ConfigDict, Field, model_validator
+from typing_extensions import Self
 
 from langchain.chains.conversation.prompt import PROMPT
 from langchain.chains.llm import LLMChain
@@ -110,9 +111,10 @@ class ConversationChain(LLMChain):
     input_key: str = "input"  #: :meta private:
     output_key: str = "response"  #: :meta private:
 
-    class Config:
-        arbitrary_types_allowed = True
-        extra = "forbid"
+    model_config = ConfigDict(
+        arbitrary_types_allowed=True,
+        extra="forbid",
+    )
 
     @classmethod
     def is_lc_serializable(cls) -> bool:
@@ -123,17 +125,17 @@ class ConversationChain(LLMChain):
         """Use this since so some prompt vars come from history."""
         return [self.input_key]
 
-    @root_validator(pre=False, skip_on_failure=True)
-    def validate_prompt_input_variables(cls, values: Dict) -> Dict:
+    @model_validator(mode="after")
+    def validate_prompt_input_variables(self) -> Self:
         """Validate that prompt input variables are consistent."""
-        memory_keys = values["memory"].memory_variables
-        input_key = values["input_key"]
+        memory_keys = self.memory.memory_variables
+        input_key = self.input_key
         if input_key in memory_keys:
             raise ValueError(
                 f"The input key {input_key} was also found in the memory keys "
                 f"({memory_keys}) - please provide keys that don't overlap."
             )
-        prompt_variables = values["prompt"].input_variables
+        prompt_variables = self.prompt.input_variables
         expected_keys = memory_keys + [input_key]
         if set(expected_keys) != set(prompt_variables):
             raise ValueError(
@@ -141,4 +143,4 @@ class ConversationChain(LLMChain):
                 f"{prompt_variables}, but got {memory_keys} as inputs from "
                 f"memory, and {input_key} as the normal input key."
             )
-        return values
+        return self

langchain/chains/conversational_retrieval/base.py

@@ -18,10 +18,10 @@ from langchain_core.documents import Document
 from langchain_core.language_models import BaseLanguageModel
 from langchain_core.messages import BaseMessage
 from langchain_core.prompts import BasePromptTemplate
-from langchain_core.pydantic_v1 import BaseModel, Field, root_validator
 from langchain_core.retrievers import BaseRetriever
 from langchain_core.runnables import RunnableConfig
 from langchain_core.vectorstores import VectorStore
+from pydantic import BaseModel, ConfigDict, Field, model_validator
 
 from langchain.chains.base import Chain
 from langchain.chains.combine_documents.base import BaseCombineDocumentsChain
@@ -92,14 +92,15 @@ class BaseConversationalRetrievalChain(Chain):
     get_chat_history: Optional[Callable[[List[CHAT_TURN_TYPE]], str]] = None
     """An optional function to get a string of the chat history.
     If None is provided, will use a default."""
-    response_if_no_docs_found: Optional[str]
+    response_if_no_docs_found: Optional[str] = None
     """If specified, the chain will return a fixed response if no docs 
     are found for the question. """
 
-    class Config:
-        allow_population_by_field_name = True
-        arbitrary_types_allowed = True
-        extra = "forbid"
+    model_config = ConfigDict(
+        populate_by_name=True,
+        arbitrary_types_allowed=True,
+        extra="forbid",
+    )
 
     @property
     def input_keys(self) -> List[str]:
@@ -482,8 +483,9 @@ class ChatVectorDBChain(BaseConversationalRetrievalChain):
     def _chain_type(self) -> str:
         return "chat-vector-db"
 
-    @root_validator(pre=True)
-    def raise_deprecation(cls, values: Dict) -> Dict:
+    @model_validator(mode="before")
+    @classmethod
+    def raise_deprecation(cls, values: Dict) -> Any:
         warnings.warn(
             "`ChatVectorDBChain` is deprecated - "
             "please use `from langchain.chains import ConversationalRetrievalChain`"

langchain/chains/elasticsearch_database/base.py

@@ -9,8 +9,9 @@ from langchain_core.language_models import BaseLanguageModel
 from langchain_core.output_parsers import BaseOutputParser, StrOutputParser
 from langchain_core.output_parsers.json import SimpleJsonOutputParser
 from langchain_core.prompts import BasePromptTemplate
-from langchain_core.pydantic_v1 import root_validator
 from langchain_core.runnables import Runnable
+from pydantic import ConfigDict, model_validator
+from typing_extensions import Self
 
 from langchain.chains.base import Chain
 from langchain.chains.elasticsearch_database.prompts import ANSWER_PROMPT, DSL_PROMPT
@@ -39,7 +40,7 @@ class ElasticsearchDatabaseChain(Chain):
     """Chain for creating the ES query."""
     answer_chain: Runnable
     """Chain for answering the user question."""
-    database: Any
+    database: Any = None
     """Elasticsearch database to connect to of type elasticsearch.Elasticsearch."""
     top_k: int = 10
     """Number of results to return from the query"""
@@ -51,17 +52,18 @@ class ElasticsearchDatabaseChain(Chain):
     return_intermediate_steps: bool = False
     """Whether or not to return the intermediate steps along with the final answer."""
 
-    class Config:
-        arbitrary_types_allowed = True
-        extra = "forbid"
+    model_config = ConfigDict(
+        arbitrary_types_allowed=True,
+        extra="forbid",
+    )
 
-    @root_validator(pre=False, skip_on_failure=True)
-    def validate_indices(cls, values: dict) -> dict:
-        if values["include_indices"] and values["ignore_indices"]:
+    @model_validator(mode="after")
+    def validate_indices(self) -> Self:
+        if self.include_indices and self.ignore_indices:
             raise ValueError(
                 "Cannot specify both 'include_indices' and 'ignore_indices'."
             )
-        return values
+        return self
 
     @property
     def input_keys(self) -> List[str]:

langchain/chains/flare/base.py

@@ -11,9 +11,9 @@ from langchain_core.language_models import BaseLanguageModel
 from langchain_core.messages import AIMessage
 from langchain_core.output_parsers import StrOutputParser
 from langchain_core.prompts import BasePromptTemplate
-from langchain_core.pydantic_v1 import Field
 from langchain_core.retrievers import BaseRetriever
 from langchain_core.runnables import Runnable
+from pydantic import Field
 
 from langchain.chains.base import Chain
 from langchain.chains.flare.prompts import (

langchain/chains/hyde/base.py

@@ -14,6 +14,7 @@ from langchain_core.language_models import BaseLanguageModel
 from langchain_core.output_parsers import StrOutputParser
 from langchain_core.prompts import BasePromptTemplate
 from langchain_core.runnables import Runnable
+from pydantic import ConfigDict
 
 from langchain.chains.base import Chain
 from langchain.chains.hyde.prompts import PROMPT_MAP
@@ -29,14 +30,15 @@ class HypotheticalDocumentEmbedder(Chain, Embeddings):
     base_embeddings: Embeddings
     llm_chain: Runnable
 
-    class Config:
-        arbitrary_types_allowed = True
-        extra = "forbid"
+    model_config = ConfigDict(
+        arbitrary_types_allowed=True,
+        extra="forbid",
+    )
 
     @property
     def input_keys(self) -> List[str]:
         """Input keys for Hyde's LLM chain."""
-        return self.llm_chain.input_schema.schema()["required"]
+        return self.llm_chain.input_schema.model_json_schema()["required"]
 
     @property
     def output_keys(self) -> List[str]:

langchain/chains/llm.py

@@ -17,13 +17,11 @@ from langchain_core.language_models import (
     BaseLanguageModel,
     LanguageModelInput,
 )
-from langchain_core.load.dump import dumpd
 from langchain_core.messages import BaseMessage
 from langchain_core.output_parsers import BaseLLMOutputParser, StrOutputParser
 from langchain_core.outputs import ChatGeneration, Generation, LLMResult
 from langchain_core.prompt_values import PromptValue
 from langchain_core.prompts import BasePromptTemplate, PromptTemplate
-from langchain_core.pydantic_v1 import Field
 from langchain_core.runnables import (
     Runnable,
     RunnableBinding,
@@ -32,6 +30,7 @@ from langchain_core.runnables import (
 )
 from langchain_core.runnables.configurable import DynamicRunnable
 from langchain_core.utils.input import get_colored_text
+from pydantic import ConfigDict, Field
 
 from langchain.chains.base import Chain
 
@@ -95,9 +94,10 @@ class LLMChain(Chain):
     If false, will return a bunch of extra information about the generation."""
     llm_kwargs: dict = Field(default_factory=dict)
 
-    class Config:
-        arbitrary_types_allowed = True
-        extra = "forbid"
+    model_config = ConfigDict(
+        arbitrary_types_allowed=True,
+        extra="forbid",
+    )
 
     @property
     def input_keys(self) -> List[str]:
@@ -240,7 +240,7 @@ class LLMChain(Chain):
             callbacks, self.callbacks, self.verbose
         )
         run_manager = callback_manager.on_chain_start(
-            dumpd(self),
+            None,
             {"input_list": input_list},
         )
         try:
@@ -260,7 +260,7 @@ class LLMChain(Chain):
             callbacks, self.callbacks, self.verbose
         )
         run_manager = await callback_manager.on_chain_start(
-            dumpd(self),
+            None,
             {"input_list": input_list},
         )
         try:

langchain/chains/llm_checker/base.py

@@ -9,7 +9,7 @@ from langchain_core._api import deprecated
 from langchain_core.callbacks import CallbackManagerForChainRun
 from langchain_core.language_models import BaseLanguageModel
 from langchain_core.prompts import PromptTemplate
-from langchain_core.pydantic_v1 import root_validator
+from pydantic import ConfigDict, model_validator
 
 from langchain.chains.base import Chain
 from langchain.chains.llm import LLMChain
@@ -100,12 +100,14 @@ class LLMCheckerChain(Chain):
     input_key: str = "query"  #: :meta private:
     output_key: str = "result"  #: :meta private:
 
-    class Config:
-        arbitrary_types_allowed = True
-        extra = "forbid"
+    model_config = ConfigDict(
+        arbitrary_types_allowed=True,
+        extra="forbid",
+    )
 
-    @root_validator(pre=True)
-    def raise_deprecation(cls, values: Dict) -> Dict:
+    @model_validator(mode="before")
+    @classmethod
+    def raise_deprecation(cls, values: Dict) -> Any:
         if "llm" in values:
             warnings.warn(
                 "Directly instantiating an LLMCheckerChain with an llm is deprecated. "

langchain/chains/llm_math/base.py

@@ -14,7 +14,7 @@ from langchain_core.callbacks import (
 )
 from langchain_core.language_models import BaseLanguageModel
 from langchain_core.prompts import BasePromptTemplate
-from langchain_core.pydantic_v1 import root_validator
+from pydantic import ConfigDict, model_validator
 
 from langchain.chains.base import Chain
 from langchain.chains.llm import LLMChain
@@ -156,12 +156,14 @@ class LLMMathChain(Chain):
     input_key: str = "question"  #: :meta private:
     output_key: str = "answer"  #: :meta private:
 
-    class Config:
-        arbitrary_types_allowed = True
-        extra = "forbid"
+    model_config = ConfigDict(
+        arbitrary_types_allowed=True,
+        extra="forbid",
+    )
 
-    @root_validator(pre=True)
-    def raise_deprecation(cls, values: Dict) -> Dict:
+    @model_validator(mode="before")
+    @classmethod
+    def raise_deprecation(cls, values: Dict) -> Any:
         try:
             import numexpr  # noqa: F401
         except ImportError:

langchain/chains/llm_summarization_checker/base.py

@@ -10,7 +10,7 @@ from langchain_core._api import deprecated
 from langchain_core.callbacks import CallbackManagerForChainRun
 from langchain_core.language_models import BaseLanguageModel
 from langchain_core.prompts.prompt import PromptTemplate
-from langchain_core.pydantic_v1 import root_validator
+from pydantic import ConfigDict, model_validator
 
 from langchain.chains.base import Chain
 from langchain.chains.llm import LLMChain
@@ -105,12 +105,14 @@ class LLMSummarizationCheckerChain(Chain):
     max_checks: int = 2
     """Maximum number of times to check the assertions. Default to double-checking."""
 
-    class Config:
-        arbitrary_types_allowed = True
-        extra = "forbid"
+    model_config = ConfigDict(
+        arbitrary_types_allowed=True,
+        extra="forbid",
+    )
 
-    @root_validator(pre=True)
-    def raise_deprecation(cls, values: Dict) -> Dict:
+    @model_validator(mode="before")
+    @classmethod
+    def raise_deprecation(cls, values: Dict) -> Any:
         if "llm" in values:
             warnings.warn(
                 "Directly instantiating an LLMSummarizationCheckerChain with an llm is "

langchain/chains/mapreduce.py

@@ -14,6 +14,7 @@ from langchain_core.documents import Document
 from langchain_core.language_models import BaseLanguageModel
 from langchain_core.prompts import BasePromptTemplate
 from langchain_text_splitters import TextSplitter
+from pydantic import ConfigDict
 
 from langchain.chains import ReduceDocumentsChain
 from langchain.chains.base import Chain
@@ -77,9 +78,10 @@ class MapReduceChain(Chain):
             **kwargs,
         )
 
-    class Config:
-        arbitrary_types_allowed = True
-        extra = "forbid"
+    model_config = ConfigDict(
+        arbitrary_types_allowed=True,
+        extra="forbid",
+    )
 
     @property
     def input_keys(self) -> List[str]:

langchain/chains/moderation.py

@@ -6,8 +6,8 @@ from langchain_core.callbacks import (
     AsyncCallbackManagerForChainRun,
     CallbackManagerForChainRun,
 )
-from langchain_core.pydantic_v1 import Field, root_validator
 from langchain_core.utils import check_package_version, get_from_dict_or_env
+from pydantic import Field, model_validator
 
 from langchain.chains.base import Chain
 
@@ -28,8 +28,8 @@ class OpenAIModerationChain(Chain):
             moderation = OpenAIModerationChain()
     """
 
-    client: Any  #: :meta private:
-    async_client: Any  #: :meta private:
+    client: Any = None  #: :meta private:
+    async_client: Any = None  #: :meta private:
     model_name: Optional[str] = None
     """Moderation model name to use."""
     error: bool = False
@@ -40,8 +40,9 @@ class OpenAIModerationChain(Chain):
     openai_organization: Optional[str] = None
     openai_pre_1_0: bool = Field(default=None)
 
-    @root_validator(pre=True)
-    def validate_environment(cls, values: Dict) -> Dict:
+    @model_validator(mode="before")
+    @classmethod
+    def validate_environment(cls, values: Dict) -> Any:
         """Validate that api key and python package exists in environment."""
         openai_api_key = get_from_dict_or_env(
             values, "openai_api_key", "OPENAI_API_KEY"

langchain/chains/natbot/base.py

@@ -9,8 +9,8 @@ from langchain_core._api import deprecated
 from langchain_core.callbacks import CallbackManagerForChainRun
 from langchain_core.language_models import BaseLanguageModel
 from langchain_core.output_parsers import StrOutputParser
-from langchain_core.pydantic_v1 import root_validator
 from langchain_core.runnables import Runnable
+from pydantic import ConfigDict, model_validator
 
 from langchain.chains.base import Chain
 from langchain.chains.natbot.prompt import PROMPT
@@ -59,12 +59,14 @@ class NatBotChain(Chain):
     previous_command: str = ""  #: :meta private:
     output_key: str = "command"  #: :meta private:
 
-    class Config:
-        arbitrary_types_allowed = True
-        extra = "forbid"
+    model_config = ConfigDict(
+        arbitrary_types_allowed=True,
+        extra="forbid",
+    )
 
-    @root_validator(pre=True)
-    def raise_deprecation(cls, values: Dict) -> Dict:
+    @model_validator(mode="before")
+    @classmethod
+    def raise_deprecation(cls, values: Dict) -> Any:
         if "llm" in values:
             warnings.warn(
                 "Directly instantiating an NatBotChain with an llm is deprecated. "

langchain/chains/openai_functions/base.py

@@ -19,11 +19,11 @@ from langchain_core.output_parsers.openai_functions import (
     PydanticAttrOutputFunctionsParser,
 )
 from langchain_core.prompts import BasePromptTemplate
-from langchain_core.pydantic_v1 import BaseModel
 from langchain_core.utils.function_calling import (
     PYTHON_TO_JSON_TYPES,
     convert_to_openai_function,
 )
+from pydantic import BaseModel
 
 from langchain.chains import LLMChain
 from langchain.chains.structured_output.base import (
@@ -93,7 +93,7 @@ def create_openai_fn_chain(
                 from langchain_community.chat_models import ChatOpenAI
                 from langchain_core.prompts import ChatPromptTemplate
 
-                from langchain_core.pydantic_v1 import BaseModel, Field
+                from pydantic import BaseModel, Field
 
 
                 class RecordPerson(BaseModel):
@@ -183,7 +183,7 @@ def create_structured_output_chain(
                 from langchain_community.chat_models import ChatOpenAI
                 from langchain_core.prompts import ChatPromptTemplate
 
-                from langchain_core.pydantic_v1 import BaseModel, Field
+                from pydantic import BaseModel, Field
 
                 class Dog(BaseModel):
                     \"\"\"Identifying information about a dog.\"\"\"

langchain/chains/openai_functions/citation_fuzzy_match.py

@@ -5,8 +5,8 @@ from langchain_core.language_models import BaseChatModel, BaseLanguageModel
 from langchain_core.messages import HumanMessage, SystemMessage
 from langchain_core.output_parsers.openai_functions import PydanticOutputFunctionsParser
 from langchain_core.prompts.chat import ChatPromptTemplate, HumanMessagePromptTemplate
-from langchain_core.pydantic_v1 import BaseModel, Field
 from langchain_core.runnables import Runnable
+from pydantic import BaseModel, Field
 
 from langchain.chains.llm import LLMChain
 from langchain.chains.openai_functions.utils import get_llm_kwargs

langchain/chains/openai_functions/extraction.py

@@ -7,7 +7,7 @@ from langchain_core.output_parsers.openai_functions import (
     PydanticAttrOutputFunctionsParser,
 )
 from langchain_core.prompts import BasePromptTemplate, ChatPromptTemplate
-from langchain_core.pydantic_v1 import BaseModel
+from pydantic import BaseModel
 
 from langchain.chains.base import Chain
 from langchain.chains.llm import LLMChain
@@ -61,7 +61,7 @@ Passage:
     removal="1.0",
     alternative=(
         """
-            from langchain_core.pydantic_v1 import BaseModel, Field
+            from pydantic import BaseModel, Field
             from langchain_anthropic import ChatAnthropic
     
             class Joke(BaseModel):
@@ -131,7 +131,7 @@ def create_extraction_chain(
     removal="1.0",
     alternative=(
         """
-            from langchain_core.pydantic_v1 import BaseModel, Field
+            from pydantic import BaseModel, Field
             from langchain_anthropic import ChatAnthropic
     
             class Joke(BaseModel):
@@ -172,7 +172,11 @@ def create_extraction_chain_pydantic(
     class PydanticSchema(BaseModel):
         info: List[pydantic_schema]  # type: ignore
 
-    openai_schema = pydantic_schema.schema()
+    if hasattr(pydantic_schema, "model_json_schema"):
+        openai_schema = pydantic_schema.model_json_schema()
+    else:
+        openai_schema = pydantic_schema.schema()
+
     openai_schema = _resolve_schema_references(
         openai_schema, openai_schema.get("definitions", {})
     )

langchain/chains/openai_functions/qa_with_structure.py

@@ -10,8 +10,8 @@ from langchain_core.output_parsers.openai_functions import (
 )
 from langchain_core.prompts import PromptTemplate
 from langchain_core.prompts.chat import ChatPromptTemplate, HumanMessagePromptTemplate
-from langchain_core.pydantic_v1 import BaseModel, Field
 from langchain_core.utils.pydantic import is_basemodel_subclass
+from pydantic import BaseModel, Field
 
 from langchain.chains.llm import LLMChain
 from langchain.chains.openai_functions.utils import get_llm_kwargs
@@ -72,7 +72,10 @@ def create_qa_with_structure_chain(
             f"Should be one of `pydantic` or `base`."
         )
     if isinstance(schema, type) and is_basemodel_subclass(schema):
-        schema_dict = cast(dict, schema.schema())
+        if hasattr(schema, "model_json_schema"):
+            schema_dict = cast(dict, schema.model_json_schema())
+        else:
+            schema_dict = cast(dict, schema.schema())
     else:
         schema_dict = cast(dict, schema)
     function = {

langchain/chains/openai_functions/tagging.py

@@ -130,7 +130,7 @@ def create_tagging_chain_pydantic(
 
      .. code-block:: python
 
-            from langchain_core.pydantic_v1 import BaseModel, Field
+            from pydantic import BaseModel, Field
             from langchain_anthropic import ChatAnthropic
 
             class Joke(BaseModel):
@@ -156,7 +156,10 @@ def create_tagging_chain_pydantic(
     Returns:
         Chain (LLMChain) that can be used to extract information from a passage.
     """
-    openai_schema = pydantic_schema.schema()
+    if hasattr(pydantic_schema, "model_json_schema"):
+        openai_schema = pydantic_schema.model_json_schema()
+    else:
+        openai_schema = pydantic_schema.schema()
     function = _get_tagging_function(openai_schema)
     prompt = prompt or ChatPromptTemplate.from_template(_TAGGING_TEMPLATE)
     output_parser = PydanticOutputFunctionsParser(pydantic_schema=pydantic_schema)