mirror of
https://github.com/langgenius/dify.git
synced 2025-07-11 11:13:35 +00:00
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
|