RemoteClient should handle metadata event on streaming (#210)

* Handle metadata event on streaming * Make sure not to duplicate logger error messages Resolve: #208
langchain-ai · Nov 11, 2023 · c93e3fd · c93e3fd
1 parent 7e1a1a1
commit c93e3fd
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/langserve/client.py b/langserve/client.py
@@ -512,10 +512,13 @@ def stream(
                         _raise_exception_from_data(
                             sse.data, httpx.Request(method="POST", url=endpoint)
                         )
+                    elif sse.event == "metadata":
+                        # Nothing to do for metadata for the regular remote client.
+                        continue
                     elif sse.event == "end":
                         break
                     else:
-                        logger.error(
+                        _log_error_message_once(
                             f"Encountered an unsupported event type: `{sse.event}`. "
                             f"Try upgrading the remote client to the latest version."
                             f"Ignoring events of type `{sse.event}`."
@@ -593,10 +596,13 @@ async def astream(
                         _raise_exception_from_data(
                             sse.data, httpx.Request(method="POST", url=endpoint)
                         )
+                    elif sse.event == "metadata":
+                        # Nothing to do for metadata for the regular remote client.
+                        continue
                     elif sse.event == "end":
                         break
                     else:
-                        logger.error(
+                        _log_error_message_once(
                             f"Encountered an unsupported event type: `{sse.event}`. "
                             f"Try upgrading the remote client to the latest version."
                             f"Ignoring events of type `{sse.event}`."