feat: 新增Client简单对话调用

This commit is contained in:
hhs
2026-04-19 15:58:41 +08:00
parent 004a5d685a
commit d7ac8889c0
2 changed files with 76 additions and 0 deletions
+19
View File
@@ -1,8 +1,10 @@
package llm
import (
"context"
"fmt"
"knowledge-graph-backend/internal/config"
"log"
"github.com/tmc/langchaingo/llms"
"github.com/tmc/langchaingo/llms/openai"
@@ -26,4 +28,21 @@ func NewClient(cfg config.LLMConfig) (*Client, error) {
}
return &Client{config: cfg, llm: model}, nil
}
// Client简单对话调用
func (c *Client) Chat(ctx context.Context, messages []llms.MessageContent) (string, error) {
resp, err := c.llm.GenerateContent(ctx, messages)
if err != nil {
log.Printf("[ERROR] GenerateContent failed: %v", err)
return "", fmt.Errorf("generate content error: %w", err)
}
if resp == nil || len(resp.Choices) == 0 {
log.Printf("[WARN] Response is empty or has no choices")
return "", fmt.Errorf("no content choices returned")
}
log.Printf("[INFO] Chat completed successfully, choice length: %d", len(resp.Choices[0].Content))
return resp.Choices[0].Content, nil
}