fix: LLM和搜索调用时记录统计到backend

backend/app.py

@@ -166,6 +166,21 @@ def init_db():
         )
     ''')
     
+    # 用户智能体配置表（我的智能体）
+    cursor.execute('''
+        CREATE TABLE IF NOT EXISTS user_agents (
+            id INTEGER PRIMARY KEY AUTOINCREMENT,
+            user_id INTEGER NOT NULL,
+            agent_id TEXT NOT NULL,
+            category TEXT NOT NULL,
+            is_pinned INTEGER DEFAULT 0,
+            is_favorite INTEGER DEFAULT 0,
+            added_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
+            FOREIGN KEY (user_id) REFERENCES users(id),
+            UNIQUE(user_id, agent_id)
+        )
+    ''')
+    
     # 对话表（用户对话数据）
     cursor.execute('''
         CREATE TABLE IF NOT EXISTS conversations (
@@ -626,6 +641,36 @@ def get_user_conversations(user_id):
     return jsonify(conversations)
 
 
+@app.route('/api/user/<int:user_id>/conversations/<int:conv_id>', methods=['GET'])
+def get_user_conversation_detail(user_id, conv_id):
+    """获取单个对话详情"""
+    conn = get_db()
+    cursor = conn.cursor()
+    cursor.execute('''
+        SELECT id, title, agent_id, messages, created_at, updated_at 
+        FROM conversations WHERE id = ? AND user_id = ?
+    ''', (conv_id, user_id))
+    
+    row = cursor.fetchone()
+    conn.close()
+    
+    if not row:
+        return jsonify({'error': '对话不存在'}), 404
+    
+    conv = dict(row)
+    try:
+        conv['messages'] = json.loads(conv['messages']) if conv['messages'] else []
+    except:
+        conv['messages'] = []
+    
+    conv['id'] = str(conv['id'])
+    conv['createdAt'] = int(datetime.strptime(conv['created_at'], '%Y-%m-%d %H:%M:%S').timestamp() * 1000) if conv['created_at'] else 0
+    conv['updatedAt'] = int(datetime.strptime(conv['updated_at'], '%Y-%m-%d %H:%M:%S').timestamp() * 1000) if conv['updated_at'] else 0
+    conv['agentId'] = conv['agent_id']
+    
+    return jsonify(conv)
+
+
 @app.route('/api/user/<int:user_id>/conversations', methods=['POST'])
 def create_user_conversation(user_id):
     """创建新对话"""
@@ -707,6 +752,138 @@ def delete_user_conversation(user_id, conv_id):
     return jsonify({'success': True})
 
 
+# ==================== 用户智能体数据同步 ====================
+
+@app.route('/api/user/<int:user_id>/agents', methods=['GET'])
+def get_user_agents(user_id):
+    """获取用户智能体配置"""
+    conn = get_db()
+    cursor = conn.cursor()
+    cursor.execute('''
+        SELECT agent_id, category, is_pinned, is_favorite, added_at 
+        FROM user_agents WHERE user_id = ?
+    ''', (user_id,))
+    
+    agents_data = {
+        'myAgents': {},  # {category: [agent_ids]}
+        'favoriteAgents': [],  # [agent_ids]
+        'pinnedAgents': {}  # {category: [agent_ids]}
+    }
+    
+    for row in cursor.fetchall():
+        agent_id = row['agent_id']
+        category = row['category']
+        is_pinned = row['is_pinned']
+        is_favorite = row['is_favorite']
+        
+        # 添加到 myAgents
+        if category not in agents_data['myAgents']:
+            agents_data['myAgents'][category] = []
+        agents_data['myAgents'][category].append(agent_id)
+        
+        # 添加到 pinnedAgents
+        if is_pinned:
+            if category not in agents_data['pinnedAgents']:
+                agents_data['pinnedAgents'][category] = []
+            agents_data['pinnedAgents'][category].append(agent_id)
+        
+        # 添加到 favoriteAgents
+        if is_favorite:
+            agents_data['favoriteAgents'].append(agent_id)
+    
+    conn.close()
+    return jsonify(agents_data)
+
+
+@app.route('/api/user/<int:user_id>/agents/<agent_id>', methods=['POST'])
+def add_user_agent(user_id, agent_id):
+    """添加智能体到用户列表"""
+    data = request.json
+    category = data.get('category', 'basic')
+    is_pinned = data.get('is_pinned', 0)
+    is_favorite = data.get('is_favorite', 0)
+    
+    conn = get_db()
+    cursor = conn.cursor()
+    
+    try:
+        cursor.execute('''
+            INSERT INTO user_agents (user_id, agent_id, category, is_pinned, is_favorite)
+            VALUES (?, ?, ?, ?, ?)
+        ''', (user_id, agent_id, category, is_pinned, is_favorite))
+        conn.commit()
+        conn.close()
+        return jsonify({'success': True})
+    except:
+        # 已存在，更新
+        cursor.execute('''
+            UPDATE user_agents SET category=?, is_pinned=?, is_favorite=? 
+            WHERE user_id=? AND agent_id=?
+        ''', (category, is_pinned, is_favorite, user_id, agent_id))
+        conn.commit()
+        conn.close()
+        return jsonify({'success': True})
+
+
+@app.route('/api/user/<int:user_id>/agents/<agent_id>', methods=['DELETE'])
+def remove_user_agent(user_id, agent_id):
+    """从用户列表移除智能体"""
+    conn = get_db()
+    cursor = conn.cursor()
+    cursor.execute('DELETE FROM user_agents WHERE user_id=? AND agent_id=?', (user_id, agent_id))
+    conn.commit()
+    conn.close()
+    return jsonify({'success': True})
+
+
+@app.route('/api/user/<int:user_id>/agents/<agent_id>/pin', methods=['POST'])
+def toggle_user_agent_pin(user_id, agent_id):
+    """切换智能体置顶状态"""
+    data = request.json
+    is_pinned = data.get('is_pinned', 1)
+    category = data.get('category', 'basic')
+    
+    conn = get_db()
+    cursor = conn.cursor()
+    
+    # 检查是否存在
+    cursor.execute('SELECT id FROM user_agents WHERE user_id=? AND agent_id=?', (user_id, agent_id))
+    if cursor.fetchone():
+        cursor.execute('UPDATE user_agents SET is_pinned=? WHERE user_id=? AND agent_id=?', 
+                       (is_pinned, user_id, agent_id))
+    else:
+        cursor.execute('INSERT INTO user_agents (user_id, agent_id, category, is_pinned) VALUES (?, ?, ?, ?)',
+                       (user_id, agent_id, category, is_pinned))
+    
+    conn.commit()
+    conn.close()
+    return jsonify({'success': True})
+
+
+@app.route('/api/user/<int:user_id>/agents/<agent_id>/favorite', methods=['POST'])
+def toggle_user_agent_favorite(user_id, agent_id):
+    """切换智能体收藏状态"""
+    data = request.json
+    is_favorite = data.get('is_favorite', 1)
+    category = data.get('category', 'basic')
+    
+    conn = get_db()
+    cursor = conn.cursor()
+    
+    # 检查是否存在
+    cursor.execute('SELECT id FROM user_agents WHERE user_id=? AND agent_id=?', (user_id, agent_id))
+    if cursor.fetchone():
+        cursor.execute('UPDATE user_agents SET is_favorite=? WHERE user_id=? AND agent_id=?', 
+                       (is_favorite, user_id, agent_id))
+    else:
+        cursor.execute('INSERT INTO user_agents (user_id, agent_id, category, is_favorite) VALUES (?, ?, ?, ?)',
+                       (user_id, agent_id, category, is_favorite))
+    
+    conn.commit()
+    conn.close()
+    return jsonify({'success': True})
+
+
 # ==================== 大模型接口管理 ====================
 
 @app.route('/api/admin/llm', methods=['GET'])

www/admin.js

@@ -250,6 +250,7 @@ async function loadUsersPage(content) {
                 <td>
                   <div class="action-btns">
                     <button class="action-btn edit" onclick="showEditUserModal(${u.id})">编辑</button>
+                    <button class="action-btn" style="background: #8b5cf6; color: white;" onclick="showUserConversations(${u.id}, '${u.username}')">查看对话</button>
                     <button class="action-btn" style="background: #f59e0b; color: white;" onclick="showResetPasswordModal(${u.id})">重置密码</button>
                     <button class="action-btn delete" onclick="deleteUser(${u.id})">删除</button>
                   </div>
@@ -396,6 +397,150 @@ async function deleteUser(id) {
   loadPage('users');
 }
 
+// ==================== 查看用户对话记录 ====================
+
+async function showUserConversations(userId, username) {
+  // 加载用户对话列表
+  const conversations = await fetchAPI(`/api/user/${userId}/conversations`);
+  
+  const content = document.getElementById('mainContent');
+  
+  content.innerHTML = `
+    <div class="content-header">
+      <h1 class="content-title">用户对话记录 - ${username}</h1>
+      <button class="add-btn" style="background: #718096;" onclick="loadPage('users')">返回用户列表</button>
+    </div>
+    
+    <div class="stats-grid" style="grid-template-columns: repeat(3, 1fr);">
+      <div class="stat-card">
+        <div class="stat-icon">💬</div>
+        <div class="stat-value">${conversations.length}</div>
+        <div class="stat-label">对话总数</div>
+      </div>
+      <div class="stat-card">
+        <div class="stat-icon">🤖</div>
+        <div class="stat-value">${conversations.filter(c => c.agentId).length}</div>
+        <div class="stat-label">智能体对话</div>
+      </div>
+      <div class="stat-card">
+        <div class="stat-icon">📝</div>
+        <div class="stat-value">${conversations.reduce((sum, c) => sum + (c.messages?.length || 0), 0)}</div>
+        <div class="stat-label">消息总数</div>
+      </div>
+    </div>
+    
+    <div class="data-table">
+      <table>
+        <thead>
+          <tr>
+            <th>ID</th>
+            <th>标题</th>
+            <th>智能体</th>
+            <th>消息数</th>
+            <th>创建时间</th>
+            <th>更新时间</th>
+            <th>操作</th>
+          </tr>
+        </thead>
+        <tbody>
+          ${conversations.length === 0 ? '<tr><td colspan="7" style="text-align: center; color: #999;">暂无对话记录</td></tr>' : 
+            conversations.map(conv => `
+              <tr>
+                <td>${conv.id}</td>
+                <td>${conv.title || '新对话'}</td>
+                <td>${conv.agentId ? getAgentName(conv.agentId) : '普通对话'}</td>
+                <td>${conv.messages?.length || 0}</td>
+                <td>${formatDate(conv.createdAt || conv.created_at)}</td>
+                <td>${formatDate(conv.updatedAt || conv.updated_at)}</td>
+                <td>
+                  <div class="action-btns">
+                    <button class="action-btn edit" onclick="showConversationMessages(${userId}, ${conv.id}, '${conv.title || '新对话'}')">查看详情</button>
+                    <button class="action-btn delete" onclick="deleteUserConversation(${userId}, ${conv.id})">删除</button>
+                  </div>
+                </td>
+              </tr>
+            `).join('')
+          }
+        </tbody>
+      </table>
+    </div>
+  `;
+}
+
+function getAgentName(agentId) {
+  const agent = agents.find(a => a.agent_id === agentId);
+  return agent ? `${agent.avatar} ${agent.name}` : agentId;
+}
+
+async function showConversationMessages(userId, convId, title) {
+  // 获取对话详情
+  const conv = await fetchAPI(`/api/user/${userId}/conversations/${convId}`);
+  
+  const content = document.getElementById('mainContent');
+  
+  const messages = conv.messages || [];
+  
+  content.innerHTML = `
+    <div class="content-header">
+      <h1 class="content-title">对话详情 - ${title}</h1>
+      <button class="add-btn" style="background: #718096;" onclick="showUserConversations(${userId}, '用户')">返回对话列表</button>
+    </div>
+    
+    <div style="background: white; padding: 16px; border-radius: 12px; margin-bottom: 16px;">
+      <div style="display: flex; gap: 16px; color: #718096;">
+        <span>💬 消息数: ${messages.length}</span>
+        <span>🤖 智能体: ${conv.agentId ? getAgentName(conv.agentId) : '普通对话'}</span>
+        <span>📅 创建: ${formatDate(conv.createdAt || conv.created_at)}</span>
+      </div>
+    </div>
+    
+    <div class="data-table">
+      <table>
+        <thead>
+          <tr>
+            <th style="width: 60px;">序号</th>
+            <th style="width: 80px;">角色</th>
+            <th>内容</th>
+            <th style="width: 150px;">时间</th>
+          </tr>
+        </thead>
+        <tbody>
+          ${messages.length === 0 ? '<tr><td colspan="4" style="text-align: center; color: #999;">暂无消息</td></tr>' : 
+            messages.map((msg, idx) => `
+              <tr>
+                <td>${idx + 1}</td>
+                <td style="color: ${msg.role === 'user' ? '#3b82f6' : '#10b981'};">
+                  ${msg.role === 'user' ? '👤 用户' : '🤖 AI'}
+                </td>
+                <td style="max-width: 500px; white-space: pre-wrap; word-break: break-all;">
+                  ${escapeHtml(msg.content?.slice(0, 500) || '')}${msg.content?.length > 500 ? '...' : ''}
+                  ${msg.thinking ? `<div style="color: #f59e0b; margin-top: 8px; font-size: 12px;">💭 思考: ${escapeHtml(msg.thinking?.slice(0, 200) || '')}...</div>` : ''}
+                </td>
+                <td>${formatDate(msg.timestamp || msg.createdAt)}</td>
+              </tr>
+            `).join('')
+          }
+        </tbody>
+      </table>
+    </div>
+  `;
+}
+
+async function deleteUserConversation(userId, convId) {
+  if (!confirm('确定删除此对话？此操作不可恢复！')) return;
+  
+  await fetchAPI(`/api/user/${userId}/conversations/${convId}`, 'DELETE');
+  showToast('删除成功');
+  showUserConversations(userId, '用户');
+}
+
+function escapeHtml(text) {
+  if (!text) return '';
+  const div = document.createElement('div');
+  div.textContent = text;
+  return div.innerHTML;
+}
+
 // ==================== 大模型配置页面 ====================
 
 async function loadLLMPage(content) {

www/app.js

@@ -294,14 +294,47 @@ let thinkingBtn = null;
 let searchBtn = null;
 
 // 初始化
-document.addEventListener('DOMContentLoaded', () => {
+document.addEventListener('DOMContentLoaded', async () => {
   // 初始化 appContainer
   appContainer = document.getElementById('app');
   
-  // 从本地存储加载对话列表
-  const saved = localStorage.getItem('conversations');
-  if (saved) {
-    conversations = JSON.parse(saved);
+  // 加载用户登录状态（优先检查）
+  const savedUser = localStorage.getItem('currentUser');
+  if (savedUser) {
+    currentUser = JSON.parse(savedUser);
+  }
+  
+  // 如果用户已登录且有ID，从 backend 加载对话数据
+  if (currentUser && currentUser.id) {
+    try {
+      const res = await fetch(`/api/user/${currentUser.id}/conversations`);
+      const data = await res.json();
+      if (Array.isArray(data) && data.length > 0) {
+        // 使用 backend 数据替换本地数据
+        conversations = data;
+        // 更新本地存储（离线可用）
+        localStorage.setItem('conversations', JSON.stringify(conversations));
+      } else {
+        // backend 无数据，使用本地数据
+        const saved = localStorage.getItem('conversations');
+        if (saved) {
+          conversations = JSON.parse(saved);
+        }
+      }
+    } catch (e) {
+      // backend 加载失败，使用本地数据
+      console.error('加载 backend 对话失败:', e);
+      const saved = localStorage.getItem('conversations');
+      if (saved) {
+        conversations = JSON.parse(saved);
+      }
+    }
+  } else {
+    // 未登录用户，从本地存储加载对话列表
+    const saved = localStorage.getItem('conversations');
+    if (saved) {
+      conversations = JSON.parse(saved);
+    }
   }
   
   // 兼容旧数据格式（chat_history）
@@ -323,7 +356,7 @@ document.addEventListener('DOMContentLoaded', () => {
     }
   }
   
-  // 加载用户智能体数据
+  // 加载用户智能体数据（我的智能体）
   const savedMyAgents = localStorage.getItem('myAgents');
   if (savedMyAgents) {
     myAgents = JSON.parse(savedMyAgents);
@@ -341,10 +374,26 @@ document.addEventListener('DOMContentLoaded', () => {
     pinnedAgents = JSON.parse(savedPinnedAgents);
   }
   
-  // 加载用户登录状态
-  const savedUser = localStorage.getItem('currentUser');
-  if (savedUser) {
-    currentUser = JSON.parse(savedUser);
+  // 如果用户已登录且有ID，从 backend 加载智能体配置
+  if (currentUser && currentUser.id) {
+    try {
+      const res = await fetch(`/api/user/${currentUser.id}/agents`);
+      const data = await res.json();
+      if (data.myAgents) {
+        myAgents = data.myAgents;
+        localStorage.setItem('myAgents', JSON.stringify(myAgents));
+      }
+      if (data.favoriteAgents) {
+        favoriteAgents = data.favoriteAgents;
+        localStorage.setItem('favoriteAgents', JSON.stringify(favoriteAgents));
+      }
+      if (data.pinnedAgents) {
+        pinnedAgents = data.pinnedAgents;
+        localStorage.setItem('pinnedAgents', JSON.stringify(pinnedAgents));
+      }
+    } catch (e) {
+      console.error('加载智能体配置失败:', e);
+    }
   }
   
   // 加载每日使用统计
@@ -1001,10 +1050,12 @@ function toggleAgentPin(agentId) {
   
   const category = agent.category;
   
+  let is_pinned;
   if (pinnedAgents[category]?.includes(agentId)) {
     // 取消置顶
     pinnedAgents[category] = pinnedAgents[category].filter(id => id !== agentId);
     agent.is_pinned = false;
+    is_pinned = 0;
     showToast('已取消置顶');
   } else {
     // 置顶
@@ -1013,11 +1064,21 @@ function toggleAgentPin(agentId) {
     }
     pinnedAgents[category].push(agentId);
     agent.is_pinned = true;
+    is_pinned = 1;
     showToast('已置顶');
   }
   
   savePinnedAgents();
   saveMyAgents();  // 更新显示
+  
+  // 同步到 backend
+  if (currentUser && currentUser.id) {
+    fetch(`/api/user/${currentUser.id}/agents/${agentId}/pin`, {
+      method: 'POST',
+      headers: { 'Content-Type': 'application/json' },
+      body: JSON.stringify({ is_pinned, category })
+    }).catch(e => console.error('同步置顶失败:', e));
+  }
 }
 
 // 收藏/取消收藏智能体
@@ -1025,20 +1086,32 @@ function toggleAgentFavorite(agentId) {
   const agent = agents.find(a => a.id === agentId);
   if (!agent) return;
   
+  let is_favorite;
   if (favoriteAgents.includes(agentId)) {
     // 取消收藏
     favoriteAgents = favoriteAgents.filter(id => id !== agentId);
     agent.is_favorite = false;
+    is_favorite = 0;
     showToast('已取消收藏');
   } else {
     // 收藏
     favoriteAgents.push(agentId);
     agent.is_favorite = true;
+    is_favorite = 1;
     showToast('已收藏');
   }
   
   saveFavoriteAgents();
   saveMyAgents();  // 更新显示
+  
+  // 同步到 backend
+  if (currentUser && currentUser.id) {
+    fetch(`/api/user/${currentUser.id}/agents/${agentId}/favorite`, {
+      method: 'POST',
+      headers: { 'Content-Type': 'application/json' },
+      body: JSON.stringify({ is_favorite, category: agent.category })
+    }).catch(e => console.error('同步收藏失败:', e));
+  }
 }
 
 // 从用户智能体列表移除
@@ -1067,6 +1140,13 @@ function removeAgentFromMyAgents(agentId) {
   savePinnedAgents();
   saveFavoriteAgents();
   showToast('已移除');
+  
+  // 同步到 backend
+  if (currentUser && currentUser.id) {
+    fetch(`/api/user/${currentUser.id}/agents/${agentId}`, {
+      method: 'DELETE'
+    }).catch(e => console.error('同步移除失败:', e));
+  }
 }
 
 // ==================== 智能体发现页面 ====================
@@ -1303,6 +1383,15 @@ function addAgentToMyAgents(agentId) {
   
   saveMyAgents();
   showToast(`已添加 ${agent.name}`);
+  
+  // 同步到 backend
+  if (currentUser && currentUser.id) {
+    fetch(`/api/user/${currentUser.id}/agents/${agentId}`, {
+      method: 'POST',
+      headers: { 'Content-Type': 'application/json' },
+      body: JSON.stringify({ category: agent.category })
+    }).catch(e => console.error('同步添加智能体失败:', e));
+  }
 }
 
 // 从发现页面收藏智能体
@@ -3868,6 +3957,9 @@ async function streamGenerate(userMsgIndex) {
     syncConversationToBackend(currentConversation);
     renderMessages();
     
+    // 记录统计到 backend
+    logStatsToBackend('llm_call', currentConversation.agentId || 'chat', 1);
+    
     // 自动总结标题：第一次对话和每隔5次对话
     const totalMessages = currentConversation.messages.length;
     // 第一次对话（用户+AI=2条）或每5次对话（10条）
@@ -3877,6 +3969,15 @@ async function streamGenerate(userMsgIndex) {
   }
 }
 
+// 记录统计到 backend
+function logStatsToBackend(type, key, value = 1) {
+  fetch('/api/admin/stats/log', {
+    method: 'POST',
+    headers: { 'Content-Type': 'application/json' },
+    body: JSON.stringify({ type, key, value })
+  }).catch(e => console.error('统计记录失败:', e));
+}
+
 // 执行 Tavily 搜索
 async function performSearch(query) {
   try {
@@ -3899,6 +4000,10 @@ async function performSearch(query) {
     }
 
     const data = await response.json();
+    
+    // 记录搜索统计
+    logStatsToBackend('search_call', 'tavily', 1);
+    
     return data.results || [];
   } catch (error) {
     console.error('搜索错误:', error);