mirror of
https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git
synced 2025-05-15 08:08:16 +08:00
86286e1ac8
Co-authored-by: chenhe <guchenhe@gmail.com> Co-authored-by: Pascal M <11357019+perzeuss@users.noreply.github.com>
34 lines
920 B
Python
34 lines
920 B
Python
from pydantic import BaseModel
|
|
from typing import Dict, Optional, Any, List
|
|
|
|
from core.tools.entities.tool_entities import ToolProviderType, ToolParamter
|
|
|
|
class ApiBasedToolBundle(BaseModel):
|
|
"""
|
|
This class is used to store the schema information of an api based tool. such as the url, the method, the parameters, etc.
|
|
"""
|
|
# server_url
|
|
server_url: str
|
|
# method
|
|
method: str
|
|
# summary
|
|
summary: Optional[str] = None
|
|
# operation_id
|
|
operation_id: str = None
|
|
# parameters
|
|
parameters: Optional[List[ToolParamter]] = None
|
|
# author
|
|
author: str
|
|
# icon
|
|
icon: Optional[str] = None
|
|
# openapi operation
|
|
openapi: dict
|
|
|
|
class AppToolBundle(BaseModel):
|
|
"""
|
|
This class is used to store the schema information of an tool for an app.
|
|
"""
|
|
type: ToolProviderType
|
|
credential: Optional[Dict[str, Any]] = None
|
|
provider_id: str
|
|
tool_name: str |