From d19fd0cfaeeb164dfee6dfc90982b14e44b79d22 Mon Sep 17 00:00:00 2001 From: Eugene Yurtsev Date: Tue, 26 Sep 2023 15:38:51 -0400 Subject: [PATCH] LogEntry/LogStream use str instead of uuid for id (#11080) Cast the UUID to a string --- libs/langchain/langchain/callbacks/tracers/log_stream.py | 2 +- .../tests/unit_tests/schema/runnable/test_runnable.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) 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"