feat(bootstrap): 添加 Elasticsearch 启动配置选项
- 在 config.yml 中添加 ElasticSearch.Start 配置项,用于控制是否启动 Elasticsearch 客户端 - 修改 init() 函数,根据配置项决定是否初始化 Elasticsearch 客户端 - 更新 prompts.yml,保留原有提示内容
This commit is contained in:
		
							parent
							
								
									91073fdf7f
								
							
						
					
					
						commit
						85b4b61a32
					
				@ -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())
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
@ -173,6 +173,7 @@ QiNiu:
 | 
			
		||||
  SecretKey: "UP5-GmSmAYNbMlSb6LYLuKZ-fT35nlEzGvOKKm9S"
 | 
			
		||||
 | 
			
		||||
ElasticSearch:
 | 
			
		||||
  Start: 0  # 0 不启动;1 启动
 | 
			
		||||
  Addr: "http://localhost:9200"
 | 
			
		||||
  UserName: "elastic"
 | 
			
		||||
  Password: ""
 | 
			
		||||
 | 
			
		||||
@ -18,4 +18,4 @@ Prompt:
 | 
			
		||||
    ···
 | 
			
		||||
    {context}
 | 
			
		||||
    ···
 | 
			
		||||
    如果给定的知识库无法让你做出回答,请回答知识库中未找到符合的资料,我不知道。"
 | 
			
		||||
    如果给定的知识库无法让你做出回答,请回答知识库中未找到符合的资料,我不知道。" 
 | 
			
		||||
		Loading…
	
	
			
			x
			
			
		
	
		Reference in New Issue
	
	Block a user