Upload 7 files

.gitattributes

@@ -1,35 +1,35 @@
-*.7z filter=lfs diff=lfs merge=lfs -text
-*.arrow filter=lfs diff=lfs merge=lfs -text
-*.bin filter=lfs diff=lfs merge=lfs -text
-*.bz2 filter=lfs diff=lfs merge=lfs -text
-*.ckpt filter=lfs diff=lfs merge=lfs -text
-*.ftz filter=lfs diff=lfs merge=lfs -text
-*.gz filter=lfs diff=lfs merge=lfs -text
-*.h5 filter=lfs diff=lfs merge=lfs -text
-*.joblib filter=lfs diff=lfs merge=lfs -text
-*.lfs.* filter=lfs diff=lfs merge=lfs -text
-*.mlmodel filter=lfs diff=lfs merge=lfs -text
-*.model filter=lfs diff=lfs merge=lfs -text
-*.msgpack filter=lfs diff=lfs merge=lfs -text
-*.npy filter=lfs diff=lfs merge=lfs -text
-*.npz filter=lfs diff=lfs merge=lfs -text
-*.onnx filter=lfs diff=lfs merge=lfs -text
-*.ot filter=lfs diff=lfs merge=lfs -text
-*.parquet filter=lfs diff=lfs merge=lfs -text
-*.pb filter=lfs diff=lfs merge=lfs -text
-*.pickle filter=lfs diff=lfs merge=lfs -text
-*.pkl filter=lfs diff=lfs merge=lfs -text
-*.pt filter=lfs diff=lfs merge=lfs -text
-*.pth filter=lfs diff=lfs merge=lfs -text
-*.rar filter=lfs diff=lfs merge=lfs -text
-*.safetensors filter=lfs diff=lfs merge=lfs -text
-saved_model/**/* filter=lfs diff=lfs merge=lfs -text
-*.tar.* filter=lfs diff=lfs merge=lfs -text
-*.tar filter=lfs diff=lfs merge=lfs -text
-*.tflite filter=lfs diff=lfs merge=lfs -text
-*.tgz filter=lfs diff=lfs merge=lfs -text
-*.wasm filter=lfs diff=lfs merge=lfs -text
-*.xz filter=lfs diff=lfs merge=lfs -text
-*.zip filter=lfs diff=lfs merge=lfs -text
-*.zst filter=lfs diff=lfs merge=lfs -text
-*tfevents* filter=lfs diff=lfs merge=lfs -text
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -0,0 +1 @@
+node_modules

Dockerfile

@@ -0,0 +1,11 @@
+FROM node:18-slim
+
+WORKDIR /app
+
+COPY package*.json ./
+RUN npm install
+
+COPY . .
+
+EXPOSE 3000
+CMD ["npm", "run", "dev"] 

README.md

@@ -1,11 +1,12 @@
----
-title: Gen2
-emoji: 👁
-colorFrom: pink
-colorTo: pink
-sdk: docker
-pinned: false
-license: mit
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+---
+title: Gen
+emoji: 🐢
+colorFrom: pink
+colorTo: green
+sdk: docker
+pinned: false
+license: mit
+app_port: 3000
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

package.json

@@ -0,0 +1,17 @@
+{
+  "name": "genspark2api",
+  "version": "1.0.0",
+  "main": "index.js",
+  "scripts": {
+    "dev": "node src/index.js"
+  },
+  "keywords": [],
+  "author": "",
+  "license": "ISC",
+  "description": "",
+  "dependencies": {
+    "koa": "^2.15.3",
+    "koa-bodyparser": "^4.4.1",
+    "koa-router": "^13.0.1"
+  }
+}

src/index.js

