heimoshuiyu/openai-api-route
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 2 Wiki Activity

use path as default model name

Browse Source
This commit is contained in:
黑墨水鱼
2024-02-16 22:31:32 +08:00
parent f2e32340e3
commit 46ee30ced7
2 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
main.go
View File

@@ -109,6 +109,7 @@ func main() {
CreatedAt: time.Now(), CreatedAt: time.Now(),
Authorization: c.Request.Header.Get("Authorization"), Authorization: c.Request.Header.Get("Authorization"),
UserAgent: c.Request.Header.Get("User-Agent"), UserAgent: c.Request.Header.Get("User-Agent"),
Model: c.Request.URL.Path,
} }
// check authorization header // check authorization header

5
process.go
View File

@@ -33,9 +33,6 @@ func processRequest(c *gin.Context, upstream *OPENAI_UPSTREAM, record *Record, s
path := strings.TrimPrefix(c.Request.URL.Path, "/v1") path := strings.TrimPrefix(c.Request.URL.Path, "/v1")
// recoognize whisper url // recoognize whisper url
if strings.HasPrefix(path, "/audio/transcriptions") || strings.HasPrefix(path, "/audio/translations") {
record.Model = "whisper"
}
remote.Path = upstream.URL.Path + path remote.Path = upstream.URL.Path + path
log.Println("[proxy.begin]:", remote) log.Println("[proxy.begin]:", remote)
log.Println("[proxy.begin]: shouldResposne:", shouldResponse) log.Println("[proxy.begin]: shouldResposne:", shouldResponse)
@@ -65,7 +62,7 @@ func processRequest(c *gin.Context, upstream *OPENAI_UPSTREAM, record *Record, s
record.Body = string(inBody) record.Body = string(inBody)
requestBody, requestBodyOK := ParseRequestBody(inBody) requestBody, requestBodyOK := ParseRequestBody(inBody)
// record if parse success // record if parse success
if requestBodyOK == nil && record.Model == "" { if requestBodyOK == nil && requestBody.Model != "" {
record.Model = requestBody.Model record.Model = requestBody.Model
} }