add model config for ai

Author: madneal

server/api/search_result_test.go

@@ -0,0 +1,39 @@
+package api
+
+import (
+	"encoding/json"
+	"fmt"
+	"github.com/madneal/gshark/global"
+	"github.com/madneal/gshark/initialize"
+	"github.com/madneal/gshark/model"
+	"github.com/madneal/gshark/service"
+	"testing"
+)
+
+func TestStartAITask(t *testing.T) {
+	global.GVA_VP = initialize.Viper("../config.yaml")
+	global.GVA_LOG = initialize.Zap()
+	global.GVA_DB = initialize.Gorm()
+	err, list := service.ListSearchResultByStatus(0)
+	if err != nil {
+		fmt.Println(err)
+		return
+	}
+	for _, result := range list {
+		var content string
+		textMatches := make([]model.TextMatch, 0)
+		if json.Valid(result.TextMatchesJson) {
+			err = json.Unmarshal(result.TextMatchesJson, &textMatches)
+			if err != nil {
+				fmt.Println(err)
+				continue
+			}
+			for _, textMatch := range textMatches {
+				content += *textMatch.Fragment + "\n"
+			}
+		} else {
+			content = string(result.TextMatchesJson)
+		}
+		fmt.Println(content)
+	}
+}

server/config-temp.yaml

@@ -44,6 +44,7 @@ system:
   gitlab-base:
   ai_server:
   ai_token:
+  model:
 wechat:
   url:
   enable: false

server/config/system.go

@@ -9,4 +9,5 @@ type System struct {
 	GitlabBase    string `mapstructure:"gitlab-base" yaml:"gitlab-base"`
 	AiServer      string `mapstructure:"ai_server" yaml:"ai_server"`
 	AiToken       string `mapstructure:"ai_token" yaml:"ai_token"`
+	Model         string `mapstructure:"model" json:"model" yaml:"model"`
 }

server/service/ai.go

@@ -45,7 +45,7 @@ func Question(command, question string) string {
 func callChatCompletion(command, question string) ([]byte, error) {
 	var result []byte
 	requestData := ChatCompletionRequest{
-		Model: "deepseek-r1",
+		Model: global.GVA_CONFIG.System.Model,
 		Messages: []Message{
 			{
 				Role:    "system",