@@ -0,0 +1,428 @@
+const Koa = require('koa')
+const Router = require('koa-router')
+const bodyParser = require('koa-bodyparser')
+const models = require("./models")
+const crypto = require('crypto')
+
+
+const app = new Koa()
+const router = new Router()
+
+// 使用 bodyParser 中间件
+app.use(bodyParser())
+
+// 配置 bodyParser
+app.use(bodyParser({
+  enableTypes: ['json', 'form', 'text'],
+  jsonLimit: '30mb',  // JSON 数据大小限制
+  formLimit: '30mb',  // form 数据大小限制
+  textLimit: '30mb',  // text 数据大小限制
+}))
+
+// 添加 CORS 中间件
+app.use(async (ctx, next) => {
+  ctx.set('Access-Control-Allow-Origin', '*');
+  ctx.set('Access-Control-Allow-Headers', 'Origin, X-Requested-With, Content-Type, Accept, Authorization');
+  ctx.set('Access-Control-Allow-Methods', 'POST, GET, PUT, DELETE, OPTIONS');
+  ctx.set('Access-Control-Max-Age', '3600');
+  
+  // 处理 OPTIONS 请求
+  if (ctx.method === 'OPTIONS') {
+    ctx.status = 204;
+    return;
+  }
+  
+  await next();
+});
+
+// 添加请求日志中间件
+app.use(async (ctx, next) => {
+  const start = Date.now()
+  console.log(`[${new Date().toISOString()}] ${ctx.method} ${ctx.url} - Request started`)
+  try {
+    await next()
+  } catch (err) {
+    throw err
+  } finally {
+    const ms = Date.now() - start
+    console.log(`[${new Date().toISOString()}] ${ctx.method} ${ctx.url} - Response ${ctx.status} - ${ms}ms`)
+  }
+})
+
+// 错误处理中间件
+app.use(async (ctx, next) => {
+  try {
+    await next()
+    // 只有当没有任何路由处理请求时才设置 404
+    if (ctx.status === 404 && !ctx.body) {
+      ctx.status = 404
+      ctx.body = {
+        error: 'Not Found',
+        message: `The requested path ${ctx.path} was not found`,
+        availableEndpoints: [
+          '/v1/chat/completions',
+          '/hf/v1/chat/completions',
+          '/v1/models',
+          '/hf/v1/models'
+        ]
+      }
+    }
+  } catch (err) {
+    console.error('Error details:', {
+      path: ctx.path,
+      method: ctx.method,
+      error: err.message,
+      stack: err.stack
+    })
+    ctx.status = err.status || 500
+    ctx.body = {
+      error: err.name || 'Internal Server Error',
+      message: err.message,
+      path: ctx.path
+    }
+    ctx.app.emit('error', err, ctx)
+  }
+})
+
+const makeRequest = async (session_id, requestModel, messages) => {
+  console.log('开始请求:', { 
+    session_id: '***', 
+    model: requestModel,
+    messages_length: messages?.length,
+    timestamp: new Date().toISOString()
+  })
+  try {
+    // 设置请求头
+    const myHeaders = new Headers()
+    myHeaders.append("Cookie", `session_id=${session_id}`)
+    myHeaders.append("User-Agent", "Apifox/1.0.0 (https://apifox.com)");
+    myHeaders.append("Content-Type", "application/json")
+    myHeaders.append("Accept", "*/*")
+    myHeaders.append("Host", "www.genspark.ai")
+    myHeaders.append("Connection", "keep-alive")
+
+    // 设置请求体
+    var body = JSON.stringify({
+      "type": "COPILOT_MOA_CHAT",
+      "current_query_string": "type=COPILOT_MOA_CHAT",
+      "messages": messages,
+      "action_params": {},
+      "extra_data": {
+        "models": [
+          models[`${requestModel}`] || models["claude-3-5-sonnet-20241022"]
+        ],
+        "run_with_another_model": false,
+        "writingContent": null
+      }
+    })
+
+    const requestConfig = {
+      method: 'POST',
+      headers: myHeaders,
+      body: body,
+      redirect: 'follow'
+    };
+
+    const response = await fetch("https://www.genspark.ai/api/copilot/ask", requestConfig)
+    
+    console.log('收到响应:', {
+      status: response.status,
+      ok: response.ok,
+      headers: Object.fromEntries(response.headers.entries()),
+      timestamp: new Date().toISOString()
+    })
+    
+    if (!response.ok) {
+      throw new Error(`HTTP error! status: ${response.status}`)
+    }
+    
+    return response
+  } catch (error) {
+    console.error('请求出错:', {
+      error: error.message,
+      cause: error.cause?.message,
+      stack: error.stack,
+      timestamp: new Date().toISOString()
+    })
+    throw error
+  }
+}
+
+// 创建处理函数
+const handleChatCompletions = async (ctx) => {
+  const { messages, stream = false, model = 'claude-3-5-sonnet' } = ctx.request.body
+  const session_id = ctx.get('Authorization')?.replace('Bearer ', '')
+
+  console.log('收到请求:', {
+    path: ctx.path,
+    model,
+    stream,
+    messages_count: messages?.length
+  })
+
+  if (!session_id) {
+    ctx.status = 401
+    ctx.body = { error: '未提供有效的 session_id' }
+    return
+  }
+
+  try {
+    const response = await makeRequest(session_id, model, messages)
+    
+    if (!response.body) {
+      throw new Error('Response body is null')
+    }
+
+    if (stream == "true" || stream == true) {
+      ctx.status = 200
+      ctx.set({
+        'Content-Type': 'text/event-stream',
+        'Cache-Control': 'no-cache',
+        'Connection': 'keep-alive',
+      })
+    } else {
+      ctx.status = 200
+      ctx.set({
+        'Content-Type': 'application/json',
+      })
+    }
+
+    const messageId = crypto.randomUUID()
+    const reader = response.body.getReader()
+    let resBody = {
+      id: `chatcmpl-${messageId}`,
+      object: 'chat.completion',
+      created: Math.floor(Date.now() / 1000),
+      model,
+      choices: [],
+      usage: {
+        prompt_tokens: 0,
+        completion_tokens: 0,
+        total_tokens: 0,
+      },
+    }
+
+    if (stream == "true" || stream == true) {
+      ctx.res.write(`data: ${JSON.stringify({
+        "id": `chatcmpl-${messageId}`,
+        "choices": [
+          {
+            "index": 0,
+            "delta": {
+              "content": "",
+              "role": "assistant"
+            }
+          }
+        ],
+        "created": Math.floor(Date.now() / 1000),
+        "model": model,
+        "object": "chat.completion.chunk"
+      })}\n\n`)
+    }
+
+    try {
+      while (true) {
+        const { done, value } = await reader.read()
+        if (done) {
+          console.log('响应流结束', {
+            messageId,
+            timestamp: new Date().toISOString()
+          })
+          if (stream == "true" || stream == true) {
+            ctx.res.write('data: [DONE]\n\n')
+          }
+          break
+        }
+
+        if (stream == "true" || stream == true) {
+          const text = new TextDecoder().decode(value)
+          console.log('收到流数据:', {
+            length: text.length,
+            timestamp: new Date().toISOString()
+          })
+          const textContent = [...text.matchAll(/data:.*"}/g)]
+          
+          textContent.forEach((item, index) => {
+            if (!item[0]) {
+              console.log('跳过空数据块:', { index })
+              return
+            }
+            try {
+              const content = JSON.parse(item[0].replace("data: ", ''))
+              if (!content || !content.delta) {
+                console.log('无效数据块:', { index, content })
+                return
+              }
+
+              ctx.res.write(`data: ${JSON.stringify({
+                "id": `chatcmpl-${messageId}`,
+                "choices": [
+                  {
+                    "index": 0,
+                    "delta": {
+                      "content": content.delta
+                    }
+                  }
+                ],
+                "created": Math.floor(Date.now() / 1000),
+                "model": model,
+                "object": "chat.completion.chunk"
+              })}\n\n`)
+            } catch (e) {
+              console.error('解析流数据失败:', {
+                index,
+                data: item[0],
+                error: e.message
+              })
+            }
+          })
+        } else {
+          const text = new TextDecoder().decode(value)
+          console.log('收到非流数据:', {
+            length: text.length,
+            timestamp: new Date().toISOString()
+          })
+          const textContent = [...text.matchAll(/data:.*"}/g)]
+
+          textContent.forEach((item, index) => {
+            if (!item[0]) {
+              console.log('跳过空数据块:', { index })
+              return
+            }
+            try {
+              const content = JSON.parse(item[0].replace("data: ", ''))
+              if (!content || !content.field_value || content.field_name == 'session_state.answer_is_finished' || content.field_name == 'content' || content.field_name == 'session_state' || content.delta || content.type == 'project_field') {
+                console.log('跳过无效数据:', {
+                  field_name: content?.field_name,
+                  type: content?.type
+                })
+                return
+              }
+
+              resBody.choices.push({
+                index: 0,
+                message: {
+                  role: 'assistant',
+                  content: content.field_value,
+                },
+                finish_reason: 'stop',
+              })
+              resBody.usage.total_tokens = content.field_value.length
+            } catch (e) {
+              console.error('解析非流数据失败:', {
+                index,
+                data: item[0],
+                error: e.message
+              })
+            }
+          })
+        }
+      }
+
+      if (stream == "false" || stream == false) {
+        if (resBody.choices.length === 0) {
+          resBody.choices.push({
+            index: 0,
+            message: {
+              role: 'assistant',
+              content: '抱歉，我无法处理您的请求。',
+            },
+            finish_reason: 'stop',
+          })
+        }
+        ctx.body = resBody
+      } else {
+        ctx.res.end()
+      }
+    } catch (error) {
+      console.error('响应处理出错:', {
+        error: error.message,
+        cause: error.cause?.message,
+        stack: error.stack,
+        timestamp: new Date().toISOString()
+      })
+      if (stream == "true" || stream == true) {
+        try {
+          ctx.res.write(`data: ${JSON.stringify({
+            "id": `chatcmpl-${messageId}`,
+            "choices": [
+              {
+                "index": 0,
+                "delta": {
+                  "content": "\n\n抱歉，处理您的请求时出现错误。"
+                }
+              }
+            ],
+            "created": Math.floor(Date.now() / 1000),
+            "model": model,
+            "object": "chat.completion.chunk"
+          })}\n\n`)
+          ctx.res.write('data: [DONE]\n\n')
+        } catch (e) {
+          console.error('发送错误信息失败:', e)
+        } finally {
+          ctx.res.end()
+        }
+      } else {
+        ctx.status = 500
+        ctx.body = { error: '响应处理失败', details: error.toString() }
+      }
+    }
+  } catch (error) {
+    console.error('请求处理出错:', error)
+    ctx.status = error.status || 500
+    ctx.body = { 
+      error: error.message || '请求处理失败',
+      details: error.toString()
+    }
+  }
+}
+
+// 创建获取模型列表处理函数
+const handleModels = async (ctx) => {
+  ctx.body = {
+    object: "list",
+    data: Object.keys(models).map(model => ({
+      id: model,
+      object: "model",
+      created: 1706745938,
+      owned_by: "genspark"
+    }))
+  }
+}
+
+// 注册路由 - 同时支持 /v1 和 /hf/v1 路径
+router.post('/v1/chat/completions', handleChatCompletions)
+router.post('/hf/v1/chat/completions', handleChatCompletions)
+router.get('/v1/models', handleModels)
+router.get('/hf/v1/models', handleModels)
+
+// 添加根路由处理器
+router.get('/', async (ctx) => {
+  ctx.body = {
+    status: 'ok',
+    message: 'API server is running',
+    version: '1.0.0',
+    endpoints: [
+      '/v1/chat/completions',
+      '/hf/v1/chat/completions',
+      '/v1/models',
+      '/hf/v1/models'
+    ]
+  }
+})
+
+// 注册路由中间件
+app.use(router.routes()).use(router.allowedMethods())
+
+// 启动服务器
+const PORT = process.env.PORT || 3000
+app.listen(PORT, '0.0.0.0', () => {
+  console.log('='.repeat(50))
+  console.log(`服务器启动于 http://0.0.0.0:${PORT}`)
+  console.log('环境变量:', {
+    PORT: process.env.PORT,
+    NODE_ENV: process.env.NODE_ENV
+  })
+  console.log('='.repeat(50))
+});

src/models.js

@@ -0,0 +1,14 @@
+const models = {
+  "o1-preview": "o1-preview",
+  "o1-mini": "o1-preview",
+  "claude-3-5-sonnet-20240620": "claude-3-5-sonnet",
+  "claude-3-5-sonnet-20241022": "claude-3-5-sonnet",
+  "claude-3-5-haiku-20241022": "claude-3-5-haiku",
+  "claude-3-haiku-20240307": "claude-3-5-haiku",
+  "gpt-4o": "gpt-4o",
+  "gpt-4o-mini": "gpt-4o-mini",
+  "gemini-1.5-pro-latest": "gemini-1.5-pro",
+  "gemini-1.5-flash-latest": "gemini-1.5-flash"
+}
+
+module.exports = models