AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-20 12:39:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: summary and create_file_by_url

Browse Source
This commit is contained in:
Yeuoly 2024-12-02 16:51:37 +08:00
parent fb44cd87e7
commit b10d6051ba
No known key found for this signature in database
GPG Key ID: A66E7E320FB19F61
3 changed files with 9 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -242,11 +242,13 @@ class PluginInvokeSummaryApi(Resource):
def post(self, user_model: Account | EndUser, tenant_model: Tenant, payload: RequestInvokeSummary): def post(self, user_model: Account | EndUser, tenant_model: Tenant, payload: RequestInvokeSummary):
try: try:
return BaseBackwardsInvocationResponse( return BaseBackwardsInvocationResponse(
data=PluginModelBackwardsInvocation.invoke_summary( data={
user_id=user_model.id, "summary": PluginModelBackwardsInvocation.invoke_summary(
tenant=tenant_model, user_id=user_model.id,
payload=payload, tenant=tenant_model,
) payload=payload,
)
}
).model_dump() ).model_dump()
except Exception as e: except Exception as e:
return BaseBackwardsInvocationResponse(error=str(e)).model_dump() return BaseBackwardsInvocationResponse(error=str(e)).model_dump()

5
api/core/tools/tool_file_manager.py
View File

@ -8,9 +8,8 @@ from mimetypes import guess_extension, guess_type
from typing import Optional, Union from typing import Optional, Union
from uuid import uuid4 from uuid import uuid4
from httpx import get
from configs import dify_config from configs import dify_config
from core.helper import ssrf_proxy
from extensions.ext_database import db from extensions.ext_database import db
from extensions.ext_storage import storage from extensions.ext_storage import storage
from models.model import MessageFile from models.model import MessageFile
@ -94,7 +93,7 @@ class ToolFileManager:
) -> ToolFile: ) -> ToolFile:
# try to download image # try to download image
try: try:
response = get(file_url) response = ssrf_proxy.get(file_url)
response.raise_for_status() response.raise_for_status()
blob = response.content blob = response.content
except Exception as e: except Exception as e:

3
api/core/tools/utils/message_transformer.py
View File

@ -30,9 +30,6 @@ class ToolFileMessageTransformer:
): ):
# try to download image # try to download image
try: try:
if not conversation_id:
raise
assert isinstance(message.message, ToolInvokeMessage.TextMessage) assert isinstance(message.message, ToolInvokeMessage.TextMessage)
file = ToolFileManager.create_file_by_url( file = ToolFileManager.create_file_by_url(