diff --git a/libs/langchain/langchain/callbacks/tracers/log_stream.py b/libs/langchain/langchain/callbacks/tracers/log_stream.py index c4978f88bf..08e1eedddb 100644 --- a/libs/langchain/langchain/callbacks/tracers/log_stream.py +++ b/libs/langchain/langchain/callbacks/tracers/log_stream.py @@ -193,7 +193,7 @@ class LogStreamCallbackHandler(BaseTracer): "op": "replace", "path": "", "value": RunState( - id=run.id, + id=str(run.id), streamed_output=[], final_output=None, logs=[], diff --git a/libs/langchain/tests/unit_tests/schema/runnable/test_runnable.py b/libs/langchain/tests/unit_tests/schema/runnable/test_runnable.py index 28f0045a2d..5507e63d35 100644 --- a/libs/langchain/tests/unit_tests/schema/runnable/test_runnable.py +++ b/libs/langchain/tests/unit_tests/schema/runnable/test_runnable.py @@ -379,7 +379,7 @@ async def test_prompt() -> None: assert stream_log[0].ops[0]["value"]["logs"] == [] assert stream_log[0].ops[0]["value"]["final_output"] is None assert stream_log[0].ops[0]["value"]["streamed_output"] == [] - assert type(stream_log[0].ops[0]["value"]["id"]) == UUID + assert type(stream_log[0].ops[0]["value"]["id"]) == str assert stream_log[1:] == [ RunLogPatch( @@ -1414,7 +1414,7 @@ async def test_map_astream() -> None: assert final_state.state["final_output"] == final_value assert len(final_state.state["streamed_output"]) == len(streamed_chunks) - assert isinstance(final_state.state["id"], UUID) + assert type(final_state.state["id"]) == str assert len(final_state.ops) == len(streamed_ops) assert len(final_state.state["logs"]) == 5 assert final_state.state["logs"][0]["name"] == "ChatPromptTemplate"