@lobehub/chat 0.141.2 → 0.142.0

package/src/libs/agent-runtime/error.ts

@@ -37,7 +37,10 @@ export const AgentRuntimeErrorType = {
 
   InvalidGroqAPIKey: 'InvalidGroqAPIKey',
   GroqBizError: 'GroqBizError',
-  
+
+  InvalidZeroOneAPIKey: 'InvalidZeroOneAPIKey',
+  ZeroOneBizError: 'ZeroOneBizError',
+
   InvalidOpenRouterAPIKey: 'InvalidOpenRouterAPIKey',
   OpenRouterBizError: 'OpenRouterBizError',
 } as const;

package/src/libs/agent-runtime/index.ts

@@ -13,4 +13,5 @@ export { LobeOpenRouterAI } from './openrouter';
 export { LobePerplexityAI } from './perplexity';
 export * from './types';
 export { AgentRuntimeError } from './utils/createError';
+export { LobeZeroOneAI } from './zeroone';
 export { LobeZhipuAI } from './zhipu';

package/src/libs/agent-runtime/types/type.ts

@@ -35,5 +35,6 @@ export enum ModelProvider {
   OpenRouter = 'openrouter',
   Perplexity = 'perplexity',
   Tongyi = 'tongyi',
+  ZeroOne = 'zeroone',
   ZhiPu = 'zhipu',
 }

package/src/libs/agent-runtime/zeroone/index.ts

@@ -0,0 +1,78 @@
+import { OpenAIStream, StreamingTextResponse } from 'ai';
+import OpenAI, { ClientOptions } from 'openai';
+
+import { LobeRuntimeAI } from '../BaseAI';
+import { AgentRuntimeErrorType } from '../error';
+import { ChatCompetitionOptions, ChatStreamPayload, ModelProvider } from '../types';
+import { AgentRuntimeError } from '../utils/createError';
+import { debugStream } from '../utils/debugStream';
+import { desensitizeUrl } from '../utils/desensitizeUrl';
+import { handleOpenAIError } from '../utils/handleOpenAIError';
+
+const DEFAULT_BASE_URL = 'https://api.lingyiwanwu.com/v1';
+
+export class LobeZeroOneAI implements LobeRuntimeAI {
+  private client: OpenAI;
+
+  baseURL: string;
+
+  constructor({ apiKey, baseURL = DEFAULT_BASE_URL, ...res }: ClientOptions) {
+    if (!apiKey) throw AgentRuntimeError.createError(AgentRuntimeErrorType.InvalidZeroOneAPIKey);
+
+    this.client = new OpenAI({ apiKey, baseURL, ...res });
+    this.baseURL = this.client.baseURL;
+  }
+
+  async chat(payload: ChatStreamPayload, options?: ChatCompetitionOptions) {
+    try {
+      const response = await this.client.chat.completions.create(
+        payload as unknown as OpenAI.ChatCompletionCreateParamsStreaming
+      );
+      const [prod, debug] = response.tee();
+
+      if (process.env.DEBUG_ZEROONE_CHAT_COMPLETION === '1') {
+        debugStream(debug.toReadableStream()).catch(console.error);
+      }
+
+      return new StreamingTextResponse(OpenAIStream(prod, options?.callback), {
+        headers: options?.headers,
+      });
+    } catch (error) {
+      let desensitizedEndpoint = this.baseURL;
+
+      if (this.baseURL !== DEFAULT_BASE_URL) {
+        desensitizedEndpoint = desensitizeUrl(this.baseURL);
+      }
+
+      if ('status' in (error as any)) {
+        switch ((error as Response).status) {
+          case 401: {
+            throw AgentRuntimeError.chat({
+              endpoint: desensitizedEndpoint,
+              error: error as any,
+              errorType: AgentRuntimeErrorType.InvalidZeroOneAPIKey,
+              provider: ModelProvider.ZeroOne,
+            });
+          }
+
+          default: {
+            break;
+          }
+        }
+      }
+
+      const { errorResult, RuntimeError } = handleOpenAIError(error);
+
+      const errorType = RuntimeError || AgentRuntimeErrorType.ZeroOneBizError;
+
+      throw AgentRuntimeError.chat({
+        endpoint: desensitizedEndpoint,
+        error: errorResult,
+        errorType,
+        provider: ModelProvider.ZeroOne,
+      });
+    }
+  }
+}
+
+export default LobeZeroOneAI;

package/src/locales/default/common.ts

@@ -109,6 +109,7 @@ export default {
     openai: 'OpenAI',
     openrouter: 'OpenRouter',
     perplexity: 'Perplexity',
+    zeroone: '01.AI 零一万物',
     zhipu: '智谱AI',
   },
   noDescription: '暂无描述',

package/src/locales/default/error.ts

