langchain-core 1.0.0a7__py3-none-any.whl → 1.0.0rc1__py3-none-any.whl

langchain_core/tracers/__init__.py

@@ -1,12 +1,4 @@
-"""**Tracers** are classes for tracing runs.
-
-**Class hierarchy:**
-
-.. code-block::
-
-    BaseCallbackHandler --> BaseTracer --> <name>Tracer  # Examples: LangChainTracer, RootListenersTracer
-                                       --> <name>  # Examples: LogStreamCallbackHandler
-"""  # noqa: E501
+"""**Tracers** are classes for tracing runs."""
 
 from typing import TYPE_CHECKING
 

langchain_core/tracers/base.py

@@ -67,11 +67,11 @@ class BaseTracer(_TracerCore, BaseCallbackHandler, ABC):
             serialized: The serialized model.
             messages: The messages to start the chat with.
             run_id: The run ID.
-            tags: The tags for the run. Defaults to None.
-            parent_run_id: The parent run ID. Defaults to None.
-            metadata: The metadata for the run. Defaults to None.
+            tags: The tags for the run.
+            parent_run_id: The parent run ID.
+            metadata: The metadata for the run.
             name: The name of the run.
-            kwargs: Additional arguments.
+            **kwargs: Additional arguments.
 
         Returns:
             The run.
@@ -108,11 +108,11 @@ class BaseTracer(_TracerCore, BaseCallbackHandler, ABC):
             serialized: The serialized model.
             prompts: The prompts to start the LLM with.
             run_id: The run ID.
-            tags: The tags for the run. Defaults to None.
-            parent_run_id: The parent run ID. Defaults to None.
-            metadata: The metadata for the run. Defaults to None.
+            tags: The tags for the run.
+            parent_run_id: The parent run ID.
+            metadata: The metadata for the run.
             name: The name of the run.
-            kwargs: Additional arguments.
+            **kwargs: Additional arguments.
 
         Returns:
             The run.
@@ -145,10 +145,10 @@ class BaseTracer(_TracerCore, BaseCallbackHandler, ABC):
 
         Args:
             token: The token.
-            chunk: The chunk. Defaults to None.
+            chunk: The chunk.
             run_id: The run ID.
-            parent_run_id: The parent run ID. Defaults to None.
-            kwargs: Additional arguments.
+            parent_run_id: The parent run ID.
+            **kwargs: Additional arguments.
 
         Returns:
             The run.
@@ -177,7 +177,7 @@ class BaseTracer(_TracerCore, BaseCallbackHandler, ABC):
         Args:
             retry_state: The retry state.
             run_id: The run ID.
-            kwargs: Additional arguments.
+            **kwargs: Additional arguments.
 
         Returns:
             The run.
@@ -194,7 +194,7 @@ class BaseTracer(_TracerCore, BaseCallbackHandler, ABC):
         Args:
             response: The response.
             run_id: The run ID.
-            kwargs: Additional arguments.
+            **kwargs: Additional arguments.
 
         Returns:
             The run.
@@ -221,7 +221,7 @@ class BaseTracer(_TracerCore, BaseCallbackHandler, ABC):
         Args:
             error: The error.
             run_id: The run ID.
-            kwargs: Additional arguments.
+            **kwargs: Additional arguments.
 
         Returns:
             The run.
@@ -255,12 +255,12 @@ class BaseTracer(_TracerCore, BaseCallbackHandler, ABC):
             serialized: The serialized chain.
             inputs: The inputs for the chain.
             run_id: The run ID.
-            tags: The tags for the run. Defaults to None.
-            parent_run_id: The parent run ID. Defaults to None.
-            metadata: The metadata for the run. Defaults to None.
-            run_type: The type of the run. Defaults to None.
+            tags: The tags for the run.
+            parent_run_id: The parent run ID.
+            metadata: The metadata for the run.
+            run_type: The type of the run.
             name: The name of the run.
-            kwargs: Additional arguments.
+            **kwargs: Additional arguments.
 
         Returns:
             The run.
@@ -294,8 +294,8 @@ class BaseTracer(_TracerCore, BaseCallbackHandler, ABC):
         Args:
             outputs: The outputs for the chain.
             run_id: The run ID.
-            inputs: The inputs for the chain. Defaults to None.
-            kwargs: Additional arguments.
+            inputs: The inputs for the chain.
+            **kwargs: Additional arguments.
 
         Returns:
             The run.
