fix(llm_node): Ignore file if not supported. (#11114)

pull/11122/head
-LAN- 1 year ago committed by GitHub
parent 5b7b328193
commit 044e7b63c2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -67,7 +67,6 @@ from .entities import (
ModelConfig, ModelConfig,
) )
from .exc import ( from .exc import (
FileTypeNotSupportError,
InvalidContextStructureError, InvalidContextStructureError,
InvalidVariableTypeError, InvalidVariableTypeError,
LLMModeRequiredError, LLMModeRequiredError,
@ -676,7 +675,7 @@ class LLMNode(BaseNode[LLMNodeData]):
and ModelFeature.AUDIO not in model_config.model_schema.features and ModelFeature.AUDIO not in model_config.model_schema.features
) )
): ):
raise FileTypeNotSupportError(type_name=content_item.type) continue
prompt_message_content.append(content_item) prompt_message_content.append(content_item)
if len(prompt_message_content) == 1 and prompt_message_content[0].type == PromptMessageContentType.TEXT: if len(prompt_message_content) == 1 and prompt_message_content[0].type == PromptMessageContentType.TEXT:
prompt_message.content = prompt_message_content[0].data prompt_message.content = prompt_message_content[0].data

Loading…
Cancel
Save