Timeline

• Elon Musk (@elonmusk) 2025-03-26 13:08:54.615799144 +0800 CST
  

  Gemini API 集成方案

  您提供的代码正在处理不同API的认证和路径映射，现在需要兼容Gemini API的调用格式。查看您提供的curl请求，Gemini API有以下特点：

  1. 使用查询参数(?key=)进行身份验证，而不是Authorization头
  2. 请求路径包含:generateContent这样的动作后缀
  3. 包含system_instruction和contents的请求体结构

  修改建议

  在generativelanguage.googleapis.com的case中添加对Gemini API的处理:

  case "generativelanguage.googleapis.com":
      // 检查URL中是否已有key参数
      queryParams := req.URL.Query()
      if queryParams.Get("key") == "" {
          // 没有key参数时添加到查询字符串
          queryParams.Set("key", key.SecretKey)
          req.URL.RawQuery = queryParams.Encode()
      }
      
      // 处理路径映射
      if newPath, exists := conf.PathMappers[4][req.URL.Path]; exists {
          req.URL.Path = newPath
      } else if strings.Contains(req.URL.Path, ":generateContent") {
          // 特别处理generateContent端点
          newPath := normalizePath(req.URL.Path)
          conf.PathMappers[4][req.URL.Path] = newPath
          req.URL.Path = newPath
      } else if strings.Contains(req.URL.Path, "chat") {
          newPath := normalizePath(req.URL.Path)
          conf.PathMappers[4][req.URL.Path] = newPath
          req.URL.Path = newPath
      }
      
      // 某些Gemini端点可能还是需要Authorization头
      if !strings.Contains(req.URL.RawQuery, "key=") {
          req.Header.Set("Authorization", "Bearer "+key.SecretKey)
      }
  

  注意事项

  1. 这个实现保留了原有处理逻辑，同时添加了对查询参数认证的支持
  2. 增加了对:generateContent路径模式的处理
  3. 实现了认证方式的双重兼容 - 既支持查询参数也支持Authorization头
  4. 如果URL中已经有key参数，则不会重复添加

  如果您的应用需要对Gemini API请求/响应格式进行转换以匹配内部格式，可能还需要添加中间件处理逻辑。