This commit is contained in:
tkisme 2024-05-17 15:05:22 +08:00
parent 393bad0930
commit ddbe908873
2 changed files with 14 additions and 3 deletions

View File

@ -2,7 +2,7 @@
"bind": "127.0.0.1:8181",
"proxy_url": "",
"timeout": 600,
"codex_api_base": "https://api-proxy.oaipro.com/v1",
"codex_api_base": "https://api-proxy.oaipro.com/v1/completions",
"codex_api_key": "sk-xxx",
"codex_api_organization": "",
"codex_api_project": "",

15
main.go
View File

@ -14,6 +14,7 @@ import (
"net/http"
"net/url"
"os"
"strings"
"time"
)
@ -220,9 +221,19 @@ func (s *ProxyService) codeCompletions(c *gin.Context) {
body, _ = sjson.DeleteBytes(body, "extra")
body, _ = sjson.DeleteBytes(body, "nwo")
body, _ = sjson.SetBytes(body, "model", "gpt-3.5-turbo-instruct")
body, _ = sjson.SetBytes(body, "model", s.cfg.CodexModelDefault)
proxyUrl := s.cfg.CodexApiBase + "/completions"
proxyUrl := s.cfg.CodexApiBase
if strings.HasPrefix(s.cfg.CodexModelDefault, "@") {
proxyUrl = s.cfg.CodexApiBase
message := gjson.GetBytes(body, "prompt").String()
body, _ = sjson.DeleteBytes(body, "prompt")
msg := make([]GPTMessage, 0)
msg = append(msg, GPTMessage{Role: "system", Content: "You are a helpful assistant"})
msg = append(msg, GPTMessage{Role: "user", Content: message})
body, _ = sjson.SetBytes(body, "messages", msg)
body, _ = sjson.DeleteBytes(body, "n")
}
req, err := http.NewRequestWithContext(ctx, http.MethodPost, proxyUrl, io.NopCloser(bytes.NewBuffer(body)))
if nil != err {
abortCodex(c, http.StatusInternalServerError)