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

f

Browse Source
This commit is contained in:
unintendedfraud
2023-08-27 08:59:55 +02:00
parent 769f9d047f
commit 79e95067b6
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
openai/client.go
+5
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.Printf("\ncompletions: %s", prompt)
body := &CompletionPayload{ body := &CompletionPayload{
Model: "text-davinci-003", Model: "text-davinci-003",
Prompt: cleanPrompt(prompt), Prompt: cleanPrompt(prompt),
@@ -33,6 +34,8 @@ func (c Client) Completions(prompt string) (*CompletionResponse, error) {
return nil, fmt.Errorf("failed to marshal the payload. %s", err.Error()) return nil, fmt.Errorf("failed to marshal the payload. %s", err.Error())
} }
fmt.Printf("\nsend request to openai")
res, err := c.HttpClient.NewRequest(). res, err := c.HttpClient.NewRequest().
SetHeader("Content-Type", "application/json"). SetHeader("Content-Type", "application/json").
SetBody(b). SetBody(b).
@@ -41,6 +44,8 @@ func (c Client) Completions(prompt string) (*CompletionResponse, error) {
return nil, fmt.Errorf("failed to query openai. Error: %s", err.Error()) return nil, fmt.Errorf("failed to query openai. Error: %s", err.Error())
} }
fmt.Printf("\nreceived openai response: %+v", res.StatusCode())
var r CompletionResponse var r CompletionResponse
if err := json.Unmarshal(res.Body(), &r); err != nil { if err := json.Unmarshal(res.Body(), &r); err != nil {
return nil, fmt.Errorf("failed to parse the openai response. Error: %s", err.Error()) return nil, fmt.Errorf("failed to parse the openai response. Error: %s", err.Error())