From c51001f01ee47daf97e3004fc434c84dd0e2a694 Mon Sep 17 00:00:00 2001 From: Harrison Chase Date: Mon, 4 Dec 2023 19:03:19 -0800 Subject: [PATCH] fix comet tracer (#14259) --- .../tests/unit_tests/callbacks/tracers/test_comet.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/libs/langchain/tests/unit_tests/callbacks/tracers/test_comet.py b/libs/langchain/tests/unit_tests/callbacks/tracers/test_comet.py index 537bc64e455..3dff6520d47 100644 --- a/libs/langchain/tests/unit_tests/callbacks/tracers/test_comet.py +++ b/libs/langchain/tests/unit_tests/callbacks/tracers/test_comet.py @@ -44,14 +44,14 @@ def test_comet_tracer__trace_chain_with_single_span__happyflow() -> None: # Parent run tracer.on_chain_start( {"name": "chain-input"}, - ["chain-input-prompt"], + {"input": "chain-input-prompt"}, parent_run_id=None, run_id=run_id_1, ) # Check that chain was created chain_module_mock.Chain.assert_called_once_with( - inputs={"input": ["chain-input-prompt"]}, + inputs={"input": "chain-input-prompt"}, metadata=None, experiment_info="the-experiment-info", )