feat: 智能体数据同步到backend + 初始化时从backend加载完整数据
This commit is contained in:
147
backend/app.py
147
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 (
|
||||
@@ -707,6 +722,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'])
|
||||
|
||||
Reference in New Issue
Block a user