会话

models/business_conf.go

@@ -56,6 +56,7 @@ const (
 	BusinessConfReportViewUrl                = "ReportViewUrl"                // 报告详情地址
 	BusinessConfEsIndexNameExcel             = "EsIndexNameExcel"             // ES索引名称-表格
 	BusinessConfEsIndexNameDataSource        = "EsIndexNameDataSource"        // ES索引名称-数据源
+	LLMInitConfig="llmInitConfig"
 )
 
 const (
@@ -63,6 +64,8 @@ const (
 	BusinessConfReportApproveTypeOther = "other"
 	BusinessConfClientFlagNanHua       = "nhqh" // 南华标记
 	BusinessConfEmailClientSmtp        = "smtp" // 普通邮箱标记
+
+
 )
 
 // FromSceneMap 数据源名称与数据源ID的对应关系

utils/llm/eta_llm/eta_llm_client.go

@@ -5,6 +5,7 @@ import (
 	"bytes"
 	"encoding/json"
 	"errors"
+	"eta/eta_api/models"
 	"eta/eta_api/utils"
 	"eta/eta_api/utils/llm"
 	"eta/eta_api/utils/llm/eta_llm/eta_llm_http"
@@ -33,13 +34,28 @@ type ETALLMClient struct {
 	*llm.LLMClient
 	LlmModel string
 }
+type LLMConfig struct {
+	LlmAddress string `json:"llm_server"`
+	LlmModel   string `json:"llm_model"`
+}
 
 func GetInstance() llm.LLMService {
 	dsOnce.Do(func() {
+		confStr := models.BusinessConfMap[models.LLMInitConfig]
+		if confStr == "" {
+			utils.FileLog.Error("LLM配置为空")
+			return
+		}
+
+		var config LLMConfig
+		err := json.Unmarshal([]byte(confStr), &config)
+		if err != nil {
+			utils.FileLog.Error("LLM配置错误")
+		}
 		if etaLlmClient == nil {
 			etaLlmClient = &ETALLMClient{
-				LLMClient: llm.NewLLMClient(utils.LLM_SERVER, 120),
-				LlmModel:  utils.LLM_MODEL,
+				LLMClient: llm.NewLLMClient(config.LlmAddress, 120),
+				LlmModel:  config.LlmModel,
 			}
 		}
 	})