langchain-core 0.4.0.dev0__py3-none-any.whl → 1.0.0a2__py3-none-any.whl

langchain_core/_api/beta_decorator.py

@@ -144,10 +144,9 @@ def beta(
                 obj.__init__ = functools.wraps(obj.__init__)(  # type: ignore[misc]
                     warn_if_direct_instance
                 )
-                return cast("T", obj)
+                return obj
 
         elif isinstance(obj, property):
-            # note(erick): this block doesn't seem to be used?
             if not _obj_type:
                 _obj_type = "attribute"
             wrapped = None
@@ -168,6 +167,7 @@ def beta(
                     self.__orig_fget = fget
                     self.__orig_fset = fset
                     self.__orig_fdel = fdel
+                    self.__doc__ = doc
 
                 def __get__(
                     self, instance: Any, owner: Union[type, None] = None

langchain_core/_api/deprecation.py

@@ -225,7 +225,7 @@ def deprecated(
                 obj.__init__ = functools.wraps(obj.__init__)(  # type: ignore[misc]
                     warn_if_direct_instance
                 )
-                return cast("T", obj)
+                return obj
 
         elif isinstance(obj, FieldInfoV1):
             wrapped = None

langchain_core/beta/runnables/context.py

@@ -253,7 +253,7 @@ class ContextSet(RunnableSerializable):
         """
         if key is not None:
             kwargs[key] = value
-        super().__init__(  # type: ignore[call-arg]
+        super().__init__(
             keys={
                 k: _coerce_set_value(v) if v is not None else None
                 for k, v in kwargs.items()

langchain_core/callbacks/base.py

@@ -7,8 +7,6 @@ from typing import TYPE_CHECKING, Any, Optional, Union
 
 from typing_extensions import Self
 
-from langchain_core.v1.messages import AIMessage, AIMessageChunk, MessageV1
-
 if TYPE_CHECKING:
     from collections.abc import Sequence
     from uuid import UUID
@@ -68,9 +66,7 @@ class LLMManagerMixin:
         self,
         token: str,
         *,
-        chunk: Optional[
-            Union[GenerationChunk, ChatGenerationChunk, AIMessageChunk]
-        ] = None,
+        chunk: Optional[Union[GenerationChunk, ChatGenerationChunk]] = None,
         run_id: UUID,
         parent_run_id: Optional[UUID] = None,
         **kwargs: Any,
@@ -79,8 +75,8 @@ class LLMManagerMixin:
 
         Args:
             token (str): The new token.
-            chunk (GenerationChunk | ChatGenerationChunk | AIMessageChunk): The new
-              generated chunk, containing content and other information.
+            chunk (GenerationChunk | ChatGenerationChunk): The new generated chunk,
+              containing content and other information.
             run_id (UUID): The run ID. This is the ID of the current run.
             parent_run_id (UUID): The parent run ID. This is the ID of the parent run.
             kwargs (Any): Additional keyword arguments.
@@ -88,7 +84,7 @@ class LLMManagerMixin:
 
     def on_llm_end(
         self,
-        response: Union[LLMResult, AIMessage],
+        response: LLMResult,
         *,
         run_id: UUID,
         parent_run_id: Optional[UUID] = None,
@@ -97,7 +93,7 @@ class LLMManagerMixin:
         """Run when LLM ends running.
 
         Args:
-            response (LLMResult | AIMessage): The response which was generated.
+            response (LLMResult): The response which was generated.
             run_id (UUID): The run ID. This is the ID of the current run.
             parent_run_id (UUID): The parent run ID. This is the ID of the parent run.
             kwargs (Any): Additional keyword arguments.
@@ -265,7 +261,7 @@ class CallbackManagerMixin:
     def on_chat_model_start(
         self,
         serialized: dict[str, Any],
-        messages: Union[list[list[BaseMessage]], list[MessageV1]],
+        messages: list[list[BaseMessage]],
         *,
         run_id: UUID,
         parent_run_id: Optional[UUID] = None,
@@ -443,9 +439,6 @@ class BaseCallbackHandler(
     run_inline: bool = False
     """Whether to run the callback inline."""
 
-    accepts_new_messages: bool = False
-    """Whether the callback accepts new message format."""
-
     @property
     def ignore_llm(self) -> bool:
         """Whether to ignore LLM callbacks."""
@@ -516,7 +509,7 @@ class AsyncCallbackHandler(BaseCallbackHandler):
     async def on_chat_model_start(
         self,
         serialized: dict[str, Any],
-        messages: Union[list[list[BaseMessage]], list[MessageV1]],
+        messages: list[list[BaseMessage]],
         *,
         run_id: UUID,
         parent_run_id: Optional[UUID] = None,
@@ -547,9 +540,7 @@ class AsyncCallbackHandler(BaseCallbackHandler):
         self,
         token: str,
         *,
-        chunk: Optional[
-            Union[GenerationChunk, ChatGenerationChunk, AIMessageChunk]
-        ] = None,
+        chunk: Optional[Union[GenerationChunk, ChatGenerationChunk]] = None,
         run_id: UUID,
         parent_run_id: Optional[UUID] = None,
         tags: Optional[list[str]] = None,
@@ -559,8 +550,8 @@ class AsyncCallbackHandler(BaseCallbackHandler):
 
         Args:
             token (str): The new token.
-            chunk (GenerationChunk | ChatGenerationChunk | AIMessageChunk): The new
-              generated chunk, containing content and other information.
+            chunk (GenerationChunk | ChatGenerationChunk): The new generated chunk,
+              containing content and other information.
             run_id (UUID): The run ID. This is the ID of the current run.
             parent_run_id (UUID): The parent run ID. This is the ID of the parent run.
             tags (Optional[list[str]]): The tags.
@@ -569,7 +560,7 @@ class AsyncCallbackHandler(BaseCallbackHandler):
 
     async def on_llm_end(
         self,
-        response: Union[LLMResult, AIMessage],
+        response: LLMResult,
         *,
         run_id: UUID,
         parent_run_id: Optional[UUID] = None,
@@ -579,7 +570,7 @@ class AsyncCallbackHandler(BaseCallbackHandler):
         """Run when LLM ends running.
 
         Args:
-            response (LLMResult | AIMessage): The response which was generated.
+            response (LLMResult): The response which was generated.
             run_id (UUID): The run ID. This is the ID of the current run.
             parent_run_id (UUID): The parent run ID. This is the ID of the parent run.
             tags (Optional[list[str]]): The tags.
@@ -603,8 +594,8 @@ class AsyncCallbackHandler(BaseCallbackHandler):
             parent_run_id: The parent run ID. This is the ID of the parent run.
             tags: The tags.
             kwargs (Any): Additional keyword arguments.
-                - response (LLMResult | AIMessage): The response which was generated
-                    before the error occurred.
+                - response (LLMResult): The response which was generated before
+                    the error occurred.
         """
 
     async def on_chain_start(

langchain_core/callbacks/file.py

@@ -49,7 +49,7 @@ class FileCallbackHandler(BaseCallbackHandler):
         mode: The file open mode. Defaults to ``'a'`` (append).
         color: Default color for text output. Defaults to ``None``.
 
-    Note:
+    .. note::
         When not used as a context manager, a deprecation warning will be issued
         on first use. The file will be opened immediately in ``__init__`` and closed
         in ``__del__`` or when ``close()`` is called explicitly.
@@ -65,6 +65,7 @@ class FileCallbackHandler(BaseCallbackHandler):
             filename: Path to the output file.
             mode: File open mode (e.g., ``'w'``, ``'a'``, ``'x'``). Defaults to ``'a'``.
             color: Default text color for output. Defaults to ``None``.
+
         """
         self.filename = filename
         self.mode = mode
@@ -82,9 +83,10 @@ class FileCallbackHandler(BaseCallbackHandler):
         Returns:
             The FileCallbackHandler instance.
 
-        Note:
+        .. note::
             The file is already opened in ``__init__``, so this just marks that
             the handler is being used as a context manager.
+
         """
         self._file_opened_in_context = True
         return self
@@ -101,6 +103,7 @@ class FileCallbackHandler(BaseCallbackHandler):
             exc_type: Exception type if an exception occurred.
             exc_val: Exception value if an exception occurred.
             exc_tb: Exception traceback if an exception occurred.
+
         """
         self.close()
 
@@ -113,6 +116,7 @@ class FileCallbackHandler(BaseCallbackHandler):
 
         This method is safe to call multiple times and will only close
         the file if it's currently open.
+
         """
         if hasattr(self, "file") and self.file and not self.file.closed:
             self.file.close()
@@ -133,6 +137,7 @@ class FileCallbackHandler(BaseCallbackHandler):
 
         Raises:
             RuntimeError: If the file is closed or not available.
+
         """
         global _GLOBAL_DEPRECATION_WARNED  # noqa: PLW0603
         if not self._file_opened_in_context and not _GLOBAL_DEPRECATION_WARNED:
@@ -163,6 +168,7 @@ class FileCallbackHandler(BaseCallbackHandler):
             serialized: The serialized chain information.
             inputs: The inputs to the chain.
             **kwargs: Additional keyword arguments that may contain ``'name'``.
+
         """
         name = (
             kwargs.get("name")
@@ -178,6 +184,7 @@ class FileCallbackHandler(BaseCallbackHandler):
         Args:
             outputs: The outputs of the chain.
             **kwargs: Additional keyword arguments.
+
         """
         self._write("\n> Finished chain.", end="\n")
 
@@ -192,6 +199,7 @@ class FileCallbackHandler(BaseCallbackHandler):
             color: Color override for this specific output. If ``None``, uses
                 ``self.color``.
             **kwargs: Additional keyword arguments.
+
         """
         self._write(action.log, color=color or self.color)
 
@@ -213,6 +221,7 @@ class FileCallbackHandler(BaseCallbackHandler):
             observation_prefix: Optional prefix to write before the output.
             llm_prefix: Optional prefix to write after the output.
             **kwargs: Additional keyword arguments.
+
         """
         if observation_prefix is not None:
             self._write(f"\n{observation_prefix}")
@@ -232,6 +241,7 @@ class FileCallbackHandler(BaseCallbackHandler):
                 ``self.color``.
             end: String appended after the text. Defaults to ``""``.
             **kwargs: Additional keyword arguments.
+
         """
         self._write(text, color=color or self.color, end=end)
 
@@ -246,5 +256,6 @@ class FileCallbackHandler(BaseCallbackHandler):
             color: Color override for this specific output. If ``None``, uses
                 ``self.color``.
             **kwargs: Additional keyword arguments.
+
         """
         self._write(finish.log, color=color or self.color, end="\n")