Skip to content

Commit

Permalink
fix: remove unused credential validation logic in VectorizerProvider (#…
Browse files Browse the repository at this point in the history
…12000)

Signed-off-by: -LAN- <[email protected]>
  • Loading branch information
laipz8200 authored Dec 23, 2024
1 parent 4e3d732 commit 8978a6a
Showing 1 changed file with 1 addition and 25 deletions.
26 changes: 1 addition & 25 deletions api/core/tools/provider/builtin/vectorizer/vectorizer.py
Original file line number Diff line number Diff line change
@@ -1,32 +1,8 @@
from typing import Any

from core.file import FileTransferMethod, FileType
from core.tools.errors import ToolProviderCredentialValidationError
from core.tools.provider.builtin.vectorizer.tools.vectorizer import VectorizerTool
from core.tools.provider.builtin_tool_provider import BuiltinToolProviderController
from factories import file_factory


class VectorizerProvider(BuiltinToolProviderController):
def _validate_credentials(self, credentials: dict[str, Any]) -> None:
mapping = {
"transfer_method": FileTransferMethod.TOOL_FILE,
"type": FileType.IMAGE,
"id": "test_id",
"url": "https://cloud.dify.ai/logo/logo-site.png",
}
test_img = file_factory.build_from_mapping(
mapping=mapping,
tenant_id="__test_123",
)
try:
VectorizerTool().fork_tool_runtime(
runtime={
"credentials": credentials,
}
).invoke(
user_id="",
tool_parameters={"mode": "test", "image": test_img},
)
except Exception as e:
raise ToolProviderCredentialValidationError(str(e))
return

0 comments on commit 8978a6a

Please sign in to comment.