Add chat completions (#90)

This commit is contained in:
sashabaranov
2023-03-02 00:47:25 +04:00
committed by GitHub
parent ae05ed976f
commit c8261c5638
3 changed files with 87 additions and 0 deletions

View File

@@ -53,6 +53,23 @@ func TestAPI(t *testing.T) {
t.Fatalf("Embedding error: %v", err)
}
_, err = c.CreateChatCompletion(
ctx,
ChatCompletionRequest{
Model: GPT3Dot5Turbo,
Messages: []ChatCompletionMessage{
{
Role: "user",
Content: "Hello!",
},
},
},
)
if err != nil {
t.Errorf("CreateChatCompletion returned error: %v", err)
}
stream, err := c.CreateCompletionStream(ctx, CompletionRequest{
Prompt: "Ex falso quodlibet",
Model: GPT3Ada,