@@ -84,10 +84,13 @@ export default {
 
     InvalidGroqAPIKey: 'Groq API Key 不正确或为空，请检查 Groq API Key 后重试',
     GroqBizError: '请求 Groq 服务出错，请根据以下信息排查或重试',
-    
+
     InvalidOpenRouterAPIKey: 'OpenRouter API Key 不正确或为空，请检查 OpenRouter API Key 后重试',
     OpenRouterBizError: '请求 OpenRouter AI 服务出错，请根据以下信息排查或重试',
 
+    ZeroOneBizError: '请求零一万物服务出错，请根据以下信息排查或重试',
+    InvalidZeroOneAPIKey: '零一万物 API Key 不正确或为空，请检查零一万物 API Key 后重试',
+
     InvalidOllamaArgs: 'Ollama 配置不正确，请检查 Ollama 配置后重试',
     OllamaBizError: '请求 Ollama 服务出错，请根据以下信息排查或重试',
     OllamaServiceUnavailable: '未检测到 Ollama 服务，请检查是否正常启动',
@@ -142,6 +145,10 @@ export default {
         description: '输入你的 Perplexity API Key 即可开始会话。应用不会记录你的 API Key',
         title: '使用自定义 Perplexity API Key',
       },
+      ZeroOne: {
+        description: '输入你的零一万物 API Key 即可开始会话。应用不会记录你的 API Key',
+        title: '使用自定义零一万物 API Key',
+      },
       Zhipu: {
         description: '输入你的 Zhipu API Key 即可开始会话。应用不会记录你的 API Key',
         title: '使用自定义 Zhipu API Key',

package/src/locales/default/setting.ts

@@ -202,6 +202,14 @@ export default {
         title: 'API Key',
       },
     },
+    ZeroOne: {
+      title: '01.AI 零一万物',
+      token: {
+        desc: '填入来自 01.AI 零一万物的 API Key',
+        placeholder: '01.AI 零一万物 API Key',
+        title: 'API Key',
+      }
+    },
     Zhipu: {
       title: '智谱',
       token: {

package/src/services/_auth.ts

@@ -65,11 +65,15 @@ export const getProviderAuthPayload = (provider: string) => {
     case ModelProvider.Groq: {
       return { apiKey: modelProviderSelectors.groqAPIKey(useGlobalStore.getState()) };
     }
-    
+
     case ModelProvider.OpenRouter: {
       return { apiKey: modelProviderSelectors.openrouterAPIKey(useGlobalStore.getState()) };
     }
 
+    case ModelProvider.ZeroOne: {
+      return { apiKey: modelProviderSelectors.zerooneAPIKey(useGlobalStore.getState()) };
+    }
+
     default:
     case ModelProvider.OpenAI: {
       const openai = modelProviderSelectors.openAIConfig(useGlobalStore.getState());

package/src/store/global/slices/settings/selectors/modelProvider.ts

@@ -12,6 +12,7 @@ import {
   OpenAIProvider,
   OpenRouterProvider,
   PerplexityProvider,
+  ZeroOneProvider,
   ZhiPuProvider,
 } from '@/config/modelProviders';
 import { ChatModelCard, ModelProviderCard } from '@/types/llm';
@@ -66,6 +67,9 @@ const groqAPIKey = (s: GlobalStore) => modelProvider(s).groq.apiKey;
 const enableOpenrouter = (s: GlobalStore) => modelProvider(s).openrouter.enabled;
 const openrouterAPIKey = (s: GlobalStore) => modelProvider(s).openrouter.apiKey;
 
+const enableZeroone = (s: GlobalStore) => modelProvider(s).zeroone.enabled;
+const zerooneAPIKey = (s: GlobalStore) => modelProvider(s).zeroone.apiKey;
+
 // const azureModelList = (s: GlobalStore): ModelProviderCard => {
 //   const azure = azureConfig(s);
 //   return {
@@ -165,7 +169,8 @@ const modelSelectList = (s: GlobalStore): ModelProviderCard[] => {
     { ...GroqProvider, enabled: enableGroq(s) },
     { ...ZhiPuProvider, enabled: enableZhipu(s) },
     { ...MoonshotProvider, enabled: enableMoonshot(s) },
-    { ...OpenRouterProvider, chatModels: openrouterChatModels, enabled: enableOpenrouter(s)}
+    { ...OpenRouterProvider, chatModels: openrouterChatModels, enabled: enableOpenrouter(s)},
+    { ...ZeroOneProvider, enabled: enableZeroone(s) }
   ];
 };
 
@@ -252,8 +257,12 @@ export const modelProviderSelectors = {
   // Groq
   enableGroq,
   groqAPIKey,
-  
+
   // OpenRouter
   enableOpenrouter,
   openrouterAPIKey,
+
+  // ZeroOne 零一万物
+  enableZeroone,
+  zerooneAPIKey,
 };

package/src/types/settings/modelProvider.ts

@@ -82,6 +82,12 @@ export interface OpenRouterConfig {
   enabled?: boolean;
 }
 
+export interface ZeroOneConfig {
+  apiKey?: string;
+  customModelName?: string;
+  enabled?: boolean;
+}
+
 export interface GlobalLLMConfig {
   anthropic: AnthropicConfig;
   azure: AzureOpenAIConfig;
@@ -94,6 +100,7 @@ export interface GlobalLLMConfig {
   openAI: OpenAIConfig;
   openrouter: OpenRouterConfig;
   perplexity: PerplexityConfig;
+  zeroone: ZeroOneConfig;
   zhipu: ZhiPuConfig;
 }