xidu commited on
Commit
f0c0993
·
1 Parent(s): 46bfd69

fix(logic): Use direct model name as requested

Browse files
Files changed (1) hide show
  1. main.go +8 -15
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-1.5-flash-latest",
65
- "gemini-2.5-flash": "gemini-1.5-flash-latest",
66
- "gemini-1.5-pro-latest": "gemini-1.5-pro-latest",
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
- modelName, ok := modelMapping[req.Model]
211
- if !ok {
212
- modelName = "gemini-1.5-flash-latest" // 默认模型
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