kris/ForcePilot
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

更新关于在线搜索的描述

Browse Source
This commit is contained in:
Wenjie Zhang 2025-04-09 12:05:05 +08:00
parent ede78ad390
commit a2be4c80dd
3 changed files with 18 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/agents/registry.py
View File

@ -97,7 +97,7 @@ class BaseAgent():
return
for requirement in self.requirements:
if requirement not in os.environ:
raise ValueError(f"没有配置{requirement} 环境变量，请在 src/.env 文件中配置")
raise ValueError(f"没有配置{requirement} 环境变量，请在 src/.env 文件中配置，并重新启动服务")
def stream_values(self, messages: list[str], config_schema: RunnableConfig = None, **kwargs):
graph = self.get_graph(config_schema=config_schema, **kwargs)

12
src/config/__init__.py
View File

@ -47,7 +47,7 @@ class Config(SimpleConfig):
self.add_item("enable_reranker", default=False, des="是否开启重排序")
self.add_item("enable_knowledge_base", default=False, des="是否开启知识库")
self.add_item("enable_knowledge_graph", default=False, des="是否开启知识图谱")
self.add_item("enable_web_search", default=False, des="是否开启网页搜索（需配置 TAVILY_API_KEY）")
self.add_item("enable_web_search", default=False, des="是否开启网页搜索（注：现阶段会根据 TAVILY_API_KEY 自动开启，无法手动配置，将会在下个版本移除此配置项）")
# 模型配置
## 注意这里是模型名，而不是具体的模型路径，默认使用 HuggingFace 的路径
## 如果需要自定义本地模型路径，则在 src/.env 中配置 MODEL_DIR
@ -150,9 +150,13 @@ class Config(SimpleConfig):
self.model_provider_status[provider] = all(conds_bool)
# 检查web_search的环境变量
if self.enable_web_search and not os.getenv("TAVILY_API_KEY"):
logger.warning("TAVILY_API_KEY not set, web search will be disabled")
self.enable_web_search = False
# if self.enable_web_search and not os.getenv("TAVILY_API_KEY"):
# logger.warning("TAVILY_API_KEY not set, web search will be disabled")
# self.enable_web_search = False
# 2025.04.08 修改为不手动配置，只要配置了TAVILY_API_KEY，就默认开启web_search
if os.getenv("TAVILY_API_KEY"):
self.enable_web_search = True
self.valuable_model_provider = [k for k, v in self.model_provider_status.items() if v]
assert len(self.valuable_model_provider) > 0, f"No model provider available, please check your `.env` file. API_KEY_LIST: {conds}"

17
web/src/views/SettingView.vue
View File

@ -6,7 +6,7 @@
</template>
<template #actions>
<a-button :type="isNeedRestart ? 'primary' : 'default'" @click="sendRestart" :icon="h(ReloadOutlined)">
{{ isNeedRestart ? '需要重启' : '重启服务' }}
{{ isNeedRestart ? '需要刷新' : '重新加载' }}
</a-button>
</template>
</HeaderComponent>
@ -63,10 +63,11 @@
</div>
<div class="card">
<span class="label">{{ items?.enable_web_search.des }}</span>
<a-switch
<!-- <a-switch
:checked="configStore.config.enable_web_search"
@change="handleChange('enable_web_search', !configStore.config.enable_web_search)"
/>
/> -->
<a-switch :checked="configStore.config.enable_web_search" />
</div>
<div class="card">
<span class="label">{{ items?.enable_reranker.des }}</span>
@ -264,7 +265,7 @@
</a-checkbox-group>
</div>
<div v-if="providerConfig.allModels.length === 0" class="modal-no-models">
<a-alert v-if="!modelStatus[providerConfig.provider]" type="warning" message="请在 src/.env 中配置对应的 APIKEY" />
<a-alert v-if="!modelStatus[providerConfig.provider]" type="warning" message="请在 src/.env 中配置对应的 APIKEY，并重新启动服务" />
<a-alert v-else type="warning" message="该提供商暂未适配获取模型列表的方法，如果需要添加模型，请在 src/static/models.private.yml 中添加。" />
</div>
</div>
@ -387,11 +388,11 @@ const handleChange = (key, e) => {
if (!isNeedRestart.value) {
isNeedRestart.value = true
notification.info({
message: '需要重启服务',
description: '请点击右下角按钮重启服务',
message: '需要重新加载模型',
description: '请点击右下角按钮重新加载模型',
placement: 'topLeft',
duration: 0,
btn: h(Button, { type: 'primary', onClick: sendRestart }, '立即重启')
btn: h(Button, { type: 'primary', onClick: sendRestart }, '立即重新加载')
})
}
}
@ -543,7 +544,7 @@ const openProviderConfig = (provider) => {
const saveProviderConfig = async () => {
if (!modelStatus.value[providerConfig.provider]) {
message.error('请在 src/.env 中配置对应的 APIKEY')
message.error('请在 src/.env 中配置对应的 APIKEY，并重新启动服务')
return
}