langchain-core 0.3.79__py3-none-any.whl → 1.0.0__py3-none-any.whl

langchain_core/tracers/context.py

@@ -8,8 +8,6 @@ from typing import (
     TYPE_CHECKING,
     Any,
     Literal,
-    Optional,
-    Union,
     cast,
 )
 from uuid import UUID
@@ -27,55 +25,35 @@ if TYPE_CHECKING:
 
     from langchain_core.callbacks.base import BaseCallbackHandler, Callbacks
     from langchain_core.callbacks.manager import AsyncCallbackManager, CallbackManager
-    from langchain_core.tracers.schemas import TracerSessionV1
 
 # for backwards partial compatibility if this is imported by users but unused
 tracing_callback_var: Any = None
-tracing_v2_callback_var: ContextVar[Optional[LangChainTracer]] = ContextVar(
+tracing_v2_callback_var: ContextVar[LangChainTracer | None] = ContextVar(
     "tracing_callback_v2", default=None
 )
-run_collector_var: ContextVar[Optional[RunCollectorCallbackHandler]] = ContextVar(
+run_collector_var: ContextVar[RunCollectorCallbackHandler | None] = ContextVar(
     "run_collector", default=None
 )
 
 
-@contextmanager
-def tracing_enabled(
-    session_name: str = "default",  # noqa: ARG001
-) -> Generator[TracerSessionV1, None, None]:
-    """Throw an error because this has been replaced by ``tracing_v2_enabled``.
-
-    Raises:
-        RuntimeError: Always, because this function is deprecated.
-    """
-    msg = (
-        "tracing_enabled is no longer supported. Please use tracing_enabled_v2 instead."
-    )
-    raise RuntimeError(msg)
-
-
 @contextmanager
 def tracing_v2_enabled(
-    project_name: Optional[str] = None,
+    project_name: str | None = None,
     *,
-    example_id: Optional[Union[str, UUID]] = None,
-    tags: Optional[list[str]] = None,
-    client: Optional[LangSmithClient] = None,
+    example_id: str | UUID | None = None,
+    tags: list[str] | None = None,
+    client: LangSmithClient | None = None,
 ) -> Generator[LangChainTracer, None, None]:
     """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():
@@ -107,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:
@@ -123,9 +101,9 @@ def collect_runs() -> Generator[RunCollectorCallbackHandler, None, None]:
 
 
 def _get_trace_callbacks(
-    project_name: Optional[str] = None,
-    example_id: Optional[Union[str, UUID]] = None,
-    callback_manager: Optional[Union[CallbackManager, AsyncCallbackManager]] = None,
+    project_name: str | None = None,
+    example_id: str | UUID | None = None,
+    callback_manager: CallbackManager | AsyncCallbackManager | None = None,
 ) -> Callbacks:
     if _tracing_v2_is_enabled():
         project_name_ = project_name or _get_tracer_project()
@@ -149,7 +127,7 @@ def _get_trace_callbacks(
     return cb
 
 
-def _tracing_v2_is_enabled() -> Union[bool, Literal["local"]]:
+def _tracing_v2_is_enabled() -> bool | Literal["local"]:
     if tracing_v2_callback_var.get() is not None:
         return True
     return ls_utils.tracing_is_enabled()
@@ -180,32 +158,31 @@ def _get_tracer_project() -> str:
 
 _configure_hooks: list[
     tuple[
-        ContextVar[Optional[BaseCallbackHandler]],
+        ContextVar[BaseCallbackHandler | None],
         bool,
-        Optional[type[BaseCallbackHandler]],
-        Optional[str],
+        type[BaseCallbackHandler] | None,
+        str | None,
     ]
 ] = []
 
 
 def register_configure_hook(
-    context_var: ContextVar[Optional[Any]],
+    context_var: ContextVar[Any | None],
     inheritable: bool,  # noqa: FBT001
-    handle_class: Optional[type[BaseCallbackHandler]] = None,
-    env_var: Optional[str] = None,
+    handle_class: type[BaseCallbackHandler] | None = None,
+    env_var: str | None = None,
 ) -> None:
     """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."
@@ -215,7 +192,7 @@ def register_configure_hook(
         (
             # the typings of ContextVar do not have the generic arg set as covariant
             # so we have to cast it
-            cast("ContextVar[Optional[BaseCallbackHandler]]", context_var),
+            cast("ContextVar[BaseCallbackHandler | None]", context_var),
             inheritable,
             handle_class,
             env_var,

langchain_core/tracers/core.py

@@ -3,7 +3,6 @@
 from __future__ import annotations
 
 import logging
-import sys
 import traceback
 from abc import ABC, abstractmethod
 from datetime import datetime, timezone
@@ -11,8 +10,6 @@ from typing import (
     TYPE_CHECKING,
     Any,
     Literal,
-    Optional,
-    Union,
     cast,
 )
 
@@ -82,7 +79,7 @@ class _TracerCore(ABC):
         """Map of run ID to (trace_id, dotted_order). Cleared when tracer GCed."""
 
     @abstractmethod
-    def _persist_run(self, run: Run) -> Union[Coroutine[Any, Any, None], None]:
+    def _persist_run(self, run: Run) -> Coroutine[Any, Any, None] | None:
         """Persist a run."""
 
     @staticmethod
@@ -98,17 +95,12 @@ class _TracerCore(ABC):
         """Get the stacktrace of the parent error."""
         msg = repr(error)
         try:
-            if sys.version_info < (3, 10):
-                tb = traceback.format_exception(
-                    error.__class__, error, error.__traceback__
-                )
-            else:
-                tb = traceback.format_exception(error)
+            tb = traceback.format_exception(error)
             return (msg + "\n\n".join(tb)).strip()
         except:  # noqa: E722
             return msg
 
-    def _start_trace(self, run: Run) -> Union[Coroutine[Any, Any, None], None]:  # type: ignore[return]
+    def _start_trace(self, run: Run) -> Coroutine[Any, Any, None] | None:  # type: ignore[return]
         current_dotted_order = run.start_time.strftime("%Y%m%dT%H%M%S%fZ") + str(run.id)
         if run.parent_run_id:
             if parent := self.order_map.get(run.parent_run_id):
@@ -132,9 +124,7 @@ class _TracerCore(ABC):
         self.order_map[run.id] = (run.trace_id, run.dotted_order)
         self.run_map[str(run.id)] = run
 
-    def _get_run(
-        self, run_id: UUID, run_type: Union[str, set[str], None] = None
-    ) -> Run:
+    def _get_run(self, run_id: UUID, run_type: str | set[str] | None = None) -> Run:
         try:
             run = self.run_map[str(run_id)]
         except KeyError as exc:
@@ -142,7 +132,7 @@ class _TracerCore(ABC):
             raise TracerException(msg) from exc
 
         if isinstance(run_type, str):
-            run_types: Union[set[str], None] = {run_type}
+            run_types: set[str] | None = {run_type}
         else:
             run_types = run_type
         if run_types is not None and run.run_type not in run_types:
@@ -158,10 +148,10 @@ class _TracerCore(ABC):
         serialized: dict[str, Any],
         messages: list[list[BaseMessage]],
         run_id: UUID,
-        tags: Optional[list[str]] = None,
-        parent_run_id: Optional[UUID] = None,
-        metadata: Optional[dict[str, Any]] = None,
-        name: Optional[str] = None,
+        tags: list[str] | None = None,
+        parent_run_id: UUID | None = None,
+        metadata: dict[str, Any] | None = None,
+        name: str | None = None,
         **kwargs: Any,
     ) -> Run:
         """Create a chat model run."""
@@ -202,10 +192,10 @@ class _TracerCore(ABC):
         serialized: dict[str, Any],
         prompts: list[str],
         run_id: UUID,
-        tags: Optional[list[str]] = None,
-        parent_run_id: Optional[UUID] = None,
-        metadata: Optional[dict[str, Any]] = None,
-        name: Optional[str] = None,
+        tags: list[str] | None = None,
+        parent_run_id: UUID | None = None,
+        metadata: dict[str, Any] | None = None,
+        name: str | None = None,
         **kwargs: Any,
     ) -> Run:
         """Create a llm run."""
@@ -230,8 +220,8 @@ class _TracerCore(ABC):
         self,
         token: str,
         run_id: UUID,
-        chunk: Optional[Union[GenerationChunk, ChatGenerationChunk]] = None,
-        parent_run_id: Optional[UUID] = None,  # noqa: ARG002
+        chunk: GenerationChunk | ChatGenerationChunk | None = None,
+        parent_run_id: UUID | None = None,  # noqa: ARG002
     ) -> Run:
         """Append token event to LLM run and return the run."""
         llm_run = self._get_run(run_id, run_type={"llm", "chat_model"})
@@ -297,7 +287,7 @@ class _TracerCore(ABC):
         return llm_run
 
     def _errored_llm_run(
-        self, error: BaseException, run_id: UUID, response: Optional[LLMResult] = None
+        self, error: BaseException, run_id: UUID, response: LLMResult | None = None
     ) -> Run:
         llm_run = self._get_run(run_id, run_type={"llm", "chat_model"})
         llm_run.error = self._get_stacktrace(error)
@@ -325,11 +315,11 @@ class _TracerCore(ABC):
         serialized: dict[str, Any],
         inputs: dict[str, Any],
         run_id: UUID,
-        tags: Optional[list[str]] = None,
-        parent_run_id: Optional[UUID] = None,
-        metadata: Optional[dict[str, Any]] = None,
-        run_type: Optional[str] = None,
-        name: Optional[str] = None,
+        tags: list[str] | None = None,
+        parent_run_id: UUID | None = None,
+        metadata: dict[str, Any] | None = None,
+        run_type: str | None = None,
+        name: str | None = None,
         **kwargs: Any,
     ) -> Run:
         """Create a chain Run."""
