mirror of https://github.com/hwchase17/langchain
Add collect_runs callback (#9885)
parent
3103f07e03
commit
907c57e324
@ -0,0 +1,16 @@
|
||||
"""Test the run collector."""
|
||||
|
||||
import uuid
|
||||
|
||||
from langchain.callbacks import collect_runs
|
||||
from tests.unit_tests.llms.fake_llm import FakeLLM
|
||||
|
||||
|
||||
def test_collect_runs() -> None:
|
||||
llm = FakeLLM(queries={"hi": "hello"}, sequential_responses=True)
|
||||
with collect_runs() as cb:
|
||||
llm.predict("hi")
|
||||
assert cb.traced_runs
|
||||
assert len(cb.traced_runs) == 1
|
||||
assert isinstance(cb.traced_runs[0].id, uuid.UUID)
|
||||
assert cb.traced_runs[0].inputs == {"prompts": ["hi"]}
|
Loading…
Reference in New Issue