48 lines
1.1 KiB
Go
48 lines
1.1 KiB
Go
package openai
|
|
|
|
import (
|
|
"context"
|
|
|
|
"matrix-gpt-bot/config"
|
|
|
|
goai "github.com/sashabaranov/go-openai"
|
|
)
|
|
|
|
// Message represents a single message in the conversation.
|
|
type Message struct {
|
|
Role string `json:"role"`
|
|
Content string `json:"content"`
|
|
}
|
|
|
|
// Ask sends the provided messages to the OpenAI API and returns the assistant's reply.
|
|
func Ask(messages []Message) (string, error) {
|
|
// Load OpenAI key from configuration
|
|
conf := config.Load()
|
|
client := goai.NewClient(conf.OpenAIKey)
|
|
|
|
// Build the chat history in the format expected by the API
|
|
var chatHistory []goai.ChatCompletionMessage
|
|
for _, m := range messages {
|
|
chatHistory = append(chatHistory, goai.ChatCompletionMessage{
|
|
Role: m.Role,
|
|
Content: m.Content,
|
|
})
|
|
}
|
|
|
|
// Create the chat completion request
|
|
resp, err := client.CreateChatCompletion(
|
|
context.Background(),
|
|
goai.ChatCompletionRequest{
|
|
Model: "gpt-4.1",
|
|
Messages: chatHistory,
|
|
MaxTokens: 600,
|
|
},
|
|
)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
// Return the content of the first choice
|
|
return resp.Choices[0].Message.Content, nil
|
|
}
|