yilee01/gemini-proxy repository overview

git clone <repository-url>
cd gemini-proxy

go mod tidy

go run main.go

# 构建镜像
docker build -t gemini-proxy .

# 运行容器
docker run -d -p 8080:8080 -v $(pwd)/logs:/app/logs gemini-proxy

gemini-proxy/
├── api/                      # API 处理器
│   ├── chat.go              # 聊天接口处理
│   └── index.go             # 首页接口
├── core/                     # 核心模块
│   ├── gine/                # 自定义 Gin 扩展
│   │   ├── context.go       # 上下文封装
│   │   ├── engine.go        # 引擎封装
│   │   └── routergroup.go   # 路由组封装
│   ├── logger/              # 日志模块
│   │   ├── logger.go        # 日志器定义
│   │   └── message.go       # 日志消息
│   ├── router/              # 路由配置
│   │   └── route.go         # 路由注册
│   └── setting.go           # 全局设置
├── library/                  # 工具库
│   ├── constant/            # 常量定义
│   │   ├── code/            # 常量代码
│   │   └── errno/           # 错误码
│   ├── http/                # HTTP 工具
│   │   ├── event.go         # SSE 事件
│   │   └── http.go          # HTTP 请求处理
│   └── utils/               # 通用工具
│       ├── file.go          # 文件工具
│       ├── http.go          # HTTP 工具
│       └── unique_id.go     # 唯一 ID 生成
├── middleware/              # 中间件
│   ├── cors_middleware.go   # CORS 中间件
│   └── log_middleware.go    # 日志中间件
├── models/                  # 数据模型
│   ├── request.go           # 请求模型
│   └── response.go          # 响应模型
├── initialize/              # 初始化模块
│   └── logger.go            # 日志初始化
├── main.go                  # 程序入口
├── Dockerfile               # Docker 配置
├── go.mod                   # Go 模块文件
└── README.md                # 项目文档

GET /

{
  "message": "Welcome to the Gemini Proxy API! This server proxies requests to Google Gemini API."
}

POST /v1/chat/completions

Content-Type: application/json
Authorization: Bearer YOUR_GEMINI_API_KEY

{
  "model": "gemini-pro",
  "messages": [
    {
      "role": "user",
      "content": "Hello, how are you?"
    }
  ],
  "stream": false
}

{
  "role": "user",
  "content": [
    {
      "type": "text",
      "text": "请描述这张图片"
    },
    {
      "type": "image_url",
      "image_url": {
        "url": "https://example.com/image.jpg"
      }
    }
  ]
}

{
  "role": "user",
  "content": [
    {
      "type": "input_image",
      "image_url": "https://example.com/image.jpg"
    }
  ]
}

{
  "role": "user",
  "content": [
    {
      "type": "input_text",
      "input_text": "文本内容"
    }
  ]
}

{
  "input": [
    {
      "role": "user",
      "content": "消息内容"
    }
  ]
}

// 最大上传内存限制
MaxMultipart = 100 << 20  // 100MB

// 图片大小限制
ImageMaxSize = 100 * 1024 * 1024  // 100MB

// Gemini API 地址
GeminiApiUrl = "https://generativelanguage.googleapis.com/v1beta/openai/chat/completions"

// 日志文件配置
accessLogs := &lumberjack.Logger{
    Filename:   "logs/gemini-proxy.log",
    MaxSize:    100,  // MB
    MaxBackups: 30,   // 保留文件数
    MaxAge:     30,   // 天
    LocalTime:  true,
}

version: '3.8'

services:
  gemini-proxy:
    image: gemini-proxy:latest
    ports:
      - "8080:8080"
    volumes:
      - ./logs:/app/logs
    environment:
      - PUID=1000
      - PGID=1000
      - UMASK=022
    restart: unless-stopped

docker-compose up -d

server {
    listen 80;
    server_name your-domain.com;

    location / {
        proxy_pass http://localhost:8080;
        proxy_set_header Host $host;
        proxy_set_header X-Real-IP $remote_addr;
        proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;
        proxy_set_header X-Forwarded-Proto $scheme;
        
        # SSE 支持
        proxy_buffering off;
        proxy_cache off;
        proxy_set_header Connection '';
        proxy_http_version 1.1;
        chunked_transfer_encoding off;
    }
}

type YourRequest struct {
    Field string `json:"field"`
}

func YourHandler(ctx *gine.Context) {
    // 处理逻辑
}

r.POST("/your-endpoint", api.YourHandler)

func YourMiddleware() gine.HandlerFunc {
    return func(ctx *gine.Context) {
        // 前置处理
        ctx.Next()
        // 后置处理
    }
}

r.Use(middleware.YourMiddleware())

# 启动开发服务器
go run main.go

# 运行测试
go test ./...

# 代码格式化
go fmt ./...

# 代码检查
go vet ./...

LEVEL 2024-01-01 12:00:00 logid[xxx] uri[/v1/chat/completions] method[POST] ip[127.0.0.1] cost[100ms]