From 153616bb993a2f4f72561f04ff11c844efb959a2 Mon Sep 17 00:00:00 2001 From: Eugene Yurtsev Date: Thu, 25 Apr 2024 11:37:51 -0400 Subject: [PATCH] patch: fix feedback in astream log and astream events (#620) Code was checking if feedback was enabled rather than token feedback. --- langserve/api_handler.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/langserve/api_handler.py b/langserve/api_handler.py index d72f09cb..346e89dc 100644 --- a/langserve/api_handler.py +++ b/langserve/api_handler.py @@ -1245,7 +1245,7 @@ async def _stream_log() -> AsyncIterator[dict]: } # Send a metadata event as soon as possible - if not has_sent_metadata and self._enable_feedback_endpoint: + if not has_sent_metadata and self._token_feedback_enabled: if task is None: raise AssertionError("Feedback token task was not created.") if not task.done(): @@ -1354,7 +1354,7 @@ async def _stream_events() -> AsyncIterator[dict]: } # Send a metadata event as soon as possible - if not has_sent_metadata and self._enable_feedback_endpoint: + if not has_sent_metadata and self._token_feedback_enabled: if task is None: raise AssertionError("Feedback token task was not created.") if not task.done():