langchain-core 0.3.68__py3-none-any.whl → 0.3.70__py3-none-any.whl

langchain_core/_api/deprecation.py

@@ -469,7 +469,7 @@ def warn_deprecated(
 
     if not message:
         message = ""
-        _package = (
+        package_ = (
             package or name.split(".")[0].replace("_", "-")
             if "." in name
             else "LangChain"
@@ -483,14 +483,14 @@ def warn_deprecated(
         if pending:
             message += " will be deprecated in a future version"
         else:
-            message += f" was deprecated in {_package} {since}"
+            message += f" was deprecated in {package_} {since}"
 
             if removal:
                 message += f" and will be removed {removal}"
 
         if alternative_import:
             alt_package = alternative_import.split(".")[0].replace("_", "-")
-            if alt_package == _package:
+            if alt_package == package_:
                 message += f". Use {alternative_import} instead."
             else:
                 alt_module, alt_name = alternative_import.rsplit(".", 1)

langchain_core/_import_utils.py

@@ -27,8 +27,8 @@ def import_attr(
     else:
         try:
             module = import_module(f".{module_name}", package=package)
-        except ModuleNotFoundError:
-            msg = f"module '{package!r}.{module_name!r}' not found"
+        except ModuleNotFoundError as err:
+            msg = f"module '{package!r}.{module_name!r}' not found ({err})"
             raise ImportError(msg) from None
         result = getattr(module, attr_name)
     return result

langchain_core/caches.py

@@ -194,7 +194,7 @@ class InMemoryCache(BaseCache):
         """
         if self._maxsize is not None and len(self._cache) == self._maxsize:
             del self._cache[next(iter(self._cache))]
-        self._cache[(prompt, llm_string)] = return_val
+        self._cache[prompt, llm_string] = return_val
 
     @override
     def clear(self, **kwargs: Any) -> None:

langchain_core/callbacks/manager.py

@@ -1066,7 +1066,7 @@ class CallbackManagerForToolRun(ParentRunManager, ToolManagerMixin):
 
         Args:
             output (Any): The output of the tool.
-            **kwargs (Any): Additional keyword arguments.
+            **kwargs (Any): The keyword arguments to pass to the event handler
         """
         if not self.handlers:
             return
@@ -1470,7 +1470,7 @@ class CallbackManager(BaseCallbackManager):
                 input is needed.
                 If provided, the inputs are expected to be formatted as a dict.
                 The keys will correspond to the named-arguments in the tool.
-            **kwargs (Any): Additional keyword arguments.
+            **kwargs (Any): The keyword arguments to pass to the event handler
 
         Returns:
             CallbackManagerForToolRun: The callback manager for the tool run.

langchain_core/chat_history.py

@@ -65,28 +65,32 @@ class BaseChatMessageHistory(ABC):
         .. code-block:: python
 
             class FileChatMessageHistory(BaseChatMessageHistory):
-                storage_path:  str
+                storage_path: str
                 session_id: str
 
-               @property
-               def messages(self):
-                   with open(os.path.join(storage_path, session_id), 'r:utf-8') as f:
-                       messages = json.loads(f.read())
+                @property
+                def messages(self):
+                    with open(
+                        os.path.join(storage_path, session_id),
+                        "r",
+                        encoding="utf-8",
+                    ) as f:
+                        messages = json.loads(f.read())
                     return messages_from_dict(messages)
 
-               def add_messages(self, messages: Sequence[BaseMessage]) -> None:
-                   all_messages = list(self.messages) # Existing messages
-                   all_messages.extend(messages) # Add new messages
+                def add_messages(self, messages: Sequence[BaseMessage]) -> None:
+                    all_messages = list(self.messages)  # Existing messages
+                    all_messages.extend(messages)  # Add new messages
 
-                   serialized = [message_to_dict(message) for message in all_messages]
-                   # Can be further optimized by only writing new messages
-                   # using append mode.
-                   with open(os.path.join(storage_path, session_id), 'w') as f:
-                       json.dump(f, messages)
+                    serialized = [message_to_dict(message) for message in all_messages]
+                    # Can be further optimized by only writing new messages
+                    # using append mode.
+                    with open(os.path.join(storage_path, session_id), "w") as f:
+                        json.dump(messages, f)
 
-               def clear(self):
-                   with open(os.path.join(storage_path, session_id), 'w') as f:
-                       f.write("[]")
+                def clear(self):
+                    with open(os.path.join(storage_path, session_id), "w") as f:
+                        f.write("[]")
     """
 
     messages: list[BaseMessage]

langchain_core/document_loaders/base.py

@@ -60,11 +60,11 @@ class BaseLoader(ABC):  # noqa: B024
                 )
                 raise ImportError(msg) from e
 
-            _text_splitter: TextSplitter = RecursiveCharacterTextSplitter()
+            text_splitter_: TextSplitter = RecursiveCharacterTextSplitter()
         else:
-            _text_splitter = text_splitter
+            text_splitter_ = text_splitter
         docs = self.load()
-        return _text_splitter.split_documents(docs)
+        return text_splitter_.split_documents(docs)
 
     # Attention: This method will be upgraded into an abstractmethod once it's
     #            implemented in all the existing subclasses.

langchain_core/documents/base.py

@@ -201,14 +201,14 @@ class Blob(BaseMedia):
             Blob instance
         """
         if mime_type is None and guess_type:
-            _mimetype = mimetypes.guess_type(path)[0] if guess_type else None
+            mimetype = mimetypes.guess_type(path)[0] if guess_type else None
         else:
-            _mimetype = mime_type
+            mimetype = mime_type
         # We do not load the data immediately, instead we treat the blob as a
         # reference to the underlying data.
         return cls(
             data=None,
-            mimetype=_mimetype,
+            mimetype=mimetype,
             encoding=encoding,
             path=path,
             metadata=metadata if metadata is not None else {},

langchain_core/indexing/api.py

@@ -273,7 +273,7 @@ def index(
     vector_store: Union[VectorStore, DocumentIndex],
     *,
     batch_size: int = 100,
-    cleanup: Literal["incremental", "full", "scoped_full", None] = None,
+    cleanup: Optional[Literal["incremental", "full", "scoped_full"]] = None,
     source_id_key: Union[str, Callable[[Document], str], None] = None,
     cleanup_batch_size: int = 1_000,
     force_update: bool = False,
@@ -540,10 +540,10 @@ def index(
                     )
                     raise AssertionError(msg)
 
-            _source_ids = cast("Sequence[str]", source_ids)
+            source_ids_ = cast("Sequence[str]", source_ids)
 
             while uids_to_delete := record_manager.list_keys(
-                group_ids=_source_ids, before=index_start_dt, limit=cleanup_batch_size
+                group_ids=source_ids_, before=index_start_dt, limit=cleanup_batch_size
             ):
                 # Then delete from vector store.
                 _delete(destination, uids_to_delete)
@@ -609,7 +609,7 @@ async def aindex(
     vector_store: Union[VectorStore, DocumentIndex],
     *,
     batch_size: int = 100,
-    cleanup: Literal["incremental", "full", "scoped_full", None] = None,
+    cleanup: Optional[Literal["incremental", "full", "scoped_full"]] = None,
     source_id_key: Union[str, Callable[[Document], str], None] = None,
     cleanup_batch_size: int = 1_000,
     force_update: bool = False,
@@ -881,10 +881,10 @@ async def aindex(
                     )
                     raise AssertionError(msg)
 
-            _source_ids = cast("Sequence[str]", source_ids)
+            source_ids_ = cast("Sequence[str]", source_ids)
 
             while uids_to_delete := await record_manager.alist_keys(
-                group_ids=_source_ids, before=index_start_dt, limit=cleanup_batch_size
+                group_ids=source_ids_, before=index_start_dt, limit=cleanup_batch_size
             ):
                 # Then delete from vector store.
                 await _adelete(destination, uids_to_delete)

langchain_core/language_models/_utils.py

@@ -123,7 +123,7 @@ def _normalize_messages(messages: Sequence[BaseMessage]) -> list[BaseMessage]:
                     # Subset to (PDF) files and audio, as most relevant chat models
                     # support images in OAI format (and some may not yet support the
                     # standard data block format)
-                    and block.get("type") in ("file", "input_audio")
+                    and block.get("type") in {"file", "input_audio"}
                     and _is_openai_data_block(block)
                 ):
                     if formatted_message is message:

langchain_core/language_models/base.py

@@ -130,7 +130,7 @@ class BaseLanguageModel(
     )
 
     @field_validator("verbose", mode="before")
-    def set_verbose(cls, verbose: Optional[bool]) -> bool:
+    def set_verbose(cls, verbose: Optional[bool]) -> bool:  # noqa: FBT001
         """If verbose is None, set it.
 
         This allows users to pass in None as verbose to access the global setting.

langchain_core/language_models/chat_models.py

@@ -1263,8 +1263,8 @@ class BaseChatModel(BaseLanguageModel[BaseMessage], ABC):
         Returns:
             The predicted output string.
         """
-        _stop = None if stop is None else list(stop)
-        result = self([HumanMessage(content=text)], stop=_stop, **kwargs)
+        stop_ = None if stop is None else list(stop)
+        result = self([HumanMessage(content=text)], stop=stop_, **kwargs)
         if isinstance(result.content, str):
             return result.content
         msg = "Cannot use predict when output is not a string."
@@ -1279,17 +1279,17 @@ class BaseChatModel(BaseLanguageModel[BaseMessage], ABC):
         stop: Optional[Sequence[str]] = None,
         **kwargs: Any,
     ) -> BaseMessage:
-        _stop = None if stop is None else list(stop)
-        return self(messages, stop=_stop, **kwargs)
+        stop_ = None if stop is None else list(stop)
+        return self(messages, stop=stop_, **kwargs)
 
     @deprecated("0.1.7", alternative="ainvoke", removal="1.0")
     @override
     async def apredict(
         self, text: str, *, stop: Optional[Sequence[str]] = None, **kwargs: Any
     ) -> str:
-        _stop = None if stop is None else list(stop)
+        stop_ = None if stop is None else list(stop)
         result = await self._call_async(
-            [HumanMessage(content=text)], stop=_stop, **kwargs
+            [HumanMessage(content=text)], stop=stop_, **kwargs
         )
         if isinstance(result.content, str):
             return result.content
@@ -1305,8 +1305,8 @@ class BaseChatModel(BaseLanguageModel[BaseMessage], ABC):
         stop: Optional[Sequence[str]] = None,
         **kwargs: Any,
     ) -> BaseMessage:
-        _stop = None if stop is None else list(stop)
-        return await self._call_async(messages, stop=_stop, **kwargs)
+        stop_ = None if stop is None else list(stop)
+        return await self._call_async(messages, stop=stop_, **kwargs)
 
     @property
     @abstractmethod

langchain_core/language_models/fake_chat_models.py

@@ -36,6 +36,8 @@ class FakeMessagesListChatModel(BaseChatModel):
         run_manager: Optional[CallbackManagerForLLMRun] = None,
         **kwargs: Any,
     ) -> ChatResult:
+        if self.sleep is not None:
+            time.sleep(self.sleep)
         response = self.responses[self.i]
         if self.i < len(self.responses) - 1:
             self.i += 1
@@ -61,9 +63,9 @@ class FakeListChatModel(SimpleChatModel):
     """List of responses to **cycle** through in order."""
     sleep: Optional[float] = None
     i: int = 0
-    """List of responses to **cycle** through in order."""
-    error_on_chunk_number: Optional[int] = None
     """Internally incremented after every model invocation."""
+    error_on_chunk_number: Optional[int] = None
+    """If set, raise an error on the specified chunk number during streaming."""
 
     @property
     @override
@@ -79,6 +81,8 @@ class FakeListChatModel(SimpleChatModel):
         **kwargs: Any,
     ) -> str:
         """First try to lookup in queries, else return 'foo' or 'bar'."""
+        if self.sleep is not None:
+            time.sleep(self.sleep)
         response = self.responses[self.i]
         if self.i < len(self.responses) - 1:
             self.i += 1

langchain_core/language_models/llms.py

@@ -93,10 +93,10 @@ def create_base_retry_decorator(
     Raises:
         ValueError: If the cache is not set and cache is True.
     """
-    _logging = before_sleep_log(logger, logging.WARNING)
+    logging_ = before_sleep_log(logger, logging.WARNING)
 
     def _before_sleep(retry_state: RetryCallState) -> None:
-        _logging(retry_state)
+        logging_(retry_state)
         if run_manager:
             if isinstance(run_manager, AsyncCallbackManagerForLLMRun):
                 coro = run_manager.on_retry(retry_state)
@@ -119,7 +119,7 @@ def create_base_retry_decorator(
     # 4 seconds, then up to 10 seconds, then 10 seconds afterwards
     retry_instance: retry_base = retry_if_exception_type(error_types[0])
     for error in error_types[1:]:
-        retry_instance = retry_instance | retry_if_exception_type(error)
+        retry_instance |= retry_if_exception_type(error)
     return retry(
         reraise=True,
         stop=stop_after_attempt(max_retries),
@@ -129,7 +129,7 @@ def create_base_retry_decorator(
     )
 
 
-def _resolve_cache(cache: Union[BaseCache, bool, None]) -> Optional[BaseCache]:
+def _resolve_cache(*, cache: Union[BaseCache, bool, None]) -> Optional[BaseCache]:
     """Resolve the cache."""
     if isinstance(cache, BaseCache):
         llm_cache = cache
@@ -155,7 +155,7 @@ def _resolve_cache(cache: Union[BaseCache, bool, None]) -> Optional[BaseCache]:
 def get_prompts(
     params: dict[str, Any],
     prompts: list[str],
-    cache: Optional[Union[BaseCache, bool, None]] = None,
+    cache: Union[BaseCache, bool, None] = None,  # noqa: FBT001
 ) -> tuple[dict[int, list], str, list[int], list[str]]:
     """Get prompts that are already cached.
 
@@ -176,7 +176,7 @@ def get_prompts(
     missing_prompt_idxs = []
     existing_prompts = {}
 
-    llm_cache = _resolve_cache(cache)
+    llm_cache = _resolve_cache(cache=cache)
     for i, prompt in enumerate(prompts):
         if llm_cache:
             cache_val = llm_cache.lookup(prompt, llm_string)
@@ -191,7 +191,7 @@ def get_prompts(
 async def aget_prompts(
     params: dict[str, Any],
     prompts: list[str],
-    cache: Optional[Union[BaseCache, bool, None]] = None,
+    cache: Union[BaseCache, bool, None] = None,  # noqa: FBT001
 ) -> tuple[dict[int, list], str, list[int], list[str]]:
     """Get prompts that are already cached. Async version.
 
@@ -211,7 +211,7 @@ async def aget_prompts(
     missing_prompts = []
     missing_prompt_idxs = []
     existing_prompts = {}
-    llm_cache = _resolve_cache(cache)
+    llm_cache = _resolve_cache(cache=cache)
     for i, prompt in enumerate(prompts):
         if llm_cache:
             cache_val = await llm_cache.alookup(prompt, llm_string)
@@ -224,7 +224,7 @@ async def aget_prompts(
 
 
 def update_cache(
-    cache: Union[BaseCache, bool, None],
+    cache: Union[BaseCache, bool, None],  # noqa: FBT001
     existing_prompts: dict[int, list],
     llm_string: str,
     missing_prompt_idxs: list[int],
@@ -247,7 +247,7 @@ def update_cache(
     Raises:
         ValueError: If the cache is not set and cache is True.
     """
-    llm_cache = _resolve_cache(cache)
+    llm_cache = _resolve_cache(cache=cache)
     for i, result in enumerate(new_results.generations):
         existing_prompts[missing_prompt_idxs[i]] = result
         prompt = prompts[missing_prompt_idxs[i]]
@@ -257,7 +257,7 @@ def update_cache(
 
 
 async def aupdate_cache(
-    cache: Union[BaseCache, bool, None],
+    cache: Union[BaseCache, bool, None],  # noqa: FBT001
     existing_prompts: dict[int, list],
     llm_string: str,
     missing_prompt_idxs: list[int],
@@ -280,7 +280,7 @@ async def aupdate_cache(
     Raises:
         ValueError: If the cache is not set and cache is True.
     """
-    llm_cache = _resolve_cache(cache)
+    llm_cache = _resolve_cache(cache=cache)
     for i, result in enumerate(new_results.generations):
         existing_prompts[missing_prompt_idxs[i]] = result
         prompt = prompts[missing_prompt_idxs[i]]
@@ -877,8 +877,6 @@ class BaseLLM(BaseLanguageModel[str], ABC):
                 **(metadata or {}),
                 **self._get_ls_params(stop=stop, **kwargs),
             }
-        else:
-            pass
         if (
             isinstance(callbacks, list)
             and callbacks
@@ -1132,8 +1130,6 @@ class BaseLLM(BaseLanguageModel[str], ABC):
                 **(metadata or {}),
                 **self._get_ls_params(stop=stop, **kwargs),
             }
-        else:
-            pass
         # Create callback managers
         if isinstance(callbacks, list) and (
             isinstance(callbacks[0], (list, BaseCallbackManager))
@@ -1352,8 +1348,8 @@ class BaseLLM(BaseLanguageModel[str], ABC):
     def predict(
         self, text: str, *, stop: Optional[Sequence[str]] = None, **kwargs: Any
     ) -> str:
-        _stop = None if stop is None else list(stop)
-        return self(text, stop=_stop, **kwargs)
+        stop_ = None if stop is None else list(stop)
+        return self(text, stop=stop_, **kwargs)
 
     @deprecated("0.1.7", alternative="invoke", removal="1.0")
     @override
@@ -1365,8 +1361,8 @@ class BaseLLM(BaseLanguageModel[str], ABC):
         **kwargs: Any,
     ) -> BaseMessage:
         text = get_buffer_string(messages)
-        _stop = None if stop is None else list(stop)
-        content = self(text, stop=_stop, **kwargs)
+        stop_ = None if stop is None else list(stop)
+        content = self(text, stop=stop_, **kwargs)
         return AIMessage(content=content)
 
     @deprecated("0.1.7", alternative="ainvoke", removal="1.0")
@@ -1374,8 +1370,8 @@ class BaseLLM(BaseLanguageModel[str], ABC):
     async def apredict(
         self, text: str, *, stop: Optional[Sequence[str]] = None, **kwargs: Any
     ) -> str:
-        _stop = None if stop is None else list(stop)
-        return await self._call_async(text, stop=_stop, **kwargs)
+        stop_ = None if stop is None else list(stop)
+        return await self._call_async(text, stop=stop_, **kwargs)
 
     @deprecated("0.1.7", alternative="ainvoke", removal="1.0")
     @override
@@ -1387,8 +1383,8 @@ class BaseLLM(BaseLanguageModel[str], ABC):
         **kwargs: Any,
     ) -> BaseMessage:
         text = get_buffer_string(messages)
-        _stop = None if stop is None else list(stop)
-        content = await self._call_async(text, stop=_stop, **kwargs)
+        stop_ = None if stop is None else list(stop)
+        content = await self._call_async(text, stop=stop_, **kwargs)
         return AIMessage(content=content)
 
     def __str__(self) -> str:
@@ -1418,9 +1414,10 @@ class BaseLLM(BaseLanguageModel[str], ABC):
             ValueError: If the file path is not a string or Path object.
 
         Example:
-        .. code-block:: python
 
-            llm.save(file_path="path/llm.yaml")
+            .. code-block:: python
+
+                llm.save(file_path="path/llm.yaml")
         """
         # Convert file to Path object.
         save_path = Path(file_path)

langchain_core/load/load.py

@@ -56,6 +56,8 @@ class Reviver:
         additional_import_mappings: Optional[
             dict[tuple[str, ...], tuple[str, ...]]
         ] = None,
+        *,
+        ignore_unserializable_fields: bool = False,
     ) -> None:
         """Initialize the reviver.
 
@@ -70,6 +72,8 @@ class Reviver:
             additional_import_mappings: A dictionary of additional namespace mappings
                 You can use this to override default mappings or add new mappings.
                 Defaults to None.
+            ignore_unserializable_fields: Whether to ignore unserializable fields.
+                Defaults to False.
         """
         self.secrets_from_env = secrets_from_env
         self.secrets_map = secrets_map or {}
@@ -88,6 +92,7 @@ class Reviver:
             if self.additional_import_mappings
             else ALL_SERIALIZABLE_MAPPINGS
         )
+        self.ignore_unserializable_fields = ignore_unserializable_fields
 
     def __call__(self, value: dict[str, Any]) -> Any:
         """Revive the value."""
@@ -108,6 +113,8 @@ class Reviver:
             and value.get("type") == "not_implemented"
             and value.get("id") is not None
         ):
+            if self.ignore_unserializable_fields:
+                return None
             msg = (
                 "Trying to load an object that doesn't implement "
                 f"serialization: {value}"
@@ -170,6 +177,7 @@ def loads(
     valid_namespaces: Optional[list[str]] = None,
     secrets_from_env: bool = True,
     additional_import_mappings: Optional[dict[tuple[str, ...], tuple[str, ...]]] = None,
+    ignore_unserializable_fields: bool = False,
 ) -> Any:
     """Revive a LangChain class from a JSON string.
 
@@ -187,6 +195,8 @@ def loads(
         additional_import_mappings: A dictionary of additional namespace mappings
             You can use this to override default mappings or add new mappings.
             Defaults to None.
+        ignore_unserializable_fields: Whether to ignore unserializable fields.
+            Defaults to False.
 
     Returns:
         Revived LangChain objects.
@@ -194,7 +204,11 @@ def loads(
     return json.loads(
         text,
         object_hook=Reviver(
-            secrets_map, valid_namespaces, secrets_from_env, additional_import_mappings
+            secrets_map,
+            valid_namespaces,
+            secrets_from_env,
+            additional_import_mappings,
+            ignore_unserializable_fields=ignore_unserializable_fields,
         ),
     )
 
@@ -207,6 +221,7 @@ def load(
     valid_namespaces: Optional[list[str]] = None,
     secrets_from_env: bool = True,
     additional_import_mappings: Optional[dict[tuple[str, ...], tuple[str, ...]]] = None,
+    ignore_unserializable_fields: bool = False,
 ) -> Any:
     """Revive a LangChain class from a JSON object.
 
@@ -225,12 +240,18 @@ def load(
         additional_import_mappings: A dictionary of additional namespace mappings
             You can use this to override default mappings or add new mappings.
             Defaults to None.
+        ignore_unserializable_fields: Whether to ignore unserializable fields.
+            Defaults to False.
 
     Returns:
         Revived LangChain objects.
     """
     reviver = Reviver(
-        secrets_map, valid_namespaces, secrets_from_env, additional_import_mappings
+        secrets_map,
+        valid_namespaces,
+        secrets_from_env,
+        additional_import_mappings,
+        ignore_unserializable_fields=ignore_unserializable_fields,
     )
 
     def _load(obj: Any) -> Any:

langchain_core/load/serializable.py

@@ -355,19 +355,19 @@ def to_json_not_implemented(obj: object) -> SerializedNotImplemented:
     Returns:
         SerializedNotImplemented
     """
-    _id: list[str] = []
+    id_: list[str] = []
     try:
         if hasattr(obj, "__name__"):
-            _id = [*obj.__module__.split("."), obj.__name__]
+            id_ = [*obj.__module__.split("."), obj.__name__]
         elif hasattr(obj, "__class__"):
-            _id = [*obj.__class__.__module__.split("."), obj.__class__.__name__]
+            id_ = [*obj.__class__.__module__.split("."), obj.__class__.__name__]
     except Exception:
         logger.debug("Failed to serialize object", exc_info=True)
 
     result: SerializedNotImplemented = {
         "lc": 1,
         "type": "not_implemented",
-        "id": _id,
+        "id": id_,
         "repr": None,
     }
     with contextlib.suppress(Exception):

langchain_core/messages/tool.py

@@ -126,8 +126,6 @@ class ToolMessage(BaseMessage, ToolOutputMixin):
                         raise ValueError(msg) from e
                 else:
                     values["content"].append(x)
-        else:
-            pass
 
         tool_call_id = values["tool_call_id"]
         if isinstance(tool_call_id, (UUID, int, float)):
@@ -366,4 +364,4 @@ def default_tool_chunk_parser(raw_tool_calls: list[dict]) -> list[ToolCallChunk]
 def _merge_status(
     left: Literal["success", "error"], right: Literal["success", "error"]
 ) -> Literal["success", "error"]:
-    return "error" if "error" in (left, right) else "success"
+    return "error" if "error" in {left, right} else "success"