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

fix: convert backwards invocation into BaseBackwardsResponse

Browse Source
This commit is contained in:
Yeuoly
2024-11-19 14:03:40 +08:00
parent 97a3727962
commit 8b30099672
2 changed files with 3 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
api/core/plugin/backwards_invocation/base.py
View File

@ -1,4 +1,3 @@
import json
from collections.abc import Generator
from typing import Generic, Optional, TypeVar
@ -19,10 +18,7 @@ class BaseBackwardsInvocation:
error_message = BaseBackwardsInvocationResponse(error=str(e)).model_dump_json()
yield f"{error_message}\n\n".encode()
else:
if isinstance(response, BaseModel):
yield response.model_dump_json().encode() + b"\n\n"
else:
yield json.dumps(response).encode() + b"\n\n"
yield BaseBackwardsInvocationResponse(data=response).model_dump_json().encode() + b"\n\n"
T = TypeVar("T", bound=dict | str | bool | int | BaseModel)