Procházet zdrojové kódy

fix: stream with empty line

Yeuoly před 8 měsíci
rodič
revize
fb3a701c86

+ 2 - 1
api/core/plugin/manager/base.py

@@ -56,7 +56,8 @@ class BasePluginManager:
             line = line.decode("utf-8").strip()
             if line.startswith("data:"):
                 line = line[5:].strip()
-            yield line
+            if line:
+                yield line
 
     def _stream_request_with_model(
         self,

+ 2 - 2
api/core/workflow/nodes/tool/tool_node.py

@@ -212,8 +212,8 @@ class ToolNode(BaseNode):
                     chunk_content=message.message.text, from_variable_selector=[self.node_id, "text"]
                 )
             elif message.type == ToolInvokeMessage.MessageType.JSON:
-                assert isinstance(message, ToolInvokeMessage.JsonMessage)
-                json.append(message.json_object)
+                assert isinstance(message.message, ToolInvokeMessage.JsonMessage)
+                json.append(message.message.json_object)
             elif message.type == ToolInvokeMessage.MessageType.LINK:
                 assert isinstance(message.message, ToolInvokeMessage.TextMessage)
                 stream_text = f"Link: {message.message.text}\n"