@@ -376,7 +366,7 @@ class _TracerCore(ABC):
         self,
         outputs: dict[str, Any],
         run_id: UUID,
-        inputs: Optional[dict[str, Any]] = None,
+        inputs: dict[str, Any] | None = None,
     ) -> Run:
         """Update a chain run with outputs and end time."""
         chain_run = self._get_run(run_id)
@@ -395,7 +385,7 @@ class _TracerCore(ABC):
     def _errored_chain_run(
         self,
         error: BaseException,
-        inputs: Optional[dict[str, Any]],
+        inputs: dict[str, Any] | None,
         run_id: UUID,
     ) -> Run:
         chain_run = self._get_run(run_id)
@@ -411,11 +401,11 @@ class _TracerCore(ABC):
         serialized: dict[str, Any],
         input_str: str,
         run_id: UUID,
-        tags: Optional[list[str]] = None,
-        parent_run_id: Optional[UUID] = None,
-        metadata: Optional[dict[str, Any]] = None,
-        name: Optional[str] = None,
-        inputs: Optional[dict[str, Any]] = None,
+        tags: list[str] | None = None,
+        parent_run_id: UUID | None = None,
+        metadata: dict[str, Any] | None = None,
+        name: str | None = None,
+        inputs: dict[str, Any] | None = None,
         **kwargs: Any,
     ) -> Run:
         """Create a tool run."""
