patch: fix feedback in astream log and astream events (#620)

Code was checking if feedback was enabled rather than token feedback.
langchain-ai · Apr 25, 2024 · 153616b · 153616b
1 parent 109445b
commit 153616b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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():