From 0354b3f24934b8c4243681b6d53e4c0db6f3c5f7 Mon Sep 17 00:00:00 2001
From: unintendedfraud <briceunintendedfraud@fastmail.com>
Date: Sat, 6 May 2023 12:26:22 +0200
Subject: [PATCH] ff

---
 openai/client.go | 12 ++++--------
 1 file changed, 4 insertions(+), 8 deletions(-)

diff --git a/openai/client.go b/openai/client.go
index bcd21c2..78f7345 100644
--- a/openai/client.go
+++ b/openai/client.go
@@ -20,6 +20,7 @@ func NewClient(token string) *Client {
 }
 
 func (c Client) Completions(prompt string) (*CompletionResponse, error) {
+	fmt.Println("prompting: ", prompt)
 	body := &CompletionPayload{
 		Model:       "gpt-3.5-turbo",
 		Prompt:      cleanPrompt(prompt),
@@ -85,12 +86,7 @@ type CompletionUsage struct {
 }
 
 type CompletionChoice struct {
-	Message      CompletionMessage `json:"message"`
-	FinishReason string            `json:"finish_reason"`
-	Index        int               `json:"index"`
-}
-
-type CompletionMessage struct {
-	Role    string `json:"role"`
-	Content string `json:"content"`
+	Text         string `json:"text"`
+	FinishReason string `json:"finish_reason"`
+	Index        int    `json:"index"`
 }