langchain 0.3.0.dev1__tar.gz → 0.3.1__tar.gz

{langchain-0.3.0.dev1 → langchain-0.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langchain
-Version: 0.3.0.dev1
+Version: 0.3.1
 Summary: Building applications with LLMs through composability
 Home-page: https://github.com/langchain-ai/langchain
 License: MIT
@@ -15,12 +15,12 @@ Requires-Dist: PyYAML (>=5.3)
 Requires-Dist: SQLAlchemy (>=1.4,<3)
 Requires-Dist: aiohttp (>=3.8.3,<4.0.0)
 Requires-Dist: async-timeout (>=4.0.0,<5.0.0) ; python_version < "3.11"
-Requires-Dist: langchain-core (>=0.3.0.dev2,<0.4.0)
-Requires-Dist: langchain-text-splitters (>=0.3.0.dev1,<0.4.0)
+Requires-Dist: langchain-core (>=0.3.6,<0.4.0)
+Requires-Dist: langchain-text-splitters (>=0.3.0,<0.4.0)
 Requires-Dist: langsmith (>=0.1.17,<0.2.0)
 Requires-Dist: numpy (>=1,<2) ; python_version < "3.12"
 Requires-Dist: numpy (>=1.26.0,<2.0.0) ; python_version >= "3.12"
-Requires-Dist: pydantic (>=1,<3)
+Requires-Dist: pydantic (>=2.7.4,<3.0.0)
 Requires-Dist: requests (>=2,<3)
 Requires-Dist: tenacity (>=8.1.0,<9.0.0,!=8.4.0)
 Project-URL: Repository, https://github.com/langchain-ai/langchain

{langchain-0.3.0.dev1 → langchain-0.3.1}/langchain/agents/agent.py

@@ -176,7 +176,7 @@ class BaseSingleActionAgent(BaseModel):
         Returns:
             Dict: Dictionary representation of agent.
         """
-        _dict = super().dict()
+        _dict = super().model_dump()
         try:
             _type = self._agent_type
         except NotImplementedError:
@@ -324,7 +324,7 @@ class BaseMultiActionAgent(BaseModel):
 
     def dict(self, **kwargs: Any) -> Dict:
         """Return dictionary representation of agent."""
-        _dict = super().dict()
+        _dict = super().model_dump()
         try:
             _dict["_type"] = str(self._agent_type)
         except NotImplementedError:

{langchain-0.3.0.dev1 → langchain-0.3.1}/langchain/agents/openai_assistant/base.py

@@ -310,7 +310,7 @@ class OpenAIAssistantRunnable(RunnableSerializable[Dict, OutputType]):
             inheritable_metadata=config.get("metadata"),
         )
         run_manager = callback_manager.on_chain_start(
-            dumpd(self), input, name=config.get("run_name")
+            dumpd(self), input, name=config.get("run_name") or self.get_name()
         )
         try:
             # Being run within AgentExecutor and there are tool outputs to submit.
@@ -429,7 +429,7 @@ class OpenAIAssistantRunnable(RunnableSerializable[Dict, OutputType]):
             inheritable_metadata=config.get("metadata"),
         )
         run_manager = callback_manager.on_chain_start(
-            dumpd(self), input, name=config.get("run_name")
+            dumpd(self), input, name=config.get("run_name") or self.get_name()
         )
         try:
             # Being run within AgentExecutor and there are tool outputs to submit.

{langchain-0.3.0.dev1 → langchain-0.3.1}/langchain/chains/api/base.py

@@ -198,7 +198,7 @@ try:
         api_docs: str
         question_key: str = "question"  #: :meta private:
         output_key: str = "output"  #: :meta private:
-        limit_to_domains: Optional[Sequence[str]]
+        limit_to_domains: Optional[Sequence[str]] = Field(default_factory=list)
         """Use to limit the domains that can be accessed by the API chain.
         
         * For example, to limit to just the domain `https://www.example.com`, set

{langchain-0.3.0.dev1 → langchain-0.3.1}/langchain/chains/base.py

@@ -18,7 +18,6 @@ from langchain_core.callbacks import (
     CallbackManagerForChainRun,
     Callbacks,
 )
-from langchain_core.load.dump import dumpd
 from langchain_core.memory import BaseMemory
 from langchain_core.outputs import RunInfo
 from langchain_core.runnables import (
@@ -32,8 +31,8 @@ from pydantic import (
     BaseModel,
     ConfigDict,
     Field,
+    field_validator,
     model_validator,
-    validator,
 )
 
 from langchain.schema import RUN_KEY
@@ -150,7 +149,7 @@ class Chain(RunnableSerializable[Dict[str, Any], Dict[str, Any]], ABC):
         new_arg_supported = inspect.signature(self._call).parameters.get("run_manager")
 
         run_manager = callback_manager.on_chain_start(
-            dumpd(self),
+            None,
             inputs,
             run_id,
             name=run_name,
@@ -202,7 +201,7 @@ class Chain(RunnableSerializable[Dict[str, Any], Dict[str, Any]], ABC):
         )
         new_arg_supported = inspect.signature(self._acall).parameters.get("run_manager")
         run_manager = await callback_manager.on_chain_start(
-            dumpd(self),
+            None,
             inputs,
             run_id,
             name=run_name,
@@ -248,7 +247,8 @@ class Chain(RunnableSerializable[Dict[str, Any], Dict[str, Any]], ABC):
             values["callbacks"] = values.pop("callback_manager", None)
         return values
 
-    @validator("verbose", pre=True, always=True)
+    @field_validator("verbose", mode="before")
+    @classmethod
     def set_verbose(cls, verbose: Optional[bool]) -> bool:
         """Set the chain verbosity.
 

{langchain-0.3.0.dev1 → langchain-0.3.1}/langchain/chains/combine_documents/map_reduce.py

@@ -4,6 +4,7 @@ from __future__ import annotations
 
 from typing import Any, Dict, List, Optional, Tuple, Type
 
+from langchain_core._api import deprecated
 from langchain_core.callbacks import Callbacks
 from langchain_core.documents import Document
 from langchain_core.runnables.config import RunnableConfig
@@ -15,6 +16,15 @@ from langchain.chains.combine_documents.reduce import ReduceDocumentsChain
 from langchain.chains.llm import LLMChain
 
 
+@deprecated(
+    since="0.3.1",
+    removal="1.0",
+    message=(
+        "This class is deprecated. Please see the migration guide here for "
+        "a recommended replacement: "
+        "https://python.langchain.com/docs/versions/migrating_chains/map_reduce_chain/"
+    ),
+)
 class MapReduceDocumentsChain(BaseCombineDocumentsChain):
     """Combining documents by mapping a chain over them, then combining results.
 

{langchain-0.3.0.dev1 → langchain-0.3.1}/langchain/chains/combine_documents/map_rerank.py

@@ -4,6 +4,7 @@ from __future__ import annotations
 
 from typing import Any, Dict, List, Optional, Sequence, Tuple, Type, Union, cast
 
+from langchain_core._api import deprecated
 from langchain_core.callbacks import Callbacks
 from langchain_core.documents import Document
 from langchain_core.runnables.config import RunnableConfig
@@ -16,6 +17,15 @@ from langchain.chains.llm import LLMChain
 from langchain.output_parsers.regex import RegexParser
 
 
+@deprecated(
+    since="0.3.1",
+    removal="1.0",
+    message=(
+        "This class is deprecated. Please see the migration guide here for "
+        "a recommended replacement: "
+        "https://python.langchain.com/docs/versions/migrating_chains/map_rerank_docs_chain/"  # noqa: E501
+    ),
+)
 class MapRerankDocumentsChain(BaseCombineDocumentsChain):
     """Combining documents by mapping a chain over them, then reranking results.
 

{langchain-0.3.0.dev1 → langchain-0.3.1}/langchain/chains/combine_documents/reduce.py

@@ -4,6 +4,7 @@ from __future__ import annotations
 
 from typing import Any, Callable, List, Optional, Protocol, Tuple
 
+from langchain_core._api import deprecated
 from langchain_core.callbacks import Callbacks
 from langchain_core.documents import Document
 from pydantic import ConfigDict
@@ -121,6 +122,15 @@ async def acollapse_docs(
     return Document(page_content=result, metadata=combined_metadata)
 
 
+@deprecated(
+    since="0.3.1",
+    removal="1.0",
+    message=(
+        "This class is deprecated. Please see the migration guide here for "
+        "a recommended replacement: "
+        "https://python.langchain.com/docs/versions/migrating_chains/map_reduce_chain/"
+    ),
+)
 class ReduceDocumentsChain(BaseCombineDocumentsChain):
     """Combine documents by recursively reducing them.
 

{langchain-0.3.0.dev1 → langchain-0.3.1}/langchain/chains/combine_documents/refine.py

@@ -4,6 +4,7 @@ from __future__ import annotations
 
 from typing import Any, Dict, List, Tuple
 
+from langchain_core._api import deprecated
 from langchain_core.callbacks import Callbacks
 from langchain_core.documents import Document
 from langchain_core.prompts import BasePromptTemplate, format_document
@@ -20,6 +21,15 @@ def _get_default_document_prompt() -> PromptTemplate:
     return PromptTemplate(input_variables=["page_content"], template="{page_content}")
 
 
+@deprecated(
+    since="0.3.1",
+    removal="1.0",
+    message=(
+        "This class is deprecated. Please see the migration guide here for "
+        "a recommended replacement: "
+        "https://python.langchain.com/docs/versions/migrating_chains/refine_docs_chain/"  # noqa: E501
+    ),
+)
 class RefineDocumentsChain(BaseCombineDocumentsChain):
     """Combine documents by doing a first pass and then refining on more documents.
 

{langchain-0.3.0.dev1 → langchain-0.3.1}/langchain/chains/conversational_retrieval/base.py

@@ -92,7 +92,7 @@ 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. """
 

{langchain-0.3.0.dev1 → langchain-0.3.1}/langchain/chains/elasticsearch_database/base.py

@@ -40,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"""

{langchain-0.3.0.dev1 → langchain-0.3.1}/langchain/chains/hyde/base.py

@@ -38,7 +38,7 @@ class HypotheticalDocumentEmbedder(Chain, Embeddings):
     @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-0.3.0.dev1 → langchain-0.3.1}/langchain/chains/llm.py

@@ -17,7 +17,6 @@ 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
@@ -241,8 +240,9 @@ class LLMChain(Chain):
             callbacks, self.callbacks, self.verbose
         )
         run_manager = callback_manager.on_chain_start(
-            dumpd(self),
+            None,
             {"input_list": input_list},
+            name=self.get_name(),
         )
         try:
             response = self.generate(input_list, run_manager=run_manager)
@@ -261,8 +261,9 @@ class LLMChain(Chain):
             callbacks, self.callbacks, self.verbose
         )
         run_manager = await callback_manager.on_chain_start(
-            dumpd(self),
+            None,
             {"input_list": input_list},
+            name=self.get_name(),
         )
         try:
             response = await self.agenerate(input_list, run_manager=run_manager)

{langchain-0.3.0.dev1 → langchain-0.3.1}/langchain/chains/moderation.py

@@ -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

{langchain-0.3.0.dev1 → langchain-0.3.1}/langchain/chains/openai_functions/base.py

@@ -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-0.3.0.dev1 → langchain-0.3.1}/langchain/chains/openai_functions/extraction.py

@@ -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-0.3.0.dev1 → langchain-0.3.1}/langchain/chains/openai_functions/qa_with_structure.py

@@ -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-0.3.0.dev1 → langchain-0.3.1}/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)

{langchain-0.3.0.dev1 → langchain-0.3.1}/langchain/chains/openai_tools/extraction.py

@@ -32,7 +32,7 @@ If a property is not present and is not required in the function parameters, do
     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):

{langchain-0.3.0.dev1 → langchain-0.3.1}/langchain/chains/structured_output/base.py

@@ -44,7 +44,7 @@ from pydantic import BaseModel
     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):
@@ -108,7 +108,7 @@ def create_openai_fn_runnable(
 
                 from langchain.chains.structured_output import create_openai_fn_runnable
                 from langchain_openai import ChatOpenAI
-                from langchain_core.pydantic_v1 import BaseModel, Field
+                from pydantic import BaseModel, Field
 
 
                 class RecordPerson(BaseModel):
@@ -162,7 +162,7 @@ def create_openai_fn_runnable(
     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):
@@ -237,7 +237,7 @@ def create_structured_output_runnable(
 
                 from langchain.chains import create_structured_output_runnable
                 from langchain_openai import ChatOpenAI
-                from langchain_core.pydantic_v1 import BaseModel, Field
+                from pydantic import BaseModel, Field
 
 
                 class RecordDog(BaseModel):
@@ -318,7 +318,7 @@ def create_structured_output_runnable(
 
                 from langchain.chains import create_structured_output_runnable
                 from langchain_openai import ChatOpenAI
-                from langchain_core.pydantic_v1 import BaseModel, Field
+                from pydantic import BaseModel, Field
 
                 class Dog(BaseModel):
                     '''Identifying information about a dog.'''
@@ -340,7 +340,7 @@ def create_structured_output_runnable(
                 from langchain.chains import create_structured_output_runnable
                 from langchain_openai 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.'''
@@ -366,7 +366,7 @@ def create_structured_output_runnable(
                 from langchain.chains import create_structured_output_runnable
                 from langchain_openai 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-0.3.0.dev1 → langchain-0.3.1}/langchain/chat_models/base.py

@@ -98,48 +98,42 @@ def init_chat_model(
 
     Must have the integration package corresponding to the model provider installed.
 
-    .. versionadded:: 0.2.7
-
-    .. versionchanged:: 0.2.8
-
-        Support for ``configurable_fields`` and ``config_prefix`` added.
-
-    .. versionchanged:: 0.2.12
-
-        Support for Ollama via langchain-ollama package added. Previously
-        langchain-community version of Ollama (now deprecated) was installed by default.
-
     Args:
         model: The name of the model, e.g. "gpt-4o", "claude-3-opus-20240229".
         model_provider: The model provider. Supported model_provider values and the
             corresponding integration package:
-                - openai (langchain-openai)
-                - anthropic (langchain-anthropic)
-                - azure_openai (langchain-openai)
-                - google_vertexai (langchain-google-vertexai)
-                - google_genai (langchain-google-genai)
-                - bedrock (langchain-aws)
-                - cohere (langchain-cohere)
-                - fireworks (langchain-fireworks)
-                - together (langchain-together)
-                - mistralai (langchain-mistralai)
-                - huggingface (langchain-huggingface)
-                - groq (langchain-groq)
-                - ollama (langchain-ollama)  [support added in langchain==0.2.12]
+
+            - openai (langchain-openai)
+            - anthropic (langchain-anthropic)
+            - azure_openai (langchain-openai)
+            - google_vertexai (langchain-google-vertexai)
+            - google_genai (langchain-google-genai)
+            - bedrock (langchain-aws)
+            - bedrock_converse (langchain-aws)
+            - cohere (langchain-cohere)
+            - fireworks (langchain-fireworks)
+            - together (langchain-together)
+            - mistralai (langchain-mistralai)
+            - huggingface (langchain-huggingface)
+            - groq (langchain-groq)
+            - ollama (langchain-ollama)  [support added in langchain==0.2.12]
 
             Will attempt to infer model_provider from model if not specified. The
             following providers will be inferred based on these model prefixes:
-                - gpt-3... or gpt-4... -> openai
-                - claude... -> anthropic
-                - amazon.... -> bedrock
-                - gemini... -> google_vertexai
-                - command... -> cohere
-                - accounts/fireworks... -> fireworks
+
+            - gpt-3..., gpt-4..., or o1... -> openai
+            - claude... -> anthropic
+            - amazon.... -> bedrock
+            - gemini... -> google_vertexai
+            - command... -> cohere
+            - accounts/fireworks... -> fireworks
+            - mistral... -> mistralai
         configurable_fields: Which model parameters are
             configurable:
-                - None: No configurable fields.
-                - "any": All fields are configurable. *See Security Note below.*
-                - Union[List[str], Tuple[str, ...]]: Specified fields are configurable.
+
+            - None: No configurable fields.
+            - "any": All fields are configurable. *See Security Note below.*
+            - Union[List[str], Tuple[str, ...]]: Specified fields are configurable.
 
             Fields are assumed to have config_prefix stripped if there is a
             config_prefix. If model is specified, then defaults to None. If model is
@@ -168,7 +162,9 @@ def init_chat_model(
         ValueError: If model_provider cannot be inferred or isn't supported.
         ImportError: If the model provider integration package is not installed.
 
-    Initialize non-configurable models:
+    .. dropdown:: Init non-configurable model
+        :open:
+
         .. code-block:: python
 
             # pip install langchain langchain-openai langchain-anthropic langchain-google-vertexai
@@ -183,7 +179,8 @@ def init_chat_model(
             gemini_15.invoke("what's your name")
 
 
-    Create a partially configurable model with no default model:
+    .. dropdown:: Partially configurable model with no default
+
         .. code-block:: python
 
             # pip install langchain langchain-openai langchain-anthropic
@@ -204,7 +201,8 @@ def init_chat_model(
             )
             # claude-3.5 sonnet response
 
-    Create a fully configurable model with a default model and a config prefix:
+    .. dropdown:: Fully configurable model with a default
+
         .. code-block:: python
 
             # pip install langchain langchain-openai langchain-anthropic
@@ -233,7 +231,8 @@ def init_chat_model(
             )
             # Claude-3.5 sonnet response with temperature 0.6
 
-    Bind tools to a configurable model:
+    .. dropdown:: Bind tools to a configurable model
+
         You can call any ChatModel declarative methods on a configurable model in the
         same way that you would with a normal model.
 
@@ -241,7 +240,7 @@ def init_chat_model(
 
             # pip install langchain langchain-openai langchain-anthropic
             from langchain.chat_models import init_chat_model
-            from langchain_core.pydantic_v1 import BaseModel, Field
+            from pydantic import BaseModel, Field
 
             class GetWeather(BaseModel):
                 '''Get the current weather in a given location'''
@@ -270,6 +269,23 @@ def init_chat_model(
                 config={"configurable": {"model": "claude-3-5-sonnet-20240620"}}
             )
             # Claude-3.5 sonnet response with tools
+
+    .. versionadded:: 0.2.7
+
+    .. versionchanged:: 0.2.8
+
+        Support for ``configurable_fields`` and ``config_prefix`` added.
+
+    .. versionchanged:: 0.2.12
+
+        Support for ChatOllama via langchain-ollama package added
+        (langchain_ollama.ChatOllama). Previously,
+        the now-deprecated langchain-community version of Ollama was imported
+        (langchain_community.chat_models.ChatOllama).
+
+        Support for langchain_aws.ChatBedrockConverse added
+        (model_provider="bedrock_converse").
+
     """  # noqa: E501
     if not model and not configurable_fields:
         configurable_fields = ("model", "model_provider")
@@ -415,7 +431,7 @@ _SUPPORTED_PROVIDERS = {
 
 
 def _attempt_infer_model_provider(model_name: str) -> Optional[str]:
-    if model_name.startswith("gpt-3") or model_name.startswith("gpt-4"):
+    if any(model_name.startswith(pre) for pre in ("gpt-3", "gpt-4", "o1")):
         return "openai"
     elif model_name.startswith("claude"):
         return "anthropic"
@@ -427,6 +443,8 @@ def _attempt_infer_model_provider(model_name: str) -> Optional[str]:
         return "google_vertexai"
     elif model_name.startswith("amazon."):
         return "bedrock"
+    elif model_name.startswith("mistral"):
+        return "mistralai"
     else:
         return None