mirror of
https://github.com/0xJacky/nginx-ui.git
synced 2025-05-11 02:15:48 +02:00
26 lines
485 B
Go
26 lines
485 B
Go
package chatbot
|
|
|
|
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)
|
|
}
|