Merge branch 'fix/chore-fix' into dev/plugin-deploy
Some checks are pending
Build and Push API & Web / build (api, DIFY_API_IMAGE_NAME, linux/amd64, build-api-amd64) (push) Waiting to run
Build and Push API & Web / build (api, DIFY_API_IMAGE_NAME, linux/arm64, build-api-arm64) (push) Waiting to run
Build and Push API & Web / build (web, DIFY_WEB_IMAGE_NAME, linux/amd64, build-web-amd64) (push) Waiting to run
Build and Push API & Web / build (web, DIFY_WEB_IMAGE_NAME, linux/arm64, build-web-arm64) (push) Waiting to run
Build and Push API & Web / create-manifest (api, DIFY_API_IMAGE_NAME, merge-api-images) (push) Blocked by required conditions
Build and Push API & Web / create-manifest (web, DIFY_WEB_IMAGE_NAME, merge-web-images) (push) Blocked by required conditions

This commit is contained in:
Yeuoly 2024-11-15 19:54:56 +08:00
commit 2698b1c222
No known key found for this signature in database
GPG Key ID: A66E7E320FB19F61
3 changed files with 28 additions and 10 deletions

View File

@ -0,0 +1,18 @@
from flask_restful import Resource, reqparse
from controllers.files import api
class PluginFilePreviewApi(Resource):
def get(self, file_id: str, file_type: str):
parser = reqparse.RequestParser()
parser.add_argument("timestamp", type=str, required=True, location="args")
parser.add_argument("nonce", type=str, required=True, location="args")
parser.add_argument("sign", type=str, required=True, location="args")
parser.add_argument("as_attachment", type=bool, required=False, default=False, location="args")
args = parser.parse_args()
api.add_resource(PluginFilePreviewApi, "/files/<path:file_id>/<path:file_type>/plugin-file-preview")

View File

@ -45,15 +45,9 @@ class BuiltinToolProviderController(ToolProviderController):
), ),
) )
def _get_builtin_tools(self) -> list[BuiltinTool]: self._load_tools()
"""
returns a list of tools that the provider can provide
:return: list of tools
"""
if self.tools:
return self.tools
def _load_tools(self):
provider = self.entity.identity.name provider = self.entity.identity.name
tool_path = path.join(path.dirname(path.realpath(__file__)), "providers", provider, "tools") tool_path = path.join(path.dirname(path.realpath(__file__)), "providers", provider, "tools")
# get all the yaml files in the tool path # get all the yaml files in the tool path
@ -86,7 +80,14 @@ class BuiltinToolProviderController(ToolProviderController):
) )
self.tools = tools self.tools = tools
return tools
def _get_builtin_tools(self) -> list[BuiltinTool]:
"""
returns a list of tools that the provider can provide
:return: list of tools
"""
return self.tools
def get_credentials_schema(self) -> list[ProviderConfig]: def get_credentials_schema(self) -> list[ProviderConfig]:
""" """

View File

@ -18,7 +18,6 @@ class PluginTool(Tool):
self.tenant_id = tenant_id self.tenant_id = tenant_id
self.runtime_parameters = None self.runtime_parameters = None
@property
def tool_provider_type(self) -> ToolProviderType: def tool_provider_type(self) -> ToolProviderType:
return ToolProviderType.PLUGIN return ToolProviderType.PLUGIN