nomid2 commited on
Commit
a4592d6
·
verified ·
1 Parent(s): d6427c6

Upload model.js

Browse files
Files changed (1) hide show
  1. src/router/model.js +28 -0
src/router/model.js ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ const express = require('express')
2
+ const { MODEL_MAPPING } = require('../config')
3
+
4
+ const router = express.Router()
5
+
6
+ // 根据映射动态生成可用模型列表
7
+ function getAvailableModels() {
8
+ const timestamp = Math.floor(Date.now() / 1000)
9
+ return Object.entries(MODEL_MAPPING).map(([modelId, mammouthId]) => {
10
+ return {
11
+ id: modelId,
12
+ object: 'model',
13
+ created: timestamp,
14
+ owned_by: 'mammouth',
15
+ provider: 'mammouth'
16
+ }
17
+ })
18
+ }
19
+
20
+ // 获取模型列表
21
+ router.get('/', (req, res) => {
22
+ res.json({
23
+ object: 'list',
24
+ data: getAvailableModels()
25
+ })
26
+ })
27
+
28
+ module.exports = router