From f28711a32e6d155c8ca43c2b69253214e52f7b07 Mon Sep 17 00:00:00 2001
From: Maple Gao <esanisa@gmail.com>
Date: Tue, 14 May 2024 19:18:49 +0800
Subject: [PATCH] =?UTF-8?q?=F0=9F=92=84=20style:=20updates=2001.AI=20model?=
 =?UTF-8?q?=20list=20(#2471)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

https://platform.lingyiwanwu.com/docs#%E6%A8%A1%E5%9E%8B
---
 src/config/modelProviders/zeroone.ts | 67 ++++++++++++++++++++++------
 1 file changed, 53 insertions(+), 14 deletions(-)

diff --git a/src/config/modelProviders/zeroone.ts b/src/config/modelProviders/zeroone.ts
index 1192645287b5..d4ca665820b0 100644
--- a/src/config/modelProviders/zeroone.ts
+++ b/src/config/modelProviders/zeroone.ts
@@ -1,31 +1,70 @@
 import { ModelProviderCard } from '@/types/llm';
 
-// ref https://platform.lingyiwanwu.com/
+// ref https://platform.lingyiwanwu.com/docs#%E6%A8%A1%E5%9E%8B
 const ZeroOne: ModelProviderCard = {
   chatModels: [
     {
-      description: '支持聊天、问答、对话、写作、翻译等功能。',
-      displayName: 'YI 34B Chat',
+      description: '全新千亿参数模型，提供超强问答及文本生成能力。',
+      displayName: 'Yi Large',
       enabled: true,
-      id: 'yi-34b-chat-0205',
-      tokens: 4096, // https://huggingface.co/01-ai/Yi-34B-Chat/blob/main/config.json
+      id: 'yi-large',
+      tokens: 16_384,
     },
     {
-      description:
-        '支持通用图片问答、图表理解、OCR、视觉推理，能处理高分辨率（1024*1024）的图像，能在复杂视觉任务上提供优秀性能，同时支持多种语言。',
-      displayName: 'YI Vision Plus',
+      description: '中型尺寸模型升级微调，能力均衡，性价比高。深度优化指令遵循能力。',
+      displayName: 'Yi Medium',
       enabled: true,
-      id: 'yi-vl-plus',
+      id: 'yi-medium',
+      tokens: 16_384,
+    },
+    {
+      description: '复杂视觉任务模型，提供高性能图片理解、分析能力。',
+      displayName: 'Yi Vision',
+      enabled: true,
+      id: 'yi-vision',
       tokens: 4096,
-      vision: true,
     },
     {
-      description: '增强了问答对话交互和深度内容创作能力。文档问答和构建知识库小能手。',
-      displayName: 'YI 34B Chat 200k',
+      description: '200K 超长上下文窗口，提供长文本深度理解和生成能力。',
+      displayName: 'Yi 200K',
       enabled: true,
-      id: 'yi-34b-chat-200k',
-      tokens: 200_000, // https://huggingface.co/01-ai/Yi-34B-200K/blob/main/config.json
+      id: 'yi-medium-200k',
+      tokens: 200_000,
     },
+    {
+      description: '小而精悍，轻量极速模型。提供强化数学运算和代码编写能力。',
+      displayName: 'Yi Spark',
+      enabled: true,
+      id: 'yi-spark',
+      tokens: 16_384,
+    },
+    {
+      description: '基于Yi-Large超强模型的高阶服务，结合检索与生成技术提供精准答案，支持客⼾私有知识库（请联系客服申请）。',
+      displayName: 'Yi Large RAG',
+      id: 'yi-large-rag',
+      tokens: 16_384,
+    },
+    {
+      description: '超高性价比、卓越性能。根据性能和推理速度、成本，进行平衡性高精度调优。',
+      displayName: 'Yi Large Turbo',
+      enabled: true,
+      id: 'yi-large-turbo',
+      tokens: 16_384,
+    },
+    {
+      description: '「兼容版本模型」文本推理能力增强。',
+      displayName: 'Yi Large Preview',
+      enabled: true,
+      id: 'yi-large-preview',
+      tokens: 16_384,
+    },
+    {
+      description: '「兼容版本模型」实时信息获取，以及文本推理能力增强。',
+      displayName: 'Yi Large RAG Preview',
+      id: 'yi-large-rag-preview',
+      tokens: 16_384,
+    },
+
   ],
   id: 'zeroone',
 };