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

integrate model provider with plugin daemon

Browse Source
This commit is contained in:
takatost
2024-10-30 18:56:52 -07:00
parent 459cb9dd72
commit 18edeb8e0a
9 changed files with 58 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
api/core/plugin/manager/base.py
View File

@ -1,4 +1,5 @@
import json
import logging
from collections.abc import Callable, Generator
from typing import Optional, TypeVar
@ -21,6 +22,8 @@ plugin_daemon_inner_api_key = dify_config.PLUGIN_API_KEY
T = TypeVar("T", bound=(BaseModel | dict | list | bool | str))
logger = logging.getLogger(__name__)
class BasePluginManager:
def _request(
@ -44,9 +47,14 @@ class BasePluginManager:
if headers.get("Content-Type") == "application/json" and isinstance(data, dict):
data = json.dumps(data)
response = requests.request(
method=method, url=str(url), headers=headers, data=data, params=params, stream=stream, files=files
)
try:
response = requests.request(
method=method, url=str(url), headers=headers, data=data, params=params, stream=stream, files=files
)
except requests.exceptions.ConnectionError as e:
logger.exception(f"Request to Plugin Daemon Service failed: {e}")
raise ValueError("Request to Plugin Daemon Service failed")
return response
def _stream_request(