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

fr

Browse Source
This commit is contained in:
unintendedfraud
2023-05-06 11:42:50 +02:00
parent 13145cad31
commit a028fb8b06
1 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
openai/client.go
+9
View File
@@ -19,6 +19,7 @@ func NewClient(token string) *Client {
} }
func (c Client) ChatCompletions(prompt string) (*CompletionResponse, error) { func (c Client) ChatCompletions(prompt string) (*CompletionResponse, error) {
fmt.Println("using prompt: ", prompt)
body := &ChatCompletionPayload{ body := &ChatCompletionPayload{
Model: "gpt-3.5-turbo", Model: "gpt-3.5-turbo",
Prompt: prompt, Prompt: prompt,
@@ -40,6 +41,14 @@ func (c Client) ChatCompletions(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())
} }
var test interface{}
if err := json.Unmarshal(res.Body(), &test); err != nil {
return nil, fmt.Errorf("failed to parse the openai test response. Error: %s", err.Error())
}
fmt.Println()
fmt.Println("inteface: ", test)
fmt.Println()
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())