diff --git a/apps/application/workflow/nodes/text_to_video_node/text_to_video_node.py b/apps/application/workflow/nodes/text_to_video_node/text_to_video_node.py index 5fe0a47d5eb..66b6cf9f6c6 100644 --- a/apps/application/workflow/nodes/text_to_video_node/text_to_video_node.py +++ b/apps/application/workflow/nodes/text_to_video_node/text_to_video_node.py @@ -93,8 +93,8 @@ def execute(self): question = self.workflow_manage.generate_prompt(prompt) self.write_context("question", question) - message_list = [*history_message, question] - self.write_context("message_list", message_list) + # message_list = [*history_message, question] + # self.write_context("message_list", [{"content": m.content, "role": m.type} for m in message_list],) self.write_context("dialogue_type", dialogue_type) self.write_context("negative_prompt", self.workflow_manage.generate_prompt(negative_prompt)) @@ -119,7 +119,7 @@ def execute(self): self.write_context("answer", video_label) self.write_context("video", video_list) - self.write_context("chat_model", ttv_model) + # self.write_context("chat_model", ttv_model) if is_result: node_info = NodeInfo(self.get_node_id(), self.get_node_name(), Status.SUCCESS) diff --git a/ui/src/workflow-canvas/icons/text-to-video-node-icon.vue b/ui/src/workflow-canvas/icons/text-to-video-node-icon.vue new file mode 100644 index 00000000000..7ea4a139f8c --- /dev/null +++ b/ui/src/workflow-canvas/icons/text-to-video-node-icon.vue @@ -0,0 +1,6 @@ + + diff --git a/ui/src/workflow-canvas/nodes/text-to-video-node/index.ts b/ui/src/workflow-canvas/nodes/text-to-video-node/index.ts new file mode 100644 index 00000000000..c23206ba290 --- /dev/null +++ b/ui/src/workflow-canvas/nodes/text-to-video-node/index.ts @@ -0,0 +1,15 @@ +import TextToVideoNodeVue from './index.vue' +import { WorkflowNodeModel, WorkflowNodeView } from '@/workflow-canvas/core/workflow-node' +import { WorkflowNodeType } from '@/workflow-canvas/types.ts' + +class TextToVideoNodeView extends WorkflowNodeView { + constructor(props: ConstructorParameters[0]) { + super(props, TextToVideoNodeVue) + } +} + +export default { + type: WorkflowNodeType.TextToVideoGenerateNode, + model: WorkflowNodeModel, + view: TextToVideoNodeView, +} diff --git a/ui/src/workflow-canvas/nodes/text-to-video-node/index.vue b/ui/src/workflow-canvas/nodes/text-to-video-node/index.vue new file mode 100644 index 00000000000..52b20250d81 --- /dev/null +++ b/ui/src/workflow-canvas/nodes/text-to-video-node/index.vue @@ -0,0 +1,170 @@ + + + +