impl: basic plugin manager

api/configs/feature/__init__.py

@@ -115,7 +115,7 @@ class PluginConfig(BaseSettings):
     """
     Plugin configs
     """
-    PLUGIN_API_URL: str = Field(
+    PLUGIN_API_URL: HttpUrl = Field(
         description='Plugin API URL',
         default='http://plugin:5002',
     )

api/core/plugin/manager/asset.py

@@ -0,0 +1,5 @@
+from core.plugin.manager.base import BasePluginManager
+
+
+class PluginAssetManager(BasePluginManager):
+    pass

api/core/plugin/manager/base.py

@@ -0,0 +1,48 @@
+import json
+from collections.abc import Generator
+from typing import TypeVar
+
+import requests
+from pydantic import BaseModel
+from yarl import URL
+
+from configs import dify_config
+
+plugin_daemon_inner_api_baseurl = dify_config.PLUGIN_API_URL
+plugin_daemon_inner_api_key = dify_config.INNER_API_KEY_FOR_PLUGIN
+
+T = TypeVar("T", bound=(BaseModel | dict))
+
+
+class BasePluginManager:
+    def _request(self, method: str, path: str, headers: dict, data: bytes, stream: bool = False) -> requests.Response:
+        """
+        Make a request to the plugin daemon inner API.
+        """
+        url = URL(str(plugin_daemon_inner_api_baseurl)) / path
+        headers["X-Api-Key"] = plugin_daemon_inner_api_key
+        response = requests.request(method=method, url=str(url), headers=headers, data=data, stream=stream)
+        return response
+
+    def _stream_request(self, method: str, path: str, headers: dict, data: bytes) -> Generator[bytes, None, None]:
+        """
+        Make a stream request to the plugin daemon inner API
+        """
+        response = self._request(method, path, headers, data, stream=True)
+        yield from response.iter_lines()
+
+    def _stream_request_with_model(
+        self, method: str, path: str, headers: dict, data: bytes, type: type[T]
+    ) -> Generator[T, None, None]:
+        """
+        Make a stream request to the plugin daemon inner API and yield the response as a model.
+        """
+        for line in self._stream_request(method, path, headers, data):
+            yield type(**json.loads(line))
+
+    def _request_with_model(self, method: str, path: str, headers: dict, data: bytes, type: type[T]) -> T:
+        """
+        Make a request to the plugin daemon inner API and return the response as a model.
+        """
+        response = self._request(method, path, headers, data)
+        return type(**response.json())

api/core/plugin/manager/endpoint.py

@@ -0,0 +1,5 @@
+from core.plugin.manager.base import BasePluginManager
+
+
+class PluginEndpointManager(BasePluginManager):
+    pass

api/core/plugin/manager/model.py

@@ -0,0 +1,5 @@
+from core.plugin.manager.base import BasePluginManager
+
+
+class PluginModelManager(BasePluginManager):
+    pass

api/core/plugin/manager/tool.py

@@ -0,0 +1,5 @@
+from core.plugin.manager.base import BasePluginManager
+
+
+class PluginToolManager(BasePluginManager):
+    pass