feat: 新增Client简单对话调用
This commit is contained in:
@@ -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
|
||||
}
|
||||
Reference in New Issue
Block a user