fuzhongyun
/
ai_scheduler
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

结构修改

This commit is contained in:
renzhiyuan 2025-09-30 19:30:22 +08:00
parent 774701eae4
commit 91242c183c
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
internal/biz/llm_service/ollama.go
View File

@ -9,6 +9,7 @@ import (
"encoding/json" "encoding/json"
"fmt" "fmt"
"github.com/gofiber/fiber/v2/log"
"github.com/ollama/ollama/api" "github.com/ollama/ollama/api"
) )
@ -28,6 +29,7 @@ func (r *OllamaService) IntentRecognize(ctx context.Context, requireData *entity
prompt := r.getPrompt(requireData.Sys, requireData.Histories, requireData.UserInput, requireData.Tasks) prompt := r.getPrompt(requireData.Sys, requireData.Histories, requireData.UserInput, requireData.Tasks)
toolDefinitions := r.registerToolsOllama(requireData.Tasks) toolDefinitions := r.registerToolsOllama(requireData.Tasks)
match, err := r.client.ToolSelect(context.TODO(), prompt, toolDefinitions) match, err := r.client.ToolSelect(context.TODO(), prompt, toolDefinitions)
log.Info("意图识别结果: %v", match)
if err != nil { if err != nil {
return return
} }