langchain 0.3.26__py3-none-any.whl → 0.4.0.dev0__py3-none-any.whl

langchain/schema/callbacks/tracers/langchain.py

@@ -5,4 +5,4 @@ from langchain_core.tracers.langchain import (
     wait_for_all_tracers,
 )
 
-__all__ = ["log_error_once", "wait_for_all_tracers", "get_client", "LangChainTracer"]
+__all__ = ["LangChainTracer", "get_client", "log_error_once", "wait_for_all_tracers"]

langchain/schema/callbacks/tracers/langchain_v1.py

@@ -1,3 +1,3 @@
 from langchain_core.tracers.langchain_v1 import LangChainTracerV1, get_headers
 
-__all__ = ["get_headers", "LangChainTracerV1"]
+__all__ = ["LangChainTracerV1", "get_headers"]

langchain/schema/callbacks/tracers/log_stream.py

@@ -6,4 +6,4 @@ from langchain_core.tracers.log_stream import (
     RunState,
 )
 
-__all__ = ["LogEntry", "RunState", "RunLogPatch", "RunLog", "LogStreamCallbackHandler"]
+__all__ = ["LogEntry", "LogStreamCallbackHandler", "RunLog", "RunLogPatch", "RunState"]

langchain/schema/callbacks/tracers/schemas.py

@@ -13,15 +13,15 @@ from langchain_core.tracers.schemas import (
 )
 
 __all__ = [
-    "RunTypeEnum",
-    "TracerSessionV1Base",
-    "TracerSessionV1Create",
-    "TracerSessionV1",
-    "TracerSessionBase",
-    "TracerSession",
     "BaseRun",
-    "LLMRun",
     "ChainRun",
-    "ToolRun",
+    "LLMRun",
     "Run",
+    "RunTypeEnum",
+    "ToolRun",
+    "TracerSession",
+    "TracerSessionBase",
+    "TracerSessionV1",
+    "TracerSessionV1Base",
+    "TracerSessionV1Create",
 ]

langchain/schema/callbacks/tracers/stdout.py

@@ -6,8 +6,8 @@ from langchain_core.tracers.stdout import (
 )
 
 __all__ = [
-    "try_json_stringify",
-    "elapsed",
-    "FunctionCallbackHandler",
     "ConsoleCallbackHandler",
+    "FunctionCallbackHandler",
+    "elapsed",
+    "try_json_stringify",
 ]

langchain/schema/document.py

@@ -1,3 +1,3 @@
 from langchain_core.documents import BaseDocumentTransformer, Document
 
-__all__ = ["Document", "BaseDocumentTransformer"]
+__all__ = ["BaseDocumentTransformer", "Document"]

langchain/schema/language_model.py

@@ -7,9 +7,9 @@ from langchain_core.language_models import (
 from langchain_core.language_models.base import _get_token_ids_default_method
 
 __all__ = [
-    "get_tokenizer",
     "BaseLanguageModel",
-    "_get_token_ids_default_method",
     "LanguageModelInput",
     "LanguageModelOutput",
+    "_get_token_ids_default_method",
+    "get_tokenizer",
 ]

langchain/schema/messages.py

@@ -26,26 +26,26 @@ from langchain_core.messages import (
 _message_to_dict = message_to_dict
 
 __all__ = [
-    "get_buffer_string",
+    "AIMessage",
+    "AIMessageChunk",
+    "AnyMessage",
     "BaseMessage",
-    "merge_content",
     "BaseMessageChunk",
+    "ChatMessage",
+    "ChatMessageChunk",
+    "FunctionMessage",
+    "FunctionMessageChunk",
     "HumanMessage",
     "HumanMessageChunk",
-    "AIMessage",
-    "AIMessageChunk",
     "SystemMessage",
     "SystemMessageChunk",
-    "FunctionMessage",
-    "FunctionMessageChunk",
     "ToolMessage",
     "ToolMessageChunk",
-    "ChatMessage",
-    "ChatMessageChunk",
-    "messages_to_dict",
-    "messages_from_dict",
-    "_message_to_dict",
     "_message_from_dict",
+    "_message_to_dict",
+    "get_buffer_string",
+    "merge_content",
     "message_to_dict",
-    "AnyMessage",
+    "messages_from_dict",
+    "messages_to_dict",
 ]

langchain/schema/output.py

@@ -9,11 +9,11 @@ from langchain_core.outputs import (
 )
 
 __all__ = [
-    "Generation",
-    "GenerationChunk",
     "ChatGeneration",
     "ChatGenerationChunk",
-    "RunInfo",
     "ChatResult",
+    "Generation",
+    "GenerationChunk",
     "LLMResult",
+    "RunInfo",
 ]

langchain/schema/output_parser.py

@@ -13,13 +13,13 @@ from langchain_core.output_parsers.base import T
 NoOpOutputParser = StrOutputParser
 
 __all__ = [
-    "BaseLLMOutputParser",
+    "BaseCumulativeTransformOutputParser",
     "BaseGenerationOutputParser",
+    "BaseLLMOutputParser",
     "BaseOutputParser",
     "BaseTransformOutputParser",
-    "BaseCumulativeTransformOutputParser",
     "NoOpOutputParser",
-    "StrOutputParser",
     "OutputParserException",
+    "StrOutputParser",
     "T",
 ]

langchain/schema/runnable/__init__.py

@@ -38,13 +38,11 @@ from langchain_core.runnables.utils import (
 
 __all__ = [
     "ConfigurableField",
-    "ConfigurableFieldSingleOption",
     "ConfigurableFieldMultiOption",
-    "patch_config",
+    "ConfigurableFieldSingleOption",
     "RouterInput",
     "RouterRunnable",
     "Runnable",
-    "RunnableSerializable",
     "RunnableBinding",
     "RunnableBranch",
     "RunnableConfig",
@@ -54,5 +52,7 @@ __all__ = [
     "RunnableParallel",
     "RunnablePassthrough",
     "RunnableSequence",
+    "RunnableSerializable",
     "RunnableWithFallbacks",
+    "patch_config",
 ]

langchain/schema/runnable/base.py

@@ -20,19 +20,19 @@ RunnableMap = RunnableParallel
 
 __all__ = [
     "Input",
-    "Output",
-    "RunnableLike",
     "Other",
+    "Output",
     "Runnable",
-    "RunnableSerializable",
-    "RunnableSequence",
-    "RunnableParallel",
+    "RunnableBinding",
+    "RunnableBindingBase",
+    "RunnableEach",
+    "RunnableEachBase",
     "RunnableGenerator",
     "RunnableLambda",
-    "RunnableEachBase",
-    "RunnableEach",
-    "RunnableBindingBase",
-    "RunnableBinding",
+    "RunnableLike",
     "RunnableMap",
+    "RunnableParallel",
+    "RunnableSequence",
+    "RunnableSerializable",
     "coerce_to_runnable",
 ]

langchain/schema/runnable/config.py

@@ -15,13 +15,13 @@ from langchain_core.runnables.config import (
 __all__ = [
     "EmptyDict",
     "RunnableConfig",
-    "ensure_config",
-    "get_config_list",
-    "patch_config",
-    "merge_configs",
     "acall_func_with_variable_args",
     "call_func_with_variable_args",
-    "get_callback_manager_for_config",
+    "ensure_config",
     "get_async_callback_manager_for_config",
+    "get_callback_manager_for_config",
+    "get_config_list",
     "get_executor_for_config",
+    "merge_configs",
+    "patch_config",
 ]

langchain/schema/runnable/configurable.py

@@ -8,8 +8,8 @@ from langchain_core.runnables.configurable import (
 
 __all__ = [
     "DynamicRunnable",
+    "RunnableConfigurableAlternatives",
     "RunnableConfigurableFields",
     "StrEnum",
-    "RunnableConfigurableAlternatives",
     "make_options_spec",
 ]

langchain/schema/runnable/history.py

@@ -5,7 +5,7 @@ from langchain_core.runnables.history import (
 )
 
 __all__ = [
-    "RunnableWithMessageHistory",
     "GetSessionHistoryCallable",
     "MessagesOrDictWithMessages",
+    "RunnableWithMessageHistory",
 ]

langchain/schema/runnable/passthrough.py

@@ -5,4 +5,4 @@ from langchain_core.runnables.passthrough import (
     identity,
 )
 
-__all__ = ["aidentity", "identity", "RunnablePassthrough", "RunnableAssign"]
+__all__ = ["RunnableAssign", "RunnablePassthrough", "aidentity", "identity"]

langchain/schema/runnable/utils.py

@@ -25,27 +25,27 @@ from langchain_core.runnables.utils import (
 )
 
 __all__ = [
-    "accepts_run_manager",
-    "accepts_config",
-    "IsLocalDict",
-    "IsFunctionArgDict",
-    "GetLambdaSource",
-    "get_function_first_arg_dict_keys",
-    "get_lambda_source",
-    "indent_lines_after_first",
+    "Addable",
     "AddableDict",
-    "SupportsAdd",
-    "add",
+    "AnyConfigurableField",
     "ConfigurableField",
-    "ConfigurableFieldSingleOption",
     "ConfigurableFieldMultiOption",
+    "ConfigurableFieldSingleOption",
     "ConfigurableFieldSpec",
-    "get_unique_config_specs",
+    "GetLambdaSource",
+    "Input",
+    "IsFunctionArgDict",
+    "IsLocalDict",
+    "Output",
+    "SupportsAdd",
     "aadd",
+    "accepts_config",
+    "accepts_run_manager",
+    "add",
     "gated_coro",
     "gather_with_concurrency",
-    "Input",
-    "Output",
-    "Addable",
-    "AnyConfigurableField",
+    "get_function_first_arg_dict_keys",
+    "get_lambda_source",
+    "get_unique_config_specs",
+    "indent_lines_after_first",
 ]

langchain/schema/vectorstore.py

@@ -1,3 +1,3 @@
 from langchain_core.vectorstores import VST, VectorStore, VectorStoreRetriever
 
-__all__ = ["VectorStore", "VectorStoreRetriever", "VST"]
+__all__ = ["VST", "VectorStore", "VectorStoreRetriever"]

langchain/smith/__init__.py

@@ -87,6 +87,7 @@ or LangSmith's `RunEvaluator` classes.
 - :func:`arun_on_dataset <langchain.smith.evaluation.runner_utils.arun_on_dataset>`: Asynchronous function to evaluate a chain, agent, or other LangChain component over a dataset.
 - :func:`run_on_dataset <langchain.smith.evaluation.runner_utils.run_on_dataset>`: Function to evaluate a chain, agent, or other LangChain component over a dataset.
 - :class:`RunEvalConfig <langchain.smith.evaluation.config.RunEvalConfig>`: Class representing the configuration for running evaluation. You can select evaluators by :class:`EvaluatorType <langchain.evaluation.schema.EvaluatorType>` or config, or you can pass in `custom_evaluators`
+
 """  # noqa: E501
 
 from langchain.smith.evaluation import (
@@ -96,7 +97,7 @@ from langchain.smith.evaluation import (
 )
 
 __all__ = [
+    "RunEvalConfig",
     "arun_on_dataset",
     "run_on_dataset",
-    "RunEvalConfig",
 ]

langchain/smith/evaluation/__init__.py

@@ -61,8 +61,8 @@ from langchain.smith.evaluation.string_run_evaluator import StringRunEvaluatorCh
 
 __all__ = [
     "InputFormatError",
+    "RunEvalConfig",
+    "StringRunEvaluatorChain",
     "arun_on_dataset",
     "run_on_dataset",
-    "StringRunEvaluatorChain",
-    "RunEvalConfig",
 ]

langchain/smith/evaluation/config.py

@@ -10,6 +10,7 @@ from langsmith import RunEvaluator
 from langsmith.evaluation.evaluator import EvaluationResult, EvaluationResults
 from langsmith.schemas import Example, Run
 from pydantic import BaseModel, ConfigDict, Field
+from typing_extensions import override
 
 from langchain.evaluation.criteria.eval_chain import CRITERIA_TYPE
 from langchain.evaluation.embedding_distance.base import (
@@ -21,7 +22,8 @@ from langchain.evaluation.string_distance.base import (
 )
 
 RUN_EVALUATOR_LIKE = Callable[
-    [Run, Optional[Example]], Union[EvaluationResult, EvaluationResults, dict]
+    [Run, Optional[Example]],
+    Union[EvaluationResult, EvaluationResults, dict],
 ]
 BATCH_EVALUATOR_LIKE = Callable[
     [Sequence[Run], Optional[Sequence[Example]]],
@@ -57,9 +59,7 @@ class EvalConfig(BaseModel):
         """
         kwargs = {}
         for field, val in self:
-            if field == "evaluator_type":
-                continue
-            elif val is None:
+            if field == "evaluator_type" or val is None:
                 continue
             kwargs[field] = val
         return kwargs
@@ -79,6 +79,7 @@ class SingleKeyEvalConfig(EvalConfig):
     """The key from the traced run's inputs dictionary to use to represent the
     input. If not provided, it will be inferred automatically."""
 
+    @override
     def get_kwargs(self) -> dict[str, Any]:
         kwargs = super().get_kwargs()
         # Filer out the keys that are not needed for the evaluator.
@@ -133,7 +134,7 @@ class RunEvalConfig(BaseModel):
     :class:`EvaluatorType <langchain.evaluation.schema.EvaluatorType>`, such
     as `EvaluatorType.QA`, the evaluator type string ("qa"), or a configuration for a
     given evaluator
-    (e.g., 
+    (e.g.,
     :class:`RunEvalConfig.QA <langchain.smith.evaluation.config.RunEvalConfig.QA>`)."""
     custom_evaluators: Optional[list[CUSTOM_EVALUATOR_TYPE]] = None
     """Custom evaluators to apply to the dataset run."""
@@ -177,11 +178,6 @@ class RunEvalConfig(BaseModel):
         llm: Optional[BaseLanguageModel] = None
         evaluator_type: EvaluatorType = EvaluatorType.CRITERIA
 
-        def __init__(
-            self, criteria: Optional[CRITERIA_TYPE] = None, **kwargs: Any
-        ) -> None:
-            super().__init__(criteria=criteria, **kwargs)  # type: ignore[call-arg]
-
     class LabeledCriteria(SingleKeyEvalConfig):
         """Configuration for a labeled (with references) criteria evaluator.
 
@@ -197,11 +193,6 @@ class RunEvalConfig(BaseModel):
         llm: Optional[BaseLanguageModel] = None
         evaluator_type: EvaluatorType = EvaluatorType.LABELED_CRITERIA
 
-        def __init__(
-            self, criteria: Optional[CRITERIA_TYPE] = None, **kwargs: Any
-        ) -> None:
-            super().__init__(criteria=criteria, **kwargs)  # type: ignore[call-arg]
-
     class EmbeddingDistance(SingleKeyEvalConfig):
         """Configuration for an embedding distance evaluator.
 
@@ -342,6 +333,7 @@ class RunEvalConfig(BaseModel):
 
     class ScoreString(SingleKeyEvalConfig):
         """Configuration for a score string evaluator.
+
         This is like the criteria evaluator but it is configured by
         default to return a score on the scale from 1-10.
 
@@ -367,13 +359,7 @@ class RunEvalConfig(BaseModel):
         normalize_by: Optional[float] = None
         prompt: Optional[BasePromptTemplate] = None
 
-        def __init__(
-            self,
-            criteria: Optional[CRITERIA_TYPE] = None,
-            normalize_by: Optional[float] = None,
-            **kwargs: Any,
-        ) -> None:
-            super().__init__(criteria=criteria, normalize_by=normalize_by, **kwargs)  # type: ignore[call-arg]
-
     class LabeledScoreString(ScoreString):
+        """ "Configuration for a labeled score string evaluator."""
+
         evaluator_type: EvaluatorType = EvaluatorType.LABELED_SCORE_STRING

langchain/smith/evaluation/name_generation.py

@@ -721,7 +721,7 @@ nouns = [
 
 def random_name() -> str:
     """Generate a random name."""
-    adjective = random.choice(adjectives)
-    noun = random.choice(nouns)
-    number = random.randint(1, 100)
+    adjective = random.choice(adjectives)  # noqa: S311
+    noun = random.choice(nouns)  # noqa: S311
+    number = random.randint(1, 100)  # noqa: S311
     return f"{adjective}-{noun}-{number}"

langchain/smith/evaluation/progress.py

@@ -2,26 +2,35 @@
 
 import threading
 from collections.abc import Sequence
-from typing import Any, Optional
+from typing import Any, Optional, Union
 from uuid import UUID
 
 from langchain_core.callbacks import base as base_callbacks
 from langchain_core.documents import Document
 from langchain_core.outputs import LLMResult
+from langchain_core.v1.messages import AIMessage
+from typing_extensions import override
 
 
 class ProgressBarCallback(base_callbacks.BaseCallbackHandler):
     """A simple progress bar for the console."""
 
-    def __init__(self, total: int, ncols: int = 50, **kwargs: Any):
+    def __init__(
+        self,
+        total: int,
+        ncols: int = 50,
+        end_with: str = "\n",
+    ):
         """Initialize the progress bar.
 
         Args:
             total: int, the total number of items to be processed.
             ncols: int, the character width of the progress bar.
+            end_with: str, last string to print after progress bar reaches end.
         """
         self.total = total
         self.ncols = ncols
+        self.end_with = end_with
         self.counter = 0
         self.lock = threading.Lock()
         self._print_bar()
@@ -37,8 +46,10 @@ class ProgressBarCallback(base_callbacks.BaseCallbackHandler):
         progress = self.counter / self.total
         arrow = "-" * int(round(progress * self.ncols) - 1) + ">"
         spaces = " " * (self.ncols - len(arrow))
-        print(f"\r[{arrow + spaces}] {self.counter}/{self.total}", end="")  # noqa: T201
+        end = "" if self.counter < self.total else self.end_with
+        print(f"\r[{arrow + spaces}] {self.counter}/{self.total}", end=end)  # noqa: T201
 
+    @override
     def on_chain_error(
         self,
         error: BaseException,
@@ -50,6 +61,7 @@ class ProgressBarCallback(base_callbacks.BaseCallbackHandler):
         if parent_run_id is None:
             self.increment()
 
+    @override
     def on_chain_end(
         self,
         outputs: dict[str, Any],
@@ -61,6 +73,7 @@ class ProgressBarCallback(base_callbacks.BaseCallbackHandler):
         if parent_run_id is None:
             self.increment()
 
+    @override
     def on_retriever_error(
         self,
         error: BaseException,
@@ -72,6 +85,7 @@ class ProgressBarCallback(base_callbacks.BaseCallbackHandler):
         if parent_run_id is None:
             self.increment()
 
+    @override
     def on_retriever_end(
         self,
         documents: Sequence[Document],
@@ -83,6 +97,7 @@ class ProgressBarCallback(base_callbacks.BaseCallbackHandler):
         if parent_run_id is None:
             self.increment()
 
+    @override
     def on_llm_error(
         self,
         error: BaseException,
@@ -94,9 +109,10 @@ class ProgressBarCallback(base_callbacks.BaseCallbackHandler):
         if parent_run_id is None:
             self.increment()
 
+    @override
     def on_llm_end(
         self,
-        response: LLMResult,
+        response: Union[LLMResult, AIMessage],
         *,
         run_id: UUID,
         parent_run_id: Optional[UUID] = None,
@@ -105,6 +121,7 @@ class ProgressBarCallback(base_callbacks.BaseCallbackHandler):
         if parent_run_id is None:
             self.increment()
 
+    @override
     def on_tool_error(
         self,
         error: BaseException,
@@ -116,6 +133,7 @@ class ProgressBarCallback(base_callbacks.BaseCallbackHandler):
         if parent_run_id is None:
             self.increment()
 
+    @override
     def on_tool_end(
         self,
         output: str,