@@ -478,10 +468,10 @@ class _TracerCore(ABC):
         serialized: dict[str, Any],
         query: str,
         run_id: UUID,
-        parent_run_id: Optional[UUID] = None,
-        tags: Optional[list[str]] = None,
-        metadata: Optional[dict[str, Any]] = None,
-        name: Optional[str] = None,
+        parent_run_id: UUID | None = None,
+        tags: list[str] | None = None,
+        metadata: dict[str, Any] | None = None,
+        name: str | None = None,
         **kwargs: Any,
     ) -> Run:
         """Create a retrieval run."""
@@ -538,7 +528,7 @@ class _TracerCore(ABC):
         """Return self copied."""
         return self
 
-    def _end_trace(self, run: Run) -> Union[Coroutine[Any, Any, None], None]:  # noqa: ARG002
+    def _end_trace(self, run: Run) -> Coroutine[Any, Any, None] | None:  # noqa: ARG002
         """End a trace for a run.
 
         Args:
@@ -546,7 +536,7 @@ class _TracerCore(ABC):
         """
         return None
 
-    def _on_run_create(self, run: Run) -> Union[Coroutine[Any, Any, None], None]:  # noqa: ARG002
+    def _on_run_create(self, run: Run) -> Coroutine[Any, Any, None] | None:  # noqa: ARG002
         """Process a run upon creation.
 
         Args:
@@ -554,7 +544,7 @@ class _TracerCore(ABC):
         """
         return None
 
-    def _on_run_update(self, run: Run) -> Union[Coroutine[Any, Any, None], None]:  # noqa: ARG002
+    def _on_run_update(self, run: Run) -> Coroutine[Any, Any, None] | None:  # noqa: ARG002
         """Process a run upon update.
 
         Args:
@@ -562,7 +552,7 @@ class _TracerCore(ABC):
         """
         return None
 
