💄 style: Add Yi-Lightning model (#4390)

* Update zeroone.ts * Update zeroone.ts * Update zeroone.ts * Update zeroone.ts * Update zeroone.ts
lobehub · Oct 17, 2024 · 9e9fb9a · 9e9fb9a
1 parent 2b7bc50
commit 9e9fb9a
Showing 1 changed file with 74 additions and 6 deletions.
diff --git a/src/config/modelProviders/zeroone.ts b/src/config/modelProviders/zeroone.ts
@@ -1,13 +1,30 @@
 import { ModelProviderCard } from '@/types/llm';
 
-// ref :https://platform.lingyiwanwu.com/docs#%E6%A8%A1%E5%9E%8B
+// ref: https://platform.lingyiwanwu.com/docs#%E6%A8%A1%E5%9E%8B%E4%B8%8E%E8%AE%A1%E8%B4%B9
 const ZeroOne: ModelProviderCard = {
   chatModels: [
+    {
+      description: '最新高性能模型，保证高质量输出同时，推理速度大幅提升。',
+      displayName: 'Yi Lightning',
+      enabled: true,
+      id: 'yi-lightning',
+      pricing: {
+        currency: 'CNY',
+        input: 0.99,
+        output: 0.99,
+      },
+      tokens: 16_384,
+    },
     {
       description: '全新千亿参数模型，提供超强问答及文本生成能力。',
       displayName: 'Yi Large',
       enabled: true,
       id: 'yi-large',
+      pricing: {
+        currency: 'CNY',
+        input: 20,
+        output: 20,
+      },
       tokens: 32_768,
     },
     {
@@ -17,6 +34,11 @@ const ZeroOne: ModelProviderCard = {
       enabled: true,
       functionCall: true,
       id: 'yi-large-fc',
+      pricing: {
+        currency: 'CNY',
+        input: 20,
+        output: 20,
+      },
       tokens: 32_768,
     },
     {
@@ -25,58 +47,104 @@ const ZeroOne: ModelProviderCard = {
       displayName: 'Yi Large RAG',
       enabled: true,
       id: 'yi-large-rag',
+      pricing: {
+        currency: 'CNY',
+        input: 25,
+        output: 25,
+      },
       tokens: 16_384,
     },
     {
       description: '超高性价比、卓越性能。根据性能和推理速度、成本，进行平衡性高精度调优。',
       displayName: 'Yi Large Turbo',
       enabled: true,
       id: 'yi-large-turbo',
+      pricing: {
+        currency: 'CNY',
+        input: 12,
+        output: 12,
+      },
       tokens: 16_384,
     },
     {
       description: '中型尺寸模型升级微调，能力均衡，性价比高。深度优化指令遵循能力。',
       displayName: 'Yi Medium',
       enabled: true,
       id: 'yi-medium',
+      pricing: {
+        currency: 'CNY',
+        input: 2.5,
+        output: 2.5,
+      },
       tokens: 16_384,
     },
     {
       description: '200K 超长上下文窗口，提供长文本深度理解和生成能力。',
       displayName: 'Yi Medium 200K',
       enabled: true,
       id: 'yi-medium-200k',
+      pricing: {
+        currency: 'CNY',
+        input: 12,
+        output: 12,
+      },
       tokens: 200_000,
     },
     {
       description: '小而精悍，轻量极速模型。提供强化数学运算和代码编写能力。',
       displayName: 'Yi Spark',
       enabled: true,
       id: 'yi-spark',
+      pricing: {
+        currency: 'CNY',
+        input: 1,
+        output: 1,
+      },
       tokens: 16_384,
     },
     {
       description: '复杂视觉任务模型，提供高性能图片理解、分析能力。',
       displayName: 'Yi Vision',
       enabled: true,
       id: 'yi-vision',
+      pricing: {
+        currency: 'CNY',
+        input: 6,
+        output: 6,
+      },
       tokens: 16_384,
       vision: true,
     },
     {
-      description: '初期版本，推荐使用 yi-large（新版本）',
+      description: '初期版本，推荐使用 yi-large（新版本）。',
       displayName: 'Yi Large Preview',
       id: 'yi-large-preview',
+      pricing: {
+        currency: 'CNY',
+        input: 20,
+        output: 20,
+      },
+      tokens: 16_384,
+    },
+    {
+      description: '轻量化版本，推荐使用 yi-lightning。',
+      displayName: 'Yi Lightning Lite',
+      id: 'yi-lightning-lite',
+      pricing: {
+        currency: 'CNY',
+        input: 0.99,
+        output: 0.99,
+      },
       tokens: 16_384,
     },
   ],
-  checkModel: 'yi-spark',
+  checkModel: 'yi-lightning',
   description:
-    '01.AI 专注于AI 2.0时代的人工智能技术，大力推动“人+人工智能”的创新和应用，采用超强大模型和先进AI技术以提升人类生产力，实现技术赋能。',
+    '零一万物致力于推动以人为本的AI 2.0技术革命，旨在通过大语言模型创造巨大的经济和社会价值，并开创新的AI生态与商业模式。',
   id: 'zeroone',
-  modelsUrl: 'https://platform.lingyiwanwu.com/docs',
+  modelsUrl: 'https://platform.lingyiwanwu.com/docs#模型与计费',
   name: '01.AI',
-  url: 'https://01.ai',
+  url: 'https://www.lingyiwanwu.com/',
 };
 
 export default ZeroOne;