fix comet tracer (#14259)

This commit is contained in:
Harrison Chase 2023-12-04 19:03:19 -08:00 committed by GitHub
parent 4351b99d2b
commit c51001f01e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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",
)