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

fix service api blocking mode

Browse Source
This commit is contained in:
takatost
2024-03-20 21:55:06 +08:00
parent a0dde6e4da
commit c3e7299494
13 changed files with 262 additions and 263 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
api/controllers/service_api/app/workflow.py
View File

@ -36,15 +36,18 @@ class WorkflowRunApi(Resource):
parser = reqparse.RequestParser()
parser.add_argument('inputs', type=dict, required=True, nullable=False, location='json')
parser.add_argument('files', type=list, required=False, location='json')
parser.add_argument('response_mode', type=str, choices=['blocking', 'streaming'], location='json')
args = parser.parse_args()
streaming = args.get('response_mode') == 'streaming'
try:
response = AppGenerateService.generate(
app_model=app_model,
user=end_user,
args=args,
invoke_from=InvokeFrom.SERVICE_API,
streaming=True
streaming=streaming
)
return helper.compact_generate_response(response)