【Dify】 自定义vanna插件,执行报错问题修复

pull/22121/head
liuchangsheng@wisdomidata.com 11 months ago
parent f6a3f7dfce
commit 2339b6a80c

@ -553,8 +553,7 @@ class LLMNode(BaseNode[LLMNodeData]):
model_schema = model.model_type_instance.get_model_schema(node_data_model.name, model.credentials) model_schema = model.model_type_instance.get_model_schema(node_data_model.name, model.credentials)
if not model_schema: if not model_schema:
raise ModelNotExistError(f"Model {node_data_model.name} not exist.") raise ModelNotExistError(f"Model {node_data_model.name} not exist.")
if "structured_output_enabled" in self.node_data and self.node_data.structured_output_enabled:
if self.node_data.structured_output_enabled:
if model_schema.support_structure_output: if model_schema.support_structure_output:
node_data_model.completion_params = self._handle_native_json_schema( node_data_model.completion_params = self._handle_native_json_schema(
node_data_model.completion_params, model_schema.parameter_rules node_data_model.completion_params, model_schema.parameter_rules
@ -784,7 +783,7 @@ class LLMNode(BaseNode[LLMNodeData]):
) )
if not model_schema: if not model_schema:
raise ModelNotExistError(f"Model {self.node_data.model.name} not exist.") raise ModelNotExistError(f"Model {self.node_data.model.name} not exist.")
if self.node_data.structured_output_enabled: if "structured_output_enabled" in self.node_data and self.node_data.structured_output_enabled:
if not model_schema.support_structure_output: if not model_schema.support_structure_output:
filtered_prompt_messages = self._handle_prompt_based_schema( filtered_prompt_messages = self._handle_prompt_based_schema(
prompt_messages=filtered_prompt_messages, prompt_messages=filtered_prompt_messages,

Loading…
Cancel
Save