@lobehub/chat 1.126.2 → 1.126.3

package/packages/model-runtime/src/cometapi/index.ts

@@ -0,0 +1,49 @@
+import { ModelProvider } from '../types';
+import { processMultiProviderModelList } from '../utils/modelParse';
+import { createOpenAICompatibleRuntime } from '../utils/openaiCompatibleFactory';
+
+export interface CometAPIModelCard {
+  id: string;
+  object: string;
+  owned_by: string;
+}
+
+export const LobeCometAPIAI = createOpenAICompatibleRuntime({
+  baseURL: 'https://api.cometapi.com/v1',
+  chatCompletion: {
+    handlePayload: (payload) => {
+      const { model, ...rest } = payload;
+
+      return {
+        ...rest,
+        model,
+        stream: true,
+      } as any;
+    },
+  },
+  debug: {
+    chatCompletion: () => process.env.DEBUG_COMETAPI_COMPLETION === '1',
+  },
+  models: async ({ client }) => {
+    try {
+      const modelsPage = (await client.models.list()) as any;
+      const rawList: any[] = modelsPage.data || [];
+
+      // 处理模型列表，移除不必要的字段
+      const modelList: CometAPIModelCard[] = rawList.map((model) => ({
+        id: model.id,
+        object: model.object,
+        owned_by: model.owned_by,
+      }));
+
+      return await processMultiProviderModelList(modelList, 'cometapi');
+    } catch (error) {
+      console.warn(
+        'Failed to fetch CometAPI models. Please ensure your CometAPI API key is valid:',
+        error,
+      );
+      return [];
+    }
+  },
+  provider: ModelProvider.CometAPI,
+});

package/packages/model-runtime/src/index.ts

@@ -5,6 +5,7 @@ export { LobeAzureOpenAI } from './azureOpenai';
 export * from './BaseAI';
 export { LobeBedrockAI } from './bedrock';
 export { LobeBflAI } from './bfl';
+export { LobeCometAPIAI } from './cometapi';
 export { LobeDeepSeekAI } from './deepseek';
 export { LobeGoogleAI } from './google';
 export { LobeGroq } from './groq';

package/packages/model-runtime/src/runtimeMap.ts

@@ -11,6 +11,7 @@ import { LobeBedrockAI } from './bedrock';
 import { LobeBflAI } from './bfl';
 import { LobeCloudflareAI } from './cloudflare';
 import { LobeCohereAI } from './cohere';
+import { LobeCometAPIAI } from './cometapi';
 import { LobeDeepSeekAI } from './deepseek';
 import { LobeFalAI } from './fal';
 import { LobeFireworksAI } from './fireworksai';
@@ -73,6 +74,7 @@ export const providerRuntimeMap = {
   bfl: LobeBflAI,
   cloudflare: LobeCloudflareAI,
   cohere: LobeCohereAI,
+  cometapi: LobeCometAPIAI,
   deepseek: LobeDeepSeekAI,
   fal: LobeFalAI,
   fireworksai: LobeFireworksAI,

package/packages/model-runtime/src/types/type.ts

@@ -40,6 +40,7 @@ export enum ModelProvider {
   Bedrock = 'bedrock',
   Cloudflare = 'cloudflare',
   Cohere = 'cohere',
+  CometAPI = 'cometapi',
   DeepSeek = 'deepseek',
   Fal = 'fal',
   FireworksAI = 'fireworksai',

package/packages/types/src/user/settings/keyVaults.ts

@@ -49,6 +49,7 @@ export interface UserKeyVaults extends SearchEngineKeyVaults {
   bedrock?: AWSBedrockKeyVault;
   cloudflare?: CloudflareKeyVault;
   cohere?: OpenAICompatibleKeyVault;
+  cometapi?: OpenAICompatibleKeyVault;
   deepseek?: OpenAICompatibleKeyVault;
   fal?: FalKeyVault;
   fireworksai?: OpenAICompatibleKeyVault;

package/src/config/llm.ts

@@ -184,6 +184,9 @@ export const getLLMConfig = () => {
       ENABLED_AKASHCHAT: z.boolean(),
       AKASHCHAT_API_KEY: z.string().optional(),
 
+      ENABLED_COMETAPI: z.boolean(),
+      COMETAPI_KEY: z.string().optional(),
+
       ENABLED_AIHUBMIX: z.boolean(),
       AIHUBMIX_API_KEY: z.string().optional(),
 
@@ -369,6 +372,9 @@ export const getLLMConfig = () => {
       ENABLED_AKASHCHAT: !!process.env.AKASHCHAT_API_KEY,
       AKASHCHAT_API_KEY: process.env.AKASHCHAT_API_KEY,
 
+      ENABLED_COMETAPI: !!process.env.COMETAPI_KEY,
+      COMETAPI_KEY: process.env.COMETAPI_KEY,
+
       ENABLED_AIHUBMIX: !!process.env.AIHUBMIX_API_KEY,
       AIHUBMIX_API_KEY: process.env.AIHUBMIX_API_KEY,
 

package/src/config/modelProviders/cometapi.ts

@@ -0,0 +1,24 @@
+import { ModelProviderCard } from '@/types/llm';
+
+// ref: https://api.cometapi.com/pricing
+const CometAPI: ModelProviderCard = {
+  chatModels: [],
+  checkModel: 'gpt-5-mini',
+  description:
+    'CometAPI 是一个提供多种前沿大模型接口的服务平台，支持 OpenAI、Anthropic、Google 及更多，适合多样化的开发和应用需求。用户可根据自身需求灵活选择最优的模型和价格，助力AI体验的提升。',
+  enabled: true,
+  id: 'cometapi',
+  modelList: { showModelFetcher: true },
+  modelsUrl: 'https://api.cometapi.com/v1/models',
+  name: 'CometAPI',
+  settings: {
+    proxyUrl: {
+      placeholder: 'https://api.cometapi.com/v1',
+    },
+    sdkType: 'openai',
+    showModelFetcher: true,
+  },
+  url: 'https://cometapi.com',
+};
+
+export default CometAPI;

package/src/config/modelProviders/index.ts

@@ -13,6 +13,7 @@ import BedrockProvider from './bedrock';
 import BflProvider from './bfl';
 import CloudflareProvider from './cloudflare';
 import CohereProvider from './cohere';
+import CometAPIProvider from './cometapi';
 import DeepSeekProvider from './deepseek';
 import FalProvider from './fal';
 import FireworksAIProvider from './fireworksai';
@@ -179,6 +180,7 @@ export const DEFAULT_MODEL_PROVIDER_LIST = [
   AkashChatProvider,
   QiniuProvider,
   NebiusProvider,
+  CometAPIProvider,
 ];
 
 export const filterEnabledModels = (provider: ModelProviderCard) => {
@@ -203,6 +205,7 @@ export { default as BedrockProviderCard } from './bedrock';
 export { default as BflProviderCard } from './bfl';
 export { default as CloudflareProviderCard } from './cloudflare';
 export { default as CohereProviderCard } from './cohere';
+export { default as CometAPIProviderCard } from './cometapi';
 export { default as DeepSeekProviderCard } from './deepseek';
 export { default as FalProviderCard } from './fal';
 export { default as FireworksAIProviderCard } from './fireworksai';