Add initial implementation for Cohere provider support

package.json

@@ -132,6 +132,7 @@
     "antd-style": "^3.6.2",
     "brotli-wasm": "^3.0.1",
     "chroma-js": "^2.4.2",
+    "cohere-ai": "^7.10.5",
     "dayjs": "^1.11.11",
     "debug": "^4.3.5",
     "dexie": "^3.2.7",

src/app/api/chat/agentRuntime.test.ts

@@ -6,6 +6,7 @@ import {
   LobeAnthropicAI,
   LobeAzureOpenAI,
   LobeBedrockAI,
+  LobeCohereAI,
   LobeDeepSeekAI,
   LobeGoogleAI,
   LobeGroq,
@@ -37,7 +38,7 @@ vi.mock('@/config/llm', () => ({
 
     AZURE_API_KEY: 'test-azure-key',
     AZURE_ENDPOINT: 'endpoint',
-
+    COHERE_API_KEY: 'test-cohere-key',
     ZHIPU_API_KEY: 'test.zhipu-key',
     MOONSHOT_API_KEY: 'test-moonshot-key',
     AWS_SECRET_ACCESS_KEY: 'test-aws-secret',
@@ -145,6 +146,13 @@ describe('initAgentRuntimeWithUserPayload method', () => {
       expect(runtime['_runtime']).toBeInstanceOf(LobeAnthropicAI);
     });
 
+    it('Cohere AI provider: with apikey', async () => {
+      const jwtPayload: JWTPayload = { apiKey: 'user-cohere-key' };
+      const runtime = await initAgentRuntimeWithUserPayload(ModelProvider.Cohere, jwtPayload);
+      expect(runtime).toBeInstanceOf(AgentRuntime);
+      expect(runtime['_runtime']).toBeInstanceOf(LobeCohereAI);
+    });
+
     it('Minimax AI provider: with apikey', async () => {
       const jwtPayload: JWTPayload = { apiKey: 'user-minimax-key' };
       const runtime = await initAgentRuntimeWithUserPayload(ModelProvider.Minimax, jwtPayload);
@@ -266,6 +274,13 @@ describe('initAgentRuntimeWithUserPayload method', () => {
       expect(runtime['_runtime']).toBeInstanceOf(LobeBedrockAI);
     });
 
+    it('Cohere AI provider: without apikey', async () => {
+      const jwtPayload = {};
+      const runtime = await initAgentRuntimeWithUserPayload(ModelProvider.Cohere, jwtPayload);
+
+      expect(runtime['_runtime']).toBeInstanceOf(LobeCohereAI);
+    });
+
     it('Ollama provider: without endpoint', async () => {
       const jwtPayload = {};
       const runtime = await initAgentRuntimeWithUserPayload(ModelProvider.Ollama, jwtPayload);

src/config/modelProviders/cohere.ts

@@ -0,0 +1,33 @@
+import { ModelProviderCard } from '@/types/llm';
+
+// ref https://cohere.com/command
+const Cohere: ModelProviderCard = {
+  chatModels: [
+    {
+      description:
+        'An instruction-following conversational model that performs language tasks at a higher quality, more reliably, and with a longer context than previous models. Best suited for complex RAG workflows and multi-step tool use.',
+      displayName: 'Command R+',
+      enabled: true,
+      functionCall: true,
+      id: 'command-r-plus',
+      maxOutput: 4096,
+      tokens: 128_000,
+      vision: false,
+    },
+    {
+      description:
+        'An instruction-following conversational model that performs language tasks at a higher quality, more reliably, and with a longer context than previous models. It can be used for complex workflows like code generation, retrieval augmented generation (RAG), tool use, and agents.',
+      displayName: 'Command R',
+      enabled: true,
+      functionCall: true,
+      id: 'command-r',
+      maxOutput: 4096,
+      tokens: 128_000,
+      vision: false,
+    }
+  ], // TODO: what about embed models? I only see chatModels on other modelProviders
+  id: 'cohere',
+  name: 'Cohere',
+};
+
+export default Cohere;

src/config/modelProviders/index.ts

@@ -5,6 +5,7 @@ import AnthropicProvider from './anthropic';
 import AzureProvider from './azure';
 import BaichuanProvider from './baichuan';
 import BedrockProvider from './bedrock';
+import CohereProvider from './cohere';
 import DeepSeekProvider from './deepseek';
 import GoogleProvider from './google';
 import GroqProvider from './groq';
@@ -27,6 +28,7 @@ export const LOBE_DEFAULT_MODEL_LIST: ChatModelCard[] = [
   QwenProvider.chatModels,
   ZhiPuProvider.chatModels,
   BedrockProvider.chatModels,
+  CohereProvider.chatModels,
   DeepSeekProvider.chatModels,
   GoogleProvider.chatModels,
   GroqProvider.chatModels,
@@ -51,6 +53,7 @@ export const DEFAULT_MODEL_PROVIDER_LIST = [
   QwenProvider,
   OllamaProvider,
   AnthropicProvider,
+  CohereProvider,
   DeepSeekProvider,
   GoogleProvider,
   OpenRouterProvider,
@@ -83,6 +86,7 @@ export { default as AnthropicProviderCard } from './anthropic';
 export { default as AzureProviderCard } from './azure';
 export { default as BaichuanProviderCard } from './baichuan';
 export { default as BedrockProviderCard } from './bedrock';
+export { default as CohereProviderCard } from './cohere';
 export { default as DeepSeekProviderCard } from './deepseek';
 export { default as GoogleProviderCard } from './google';
 export { default as GroqProviderCard } from './groq';

src/const/settings/llm.ts

@@ -3,6 +3,7 @@ import {
   AnthropicProviderCard,
   BaichuanProviderCard,
   BedrockProviderCard,
+  CohereProviderCard,
   DeepSeekProviderCard,
   GoogleProviderCard,
   GroqProviderCard,
@@ -44,6 +45,10 @@ export const DEFAULT_LLM_CONFIG: UserModelProviderConfig = {
     enabled: false,
     enabledModels: filterEnabledModels(BedrockProviderCard),
   },
+  cohere: {
+    enabled: false,
+    enabledModels: filterEnabledModels(CohereProviderCard),
+  },
   deepseek: {
     enabled: false,
     enabledModels: filterEnabledModels(DeepSeekProviderCard),

src/features/ChatInput/ActionBar/Clear.tsx

@@ -39,7 +39,8 @@ const Clear = memo(() => {
     >
       <ActionIcon 
         icon={Eraser} 
-        overlayStyle={{ maxWidth: 'none' }}
+        // TODO: this property caused validation errors
+        // overlayStyle={{ maxWidth: 'none' }}
         placement={'bottom'} 
         title={actionTitle} />
     </Popconfirm>

src/libs/agent-runtime/AgentRuntime.test.ts

@@ -14,6 +14,7 @@ import {
   LobeAnthropicAI,
   LobeAzureOpenAI,
   LobeBedrockAI,
+  LobeCohereAI,
   LobeGoogleAI,
   LobeMistralAI,
   LobeMoonshotAI,
@@ -41,6 +42,7 @@ vi.mock('@/config/server', () => ({
     AZURE_API_KEY: 'test-azure-key',
     AZURE_ENDPOINT: 'endpoint',
 
+    COHERE_API_KEY: 'test-cohere-key',
     ZHIPU_API_KEY: 'test.zhipu-key',
     MOONSHOT_API_KEY: 'test-moonshot-key',
     AWS_SECRET_ACCESS_KEY: 'test-aws-secret',
@@ -204,6 +206,17 @@ describe('AgentRuntime', () => {
       });
     });
 
+    describe('Cohere AI provider', () => {
+      it('should initialize correctly', async () => {
+        const jwtPayload: JWTPayload = { apiKey: 'user-cohere-key' };
+        const runtime = await AgentRuntime.initializeWithProviderOptions(ModelProvider.Cohere, {
+          cohere: jwtPayload,
+        });
+
+        expect(runtime['_runtime']).toBeInstanceOf(LobeCohereAI);
+      });
+    });
+
     describe('Mistral AI provider', () => {
       it('should initialize correctly', async () => {
         const jwtPayload: JWTPayload = { apiKey: 'user-mistral-key' };

src/libs/agent-runtime/AgentRuntime.ts

@@ -8,6 +8,7 @@ import { LobeAnthropicAI } from './anthropic';
 import { LobeAzureOpenAI } from './azureOpenai';
 import { LobeBaichuanAI } from './baichuan';
 import { LobeBedrockAI, LobeBedrockAIParams } from './bedrock';
+import { LobeCohereAI } from './cohere';
 import { LobeDeepSeekAI } from './deepseek';
 import { LobeGoogleAI } from './google';
 import { LobeGroq } from './groq';
@@ -109,6 +110,7 @@ class AgentRuntime {
       azure: { apiVersion?: string; apikey?: string; endpoint?: string };
       baichuan: Partial<ClientOptions>;
       bedrock: Partial<LobeBedrockAIParams>;
+      cohere: Partial<ClientOptions>;
       deepseek: Partial<ClientOptions>;
       google: { apiKey?: string; baseURL?: string };
       groq: Partial<ClientOptions>;
@@ -181,6 +183,11 @@ class AgentRuntime {
         break;
       }
 
+      case ModelProvider.Cohere: {
+        runtimeModel = new LobeCohereAI(params.cohere ?? {});
+        break;
+      }
+
       case ModelProvider.DeepSeek: {
         runtimeModel = new LobeDeepSeekAI(params.deepseek);
         break;