-    def _on_llm_start(self, run: Run) -> Union[Coroutine[Any, Any, None], None]:  # noqa: ARG002
+    def _on_llm_start(self, run: Run) -> Coroutine[Any, Any, None] | None:  # noqa: ARG002
         """Process the LLM Run upon start.
 
         Args:
@@ -574,8 +564,8 @@ class _TracerCore(ABC):
         self,
         run: Run,  # noqa: ARG002
         token: str,  # noqa: ARG002
-        chunk: Optional[Union[GenerationChunk, ChatGenerationChunk]],  # noqa: ARG002
-    ) -> Union[Coroutine[Any, Any, None], None]:
+        chunk: GenerationChunk | ChatGenerationChunk | None,  # noqa: ARG002
+    ) -> Coroutine[Any, Any, None] | None:
         """Process new LLM token.
 
         Args:
@@ -585,7 +575,7 @@ class _TracerCore(ABC):
         """
         return None
 
-    def _on_llm_end(self, run: Run) -> Union[Coroutine[Any, Any, None], None]:  # noqa: ARG002
+    def _on_llm_end(self, run: Run) -> Coroutine[Any, Any, None] | None:  # noqa: ARG002
         """Process the LLM Run.
 
         Args:
@@ -593,7 +583,7 @@ class _TracerCore(ABC):
         """
         return None
 
-    def _on_llm_error(self, run: Run) -> Union[Coroutine[Any, Any, None], None]:  # noqa: ARG002
+    def _on_llm_error(self, run: Run) -> Coroutine[Any, Any, None] | None:  # noqa: ARG002
         """Process the LLM Run upon error.
 
         Args:
@@ -601,7 +591,7 @@ class _TracerCore(ABC):
         """
         return None
 
-    def _on_chain_start(self, run: Run) -> Union[Coroutine[Any, Any, None], None]:  # noqa: ARG002
+    def _on_chain_start(self, run: Run) -> Coroutine[Any, Any, None] | None:  # noqa: ARG002
         """Process the Chain Run upon start.
 
         Args:
@@ -609,7 +599,7 @@ class _TracerCore(ABC):
         """
         return None
 
-    def _on_chain_end(self, run: Run) -> Union[Coroutine[Any, Any, None], None]:  # noqa: ARG002
+    def _on_chain_end(self, run: Run) -> Coroutine[Any, Any, None] | None:  # noqa: ARG002
         """Process the Chain Run.
 
         Args:
@@ -617,7 +607,7 @@ class _TracerCore(ABC):
         """
         return None
 
-    def _on_chain_error(self, run: Run) -> Union[Coroutine[Any, Any, None], None]:  # noqa: ARG002
+    def _on_chain_error(self, run: Run) -> Coroutine[Any, Any, None] | None:  # noqa: ARG002
         """Process the Chain Run upon error.
 
         Args:
@@ -625,7 +615,7 @@ class _TracerCore(ABC):
         """
         return None
 
-    def _on_tool_start(self, run: Run) -> Union[Coroutine[Any, Any, None], None]:  # noqa: ARG002
+    def _on_tool_start(self, run: Run) -> Coroutine[Any, Any, None] | None:  # noqa: ARG002
         """Process the Tool Run upon start.
 
         Args:
@@ -633,7 +623,7 @@ class _TracerCore(ABC):
         """
         return None
 
-    def _on_tool_end(self, run: Run) -> Union[Coroutine[Any, Any, None], None]:  # noqa: ARG002
+    def _on_tool_end(self, run: Run) -> Coroutine[Any, Any, None] | None:  # noqa: ARG002
         """Process the Tool Run.
 
         Args:
@@ -641,7 +631,7 @@ class _TracerCore(ABC):
         """
         return None
 
-    def _on_tool_error(self, run: Run) -> Union[Coroutine[Any, Any, None], None]:  # noqa: ARG002
+    def _on_tool_error(self, run: Run) -> Coroutine[Any, Any, None] | None:  # noqa: ARG002
         """Process the Tool Run upon error.
 
         Args:
@@ -649,7 +639,7 @@ class _TracerCore(ABC):
         """
         return None
 
-    def _on_chat_model_start(self, run: Run) -> Union[Coroutine[Any, Any, None], None]:  # noqa: ARG002
+    def _on_chat_model_start(self, run: Run) -> Coroutine[Any, Any, None] | None:  # noqa: ARG002
         """Process the Chat Model Run upon start.
 
         Args:
