fix: add langgenius to list tool api (#13578)

pull/13596/head
Yeuoly 1 year ago committed by GitHub
parent 7c1d842cfe
commit 56c7f49625
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

@ -176,6 +176,14 @@ class ModelProviderID(GenericProviderID):
self.provider_name = "gemini" self.provider_name = "gemini"
class ToolProviderID(GenericProviderID):
def __init__(self, value: str, is_hardcoded: bool = False) -> None:
super().__init__(value, is_hardcoded)
if self.organization == "langgenius":
if self.provider_name in ["jina", "siliconflow"]:
self.provider_name = f"{self.provider_name}_tool"
class PluginDependency(BaseModel): class PluginDependency(BaseModel):
class Type(enum.StrEnum): class Type(enum.StrEnum):
Github = PluginInstallationSource.Github.value Github = PluginInstallationSource.Github.value

@ -1,7 +1,6 @@
import datetime import datetime
import json import json
import logging import logging
import sys
import time import time
from collections.abc import Mapping, Sequence from collections.abc import Mapping, Sequence
from concurrent.futures import ThreadPoolExecutor from concurrent.futures import ThreadPoolExecutor
@ -418,8 +417,6 @@ class PluginMigration:
logger.info("Uninstall plugins") logger.info("Uninstall plugins")
sys.exit(-1)
# get installation # get installation
try: try:
installation = manager.list_plugins(fake_tenant_id) installation = manager.list_plugins(fake_tenant_id)

@ -7,7 +7,7 @@ from sqlalchemy.orm import Session
from configs import dify_config from configs import dify_config
from core.helper.position_helper import is_filtered from core.helper.position_helper import is_filtered
from core.model_runtime.utils.encoders import jsonable_encoder from core.model_runtime.utils.encoders import jsonable_encoder
from core.plugin.entities.plugin import GenericProviderID from core.plugin.entities.plugin import GenericProviderID, ToolProviderID
from core.plugin.manager.exc import PluginDaemonClientSideError from core.plugin.manager.exc import PluginDaemonClientSideError
from core.tools.builtin_tool.providers._positions import BuiltinToolProviderSort from core.tools.builtin_tool.providers._positions import BuiltinToolProviderSort
from core.tools.entities.api_entities import ToolApiEntity, ToolProviderApiEntity from core.tools.entities.api_entities import ToolApiEntity, ToolProviderApiEntity
@ -240,10 +240,7 @@ class BuiltinToolManageService:
# rewrite db_providers # rewrite db_providers
for db_provider in db_providers: for db_provider in db_providers:
try: db_provider.provider = str(ToolProviderID(db_provider.provider))
GenericProviderID(db_provider.provider)
except Exception:
db_provider.provider = f"langgenius/{db_provider.provider}/{db_provider.provider}"
# find provider # find provider
def find_provider(provider): def find_provider(provider):

Loading…
Cancel
Save