Spaces:
Paused
Paused
| const express = require('express'); | |
| const morgan = require('morgan'); | |
| const { createProxyMiddleware } = require('http-proxy-middleware'); | |
| const axios = require('axios'); | |
| const fs = require('fs'); | |
| const app = express(); | |
| app.use(morgan('dev')); | |
| app.use(express.json()); | |
| // 生成随机字符串的函数 | |
| function getRandomIDPro({ size, dictType, customDict }) { | |
| let random = ''; | |
| if (!customDict) { | |
| switch (dictType) { | |
| case 'alphabet': | |
| customDict = 'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ'; | |
| break; | |
| case 'max': | |
| customDict = '0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ_-'; | |
| break; | |
| default: | |
| customDict = '0123456789'; | |
| } | |
| } | |
| for (; size--; ) random += customDict[(Math.random() * customDict.length) | 0]; | |
| return random; | |
| } | |
| // 模型列表路由 | |
| app.get('/hf/v1/models', (req, res) => { | |
| const models = { | |
| "object": "list", | |
| "data": [ | |
| { | |
| "id": "claude-3.5-sonnet", | |
| "object": "model", | |
| "created": 1706745938, | |
| "owned_by": "cursor" | |
| }, | |
| { | |
| "id": "gpt-4", | |
| "object": "model", | |
| "created": 1706745938, | |
| "owned_by": "cursor" | |
| }, | |
| { | |
| "id": "gpt-4o", | |
| "object": "model", | |
| "created": 1706745938, | |
| "owned_by": "cursor" | |
| }, | |
| { | |
| "id": "claude-3-opus", | |
| "object": "model", | |
| "created": 1706745938, | |
| "owned_by": "cursor" | |
| }, | |
| { | |
| "id": "gpt-3.5-turbo", | |
| "object": "model", | |
| "created": 1706745938, | |
| "owned_by": "cursor" | |
| }, | |
| { | |
| "id": "gpt-4-turbo-2024-04-09", | |
| "object": "model", | |
| "created": 1706745938, | |
| "owned_by": "cursor" | |
| }, | |
| { | |
| "id": "gpt-4o-128k", | |
| "object": "model", | |
| "created": 1706745938, | |
| "owned_by": "cursor" | |
| }, | |
| { | |
| "id": "gemini-1.5-flash-500k", | |
| "object": "model", | |
| "created": 1706745938, | |
| "owned_by": "cursor" | |
| }, | |
| { | |
| "id": "claude-3-haiku-200k", | |
| "object": "model", | |
| "created": 1706745938, | |
| "owned_by": "cursor" | |
| }, | |
| { | |
| "id": "claude-3-5-sonnet-200k", | |
| "object": "model", | |
| "created": 1706745938, | |
| "owned_by": "cursor" | |
| }, | |
| { | |
| "id": "claude-3-5-sonnet-20241022", | |
| "object": "model", | |
| "created": 1706745938, | |
| "owned_by": "cursor" | |
| }, | |
| { | |
| "id": "gpt-4o-mini", | |
| "object": "model", | |
| "created": 1706745938, | |
| "owned_by": "cursor" | |
| }, | |
| { | |
| "id": "o1-mini", | |
| "object": "model", | |
| "created": 1706745938, | |
| "owned_by": "cursor" | |
| }, | |
| { | |
| "id": "o1-preview", | |
| "object": "model", | |
| "created": 1706745938, | |
| "owned_by": "cursor" | |
| }, | |
| { | |
| "id": "o1", | |
| "object": "model", | |
| "created": 1706745938, | |
| "owned_by": "cursor" | |
| }, | |
| { | |
| "id": "claude-3.5-haiku", | |
| "object": "model", | |
| "created": 1706745938, | |
| "owned_by": "cursor" | |
| }, | |
| { | |
| "id": "gemini-exp-1206", | |
| "object": "model", | |
| "created": 1706745938, | |
| "owned_by": "cursor" | |
| }, | |
| { | |
| "id": "gemini-2.0-flash-thinking-exp", | |
| "object": "model", | |
| "created": 1706745938, | |
| "owned_by": "cursor" | |
| }, | |
| { | |
| "id": "gemini-2.0-flash-exp", | |
| "object": "model", | |
| "created": 1706745938, | |
| "owned_by": "cursor" | |
| }, | |
| { | |
| "id": "deepseek-v3", | |
| "object": "model", | |
| "created": 1706745938, | |
| "owned_by": "cursor" | |
| }, | |
| { | |
| "id": "deepseek-r1", | |
| "object": "model", | |
| "created": 1706745938, | |
| "owned_by": "cursor" | |
| } | |
| ] | |
| }; | |
| res.json(models); | |
| }); | |
| // 聊天完成度代理 | |
| app.post('/hf/v1/chat/completions', async (req, res) => { | |
| try { | |
| // 生成 checksum | |
| const checksum = `ICMlQsvO${getRandomIDPro({ dictType: 'max', size: 64 })}/${getRandomIDPro({ dictType: 'max', size: 64 })}`; | |
| // 从请求头获取认证信息 | |
| const authHeader = req.headers.authorization; | |
| const token = authHeader?.replace('Bearer ', '') || ''; | |
| // 构建请求头 | |
| const headers = { | |
| 'Authorization': authHeader, | |
| 'Content-Type': 'application/json', | |
| 'x-cursor-checksum': checksum, | |
| 'x-cursor-token': token, | |
| 'User-Agent': 'cursor/0.2.0', | |
| 'Accept': 'application/json' | |
| }; | |
| // 构建请求体 | |
| const requestBody = { | |
| ...req.body, | |
| stream: false | |
| }; | |
| // 转发请求到目标服务器 | |
| const response = await axios.post('http://localhost:3010/v1/chat/completions', requestBody, { | |
| headers: headers, | |
| timeout: 30000 | |
| }); | |
| // 返回响应 | |
| res.status(response.status).json(response.data); | |
| } catch (error) { | |
| console.error('Error in chat completions:', error); | |
| if (error.response) { | |
| res.status(error.response.status).json(error.response.data); | |
| } else { | |
| res.status(500).json({ | |
| error: 'Internal server error', | |
| message: error.message | |
| }); | |
| } | |
| } | |
| }); | |
| // 主页路由 | |
| app.get('/', (req, res) => { | |
| const htmlContent = ` | |
| <!DOCTYPE html> | |
| <html lang="en"> | |
| <head> | |
| <meta charset="UTF-8"> | |
| <meta name="viewport" content="width=device-width, initial-scale=1.0"> | |
| <title>Cursor To OpenAI</title> | |
| <style> | |
| :root { | |
| --primary-color: #2563eb; | |
| --bg-color: #f8fafc; | |
| --card-bg: #ffffff; | |
| } | |
| body { | |
| padding: 20px; | |
| font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, sans-serif; | |
| max-width: 1000px; | |
| margin: 0 auto; | |
| line-height: 1.6; | |
| background: var(--bg-color); | |
| color: #1a1a1a; | |
| } | |
| .container { | |
| padding: 20px; | |
| } | |
| .header { | |
| text-align: center; | |
| margin-bottom: 40px; | |
| } | |
| .header h1 { | |
| color: var(--primary-color); | |
| font-size: 2.5em; | |
| margin-bottom: 10px; | |
| } | |
| .info { | |
| background: #fff; | |
| padding: 25px; | |
| border-radius: 12px; | |
| box-shadow: 0 4px 6px -1px rgba(0, 0, 0, 0.1); | |
| margin-bottom: 30px; | |
| border: 1px solid #e5e7eb; | |
| } | |
| .info-item { | |
| margin: 15px 0; | |
| padding: 10px; | |
| background: #f8fafc; | |
| border-radius: 8px; | |
| border: 1px solid #e5e7eb; | |
| } | |
| .info-label { | |
| color: #4b5563; | |
| font-size: 0.9em; | |
| margin-bottom: 5px; | |
| } | |
| .info-value { | |
| color: var(--primary-color); | |
| font-weight: 500; | |
| } | |
| .models { | |
| background: var(--card-bg); | |
| padding: 25px; | |
| border-radius: 12px; | |
| box-shadow: 0 4px 6px -1px rgba(0, 0, 0, 0.1); | |
| border: 1px solid #e5e7eb; | |
| } | |
| .models h3 { | |
| color: #1a1a1a; | |
| margin-bottom: 20px; | |
| font-size: 1.5em; | |
| border-bottom: 2px solid #e5e7eb; | |
| padding-bottom: 10px; | |
| } | |
| .model-item { | |
| margin: 12px 0; | |
| padding: 15px; | |
| background: #f8fafc; | |
| border-radius: 8px; | |
| border: 1px solid #e5e7eb; | |
| transition: all 0.3s ease; | |
| display: flex; | |
| justify-content: space-between; | |
| align-items: center; | |
| } | |
| .model-item:hover { | |
| transform: translateY(-2px); | |
| box-shadow: 0 4px 12px rgba(0, 0, 0, 0.1); | |
| } | |
| .model-name { | |
| font-weight: 500; | |
| color: #1a1a1a; | |
| } | |
| .model-provider { | |
| color: #6b7280; | |
| font-size: 0.9em; | |
| padding: 4px 8px; | |
| background: #f1f5f9; | |
| border-radius: 4px; | |
| } | |
| @media (max-width: 768px) { | |
| body { | |
| padding: 10px; | |
| } | |
| .container { | |
| padding: 10px; | |
| } | |
| } | |
| </style> | |
| </head> | |
| <body> | |
| <div class="container"> | |
| <div class="header"> | |
| <h1>Cursor To OpenAI Server</h1> | |
| <p>高性能 AI 模型代理服务</p> | |
| </div> | |
| <div class="info"> | |
| <h2>配置信息</h2> | |
| <div class="info-item"> | |
| <div class="info-label">聊天来源</div> | |
| <div class="info-value">自定义(兼容 OpenAI)</div> | |
| </div> | |
| <div class="info-item"> | |
| <div class="info-label">自定义端点(基本URL)</div> | |
| <div class="info-value" id="endpoint-url"></div> | |
| </div> | |
| <div class="info-item"> | |
| <div class="info-label">自定义API密钥</div> | |
| <div class="info-value">抓取的Cursor Cookie，格式为user_...</div> | |
| </div> | |
| </div> | |
| <div class="models"> | |
| <h3>支持的模型列表</h3> | |
| <div id="model-list"></div> | |
| </div> | |
| </div> | |
| <script> | |
| const url = new URL(window.location.href); | |
| const link = url.protocol + '//' + url.host + '/hf/v1'; | |
| document.getElementById('endpoint-url').textContent = link; | |
| fetch(link + '/models') | |
| .then(response => response.json()) | |
| .then(data => { | |
| const modelList = document.getElementById('model-list'); | |
| data.data.forEach(model => { | |
| const div = document.createElement('div'); | |
| div.className = 'model-item'; | |
| div.innerHTML = | |
| '<span class="model-name">' + model.id + '</span>' + | |
| '<span class="model-provider">' + model.owned_by + '</span>'; | |
| modelList.appendChild(div); | |
| }); | |
| }) | |
| .catch(error => { | |
| console.error('Error fetching models:', error); | |
| document.getElementById('model-list').textContent = '获取模型列表失败'; | |
| }); | |
| </script> | |
| </body> | |
| </html> | |
| `; | |
| res.send(htmlContent); | |
| }); | |
| const port = process.env.HF_PORT || 7860; | |
| app.listen(port, () => { | |
| console.log(`HF Proxy server is running at PORT: ${port}`); | |
| }); | |