busya
c6189876b3
Auto-fixed categories: - F401: unused imports removed - I001: import blocks sorted - W293: trailing whitespace stripped - UP035: deprecated typing imports replaced - SIM: simplify suggestions applied - ARG: unused args prefixed with underscore - T201: print statements removed - F841: unused variables removed - RUF059: unpacked variables prefixed Remaining ~1500 unfixable errors (C901, B904, N806, E402) require manual work. Backend smoke tests: 13/13 passed.
70 lines
2.9 KiB
Python
70 lines
2.9 KiB
Python
# #region LlmModels [C:3] [TYPE Module] [SEMANTICS sqlalchemy, llm, model, schema, validate, provider]
|
|
# @BRIEF SQLAlchemy models for LLM provider configuration and validation results.
|
|
# @LAYER: Domain
|
|
# @RELATION INHERITS_FROM -> MappingModels:Base
|
|
|
|
import uuid
|
|
from datetime import datetime
|
|
|
|
from sqlalchemy import JSON, Boolean, Column, DateTime, String, Text, Time
|
|
|
|
from .mapping import Base
|
|
|
|
|
|
def generate_uuid():
|
|
return str(uuid.uuid4())
|
|
|
|
# #region ValidationPolicy [TYPE Class]
|
|
# @BRIEF Defines a scheduled rule for validating a group of dashboards within an execution window.
|
|
class ValidationPolicy(Base):
|
|
__tablename__ = "validation_policies"
|
|
|
|
id = Column(String, primary_key=True, default=generate_uuid)
|
|
name = Column(String, nullable=False)
|
|
environment_id = Column(String, nullable=False)
|
|
is_active = Column(Boolean, default=True)
|
|
dashboard_ids = Column(JSON, nullable=False) # Array of dashboard IDs
|
|
schedule_days = Column(JSON, nullable=False) # Array of integers (0-6)
|
|
window_start = Column(Time, nullable=False)
|
|
window_end = Column(Time, nullable=False)
|
|
notify_owners = Column(Boolean, default=True)
|
|
custom_channels = Column(JSON, nullable=True) # List of external channels
|
|
alert_condition = Column(String, default="FAIL_ONLY") # FAIL_ONLY, WARN_AND_FAIL, ALWAYS
|
|
created_at = Column(DateTime, default=datetime.utcnow)
|
|
updated_at = Column(DateTime, default=datetime.utcnow, onupdate=datetime.utcnow)
|
|
# #endregion ValidationPolicy
|
|
|
|
# #region LLMProvider [TYPE Class]
|
|
# @BRIEF SQLAlchemy model for LLM provider configuration.
|
|
class LLMProvider(Base):
|
|
__tablename__ = "llm_providers"
|
|
|
|
id = Column(String, primary_key=True, default=generate_uuid)
|
|
provider_type = Column(String, nullable=False) # openai, openrouter, kilo
|
|
name = Column(String, nullable=False)
|
|
base_url = Column(String, nullable=False)
|
|
api_key = Column(String, nullable=False) # Should be encrypted
|
|
default_model = Column(String, nullable=False)
|
|
is_active = Column(Boolean, default=True)
|
|
created_at = Column(DateTime, default=datetime.utcnow)
|
|
# #endregion LLMProvider
|
|
|
|
# #region ValidationRecord [TYPE Class]
|
|
# @BRIEF SQLAlchemy model for dashboard validation history.
|
|
class ValidationRecord(Base):
|
|
__tablename__ = "llm_validation_results"
|
|
|
|
id = Column(String, primary_key=True, default=generate_uuid)
|
|
task_id = Column(String, nullable=True, index=True) # Reference to TaskRecord
|
|
dashboard_id = Column(String, nullable=False, index=True)
|
|
environment_id = Column(String, nullable=True, index=True)
|
|
timestamp = Column(DateTime, default=datetime.utcnow)
|
|
status = Column(String, nullable=False) # PASS, WARN, FAIL, UNKNOWN
|
|
screenshot_path = Column(String, nullable=True)
|
|
issues = Column(JSON, nullable=False)
|
|
summary = Column(Text, nullable=False)
|
|
raw_response = Column(Text, nullable=True)
|
|
# #endregion ValidationRecord
|
|
|
|
# #endregion LlmModels
|