xinyin025/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-11-20 23:16:49 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge 7f6eab77ab into 505817ca17

Browse Source
This commit is contained in:
WangBooth
2024-01-01 17:52:20 +08:00
committed by GitHub
parent 505817ca17 7f6eab77ab
commit 1596751835
5 changed files with 31 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
controller/model.go
View File

@@ -244,6 +244,7 @@ func init() {
Parent: nil,
},
{
Id: "gpt-4-vision-preview",
Object: "model",
Created: 1699593571,

9
controller/relay-text.go
View File

@@ -48,6 +48,7 @@ func init() {
}
func relayTextHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode {
channelType := c.GetInt("channel")
channelId := c.GetInt("channel_id")
tokenId := c.GetInt("token_id")
@@ -130,6 +131,11 @@ func relayTextHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode {
if c.GetString("base_url") != "" {
baseURL = c.GetString("base_url")
}
Organization := ""
if c.GetString("openai_organization") != "" {
Organization = c.GetString("openai_organization")
}
fullRequestURL := getFullRequestURL(baseURL, requestURL, channelType)
switch apiType {
case APITypeOpenAI:
@@ -368,6 +374,9 @@ func relayTextHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode {
req.Header.Set("HTTP-Referer", "https://github.com/songquanpeng/one-api")
req.Header.Set("X-Title", "One API")
}
if Organization != "" {
req.Header.Set("OpenAI-Organization", Organization)
}
}
case APITypeClaude:
req.Header.Set("x-api-key", apiKey)