mirror of
https://gitee.com/dify_ai/dify.git
synced 2024-12-10 22:49:13 +08:00
d109881410
Co-authored-by: -LAN- <laipz8200@outlook.com>
28 lines
957 B
Python
28 lines
957 B
Python
import enum
|
|
|
|
from extensions.ext_database import db
|
|
|
|
from .types import StringUUID
|
|
|
|
|
|
class APIBasedExtensionPoint(enum.Enum):
|
|
APP_EXTERNAL_DATA_TOOL_QUERY = "app.external_data_tool.query"
|
|
PING = "ping"
|
|
APP_MODERATION_INPUT = "app.moderation.input"
|
|
APP_MODERATION_OUTPUT = "app.moderation.output"
|
|
|
|
|
|
class APIBasedExtension(db.Model):
|
|
__tablename__ = "api_based_extensions"
|
|
__table_args__ = (
|
|
db.PrimaryKeyConstraint("id", name="api_based_extension_pkey"),
|
|
db.Index("api_based_extension_tenant_idx", "tenant_id"),
|
|
)
|
|
|
|
id = db.Column(StringUUID, server_default=db.text("uuid_generate_v4()"))
|
|
tenant_id = db.Column(StringUUID, nullable=False)
|
|
name = db.Column(db.String(255), nullable=False)
|
|
api_endpoint = db.Column(db.String(255), nullable=False)
|
|
api_key = db.Column(db.Text, nullable=False)
|
|
created_at = db.Column(db.DateTime, nullable=False, server_default=db.text("CURRENT_TIMESTAMP(0)"))
|