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

This commit is contained in:
-LAN- 2024-11-26 14:14:14 +08:00 committed by GitHub
parent 5b7b328193
commit 044e7b63c2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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