feat: 添加搜索工具功能(Tavily Search)
- 新增 SearchToolConfig 模型:支持搜索工具配置 - Agent 增加 tools 字段:可配置可用工具列表 - 后台管理增加搜索工具配置页面 - Agent 管理增加工具启用开关 - 网页端增加搜索工具禁用复选框 - WebSocket chat 处理增加搜索调用逻辑 - 默认配置 Tavily Search API
This commit is contained in:
@@ -42,6 +42,7 @@
|
||||
<div class="nav-item active" data-page="llm-providers"><i class="ri-cloud-line"></i> 大模型池</div>
|
||||
<div class="nav-item" data-page="agents"><i class="ri-robot-2-line"></i> Agent管理</div>
|
||||
<div class="nav-item" data-page="channels"><i class="ri-chat-3-line"></i> 渠道管理</div>
|
||||
<div class="nav-item" data-page="search-tools"><i class="ri-search-line"></i> 搜索工具</div>
|
||||
<div class="nav-item" data-page="stats"><i class="ri-bar-chart-box-line"></i> 统计数据</div>
|
||||
</div>
|
||||
</div>
|
||||
@@ -109,6 +110,23 @@
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<!-- 搜索工具 -->
|
||||
<div class="page-section" id="page-search-tools">
|
||||
<div class="page-header"><h2><i class="ri-search-line"></i> 搜索工具配置</h2></div>
|
||||
<div class="card">
|
||||
<div class="card-header d-flex justify-content-between">
|
||||
<span>搜索工具列表(Tavily、Google等)</span>
|
||||
<button class="btn btn-primary btn-sm" onclick="showAddSearchToolModal()"><i class="ri-add-line"></i> 添加</button>
|
||||
</div>
|
||||
<div class="card-body">
|
||||
<table class="table">
|
||||
<thead><tr><th>名称</th><th>提供商</th><th>API Key</th><th>最大结果</th><th>默认</th><th>状态</th><th>操作</th></tr></thead>
|
||||
<tbody id="search-tools-list"><tr><td colspan="7" class="text-center">加载中...</td></tr></tbody>
|
||||
</table>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<!-- 统计 -->
|
||||
<div class="page-section" id="page-stats">
|
||||
<div class="page-header"><h2><i class="ri-bar-chart-box-line"></i> 统计数据</h2></div>
|
||||
@@ -152,6 +170,8 @@
|
||||
<div class="row mt-3"><div class="col-md-4 form-check"><input type="checkbox" class="form-check-input" id="agent-active" checked><label class="form-check-label">启用</label></div><div class="col-md-4 form-check"><input type="checkbox" class="form-check-input" id="agent-default"><label class="form-check-label">默认</label></div><div class="col-md-4"><label class="form-label">温度覆盖</label><input type="number" class="form-control" id="agent-temperature-override" step="0.1"></div></div>
|
||||
<hr><h6>思考功能</h6>
|
||||
<div class="thinking-config"><div class="form-check"><input type="checkbox" class="form-check-input" id="agent-enable-thinking" checked><label class="form-check-label">启用思考</label></div><div class="mt-2"><label class="form-label">思考提示词</label><textarea class="form-control" id="agent-thinking-prompt" rows="2"></textarea></div><div class="row mt-2"><div class="col-md-6"><label class="form-label">前缀</label><input type="text" class="form-control" id="agent-thinking-prefix"></div><div class="col-md-6"><label class="form-label">后缀</label><input type="text" class="form-control" id="agent-thinking-suffix"></div></div></div>
|
||||
<hr><h6>工具配置</h6>
|
||||
<div class="thinking-config"><div class="form-check"><input type="checkbox" class="form-check-input" id="agent-tool-search"><label class="form-check-label">启用搜索工具</label></div><small class="text-muted">启用后 Agent 可以使用搜索功能获取实时信息</small></div>
|
||||
</form></div>
|
||||
<div class="modal-footer"><button type="button" class="btn btn-secondary" data-bs-dismiss="modal">取消</button><button type="button" class="btn btn-primary" onclick="saveAgent()">保存</button></div>
|
||||
</div></div></div>
|
||||
@@ -180,6 +200,22 @@
|
||||
<div class="modal-footer"><button type="button" class="btn btn-secondary" data-bs-dismiss="modal">取消</button><button type="button" class="btn btn-primary" onclick="saveMatrixConfig()">保存</button></div>
|
||||
</div></div></div>
|
||||
|
||||
<!-- Search Tool Modal -->
|
||||
<div class="modal fade" id="searchToolModal"><div class="modal-dialog"><div class="modal-content">
|
||||
<div class="modal-header"><h5 class="modal-title">添加/编辑搜索工具</h5><button type="button" class="btn-close" data-bs-dismiss="modal"></button></div>
|
||||
<div class="modal-body"><form id="search-tool-form">
|
||||
<input type="hidden" id="search-tool-id">
|
||||
<div class="mb-3"><label class="form-label">名称</label><input type="text" class="form-control" id="search-tool-name" value="Tavily Search"></div>
|
||||
<div class="mb-3"><label class="form-label">提供商</label><select class="form-select" id="search-tool-provider"><option value="tavily">Tavily</option><option value="google">Google</option><option value="bing">Bing</option></select></div>
|
||||
<div class="mb-3"><label class="form-label">API Key *</label><input type="text" class="form-control" id="search-tool-api-key" required></div>
|
||||
<div class="mb-3"><label class="form-label">API地址(可选)</label><input type="text" class="form-control" id="search-tool-api-base"></div>
|
||||
<div class="row"><div class="col-md-6"><label class="form-label">最大结果数</label><input type="number" class="form-control" id="search-tool-max-results" value="5"></div><div class="col-md-6"><label class="form-label">搜索深度</label><select class="form-select" id="search-tool-depth"><option value="basic">Basic</option><option value="advanced">Advanced</option></select></div></div>
|
||||
<div class="mt-3 form-check"><input type="checkbox" class="form-check-input" id="search-tool-active" checked><label class="form-check-label">启用</label></div>
|
||||
<div class="mt-2 form-check"><input type="checkbox" class="form-check-input" id="search-tool-default"><label class="form-check-label">设为默认</label></div>
|
||||
</form></div>
|
||||
<div class="modal-footer"><button type="button" class="btn btn-secondary" data-bs-dismiss="modal">取消</button><button type="button" class="btn btn-primary" onclick="saveSearchTool()">保存</button></div>
|
||||
</div></div></div>
|
||||
|
||||
<script src="https://cdn.jsdelivr.net/npm/bootstrap@5.3.0/dist/js/bootstrap.bundle.min.js"></script>
|
||||
<script>
|
||||
// 全局数据
|
||||
@@ -190,6 +226,7 @@
|
||||
if (page === 'llm-providers') loadProviders();
|
||||
if (page === 'agents') loadAgents();
|
||||
if (page === 'channels') loadChannels();
|
||||
if (page === 'search-tools') loadSearchTools();
|
||||
if (page === 'stats') loadStats();
|
||||
}
|
||||
|
||||
@@ -348,11 +385,18 @@
|
||||
document.getElementById('agent-thinking-prompt').value = a.thinking_prompt || '';
|
||||
document.getElementById('agent-thinking-prefix').value = a.thinking_prefix || '';
|
||||
document.getElementById('agent-thinking-suffix').value = a.thinking_suffix || '';
|
||||
// 工具配置
|
||||
const tools = a.tools || [];
|
||||
document.getElementById('agent-tool-search').checked = tools.includes('search');
|
||||
new bootstrap.Modal(document.getElementById('agentModal')).show();
|
||||
}
|
||||
|
||||
async function saveAgent() {
|
||||
const id = document.getElementById('agent-id').value;
|
||||
// 处理工具列表
|
||||
const tools = [];
|
||||
if (document.getElementById('agent-tool-search').checked) tools.push('search');
|
||||
|
||||
const data = {
|
||||
name: document.getElementById('agent-name').value,
|
||||
display_name: document.getElementById('agent-display-name').value,
|
||||
@@ -367,7 +411,8 @@
|
||||
enable_thinking: document.getElementById('agent-enable-thinking').checked,
|
||||
thinking_prompt: document.getElementById('agent-thinking-prompt').value,
|
||||
thinking_prefix: document.getElementById('agent-thinking-prefix').value,
|
||||
thinking_suffix: document.getElementById('agent-thinking-suffix').value
|
||||
thinking_suffix: document.getElementById('agent-thinking-suffix').value,
|
||||
tools: tools // 工具列表
|
||||
};
|
||||
const res = await fetch(id ? `/api/v2/agents/${id}` : '/api/v2/agents', { method: id ? 'PUT' : 'POST', headers:{'Content-Type':'application/json'}, body: JSON.stringify(data) });
|
||||
const result = await res.json();
|
||||
@@ -482,6 +527,102 @@
|
||||
document.getElementById('stat-agents').textContent = (agentsData.agents || []).length;
|
||||
}
|
||||
|
||||
// ===== 搜索工具 =====
|
||||
async function loadSearchTools() {
|
||||
const res = await fetch('/api/v2/search-tools');
|
||||
const data = await res.json();
|
||||
const tools = data.configs || [];
|
||||
const tbody = document.getElementById('search-tools-list');
|
||||
if (tools.length === 0) {
|
||||
tbody.innerHTML = '<tr><td colspan="7" class="text-center text-muted">暂无配置,点击添加按钮创建</td></tr>';
|
||||
return;
|
||||
}
|
||||
tbody.innerHTML = tools.map(t => `
|
||||
<tr>
|
||||
<td>${t.name}</td>
|
||||
<td>${t.provider}</td>
|
||||
<td><small class="text-muted">${t.api_key ? t.api_key.substring(0, 10) + '...' : '-'}</small></td>
|
||||
<td>${t.max_results}</td>
|
||||
<td>${t.is_default ? '<i class="ri-check-line text-success"></i>' : ''}</td>
|
||||
<td>${t.is_active ? '<span class="badge bg-success">启用</span>' : '<span class="badge bg-secondary">禁用</span>'}</td>
|
||||
<td>
|
||||
<button class="btn btn-sm btn-outline-primary" onclick="editSearchTool(${t.id})"><i class="ri-edit-line"></i></button>
|
||||
<button class="btn btn-sm btn-outline-danger" onclick="deleteSearchTool(${t.id})"><i class="ri-delete-bin-line"></i></button>
|
||||
</td>
|
||||
</tr>
|
||||
`).join('');
|
||||
}
|
||||
|
||||
function showAddSearchToolModal() {
|
||||
document.getElementById('search-tool-id').value = '';
|
||||
document.getElementById('search-tool-name').value = 'Tavily Search';
|
||||
document.getElementById('search-tool-provider').value = 'tavily';
|
||||
document.getElementById('search-tool-api-key').value = '';
|
||||
document.getElementById('search-tool-api-base').value = '';
|
||||
document.getElementById('search-tool-max-results').value = 5;
|
||||
document.getElementById('search-tool-depth').value = 'basic';
|
||||
document.getElementById('search-tool-active').checked = true;
|
||||
document.getElementById('search-tool-default').checked = false;
|
||||
new bootstrap.Modal(document.getElementById('searchToolModal')).show();
|
||||
}
|
||||
|
||||
async function editSearchTool(id) {
|
||||
const res = await fetch('/api/v2/search-tools');
|
||||
const data = await res.json();
|
||||
const tool = (data.configs || []).find(t => t.id === id);
|
||||
if (!tool) return;
|
||||
document.getElementById('search-tool-id').value = tool.id;
|
||||
document.getElementById('search-tool-name').value = tool.name;
|
||||
document.getElementById('search-tool-provider').value = tool.provider;
|
||||
document.getElementById('search-tool-api-key').value = tool.api_key || '';
|
||||
document.getElementById('search-tool-api-base').value = tool.api_base || '';
|
||||
document.getElementById('search-tool-max-results').value = tool.max_results;
|
||||
document.getElementById('search-tool-depth').value = tool.search_depth;
|
||||
document.getElementById('search-tool-active').checked = tool.is_active;
|
||||
document.getElementById('search-tool-default').checked = tool.is_default;
|
||||
new bootstrap.Modal(document.getElementById('searchToolModal')).show();
|
||||
}
|
||||
|
||||
async function saveSearchTool() {
|
||||
const id = document.getElementById('search-tool-id').value;
|
||||
const data = {
|
||||
name: document.getElementById('search-tool-name').value,
|
||||
provider: document.getElementById('search-tool-provider').value,
|
||||
api_key: document.getElementById('search-tool-api-key').value,
|
||||
api_base: document.getElementById('search-tool-api-base').value,
|
||||
max_results: parseInt(document.getElementById('search-tool-max-results').value),
|
||||
search_depth: document.getElementById('search-tool-depth').value,
|
||||
is_active: document.getElementById('search-tool-active').checked,
|
||||
is_default: document.getElementById('search-tool-default').checked
|
||||
};
|
||||
let res;
|
||||
if (id) {
|
||||
res = await fetch(`/api/v2/search-tools/${id}`, { method: 'PUT', headers: {'Content-Type': 'application/json'}, body: JSON.stringify(data) });
|
||||
} else {
|
||||
res = await fetch('/api/v2/search-tools', { method: 'POST', headers: {'Content-Type': 'application/json'}, body: JSON.stringify(data) });
|
||||
}
|
||||
const result = await res.json();
|
||||
if (result.success) {
|
||||
bootstrap.Modal.getInstance(document.getElementById('searchToolModal')).hide();
|
||||
loadSearchTools();
|
||||
alert('保存成功');
|
||||
} else {
|
||||
alert('保存失败: ' + result.message);
|
||||
}
|
||||
}
|
||||
|
||||
async function deleteSearchTool(id) {
|
||||
if (!confirm('确定删除此搜索工具配置?')) return;
|
||||
const res = await fetch(`/api/v2/search-tools/${id}`, { method: 'DELETE' });
|
||||
const result = await res.json();
|
||||
if (result.success) {
|
||||
loadSearchTools();
|
||||
alert('删除成功');
|
||||
} else {
|
||||
alert('删除失败');
|
||||
}
|
||||
}
|
||||
|
||||
// ===== 初始化 =====
|
||||
document.addEventListener('DOMContentLoaded', function() {
|
||||
// 页面切换事件绑定
|
||||
|
||||
Reference in New Issue
Block a user