@@ -322,9 +322,9 @@ class BaseTracer(_TracerCore, BaseCallbackHandler, ABC):
 
         Args:
             error: The error.
-            inputs: The inputs for the chain. Defaults to None.
+            inputs: The inputs for the chain.
             run_id: The run ID.
-            kwargs: Additional arguments.
+            **kwargs: Additional arguments.
 
         Returns:
             The run.
@@ -357,12 +357,12 @@ class BaseTracer(_TracerCore, BaseCallbackHandler, ABC):
             serialized: The serialized tool.
             input_str: The input string.
             run_id: The run ID.
-            tags: The tags for the run. Defaults to None.
-            parent_run_id: The parent run ID. Defaults to None.
-            metadata: The metadata for the run. Defaults to None.
+            tags: The tags for the run.
+            parent_run_id: The parent run ID.
+            metadata: The metadata for the run.
             name: The name of the run.
             inputs: The inputs for the tool.
-            kwargs: Additional arguments.
+            **kwargs: Additional arguments.
 
         Returns:
             The run.
@@ -389,7 +389,7 @@ class BaseTracer(_TracerCore, BaseCallbackHandler, ABC):
         Args:
             output: The output for the tool.
             run_id: The run ID.
-            kwargs: Additional arguments.
+            **kwargs: Additional arguments.
 
         Returns:
             The run.
@@ -415,7 +415,7 @@ class BaseTracer(_TracerCore, BaseCallbackHandler, ABC):
         Args:
             error: The error.
             run_id: The run ID.
-            kwargs: Additional arguments.
+            **kwargs: Additional arguments.
 
         Returns:
             The run.
@@ -446,11 +446,11 @@ class BaseTracer(_TracerCore, BaseCallbackHandler, ABC):
             serialized: The serialized retriever.
             query: The query.
             run_id: The run ID.
-            parent_run_id: The parent run ID. Defaults to None.
-            tags: The tags for the run. Defaults to None.
-            metadata: The metadata for the run. Defaults to None.
+            parent_run_id: The parent run ID.
+            tags: The tags for the run.
+            metadata: The metadata for the run.
             name: The name of the run.
-            kwargs: Additional arguments.
+            **kwargs: Additional arguments.
 
         Returns:
             The run.
@@ -482,7 +482,7 @@ class BaseTracer(_TracerCore, BaseCallbackHandler, ABC):
         Args:
             error: The error.
             run_id: The run ID.
-            kwargs: Additional arguments.
+            **kwargs: Additional arguments.
 
         Returns:
             The run.
@@ -504,7 +504,7 @@ class BaseTracer(_TracerCore, BaseCallbackHandler, ABC):
         Args:
             documents: The documents.
             run_id: The run ID.
-            kwargs: Additional arguments.
+            **kwargs: Additional arguments.
 
         Returns:
             The run.

langchain_core/tracers/context.py

