mirror of
https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git
synced 2025-05-16 02:08:19 +08:00
7753ba2d37
Co-authored-by: Joel <iamjoel007@gmail.com> Co-authored-by: Yeuoly <admin@srmxy.cn> Co-authored-by: JzoNg <jzongcode@gmail.com> Co-authored-by: StyleZhang <jasonapring2015@outlook.com> Co-authored-by: jyong <jyong@dify.ai> Co-authored-by: nite-knite <nkCoding@gmail.com> Co-authored-by: jyong <718720800@qq.com>
54 lines
1.5 KiB
Python
54 lines
1.5 KiB
Python
from enum import Enum
|
|
from typing import Optional
|
|
|
|
from pydantic import BaseModel
|
|
|
|
from core.tools.entities.common_entities import I18nObject
|
|
from core.tools.entities.tool_entities import ToolProviderCredentials
|
|
from core.tools.tool.tool import ToolParameter
|
|
|
|
|
|
class UserTool(BaseModel):
|
|
author: str
|
|
name: str # identifier
|
|
label: I18nObject # label
|
|
description: I18nObject
|
|
parameters: Optional[list[ToolParameter]]
|
|
|
|
class UserToolProvider(BaseModel):
|
|
class ProviderType(Enum):
|
|
BUILTIN = "builtin"
|
|
APP = "app"
|
|
API = "api"
|
|
MODEL = "model"
|
|
|
|
id: str
|
|
author: str
|
|
name: str # identifier
|
|
description: I18nObject
|
|
icon: str
|
|
label: I18nObject # label
|
|
type: ProviderType
|
|
masked_credentials: dict = None
|
|
original_credentials: dict = None
|
|
is_team_authorization: bool = False
|
|
allow_delete: bool = True
|
|
tools: list[UserTool] = None
|
|
|
|
def to_dict(self) -> dict:
|
|
return {
|
|
'id': self.id,
|
|
'author': self.author,
|
|
'name': self.name,
|
|
'description': self.description.to_dict(),
|
|
'icon': self.icon,
|
|
'label': self.label.to_dict(),
|
|
'type': self.type.value,
|
|
'team_credentials': self.masked_credentials,
|
|
'is_team_authorization': self.is_team_authorization,
|
|
'allow_delete': self.allow_delete,
|
|
'tools': self.tools
|
|
}
|
|
|
|
class UserToolProviderCredentials(BaseModel):
|
|
credentials: dict[str, ToolProviderCredentials] |