fix(logic): Use direct model name as requested
Browse files
main.go
CHANGED
|
@@ -43,13 +43,6 @@ var supportedModels = []ModelInfo{
|
|
| 43 |
OwnedBy: "google",
|
| 44 |
Description: "gemini-2.5-flash稳定经典专业模型",
|
| 45 |
},
|
| 46 |
-
{
|
| 47 |
-
ID: "gemini-1.5-pro-latest",
|
| 48 |
-
Object: "model",
|
| 49 |
-
Created: time.Now().Unix(),
|
| 50 |
-
OwnedBy: "google",
|
| 51 |
-
Description: "gemini-1.5-pro-latest 经典专业模型",
|
| 52 |
-
},
|
| 53 |
{
|
| 54 |
ID: "gemini-2.5-pro",
|
| 55 |
Object: "model",
|
|
@@ -60,11 +53,11 @@ var supportedModels = []ModelInfo{
|
|
| 60 |
}
|
| 61 |
|
| 62 |
// 将OpenAI模型名称映射到Gemini模型名称
|
|
|
|
| 63 |
var modelMapping = map[string]string{
|
| 64 |
-
"gemini-2.5-flash-preview-05-20": "gemini-
|
| 65 |
-
"gemini-2.5-flash": "gemini-
|
| 66 |
-
|
| 67 |
-
"gemini-2.5-pro": "gemini-1.5-pro-latest",
|
| 68 |
}
|
| 69 |
|
| 70 |
// 配置安全设置 (全部禁用)
|
|
@@ -207,10 +200,10 @@ func chatCompletionsHandler(w http.ResponseWriter, r *http.Request) {
|
|
| 207 |
return
|
| 208 |
}
|
| 209 |
|
| 210 |
-
|
| 211 |
-
|
| 212 |
-
|
| 213 |
-
|
| 214 |
|
| 215 |
history, lastPrompt, systemInstruction := convertMessages(req.Messages)
|
| 216 |
|
|
|
|
| 43 |
OwnedBy: "google",
|
| 44 |
Description: "gemini-2.5-flash稳定经典专业模型",
|
| 45 |
},
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 46 |
{
|
| 47 |
ID: "gemini-2.5-pro",
|
| 48 |
Object: "model",
|
|
|
|
| 53 |
}
|
| 54 |
|
| 55 |
// 将OpenAI模型名称映射到Gemini模型名称
|
| 56 |
+
// 根据您的要求,键和值现在是相同的,不做任何转换。
|
| 57 |
var modelMapping = map[string]string{
|
| 58 |
+
"gemini-2.5-flash-preview-05-20": "gemini-2.5-flash-preview-05-20",
|
| 59 |
+
"gemini-2.5-flash": "gemini-2.5-flash",
|
| 60 |
+
"gemini-2.5-pro": "gemini-2.5-pro",
|
|
|
|
| 61 |
}
|
| 62 |
|
| 63 |
// 配置安全设置 (全部禁用)
|
|
|
|
| 200 |
return
|
| 201 |
}
|
| 202 |
|
| 203 |
+
// 根据您的要求,直接使用请求中的模型名称
|
| 204 |
+
modelName := req.Model
|
| 205 |
+
log.Printf("接收到模型请求: '%s',将直接使用该名称。", modelName)
|
| 206 |
+
|
| 207 |
|
| 208 |
history, lastPrompt, systemInstruction := convertMessages(req.Messages)
|
| 209 |
|