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

add to http request node convert tests

Browse Source
This commit is contained in:
takatost
2024-02-23 18:18:49 +08:00
parent 45621ba4d7
commit 0806b3163a
7 changed files with 210 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
api/services/app_model_config_service.py
View File

@ -205,7 +205,7 @@ class AppModelConfigService:
variables = []
for item in config["user_input_form"]:
key = list(item.keys())[0]
if key not in ["text-input", "select", "paragraph", "external_data_tool"]:
if key not in ["text-input", "select", "paragraph", "number", "external_data_tool"]:
raise ValueError("Keys in user_input_form list can only be 'text-input', 'paragraph' or 'select'")
form_item = item[key]

24
api/services/workflow/workflow_converter.py
View File

@ -190,10 +190,10 @@ class WorkflowConverter:
api_based_extension_id = tool_config.get("api_based_extension_id")
# get api_based_extension
api_based_extension = db.session.query(APIBasedExtension).filter(
APIBasedExtension.tenant_id == tenant_id,
APIBasedExtension.id == api_based_extension_id
).first()
api_based_extension = self._get_api_based_extension(
tenant_id=tenant_id,
api_based_extension_id=api_based_extension_id
)
if not api_based_extension:
raise ValueError("[External data tool] API query failed, variable: {}, "
@ -259,7 +259,6 @@ class WorkflowConverter:
}
}
}
index += 1
nodes.append(http_request_node)
@ -268,7 +267,7 @@ class WorkflowConverter:
"id": f"code-{index}",
"position": None,
"data": {
"title": f"Parse {api_based_extension.name} response",
"title": f"Parse {api_based_extension.name} Response",
"type": NodeType.CODE.value,
"variables": [{
"variable": "response_json",
@ -287,6 +286,7 @@ class WorkflowConverter:
}
nodes.append(code_node)
index += 1
return nodes
@ -513,3 +513,15 @@ class WorkflowConverter:
return AppMode.WORKFLOW
else:
return AppMode.value_of(app_model.mode)
def _get_api_based_extension(self, tenant_id: str, api_based_extension_id: str) -> APIBasedExtension:
"""
Get API Based Extension
:param tenant_id: tenant id
:param api_based_extension_id: api based extension id
:return:
"""
return db.session.query(APIBasedExtension).filter(
APIBasedExtension.tenant_id == tenant_id,
APIBasedExtension.id == api_based_extension_id
).first()