Build lightweight AI agent admin

2026-06-08 18:14:59 +08:00
commit e164840f43
2530 changed files with 435693 additions and 0 deletions
@@ -0,0 +1,77 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+"""
+Data Source Model - 数据源模型
+用于管理系统数据源配置，支持 API、SQL、静态数据等多种类型
+"""
+from sqlalchemy import Column, String, Text, Boolean, Integer, JSON
+
+from app.base_model import BaseModel
+
+
+class DataSource(BaseModel):
+    """
+    数据源模型
+    
+    支持三种数据源类型：
+    1. API - 调用外部或内部 API 接口
+    2. SQL - 执行 SQL 查询（只读）
+    3. Static - 静态数据
+    """
+    __tablename__ = "core_data_source"
+
+    # 所属应用（逻辑外键关联 core_application）
+    application_id = Column(String(21), nullable=True, index=True, comment="所属应用ID")
+
+    # 基本信息
+    name = Column(String(100), nullable=False, comment="数据源名称")
+    code = Column(String(50), unique=True, index=True, nullable=False, comment="数据源编码（唯一标识）")
+    source_type = Column(String(20), default='static', comment="数据源类型: api/sql/static")
+    description = Column(Text, default='', comment="描述说明")
+    status = Column(Boolean, default=True, comment="是否启用")
+
+    # ===== API 配置 =====
+    api_url = Column(String(500), default='', comment="API地址，支持 {param} 占位符")
+    api_method = Column(String(10), default='GET', comment="请求方法: GET/POST/PUT/DELETE/PATCH")
+    api_headers = Column(JSON, default=dict, comment="请求头配置")
+    api_query_params = Column(JSON, default=list, comment="Query参数列表 [{key, value, description, enabled}]")
+    api_body_type = Column(String(20), default='none', comment="请求体类型: none/json/form-data/x-www-form-urlencoded/raw")
+    api_body = Column(JSON, default=dict, comment="请求体模板")
+    api_content_type = Column(String(50), default='', comment="Content-Type，raw模式时使用")
+    api_timeout = Column(Integer, default=30, comment="请求超时时间（秒）")
+    api_data_path = Column(String(100), default='', comment="响应数据路径，如 data.list")
+
+    # ===== API 认证配置 =====
+    api_auth_type = Column(String(20), default='none', comment="认证类型: none/bearer_token/basic_auth/api_key")
+    api_auth_config = Column(JSON, default=dict, comment="认证配置 {token, username, password, key_name, key_value, key_position}")
+
+    # ===== API 高级配置 =====
+    api_retry_count = Column(Integer, default=0, comment="重试次数（0表示不重试）")
+    api_retry_interval = Column(Integer, default=1, comment="重试间隔（秒）")
+    api_success_condition = Column(JSON, default=dict, comment="成功条件 {status_codes, field_path, field_value}")
+    api_proxy = Column(String(200), default='', comment="代理地址")
+    api_follow_redirects = Column(Boolean, default=True, comment="是否跟随重定向")
+    api_verify_ssl = Column(Boolean, default=True, comment="是否验证SSL证书")
+
+    # ===== SQL 配置 =====
+    sql_content = Column(Text, default='', comment="SQL语句，使用 :param 作为参数占位符")
+    db_connection = Column(String(50), default='default', comment="数据库连接名称")
+
+    # ===== 静态数据 =====
+    static_data = Column(JSON, default=list, comment="静态数据（JSON 数组）")
+
+    # ===== 参数定义 =====
+    params = Column(JSON, default=list, comment="参数定义列表")
+
+    # ===== 结果处理 =====
+    result_type = Column(String(20), default='list', comment="结果类型: list/tree/object/value/chart-*")
+    tree_config = Column(JSON, default=dict, comment="树形转换配置")
+    field_mapping = Column(JSON, default=dict, comment="字段映射配置")
+    chart_config = Column(JSON, default=dict, comment="图表配置")
+
+    # ===== 缓存配置 =====
+    cache_enabled = Column(Boolean, default=False, comment="是否启用缓存")
+    cache_ttl = Column(Integer, default=300, comment="缓存时间（秒）")
+
+    def __repr__(self):
+        return f"<DataSource {self.name} ({self.code})>"