Mimu/hal
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

ff

Browse Source
This commit is contained in:
unintendedfraud
2023-05-06 12:26:22 +02:00
parent 4dd46df23e
commit 0354b3f249
1 changed files with 4 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
openai/client.go
+4 -8
View File
@@ -20,6 +20,7 @@ func NewClient(token string) *Client {
} }
func (c Client) Completions(prompt string) (*CompletionResponse, error) { func (c Client) Completions(prompt string) (*CompletionResponse, error) {
fmt.Println("prompting: ", prompt)
body := &CompletionPayload{ body := &CompletionPayload{
Model: "gpt-3.5-turbo", Model: "gpt-3.5-turbo",
Prompt: cleanPrompt(prompt), Prompt: cleanPrompt(prompt),
@@ -85,12 +86,7 @@ type CompletionUsage struct {
} }
type CompletionChoice struct { type CompletionChoice struct {
Message CompletionMessage `json:"message"` Text string `json:"text"`
FinishReason string `json:"finish_reason"` FinishReason string `json:"finish_reason"`
Index int `json:"index"` Index int `json:"index"`
}
type CompletionMessage struct {
Role string `json:"role"`
Content string `json:"content"`
} }