fix(api): Fix the issue that streaming output not working with multimodal output

pull/17372/head
QuantumGhost 1 year ago
parent 435537287e
commit 77ab325f86

@ -79,7 +79,6 @@ from extensions.ext_database import db
from models.model import Conversation from models.model import Conversation
from models.provider import Provider, ProviderType from models.provider import Provider, ProviderType
from models.workflow import WorkflowNodeExecutionStatus from models.workflow import WorkflowNodeExecutionStatus
from .entities import ( from .entities import (
LLMNodeChatModelMessage, LLMNodeChatModelMessage,
LLMNodeCompletionModelPromptTemplate, LLMNodeCompletionModelPromptTemplate,
@ -349,7 +348,7 @@ class LLMNode(BaseNode[LLMNodeData]):
text_chunk = self._image_file_to_markdown(file) text_chunk = self._image_file_to_markdown(file)
else: else:
raise UnsupportedPromptContentTypeError(type_name=str(type(content))) raise UnsupportedPromptContentTypeError(type_name=str(type(content)))
yield RunStreamChunkEvent(chunk_content=text_chunk, from_variable_selector=[self.node_id]) yield RunStreamChunkEvent(chunk_content=text_chunk, from_variable_selector=[self.node_id, "text"])
full_text += text_chunk full_text += text_chunk
# Update the whole metadata # Update the whole metadata

Loading…
Cancel
Save