106 lines
6.0 KiB
Python
106 lines
6.0 KiB
Python
from datetime import datetime
|
|
from decimal import Decimal
|
|
from typing import Any
|
|
|
|
from sqlalchemy import JSON, BigInteger, Boolean, DateTime, Integer, Numeric, String, Text
|
|
from sqlalchemy.dialects.mysql import BIGINT, DATETIME, INTEGER, TINYINT
|
|
from sqlalchemy.orm import Mapped, mapped_column
|
|
|
|
from app.model.base import Base
|
|
|
|
|
|
class ConfigRelease(Base):
|
|
__tablename__ = "config_release"
|
|
id: Mapped[int] = mapped_column(BigInteger, primary_key=True)
|
|
release_no: Mapped[str] = mapped_column(String(64), unique=True, nullable=False)
|
|
title: Mapped[str] = mapped_column(String(128), nullable=False)
|
|
change_summary: Mapped[str] = mapped_column(String(1000), nullable=False)
|
|
status: Mapped[str] = mapped_column(String(20), nullable=False, default="draft")
|
|
created_by: Mapped[int] = mapped_column(BigInteger, nullable=False)
|
|
reviewer_id: Mapped[int | None] = mapped_column(BigInteger)
|
|
reviewed_at: Mapped[datetime | None] = mapped_column(DateTime)
|
|
activated_at: Mapped[datetime | None] = mapped_column(DateTime)
|
|
supersedes_release_id: Mapped[int | None] = mapped_column(BigInteger)
|
|
rollback_of_release_id: Mapped[int | None] = mapped_column(BigInteger)
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, nullable=False)
|
|
updated_at: Mapped[datetime] = mapped_column(DateTime, nullable=False)
|
|
|
|
|
|
class ModelEndpointConfig(Base):
|
|
__tablename__ = "model_endpoint_config"
|
|
id: Mapped[int] = mapped_column(BigInteger, primary_key=True)
|
|
endpoint_code: Mapped[str] = mapped_column(String(64), unique=True, nullable=False)
|
|
provider: Mapped[str] = mapped_column(String(32), nullable=False)
|
|
model_name: Mapped[str] = mapped_column(String(128), nullable=False)
|
|
base_url: Mapped[str] = mapped_column(String(500), nullable=False)
|
|
secret_ref: Mapped[str] = mapped_column(String(256), nullable=False)
|
|
capabilities: Mapped[list[Any]] = mapped_column(JSON, nullable=False)
|
|
allowed_data_levels: Mapped[list[Any]] = mapped_column(JSON, nullable=False)
|
|
context_window: Mapped[int] = mapped_column(INTEGER(unsigned=True), nullable=False)
|
|
timeout_ms: Mapped[int] = mapped_column(Integer, nullable=False, default=15000)
|
|
status: Mapped[str] = mapped_column(String(16), nullable=False, default="disabled")
|
|
created_by: Mapped[int] = mapped_column(BIGINT(unsigned=True), nullable=False)
|
|
reviewer_id: Mapped[int | None] = mapped_column(BIGINT(unsigned=True))
|
|
reviewed_at: Mapped[datetime | None] = mapped_column(DATETIME(fsp=6))
|
|
created_at: Mapped[datetime] = mapped_column(DATETIME(fsp=6), nullable=False)
|
|
updated_at: Mapped[datetime] = mapped_column(DATETIME(fsp=6), nullable=False)
|
|
|
|
|
|
class PlatformConfigItem(Base):
|
|
__tablename__ = "platform_config_item"
|
|
id: Mapped[int] = mapped_column(BigInteger, primary_key=True)
|
|
release_id: Mapped[int] = mapped_column(BigInteger, nullable=False)
|
|
namespace: Mapped[str] = mapped_column(String(64), nullable=False)
|
|
config_key: Mapped[str] = mapped_column(String(128), nullable=False)
|
|
value_json: Mapped[dict[str, Any]] = mapped_column(JSON, nullable=False)
|
|
schema_version: Mapped[str] = mapped_column(String(32), nullable=False)
|
|
checksum: Mapped[str] = mapped_column(String(64), nullable=False)
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, nullable=False)
|
|
|
|
|
|
class PromptTemplateVersion(Base):
|
|
__tablename__ = "prompt_template_version"
|
|
id: Mapped[int] = mapped_column(BigInteger, primary_key=True)
|
|
release_id: Mapped[int] = mapped_column(BigInteger, nullable=False)
|
|
prompt_code: Mapped[str] = mapped_column(String(64), nullable=False)
|
|
task_type: Mapped[str] = mapped_column(String(48), nullable=False)
|
|
agent_type: Mapped[str | None] = mapped_column(String(32))
|
|
version: Mapped[int] = mapped_column(Integer, nullable=False)
|
|
system_prompt: Mapped[str] = mapped_column(String(65535), nullable=False)
|
|
user_prompt_template: Mapped[str] = mapped_column(String(65535), nullable=False)
|
|
input_schema: Mapped[dict[str, Any] | None] = mapped_column(JSON)
|
|
output_schema: Mapped[dict[str, Any] | None] = mapped_column(JSON)
|
|
checksum: Mapped[str] = mapped_column(String(64), nullable=False)
|
|
created_by: Mapped[int] = mapped_column(BigInteger, nullable=False)
|
|
created_at: Mapped[datetime] = mapped_column(DateTime, nullable=False)
|
|
|
|
|
|
class AgentIntentConfig(Base):
|
|
__tablename__ = "agent_intent_config"
|
|
id: Mapped[int] = mapped_column(BigInteger, primary_key=True)
|
|
agent_type: Mapped[str] = mapped_column(String(32), nullable=False)
|
|
intent_code: Mapped[str] = mapped_column(String(64), nullable=False)
|
|
intent_name: Mapped[str] = mapped_column(String(128), nullable=False)
|
|
description: Mapped[str | None] = mapped_column(String(500))
|
|
examples: Mapped[list[Any]] = mapped_column(JSON, nullable=False)
|
|
classifier_instruction: Mapped[str | None] = mapped_column(Text)
|
|
confidence_threshold: Mapped[Decimal] = mapped_column(
|
|
Numeric(5, 4), nullable=False, default=Decimal("0.6000")
|
|
)
|
|
max_clarification_rounds: Mapped[int] = mapped_column(
|
|
TINYINT(unsigned=True), nullable=False, default=2
|
|
)
|
|
transfer_on_failure: Mapped[bool] = mapped_column(Boolean, nullable=False, default=True)
|
|
collection_routes: Mapped[list[Any] | None] = mapped_column(JSON)
|
|
allowed_tools: Mapped[list[Any] | None] = mapped_column(JSON)
|
|
priority: Mapped[int] = mapped_column(INTEGER, nullable=False, default=100)
|
|
version: Mapped[int] = mapped_column(Integer, nullable=False, default=1)
|
|
status: Mapped[str] = mapped_column(String(16), nullable=False, default="draft")
|
|
effective_at: Mapped[datetime | None] = mapped_column(DATETIME(fsp=6))
|
|
expire_at: Mapped[datetime | None] = mapped_column(DATETIME(fsp=6))
|
|
created_by: Mapped[int] = mapped_column(BIGINT(unsigned=True), nullable=False)
|
|
reviewer_id: Mapped[int | None] = mapped_column(BIGINT(unsigned=True))
|
|
reviewed_at: Mapped[datetime | None] = mapped_column(DATETIME(fsp=6))
|
|
created_at: Mapped[datetime] = mapped_column(DATETIME(fsp=6), nullable=False)
|
|
updated_at: Mapped[datetime] = mapped_column(DATETIME(fsp=6), nullable=False)
|