mirror of
https://github.com/0xJacky/nginx-ui.git
synced 2025-05-10 18:05:48 +02:00
26 lines
481 B
Go
26 lines
481 B
Go
package llm
|
|
|
|
import (
|
|
"github.com/sashabaranov/go-openai"
|
|
"github.com/stretchr/testify/assert"
|
|
"testing"
|
|
)
|
|
|
|
func TestChatCompletionWithContext(t *testing.T) {
|
|
filename := "test"
|
|
messages := []openai.ChatCompletionMessage{
|
|
{
|
|
Role: openai.ChatMessageRoleSystem,
|
|
},
|
|
{
|
|
Role: openai.ChatMessageRoleUser,
|
|
},
|
|
{
|
|
Role: openai.ChatMessageRoleAssistant,
|
|
},
|
|
}
|
|
|
|
messages = ChatCompletionWithContext(filename, messages)
|
|
|
|
assert.NotNil(t, messages[1].MultiContent)
|
|
}
|