From c93e3fde2cf265d621a532d91f5083c9c3229050 Mon Sep 17 00:00:00 2001 From: Eugene Yurtsev Date: Fri, 10 Nov 2023 21:36:24 -0500 Subject: [PATCH] RemoteClient should handle metadata event on streaming (#210) * Handle metadata event on streaming * Make sure not to duplicate logger error messages Resolve: https://github.com/langchain-ai/langserve/issues/208 --- langserve/client.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/langserve/client.py b/langserve/client.py index 209bdc41..d8f17a16 100644 --- 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}`."