@@ -47,17 +47,13 @@ def tracing_v2_enabled(
     """Instruct LangChain to log all runs in context to LangSmith.
 
     Args:
-        project_name (str, optional): The name of the project.
-            Defaults to ``'default'``.
-        example_id (str or UUID, optional): The ID of the example.
-            Defaults to None.
-        tags (list[str], optional): The tags to add to the run.
-            Defaults to None.
-        client (LangSmithClient, optional): The client of the langsmith.
-            Defaults to None.
+        project_name: The name of the project. Defaults to `'default'`.
+        example_id: The ID of the example.
+        tags: The tags to add to the run.
+        client: The client of the langsmith.
 
     Yields:
-        LangChainTracer: The LangChain tracer.
+        The LangChain tracer.
 
     Example:
         >>> with tracing_v2_enabled():
@@ -89,7 +85,7 @@ def collect_runs() -> Generator[RunCollectorCallbackHandler, None, None]:
     """Collect all run traces in context.
 
     Yields:
-        run_collector.RunCollectorCallbackHandler: The run collector callback handler.
+        The run collector callback handler.
 
     Example:
         >>> with collect_runs() as runs_cb:
@@ -179,15 +175,14 @@ def register_configure_hook(
     """Register a configure hook.
 
     Args:
-        context_var (ContextVar[Optional[Any]]): The context variable.
-        inheritable (bool): Whether the context variable is inheritable.
-        handle_class (Optional[Type[BaseCallbackHandler]], optional):
-          The callback handler class. Defaults to None.
-        env_var (Optional[str], optional): The environment variable. Defaults to None.
+        context_var: The context variable.
+        inheritable: Whether the context variable is inheritable.
+        handle_class: The callback handler class.
+        env_var: The environment variable.
 
     Raises:
-        ValueError: If env_var is set, handle_class must also be set
-          to a non-None value.
+        ValueError: If env_var is set, handle_class must also be set to a non-None
+            value.
     """
     if env_var is not None and handle_class is None:
         msg = "If env_var is set, handle_class must also be set to a non-None value."

langchain_core/tracers/event_stream.py

@@ -179,7 +179,7 @@ class _AstreamEventsCallbackHandler(AsyncCallbackHandler, _StreamingCallbackHand
             output: The output of the Runnable.
 
         Yields:
-            T: The output of the Runnable.
+            The output of the Runnable.
         """
         sentinel = object()
         # atomic check and set
@@ -229,7 +229,7 @@ class _AstreamEventsCallbackHandler(AsyncCallbackHandler, _StreamingCallbackHand
             output: The output of the Runnable.
 
         Yields:
-            T: The output of the Runnable.
+            The output of the Runnable.
         """
         sentinel = object()
         # atomic check and set
@@ -473,7 +473,7 @@ class _AstreamEventsCallbackHandler(AsyncCallbackHandler, _StreamingCallbackHand
         For both chat models and non-chat models (legacy LLMs).
 
         Raises:
-            ValueError: If the run type is not ``'llm'`` or ``'chat_model'``.
+            ValueError: If the run type is not `'llm'` or `'chat_model'`.
         """
         run_info = self.run_map.pop(run_id)
         inputs_ = run_info.get("inputs")

langchain_core/tracers/langchain.py

@@ -89,7 +89,7 @@ class LangChainTracer(BaseTracer):
             project_name: The project name. Defaults to the tracer project.
             client: The client. Defaults to the global client.
             tags: The tags. Defaults to an empty list.
-            kwargs: Additional keyword arguments.
+            **kwargs: Additional keyword arguments.
         """
         super().__init__(**kwargs)
         self.example_id = (
@@ -134,14 +134,14 @@ class LangChainTracer(BaseTracer):
             serialized: The serialized model.
             messages: The messages.
             run_id: The run ID.
-            tags: The tags. Defaults to None.
-            parent_run_id: The parent run ID. Defaults to None.
-            metadata: The metadata. Defaults to None.
-            name: The name. Defaults to None.
-            kwargs: Additional keyword arguments.
+            tags: The tags.
+            parent_run_id: The parent run ID.
+            metadata: The metadata.
+            name: The name.
+            **kwargs: Additional keyword arguments.
 
         Returns:
-            Run: The run.
+            The run.
         """
         start_time = datetime.now(timezone.utc)
         if metadata:
@@ -175,7 +175,7 @@ class LangChainTracer(BaseTracer):
         """Get the LangSmith root run URL.
 
         Returns:
-            str: The LangSmith root run URL.
+            The LangSmith root run URL.
 
         Raises:
             ValueError: If no traced run is found.

langchain_core/tracers/log_stream.py

@@ -111,16 +111,16 @@ class RunLogPatch:
         self.ops = list(ops)
 
     def __add__(self, other: RunLogPatch | Any) -> RunLog:
-        """Combine two ``RunLogPatch`` instances.
+        """Combine two `RunLogPatch` instances.
 
         Args:
-            other: The other ``RunLogPatch`` to combine with.
+            other: The other `RunLogPatch` to combine with.
 
         Raises:
-            TypeError: If the other object is not a ``RunLogPatch``.
+            TypeError: If the other object is not a `RunLogPatch`.
 
         Returns:
-            A new ``RunLog`` representing the combination of the two.
+            A new `RunLog` representing the combination of the two.
         """
         if type(other) is RunLogPatch:
             ops = self.ops + other.ops
@@ -159,16 +159,16 @@ class RunLog(RunLogPatch):
         self.state = state
 
     def __add__(self, other: RunLogPatch | Any) -> RunLog:
-        """Combine two ``RunLog``s.
+        """Combine two `RunLog`s.
 
         Args:
-            other: The other ``RunLog`` or ``RunLogPatch`` to combine with.
+            other: The other `RunLog` or `RunLogPatch` to combine with.
 
         Raises:
-            TypeError: If the other object is not a ``RunLog`` or ``RunLogPatch``.
+            TypeError: If the other object is not a `RunLog` or `RunLogPatch`.
 
         Returns:
-            A new ``RunLog`` representing the combination of the two.
+            A new `RunLog` representing the combination of the two.
         """
         if type(other) is RunLogPatch:
             ops = self.ops + other.ops
@@ -184,13 +184,13 @@ class RunLog(RunLogPatch):
 
     @override
     def __eq__(self, other: object) -> bool:
-        """Check if two ``RunLog``s are equal.
+        """Check if two `RunLog`s are equal.
 
         Args:
-            other: The other ``RunLog`` to compare to.
+            other: The other `RunLog` to compare to.
 
         Returns:
-            True if the ``RunLog``s are equal, False otherwise.
+            `True` if the `RunLog`s are equal, `False` otherwise.
         """
         # First compare that the state is the same
         if not isinstance(other, RunLog):
@@ -288,8 +288,7 @@ class LogStreamCallbackHandler(BaseTracer, _StreamingCallbackHandler):
             *ops: The operations to send to the stream.
 
         Returns:
-            bool: True if the patch was sent successfully, False if the stream
-                is closed.
+            `True` if the patch was sent successfully, False if the stream is closed.
         """
         # We will likely want to wrap this in try / except at some point
         # to handle exceptions that might arise at run time.
@@ -308,7 +307,7 @@ class LogStreamCallbackHandler(BaseTracer, _StreamingCallbackHandler):
             output: The output async iterator.
 
         Yields:
-            T: The output value.
+            The output value.
         """
         async for chunk in output:
             # root run is handled in .astream_log()
@@ -339,7 +338,7 @@ class LogStreamCallbackHandler(BaseTracer, _StreamingCallbackHandler):
             output: The output iterator.
 
         Yields:
-            T: The output value.
+            The output value.
         """
         for chunk in output:
             # root run is handled in .astream_log()
@@ -369,7 +368,7 @@ class LogStreamCallbackHandler(BaseTracer, _StreamingCallbackHandler):
             run: The Run to check.
 
         Returns:
-            bool: True if the run should be included, False otherwise.
+            `True` if the run should be included, `False` otherwise.
         """
         if run.id == self.root_id:
             return False
@@ -659,7 +658,7 @@ async def _astream_log_implementation(
         stream: The stream to send the run logs to.
         diff: Whether to yield run log patches (True) or full run logs (False).
         with_streamed_output_list: Whether to include a list of all streamed
-            outputs in each patch. If False, only the final output will be included
+            outputs in each patch. If `False`, only the final output will be included
             in the patches.
         **kwargs: Additional keyword arguments to pass to the runnable.
 
@@ -667,7 +666,7 @@ async def _astream_log_implementation(
         ValueError: If the callbacks in the config are of an unexpected type.
 
     Yields:
-        The run log patches or states, depending on the value of ``diff``.
+        The run log patches or states, depending on the value of `diff`.
     """
     # Assign the stream handler to the config
     config = ensure_config(config)

langchain_core/tracers/memory_stream.py

@@ -130,7 +130,7 @@ class _MemoryStream(Generic[T]):
         """Get a writer for the channel.
 
         Returns:
-            _SendStream: The writer for the channel.
+            The writer for the channel.
         """
         return _SendStream[T](
             reader_loop=self._loop, queue=self._queue, done=self._done
@@ -140,6 +140,6 @@ class _MemoryStream(Generic[T]):
         """Get a reader for the channel.
 
         Returns:
-            _ReceiveStream: The reader for the channel.
+            The reader for the channel.
         """
         return _ReceiveStream[T](queue=self._queue, done=self._done)

langchain_core/tracers/schemas.py

@@ -2,142 +2,13 @@
 
 from __future__ import annotations
 
-import warnings
-from datetime import datetime, timezone
-from typing import Any
-from uuid import UUID
-
 from langsmith import RunTree
-from langsmith.schemas import RunTypeEnum as RunTypeEnumDep
-from pydantic import PydanticDeprecationWarning
-from pydantic.v1 import BaseModel as BaseModelV1
-from pydantic.v1 import Field as FieldV1
-
-from langchain_core._api import deprecated
-
-
-@deprecated("0.1.0", alternative="Use string instead.", removal="1.0")
-def RunTypeEnum() -> type[RunTypeEnumDep]:  # noqa: N802
-    """``RunTypeEnum``.
-
-    Returns:
-        The ``RunTypeEnum`` class.
-    """
-    warnings.warn(
-        "RunTypeEnum is deprecated. Please directly use a string instead"
-        " (e.g. 'llm', 'chain', 'tool').",
-        DeprecationWarning,
-        stacklevel=2,
-    )
-    return RunTypeEnumDep
-
-
-@deprecated("0.1.0", removal="1.0")
-class TracerSessionV1Base(BaseModelV1):
-    """Base class for TracerSessionV1."""
-
-    start_time: datetime = FieldV1(default_factory=lambda: datetime.now(timezone.utc))
-    name: str | None = None
-    extra: dict[str, Any] | None = None
-
-
-@deprecated("0.1.0", removal="1.0")
-class TracerSessionV1Create(TracerSessionV1Base):
-    """Create class for TracerSessionV1."""
-
-
-@deprecated("0.1.0", removal="1.0")
-class TracerSessionV1(TracerSessionV1Base):
-    """TracerSessionV1 schema."""
-
-    id: int
-
-
-@deprecated("0.1.0", removal="1.0")
-class TracerSessionBase(TracerSessionV1Base):
-    """Base class for TracerSession."""
-
-    tenant_id: UUID
-
-
-@deprecated("0.1.0", removal="1.0")
-class TracerSession(TracerSessionBase):
-    """TracerSessionV1 schema for the V2 API."""
-
-    id: UUID
-
-
-@deprecated("0.1.0", alternative="Run", removal="1.0")
-class BaseRun(BaseModelV1):
-    """Base class for Run."""
-
-    uuid: str
-    parent_uuid: str | None = None
-    start_time: datetime = FieldV1(default_factory=lambda: datetime.now(timezone.utc))
-    end_time: datetime = FieldV1(default_factory=lambda: datetime.now(timezone.utc))
-    extra: dict[str, Any] | None = None
-    execution_order: int
-    child_execution_order: int
-    serialized: dict[str, Any]
-    session_id: int
-    error: str | None = None
-
-
-@deprecated("0.1.0", alternative="Run", removal="1.0")
-class LLMRun(BaseRun):
-    """Class for LLMRun."""
-
-    prompts: list[str]
-    # Temporarily, remove but we will completely remove LLMRun
-    # response: Optional[LLMResult] = None
-
-
-@deprecated("0.1.0", alternative="Run", removal="1.0")
-class ChainRun(BaseRun):
-    """Class for ChainRun."""
-
-    inputs: dict[str, Any]
-    outputs: dict[str, Any] | None = None
-    child_llm_runs: list[LLMRun] = FieldV1(default_factory=list)
-    child_chain_runs: list[ChainRun] = FieldV1(default_factory=list)
-    child_tool_runs: list[ToolRun] = FieldV1(default_factory=list)
-
-
-@deprecated("0.1.0", alternative="Run", removal="1.0")
-class ToolRun(BaseRun):
-    """Class for ToolRun."""
-
-    tool_input: str
-    output: str | None = None
-    action: str
-    child_llm_runs: list[LLMRun] = FieldV1(default_factory=list)
-    child_chain_runs: list[ChainRun] = FieldV1(default_factory=list)
-    child_tool_runs: list[ToolRun] = FieldV1(default_factory=list)
-
 
 # Begin V2 API Schemas
 
 
 Run = RunTree  # For backwards compatibility
 
-# TODO: Update once langsmith moves to Pydantic V2 and we can swap Run.model_rebuild
-# for Run.update_forward_refs
-with warnings.catch_warnings():
-    warnings.simplefilter("ignore", category=PydanticDeprecationWarning)
-
-    ChainRun.update_forward_refs()
-    ToolRun.update_forward_refs()
-
 __all__ = [
-    "BaseRun",
-    "ChainRun",
-    "LLMRun",
     "Run",
-    "RunTypeEnum",
-    "ToolRun",
-    "TracerSession",
-    "TracerSessionBase",
-    "TracerSessionV1",
-    "TracerSessionV1Base",
-    "TracerSessionV1Create",
 ]