从数据库客户端到 AI Agent：NovaDB 的第二条命

pub async fn stream_chat_completion(
    config: &AiConfig,
    messages: &mut Vec<ChatMessage>,
    app: &AppHandle,
    chat_id: &str,
    connection_id: &str,
    abort_flag: &AbortFlag,
) -> AppResult<()> {
    // 1. 发起 SSE 流式请求
    // 2. 解析每个 delta：文本内容 → 实时推送到前端
    // 3. 遇到 tool_call → 执行工具，拿到结果
    // 4. 把工具结果追加到 messages 里
    // 5. 递归调用自己，把更新后的对话发回给 LLM
}

// tools.rs — 工具定义
pub fn tool_definitions() -> Vec<ToolDefinition> {
    vec![
        // 执行任意 SQL
        tool("run_sql", "Execute SQL query", vec![("sql", "string", true)]),
        // 获取完整 schema 树
        tool("get_schema", "Load full database schema", vec![]),
        // 获取单张表的详细信息
        tool("get_table_info", "Get table details", vec![("qualified_name", "string", true)]),
        // 获取数据库级别的信息
        tool("get_database_info", "Get database info", vec![("database_name", "string", true)]),
        // 获取数据库版本
        tool("get_server_version", "Get server version", vec![]),
    ]
}

fn build_system_prompt(config: &AiConfig, profile: &ConnectionProfile,
                       version: &str, nodes: &[SchemaNode]) -> String {
    let mut prompt = config.system_prompt.clone();
 
    // 注入当前连接信息
    prompt.push_str(&format!(
        "\n\nCurrent connection: {} ({}), Database: {}",
        profile.name, profile.db_type, profile.database
    ));
 
    // 注入服务器版本
    prompt.push_str(&format!("\nServer version: {}", version));
 
    // 注入 compact schema 摘要
    prompt.push_str("\n\nCurrent database schema:\n");
    prompt.push_str(&build_schema_summary(nodes, 0));
 
    prompt
}

function useLiveTimer(startedAt?: number, stoppedAt?: number): string {
    const [elapsed, setElapsed] = useState(0);
    useEffect(() => {
        if (!startedAt || stoppedAt) return;
        const id = setInterval(() => {
            setElapsed(Date.now() - startedAt);
        }, 100);
        return () => clearInterval(id);
    }, [startedAt, stoppedAt]);
    // 格式化为 "Xms" / "X.Xs" / "Xm Ys"
}

// 有外键关系的表 → Dagre 分层布局
const connectedLayout = new dagre.graphlib.Graph();
connectedLayout.setDefaultEdgeLabel(() => ({}));
connectedLayout.setGraph({ rankdir: "LR", nodesep: 80, ranksep: 160 });
 
// 没有外键关系的表 → 自适应网格
const disconnectedCols = Math.min(Math.ceil(Math.sqrt(disconnected.length)), 5);

-- 角色表
CREATE TABLE characters (
    id BIGSERIAL PRIMARY KEY,
    name VARCHAR(50) NOT NULL UNIQUE,
    class VARCHAR(20) NOT NULL,
    level INTEGER DEFAULT 1,
    experience BIGINT DEFAULT 0,
    hp INTEGER NOT NULL,
    mp INTEGER NOT NULL,
    created_at TIMESTAMPTZ DEFAULT NOW()
);
 
-- 物品表
CREATE TABLE items (
    id BIGSERIAL PRIMARY KEY,
    name VARCHAR(100) NOT NULL,
    type VARCHAR(30) NOT NULL,
    rarity VARCHAR(20) DEFAULT 'common',
    description TEXT,
    stats JSONB
);
 
-- 背包（多对多关系）
CREATE TABLE character_inventory (
    id BIGSERIAL PRIMARY KEY,
    character_id BIGINT REFERENCES characters(id) ON DELETE CASCADE,
    item_id BIGINT REFERENCES items(id),
    quantity INTEGER DEFAULT 1,
    equipped BOOLEAN DEFAULT FALSE,
    acquired_at TIMESTAMPTZ DEFAULT NOW()
);
 
-- 技能表
CREATE TABLE skills (
    id BIGSERIAL PRIMARY KEY,
    name VARCHAR(100) NOT NULL,
    type VARCHAR(20) NOT NULL,
    mana_cost INTEGER DEFAULT 0,
    cooldown_seconds REAL DEFAULT 0,
    description TEXT
);
 
-- 角色技能（多对多）
CREATE TABLE character_skills (
    character_id BIGINT REFERENCES characters(id) ON DELETE CASCADE,
    skill_id BIGINT REFERENCES skills(id),
    level INTEGER DEFAULT 1,
    PRIMARY KEY (character_id, skill_id)
);
 
-- 任务表
CREATE TABLE quests (
    id BIGSERIAL PRIMARY KEY,
    name VARCHAR(200) NOT NULL,
    description TEXT,
    required_level INTEGER DEFAULT 1,
    reward_xp BIGINT DEFAULT 0,
    reward_items JSONB
);
 
-- 角色任务进度
CREATE TABLE character_quests (
    character_id BIGINT REFERENCES characters(id) ON DELETE CASCADE,
    quest_id BIGINT REFERENCES quests(id),
    status VARCHAR(20) DEFAULT 'active',
    progress JSONB DEFAULT '{}',
    started_at TIMESTAMPTZ DEFAULT NOW(),
    completed_at TIMESTAMPTZ,
    PRIMARY KEY (character_id, quest_id)
);
 
-- 索引
CREATE INDEX idx_inventory_character ON character_inventory(character_id);
CREATE INDEX idx_inventory_item ON character_inventory(item_id);
CREATE INDEX idx_char_skills_character ON character_skills(character_id);
CREATE INDEX idx_char_quests_character ON character_quests(character_id);
CREATE INDEX idx_char_quests_status ON character_quests(status);

// tools.rs
let result = query_service::run_query(app, connection_id, sql).await?;
 
// 截断到 100 行
let truncated = result.rows.len() > 100;
let rows = if truncated {
    result.rows[..100].to_vec()
} else {
    result.rows
};
 
let response = ToolResponse {
    columns: result.columns,
    rows,
    row_count: result.row_count,
    truncated,
    note: if truncated {
        Some("Results truncated to 100 rows".to_string())
    } else {
        None
    },
    // ...
};

// 每个连接维护独立的对话列表
aiConversations: Record<string, ChatConversation[]>
// 切换连接时，对话跟着切换

interface AiConfig {
    apiKey: string;
    baseUrl: string;      // 你想用什么服务就填什么
    model: string;         // 模型名
    maxTokens: number;     // 最大输出 token
    systemPrompt: string;  // 可自定义的 system prompt
}