keystone-cli 0.4.2 → 0.4.4

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "keystone-cli",
-  "version": "0.4.2",
+  "version": "0.4.4",
   "description": "A local-first, declarative, agentic workflow orchestrator built on Bun",
   "type": "module",
   "bin": {

package/src/parser/config-schema.ts

@@ -42,28 +42,24 @@ export const ConfigSchema = z.object({
   workflows_directory: z.string().default('workflows'),
   mcp_servers: z
     .record(
-      z.discriminatedUnion('type', [
+      z.union([
+        // Local MCP server (command-based) - type is optional, defaults to 'local'
         z.object({
-          type: z.literal('local').default('local'),
+          type: z.literal('local').optional().default('local'),
           command: z.string(),
           args: z.array(z.string()).optional(),
           env: z.record(z.string()).optional(),
           url: z.string().url().optional(),
-          oauth: z
-            .object({
-              scope: z.string().optional(),
-            })
-            .optional(),
+          oauth: z.union([z.object({ scope: z.string().optional() }), z.literal(false)]).optional(),
+          timeout: z.number().optional().default(60000),
         }),
+        // Remote MCP server (URL-based)
         z.object({
           type: z.literal('remote'),
           url: z.string().url(),
           headers: z.record(z.string()).optional(),
-          oauth: z
-            .object({
-              scope: z.string().optional(),
-            })
-            .optional(),
+          oauth: z.union([z.object({ scope: z.string().optional() }), z.literal(false)]).optional(),
+          timeout: z.number().optional().default(60000),
         }),
       ])
     )

package/src/runner/mcp-client.ts

@@ -354,7 +354,7 @@ export class MCPClient {
       this.timeout = timeout;
     } else {
       this.transport = transportOrCommand;
-      this.timeout = (timeoutOrArgs as number) || 60000;
+      this.timeout = timeoutOrArgs as number;
     }
 
     this.transport.onMessage((response) => {

package/src/runner/mcp-manager.ts

@@ -13,6 +13,7 @@ export interface MCPServerConfig {
   oauth?: {
     scope?: string;
   };
+  timeout?: number;
 }
 
 export class MCPManager {
@@ -90,7 +91,7 @@ export class MCPManager {
             headers.Authorization = `Bearer ${token}`;
           }
 
-          client = await MCPClient.createRemote(config.url, headers);
+          client = await MCPClient.createRemote(config.url, headers, config.timeout);
         } else {
           if (!config.command) throw new Error('Local MCP server missing command');
 
@@ -113,7 +114,12 @@ export class MCPManager {
             env.MCP_TOKEN = token;
           }
 
-          client = await MCPClient.createLocal(config.command, config.args || [], env);
+          client = await MCPClient.createLocal(
+            config.command,
+            config.args || [],
+            env,
+            config.timeout
+          );
         }
 
         await client.initialize();