keystone-cli 0.4.1 → 0.4.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "keystone-cli",
-  "version": "0.4.1",
+  "version": "0.4.3",
   "description": "A local-first, declarative, agentic workflow orchestrator built on Bun",
   "type": "module",
   "bin": {
@@ -58,4 +58,4 @@
   "engines": {
     "bun": ">=1.0.0"
   }
-}
+}

package/src/parser/config-schema.ts

@@ -42,28 +42,22 @@ 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(),
         }),
+        // 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(),
         }),
       ])
     )

package/src/runner/llm-executor.ts

@@ -237,7 +237,7 @@ export async function executeLlmStep(
         if (step.schema && typeof output === 'string') {
           try {
             const { extractJson } = await import('../utils/json-parser');
-            output = extractJson(output);
+            output = extractJson(output) as typeof output;
           } catch (e) {
             throw new Error(
               `Failed to parse LLM output as JSON matching schema: ${e instanceof Error ? e.message : String(e)}\nOutput: ${output}`

package/src/runner/mcp-server.test.ts

@@ -81,7 +81,7 @@ describe('MCPServer', () => {
       },
     });
 
-    expect(JSON.parse(response.result.content[0].text).status).toBe('success');
+    expect(JSON.parse(response?.result?.content?.[0]?.text || '{}').status).toBe('success');
   });
 
   it('should handle run_workflow failure', async () => {
@@ -130,7 +130,7 @@ describe('MCPServer', () => {
       },
     });
 
-    const result = JSON.parse(response?.result?.content?.[0]?.text);
+    const result = JSON.parse(response?.result?.content?.[0]?.text || '{}');
     expect(result.status).toBe('paused');
     expect(result.run_id).toBe('run123');
     expect(result.message).toBe('Input needed');
@@ -163,7 +163,7 @@ describe('MCPServer', () => {
       },
     });
 
-    expect(JSON.parse(response.result.content[0].text).status).toBe('success');
+    expect(JSON.parse(response?.result?.content?.[0]?.text || '{}').status).toBe('success');
 
     // Verify DB was updated
     const steps = db.getStepsByRun(runId);
@@ -187,7 +187,7 @@ describe('MCPServer', () => {
       params: { name: 'get_run_logs', arguments: { run_id: runId } },
     });
 
-    const summary = JSON.parse(response?.result?.content?.[0]?.text);
+    const summary = JSON.parse(response?.result?.content?.[0]?.text || '{}');
     expect(summary.workflow).toBe('test-wf');
     expect(summary.steps).toHaveLength(1);
     expect(summary.steps[0].step).toBe('s1');
@@ -276,7 +276,7 @@ describe('MCPServer', () => {
       },
     });
 
-    const result = JSON.parse(response?.result?.content?.[0]?.text);
+    const result = JSON.parse(response?.result?.content?.[0]?.text || '{}');
     expect(result.status).toBe('running');
     expect(result.run_id).toBe('async-run-123');
     expect(result.hint).toContain('get_run_status');
@@ -294,7 +294,7 @@ describe('MCPServer', () => {
       params: { name: 'get_run_status', arguments: { run_id: runId } },
     });
 
-    const status = JSON.parse(response?.result?.content?.[0]?.text);
+    const status = JSON.parse(response?.result?.content?.[0]?.text || '{}');
     expect(status.run_id).toBe(runId);
     expect(status.workflow).toBe('test-wf');
     expect(status.status).toBe('running');
@@ -313,7 +313,7 @@ describe('MCPServer', () => {
       params: { name: 'get_run_status', arguments: { run_id: runId } },
     });
 
-    const status = JSON.parse(response?.result?.content?.[0]?.text);
+    const status = JSON.parse(response?.result?.content?.[0]?.text || '{}');
     expect(status.status).toBe('completed');
     expect(status.outputs).toEqual({ output: 'done' });
     expect(status.hint).toBeUndefined();
@@ -331,7 +331,7 @@ describe('MCPServer', () => {
       params: { name: 'get_run_status', arguments: { run_id: runId } },
     });
 
-    const status = JSON.parse(response?.result?.content?.[0]?.text);
+    const status = JSON.parse(response?.result?.content?.[0]?.text || '{}');
     expect(status.status).toBe('failed');
     expect(status.error).toBe('Something went wrong');
   });
@@ -348,7 +348,7 @@ describe('MCPServer', () => {
       params: { name: 'get_run_status', arguments: { run_id: runId } },
     });
 
-    const status = JSON.parse(response?.result?.content?.[0]?.text);
+    const status = JSON.parse(response?.result?.content?.[0]?.text || '{}');
     expect(status.status).toBe('paused');
     expect(status.hint).toContain('answer_human_input');
   });