feat(wip): code completion with llm

This commit is contained in:
Jacky 2025-04-15 21:54:51 +08:00
parent 63fb823344
commit a57748a432
No known key found for this signature in database
GPG key ID: 215C21B10DF38B4D
22 changed files with 623 additions and 31 deletions

View file

@ -0,0 +1,26 @@
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)
}