@@ -657,7 +647,7 @@ class _TracerCore(ABC):
         """
         return None
 
-    def _on_retriever_start(self, run: Run) -> Union[Coroutine[Any, Any, None], None]:  # noqa: ARG002
+    def _on_retriever_start(self, run: Run) -> Coroutine[Any, Any, None] | None:  # noqa: ARG002
         """Process the Retriever Run upon start.
 
         Args:
@@ -665,7 +655,7 @@ class _TracerCore(ABC):
         """
         return None
 
-    def _on_retriever_end(self, run: Run) -> Union[Coroutine[Any, Any, None], None]:  # noqa: ARG002
+    def _on_retriever_end(self, run: Run) -> Coroutine[Any, Any, None] | None:  # noqa: ARG002
         """Process the Retriever Run.
 
         Args:
@@ -673,7 +663,7 @@ class _TracerCore(ABC):
         """
         return None
 
-    def _on_retriever_error(self, run: Run) -> Union[Coroutine[Any, Any, None], None]:  # noqa: ARG002
+    def _on_retriever_error(self, run: Run) -> Coroutine[Any, Any, None] | None:  # noqa: ARG002
         """Process the Retriever Run upon error.
 
         Args:

langchain_core/tracers/evaluation.py

@@ -6,7 +6,7 @@ import logging
 import threading
 import weakref
 from concurrent.futures import Future, ThreadPoolExecutor, wait
-from typing import TYPE_CHECKING, Any, Optional, Union, cast
+from typing import TYPE_CHECKING, Any, cast
 from uuid import UUID
 
 import langsmith
@@ -43,19 +43,19 @@ class EvaluatorCallbackHandler(BaseTracer):
     """
 
     name: str = "evaluator_callback_handler"
-    example_id: Optional[UUID] = None
+    example_id: UUID | None = None
     """The example ID associated with the runs."""
     client: langsmith.Client
     """The LangSmith client instance used for evaluating the runs."""
     evaluators: Sequence[langsmith.RunEvaluator] = ()
     """The sequence of run evaluators to be executed."""
-    executor: Optional[ThreadPoolExecutor] = None
+    executor: ThreadPoolExecutor | None = None
     """The thread pool executor used for running the evaluators."""
     futures: weakref.WeakSet[Future] = weakref.WeakSet()
     """The set of futures representing the running evaluators."""
     skip_unfinished: bool = True
     """Whether to skip runs that are not finished or raised an error."""
-    project_name: Optional[str] = None
+    project_name: str | None = None
     """The LangSmith project name to be organize eval chain runs under."""
     logged_eval_results: dict[tuple[str, str], list[EvaluationResult]]
     lock: threading.Lock
@@ -63,11 +63,11 @@ class EvaluatorCallbackHandler(BaseTracer):
     def __init__(
         self,
         evaluators: Sequence[langsmith.RunEvaluator],
-        client: Optional[langsmith.Client] = None,
-        example_id: Optional[Union[UUID, str]] = None,
+        client: langsmith.Client | None = None,
+        example_id: UUID | str | None = None,
         skip_unfinished: bool = True,  # noqa: FBT001,FBT002
-        project_name: Optional[str] = "evaluators",
-        max_concurrency: Optional[int] = None,
+        project_name: str | None = "evaluators",
+        max_concurrency: int | None = None,
         **kwargs: Any,
     ) -> None:
         """Create an EvaluatorCallbackHandler.
@@ -156,7 +156,7 @@ class EvaluatorCallbackHandler(BaseTracer):
 
     def _select_eval_results(
         self,
-        results: Union[EvaluationResult, EvaluationResults],
+        results: EvaluationResult | EvaluationResults,
     ) -> list[EvaluationResult]:
         if isinstance(results, EvaluationResult):
             results_ = [results]
@@ -172,9 +172,9 @@ class EvaluatorCallbackHandler(BaseTracer):
 
     def _log_evaluation_feedback(
         self,
-        evaluator_response: Union[EvaluationResult, EvaluationResults],
+        evaluator_response: EvaluationResult | EvaluationResults,
         run: Run,
-        source_run_id: Optional[UUID] = None,
+        source_run_id: UUID | None = None,
     ) -> list[EvaluationResult]:
         results = self._select_eval_results(evaluator_response)
         for res in results: