mirror of
https://github.com/hwchase17/langchain.git
synced 2026-07-16 09:06:19 +00:00
Apply patch [skip ci]
This commit is contained in:
@@ -228,13 +228,11 @@ def test_inherit_run_name_enabled() -> None:
|
||||
|
||||
def test_inherit_run_name_with_chain() -> None:
|
||||
"""Test inherit_run_name behavior in a chain of runnables."""
|
||||
from typing import List
|
||||
|
||||
from langchain_core.callbacks.base import BaseCallbackHandler
|
||||
from langchain_core.runnables import RunnableLambda
|
||||
|
||||
# Track run names through callbacks
|
||||
captured_names: List[str] = []
|
||||
captured_names: list[str] = []
|
||||
|
||||
class TestCallbackHandler(BaseCallbackHandler):
|
||||
def on_chain_start(self, serialized: dict, inputs: Any, **kwargs: Any) -> None:
|
||||
@@ -370,3 +368,4 @@ def test_inherit_run_name_merge_configs() -> None:
|
||||
"inherit_run_name should pass through ensure_config"
|
||||
)
|
||||
assert ensured.get("run_name") == "test_run", "run_name should be preserved"
|
||||
|
||||
|
||||
Reference in New Issue
Block a user