From 1e1418d69987ceebed23a52eead70ad40bc6df86 Mon Sep 17 00:00:00 2001 From: Guangya Liu Date: Mon, 2 Oct 2023 13:47:14 -0400 Subject: [PATCH] fix callback use client_id --- src/backend/langflow/api/v1/callback.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/backend/langflow/api/v1/callback.py b/src/backend/langflow/api/v1/callback.py index bda05865b..54ba5c61a 100644 --- a/src/backend/langflow/api/v1/callback.py +++ b/src/backend/langflow/api/v1/callback.py @@ -124,8 +124,8 @@ class AsyncStreamingLLMCallbackHandler(AsyncCallbackHandler): class StreamingLLMCallbackHandler(BaseCallbackHandler): """Callback handler for streaming LLM responses.""" - def __init__(self, websocket): - self.websocket = websocket + def __init__(self, client_id: str): + self.client_id = client_id def on_llm_new_token(self, token: str, **kwargs: Any) -> None: resp = ChatResponse(message=token, type="stream", intermediate_steps="")