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

fix: types

Browse Source
This commit is contained in:
Yeuoly
2024-08-29 20:06:14 +08:00
parent ec711d094d
commit 113ff27d07
4 changed files with 44 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
api/controllers/inner_api/plugin/plugin.py
View File

@ -7,7 +7,7 @@ from controllers.console.setup import setup_required
from controllers.inner_api import api
from controllers.inner_api.plugin.wraps import get_tenant, plugin_data
from controllers.inner_api.wraps import plugin_inner_api_only
from core.plugin.backwards_invocation.model import PluginBackwardsInvocation
from core.plugin.backwards_invocation.model import PluginModelBackwardsInvocation
from core.plugin.entities.request import (
RequestInvokeLLM,
RequestInvokeModeration,
@ -29,7 +29,7 @@ class PluginInvokeLLMApi(Resource):
@plugin_data(payload_type=RequestInvokeLLM)
def post(self, user_id: str, tenant_model: Tenant, payload: RequestInvokeLLM):
def generator():
response = PluginBackwardsInvocation.invoke_llm(user_id, tenant_model, payload)
response = PluginModelBackwardsInvocation.invoke_llm(user_id, tenant_model, payload)
if isinstance(response, Generator):
for chunk in response:
yield chunk.model_dump_json().encode() + b'\n\n'