@koi-language/koi 1.0.4 → 1.0.6

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@koi-language/koi",
-  "version": "1.0.4",
+  "version": "1.0.6",
   "description": "Koi - Agent-first language. Calm orchestration.",
   "type": "module",
   "main": "src/runtime/index.js",

package/src/cli/koi.js

@@ -505,7 +505,8 @@ Agent Greeter : Worker {
     - Add a motivational message or fun fact
     - Be brief (2-3 sentences)
 
-    Return JSON: { "greeting": "your greeting here", "emoji": "an appropriate emoji" }
+    Use the return action with your greeting data:
+    { "greeting": "your greeting here", "emoji": "an appropriate emoji" }
     """
   }
 }

package/src/runtime/incremental-json-parser.js

@@ -136,6 +136,30 @@ export class IncrementalJSONParser {
       actions.push(...objects);
     }
 
+    // If no "actions" array was found, the LLM returned raw JSON
+    // Wrap it in a return action so it can be executed properly
+    if (this.actionsStartIndex === -1 && this.buffer.length > 0) {
+      try {
+        const rawJSON = JSON.parse(this.buffer.trim());
+        // Check if it's not already an action (has actionType or intent)
+        if (!rawJSON.actionType && !rawJSON.actions) {
+          if (process.env.KOI_DEBUG_LLM) {
+            console.error(`[IncrementalParser] ⚠️ No "actions" array found - wrapping raw JSON in return action`);
+          }
+          return [{
+            actionType: 'direct',
+            intent: 'return',
+            data: rawJSON
+          }];
+        }
+      } catch (e) {
+        // Not valid JSON, return whatever objects we found
+        if (process.env.KOI_DEBUG_LLM) {
+          console.error(`[IncrementalParser] ⚠️ Failed to parse raw response as JSON: ${e.message}`);
+        }
+      }
+    }
+
     return actions;
   }