feat(bootstrap): 添加 Elasticsearch 启动配置选项

- 在 config.yml 中添加 ElasticSearch.Start 配置项,用于控制是否启动 Elasticsearch 客户端
- 修改 init() 函数,根据配置项决定是否初始化 Elasticsearch 客户端
- 更新 prompts.yml,保留原有提示内容
This commit is contained in:
Havoc412 2024-11-19 22:09:30 +08:00
parent 91073fdf7f
commit 85b4b61a32
3 changed files with 10 additions and 7 deletions

View File

@ -125,12 +125,14 @@ func init() {
)
// 12. ES 客户端启动
var err error
variable.ElasticClient, err = elasticsearch.NewClient(elasticsearch.Config{
Addresses: []string{variable.ConfigYml.GetString("ElasticSearch.Addr")},
})
if err != nil {
log.Fatal(my_errors.ErrorsInitConnFail + err.Error())
if variable.ConfigYml.GetInt("ElasticSearch.Start") == 1 {
var err error
variable.ElasticClient, err = elasticsearch.NewClient(elasticsearch.Config{
Addresses: []string{variable.ConfigYml.GetString("ElasticSearch.Addr")},
})
if err != nil {
log.Fatal(my_errors.ErrorsInitConnFail + err.Error())
}
}
}

View File

@ -173,6 +173,7 @@ QiNiu:
SecretKey: "UP5-GmSmAYNbMlSb6LYLuKZ-fT35nlEzGvOKKm9S"
ElasticSearch:
Start: 0 # 0 不启动1 启动
Addr: "http://localhost:9200"
UserName: "elastic"
Password: ""

View File

@ -18,4 +18,4 @@ Prompt:
···
{context}
···
如果给定的知识库无法让你做出回答,请回答知识库中未找到符合的资料,我不知道。"
如果给定的知识库无法让你做出回答,请回答知识库中未找到符合的资料,我不知道。"