youngkingdom/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-05-02 16:38:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: workflow as tool type

Browse Source
This commit is contained in:
Yeuoly
2024-08-29 14:09:47 +08:00
parent c8b0160ea9
commit 4a8d3c54ca
3 changed files with 27 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
api/core/tools/tool/tool.py
View File

@ -1,5 +1,5 @@
from abc import ABC, abstractmethod
from collections.abc import Generator, Mapping
from collections.abc import Generator
from copy import deepcopy
from enum import Enum
from typing import TYPE_CHECKING, Any, Optional, Union

4
api/core/tools/tool/workflow_tool.py
View File

@ -48,6 +48,8 @@ class WorkflowTool(Tool):
from core.app.apps.workflow.app_generator import WorkflowAppGenerator
generator = WorkflowAppGenerator()
assert self.runtime and self.runtime.invoke_from
result = generator.generate(
app_model=app,
workflow=workflow,
@ -154,7 +156,7 @@ class WorkflowTool(Tool):
try:
file_var_list = [FileVar(**f) for f in file]
for file_var in file_var_list:
file_dict = {
file_dict: dict[str, Any] = {
'transfer_method': file_var.transfer_method.value,
'type': file_var.type.value,
}