feat: add Cohere Command A model

ai/providers.ts

@@ -3,7 +3,7 @@ import { openai } from "@ai-sdk/openai";
 import { google } from "@ai-sdk/google";
 import { groq } from "@ai-sdk/groq";
 import { customProvider, wrapLanguageModel, extractReasoningMiddleware } from "ai";
-
+import { cohere } from "@ai-sdk/cohere";
 export interface ModelInfo {
   provider: string;
   name: string;
@@ -21,14 +21,14 @@ const languageModels = {
   // "grok-3": xai("grok-3-latest"),
   // "grok-3-mini": xai("grok-3-mini-fast-latest"),
   "gpt-4.1-mini": openai("gpt-4.1-mini"),
-  "gpt-4.1-nano": openai("gpt-4.1-nano"),
   "gemini-2-flash": google("gemini-2.0-flash-001"),
   "qwen-qwq": wrapLanguageModel(
     {
       model: groq("qwen-qwq-32b"), 
       middleware
     }
-  )
+  ),
+  "command-a": cohere('command-a-03-2025')
 };
 
 export const modelDetails: Record<keyof typeof languageModels, ModelInfo> = {
@@ -53,13 +53,6 @@ export const modelDetails: Record<keyof typeof languageModels, ModelInfo> = {
     apiVersion: "gpt-4.1-mini",
     capabilities: [ "Balance", "Creative", "Vision"]
   },
-  "gpt-4.1-nano": {
-    provider: "OpenAI",
-    name: "GPT-4.1 Nano",
-    description: "Smallest and fastest GPT-4.1 variant designed for efficient rapid responses.",
-    apiVersion: "gpt-4.1-nano",
-    capabilities: ["Rapid", "Compact", "Efficient", "Vision"]
-  },  
   "gemini-2-flash": {
     provider: "Google",
     name: "Gemini 2 Flash",
@@ -73,6 +66,13 @@ export const modelDetails: Record<keyof typeof languageModels, ModelInfo> = {
     description: "Latest version of Alibaba's Qwen QWQ with strong reasoning and coding capabilities.",
     apiVersion: "qwen-qwq",
     capabilities: ["Reasoning", "Efficient", "Agentic"]
+  },
+  "command-a": {
+    provider: "Cohere",
+    name: "Command A",
+    description: "Latest version of Cohere's Command A with strong reasoning and coding capabilities.",
+    apiVersion: "command-a-03-2025",
+    capabilities: ["Smart", "Fast", "Reasoning"]
   }
 };
 

app/api/chat/route.ts

@@ -1,12 +1,10 @@
 import { model, type modelID } from "@/ai/providers";
-import { composioTools } from "@/ai/tools";
 import { streamText, type UIMessage } from "ai";
-import { openai } from '@ai-sdk/openai';
 import { appendResponseMessages } from 'ai';
 import { saveChat, saveMessages, convertToDBMessages } from '@/lib/chat-store';
 import { nanoid } from 'nanoid';
 import { db } from '@/lib/db';
-import { messages, chats } from '@/lib/db/schema';
+import { chats } from '@/lib/db/schema';
 import { eq, and } from 'drizzle-orm';
 
 import { experimental_createMCPClient as createMCPClient, MCPTransport } from 'ai';
@@ -14,7 +12,7 @@ import { Experimental_StdioMCPTransport as StdioMCPTransport } from 'ai/mcp-stdi
 import { spawn } from "child_process";
 
 // Allow streaming responses up to 30 seconds
-export const maxDuration = 30;
+export const maxDuration = 120;
 
 interface KeyValuePair {
   key: string;
@@ -79,13 +77,13 @@ export async function POST(req: Request) {
   // Initialize tools
   let tools = {};
   const mcpClients: any[] = [];
-  
+
   // Process each MCP server configuration
   for (const mcpServer of mcpServers) {
     try {
       // Create appropriate transport based on type
       let transport: MCPTransport | { type: 'sse', url: string, headers?: Record<string, string> };
-      
+
       if (mcpServer.type === 'sse') {
         // Convert headers array to object for SSE transport
         const headers: Record<string, string> = {};
@@ -94,9 +92,9 @@ export async function POST(req: Request) {
             if (header.key) headers[header.key] = header.value || '';
           });
         }
-        
-        transport = { 
-          type: 'sse' as const, 
+
+        transport = {
+          type: 'sse' as const,
           url: mcpServer.url,
           headers: Object.keys(headers).length > 0 ? headers : undefined
         };
@@ -106,7 +104,7 @@ export async function POST(req: Request) {
           console.warn("Skipping stdio MCP server due to missing command or args");
           continue;
         }
-        
+
         // Convert env array to object for stdio transport
         const env: Record<string, string> = {};
         if (mcpServer.env && mcpServer.env.length > 0) {
@@ -131,7 +129,7 @@ export async function POST(req: Request) {
             console.log("installed python package", packageName);
           });
         }
-        
+
         transport = new StdioMCPTransport({
           command: mcpServer.command,
           args: mcpServer.args,
@@ -141,14 +139,14 @@ export async function POST(req: Request) {
         console.warn(`Skipping MCP server with unsupported transport type: ${mcpServer.type}`);
         continue;
       }
-      
+
       const mcpClient = await createMCPClient({ transport });
       mcpClients.push(mcpClient);
-      
+
       const mcptools = await mcpClient.tools();
-      
+
       console.log(`MCP tools from ${mcpServer.type} transport:`, Object.keys(mcptools));
-      
+
       // Add MCP tools to tools object
       tools = { ...tools, ...mcptools };
     } catch (error) {
@@ -205,13 +203,13 @@ export async function POST(req: Request) {
         messages,
         responseMessages: response.messages,
       });
-      
+
       await saveChat({
         id,
         userId,
         messages: allMessages,
       });
-      
+
       const dbMessages = convertToDBMessages(allMessages, id);
       await saveMessages({ messages: dbMessages });
       // close all mcp clients

components/model-picker.tsx

@@ -42,6 +42,8 @@ export const ModelPicker = ({ selectedModel, setSelectedModel }: ModelPickerProp
         return <Zap className="h-3 w-3 text-red-500" />;
       case 'groq':
         return <Sparkles className="h-3 w-3 text-blue-500" />;
+      case 'cohere':
+        return <Sparkles className="h-3 w-3 text-yellow-500" />;
       default:
         return <Info className="h-3 w-3 text-blue-500" />;
     }

package.json

@@ -13,6 +13,7 @@
     "db:studio": "drizzle-kit studio"
   },
   "dependencies": {
+    "@ai-sdk/cohere": "^1.2.9",
     "@ai-sdk/google": "^1.2.12",
     "@ai-sdk/groq": "^1.2.8",
     "@ai-sdk/openai": "^1.3.16",

pnpm-lock.yaml

@@ -8,6 +8,9 @@ importers:
 
   .:
     dependencies:
+      '@ai-sdk/cohere':
+        specifier: ^1.2.9
+        version: 1.2.9([email protected])
       '@ai-sdk/google':
         specifier: ^1.2.12
         version: 1.2.12([email protected])
@@ -195,6 +198,12 @@ importers:
 
 packages:
 
+  '@ai-sdk/[email protected]':
+    resolution: {integrity: sha512-ENbQT2bDt1FN+DOLkoFT9n+cojLv7zdN5GCaDtkaqJUuRa+6lHqNO4tRvB79Jg8DY03mhDermty9EhJW2zHoTA==}
+    engines: {node: '>=18'}
+    peerDependencies:
+      zod: ^3.0.0
+
   '@ai-sdk/[email protected]':
     resolution: {integrity: sha512-A8AYqCmBs9SJFiAOP6AX0YEDHWTDrCaUDiRY2cdMSKjJiEknvwnPrAAKf3idgVqYaM2kS0qWz5v9v4pBzXDx+w==}
     engines: {node: '>=18'}
@@ -4492,6 +4501,12 @@ packages:
 
 snapshots:
 
+  '@ai-sdk/[email protected]([email protected])':
+    dependencies:
+      '@ai-sdk/provider': 1.1.3
+      '@ai-sdk/provider-utils': 2.2.7([email protected])
+      zod: 3.24.2
+
   '@ai-sdk/[email protected]([email protected])':
     dependencies:
       '@ai-sdk/provider': 1.1.3