Update api/index.js

api/index.js

@@ -59,11 +59,69 @@ const withAuth = (request) => {
 		}
 	}
 };
+
 // 返回运行信息
 const logger = (res, req) => {
 	console.log(req.method, res.status, req.url, Date.now() - req.start, 'ms');
 };
 
+// 定义模型映射信息
+const MODEL_INFO = {
+    "claude-3-sonnet-20240229": {
+        "provider": "anthropic",
+        "mapping": "claude-3-sonnet@20240229"
+    },
+    "claude-3-opus-20240229": {
+        "provider": "anthropic",
+        "mapping": "claude-3-opus@20240229"
+    },
+    "claude-3-haiku-20240307": {
+        "provider": "anthropic",
+        "mapping": "claude-3-haiku@20240307"
+    },
+    "claude-3-5-sonnet-20240620": {
+        "provider": "anthropic",
+        "mapping": "claude-3-5-sonnet@20240620"
+    },
+    "gpt-4o-mini": {
+        "provider": "openai",
+        "mapping": "gpt-4o-mini"
+    },
+    "gpt-4o": {
+        "provider": "openai",
+        "mapping": "gpt-4o"
+    },
+    "gpt-4-turbo": {
+        "provider": "openai",
+        "mapping": "gpt-4-turbo"
+    },
+    "gpt-4": {
+        "provider": "openai",
+        "mapping": "gpt-4"
+    },
+    "gpt-3.5-turbo": {
+        "provider": "openai",
+        "mapping": "gpt-3.5-turbo"
+    },
+    "gemini-1.5-pro": {
+        "provider": "google",
+        "mapping": "gemini-1.5-pro"
+    },
+    "gemini-1.5-flash": {
+        "provider": "google",
+        "mapping": "gemini-1.5-flash"
+    },
+    "chat-bison": {
+        "provider": "pieces-os",
+        "mapping": "chat-bison"
+    },
+    "codechat-bison": {
+        "provider": "pieces-os",
+        "mapping": "codechat-bison"
+    }
+};
+
+// 定义路由
 const router = AutoRouter({
   before: [preflight], // 只保留 CORS preflight 检查
   missing: () => error(404, '404 not found.'),
@@ -74,14 +132,14 @@ const router = AutoRouter({
 router.get('/', () => json({
     service: "AI Chat Completion Proxy",
     usage: {
-        endpoint: "/v1/chat/completions",
+        endpoint: "/v1/chat/completions", 
         method: "POST",
         headers: {
             "Content-Type": "application/json",
             "Authorization": "Bearer YOUR_API_KEY"
         },
         body: {
-            model: "One of: gpt-4o-mini, gpt-4o, gpt-4-turbo, gpt-4, gpt-3.5-turbo, claude-3-sonnet@20240229, claude-3-opus@20240229, claude-3-haiku@20240307, claude-3-5-sonnet@20240620, gemini-1.5-flash, gemini-1.5-pro, chat-bison, codechat-bison",
+            model: "One of: " + Object.keys(MODEL_INFO).join(", "),
             messages: [
                 { role: "system", content: "You are a helpful assistant." },
                 { role: "user", content: "Hello, who are you?" }
@@ -91,43 +149,29 @@ router.get('/', () => json({
             top_p: 1
         }
     },
-    availableModels: [
-        "gpt-4o-mini",
-        "gpt-4o",
-        "gpt-4-turbo",
-        "gpt-4",
-        "gpt-3.5-turbo",
-        "claude-3-sonnet@20240229",
-        "claude-3-opus@20240229",
-        "claude-3-haiku@20240307",
-        "claude-3-5-sonnet@20240620",
-        "gemini-1.5-flash",
-        "gemini-1.5-pro",
-        "chat-bison",
-        "codechat-bison"
-    ],
+    availableModels: Object.entries(MODEL_INFO).map(([modelId, info]) => ({
+        id: modelId,
+        provider: info.provider,
+        mapping: info.mapping
+    })),
     note: "Replace YOUR_API_KEY with your actual API key."
 }));
 
 // models 路由
 router.get(config.API_PREFIX + '/v1/models', withAuth, () =>
     json({
-            object: 'list',
-            data: [
-                    { id: "gpt-4o-mini", object: "model", owned_by: "pieces-os" },
-                    { id: "gpt-4o", object: "model", owned_by: "pieces-os" },
-                    { id: "gpt-4-turbo", object: "model", owned_by: "pieces-os" },
-                    { id: "gpt-4", object: "model", owned_by: "pieces-os" },
-                    { id: "gpt-3.5-turbo", object: "model", owned_by: "pieces-os" },
-                    { id: "claude-3-sonnet@20240229", object: "model", owned_by: "pieces-os" },
-                    { id: "claude-3-opus@20240229", object: "model", owned_by: "pieces-os" },
-                    { id: "claude-3-haiku@20240307", object: "model", owned_by: "pieces-os" },
-                    { id: "claude-3-5-sonnet@20240620", object: "model", owned_by: "pieces-os" },
-                    { id: "gemini-1.5-flash", object: "model", owned_by: "pieces-os" },
-                    { id: "gemini-1.5-pro", object: "model", owned_by: "pieces-os" },
-                    { id: "chat-bison", object: "model", owned_by: "pieces-os" },
-                    { id: "codechat-bison", object: "model", owned_by: "pieces-os" },
-            ],
+        object: "list",
+        data: Object.entries(MODEL_INFO).map(([modelId, info]) => ({
+            id: modelId,
+            object: "model",
+            created: Date.now(),
+            owned_by: "pieces-os",
+            permission: [],
+            root: modelId,
+            parent: null,
+            mapping: info.mapping,
+            provider: info.provider
+        }))
     })
 );
 
@@ -307,19 +351,25 @@ function ChatCompletionStreamWithModel(text, model) {
 }
 
 async function handleCompletion(request) {
-        try {
-                // todo stream逆向接口
-                // 解析openai格式API请求
-                const { model: inputModel, messages, stream,temperature,top_p} = await request.json();
-                console.log(inputModel,messages,stream)
-                // 解析system和user/assistant消息
-                const { rules, message:content } = await messagesProcess(messages);
-                console.log(rules,content)
-                // 响应码，回复的消息
-                return await GrpcToPieces(inputModel, content, rules, stream, temperature, top_p);
-        } catch (err) {
-                return error(500, err.message);
+    try {
+        const { model: inputModel, messages, stream, temperature, top_p } = await request.json();
+        
+        // 获取模型映射
+        const modelInfo = MODEL_INFO[inputModel];
+        if (!modelInfo) {
+            return error(400, `Unsupported model: ${inputModel}`);
         }
+        
+        const mappedModel = modelInfo.mapping;
+        
+        // 解析 system 和 user/assistant 消息
+        const { rules, message: content } = await messagesProcess(messages);
+        
+        // 使用映射后的模型名称
+        return await GrpcToPieces(mappedModel, content, rules, stream, temperature, top_p);
+    } catch (err) {
+        return error(500, err.message);
+    }
 }
 
 (async () => {