Unverified Commit ccf6af4d authored by boojack's avatar boojack Committed by GitHub

fix: request body format in openai api (#1309)

parent ce7564a9
package openai
import (
"bytes"
"encoding/json"
"errors"
"io"
"net/http"
"net/url"
"strings"
)
type ChatCompletionMessage struct {
......@@ -25,10 +25,6 @@ type ChatCompletionResponse struct {
}
func PostChatCompletion(prompt string, apiKey string, apiHost string) (string, error) {
requestBody := strings.NewReader(`{
"model": "gpt-3.5-turbo",
"messages": [{"role": "user", "content": "` + prompt + `"}]
}`)
if apiHost == "" {
apiHost = "https://api.openai.com"
}
......@@ -37,7 +33,16 @@ func PostChatCompletion(prompt string, apiKey string, apiHost string) (string, e
return "", err
}
req, err := http.NewRequest("POST", url, requestBody)
values := map[string]interface{}{
"model": "gpt-3.5-turbo",
"messages": []map[string]string{{"role": "user", "content": prompt}},
}
jsonValue, err := json.Marshal(values)
if err != nil {
return "", err
}
req, err := http.NewRequest("POST", url, bytes.NewBuffer(jsonValue))
if err != nil {
return "", err
}
......
package openai
import (
"bytes"
"encoding/json"
"errors"
"io"
"net/http"
"net/url"
"strings"
)
type TextCompletionChoice struct {
......@@ -20,13 +20,6 @@ type TextCompletionResponse struct {
}
func PostTextCompletion(prompt string, apiKey string, apiHost string) (string, error) {
requestBody := strings.NewReader(`{
"prompt": "` + prompt + `",
"temperature": 0.5,
"max_tokens": 100,
"n": 1,
"stop": "."
}`)
if apiHost == "" {
apiHost = "https://api.openai.com"
}
......@@ -35,7 +28,20 @@ func PostTextCompletion(prompt string, apiKey string, apiHost string) (string, e
return "", err
}
req, err := http.NewRequest("POST", url, requestBody)
values := map[string]interface{}{
"model": "gpt-3.5-turbo",
"prompt": prompt,
"temperature": 0.5,
"max_tokens": 100,
"n": 1,
"stop": ".",
}
jsonValue, err := json.Marshal(values)
if err != nil {
return "", err
}
req, err := http.NewRequest("POST", url, bytes.NewBuffer(jsonValue))
if err != nil {
return "", err
}
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment