@react-frameui/loki-ai 1.0.0 → 1.0.2

package/dist/core/agentRunner.js

@@ -111,14 +111,18 @@ Assistant:`;
     // Initial pass (Non-streaming to catch JSON better, or standard)
     // We use generated promise wrapper to allow abort
     let initialResponse = await provider.generate(fullPrompt, undefined, options.signal);
-    // Check for JSON tool call
-    const toolRegex = /```json\s*(\{.*"tool":.*\})\s*```/s;
+    // Check for empty response
+    if (!initialResponse || initialResponse.trim() === '') {
+        throw new Error('LLM returned an empty response. Check if the model is loaded correctly.');
+    }
+    // Check for JSON tool call (support both markdown block and raw JSON)
+    const toolRegex = /(?:```json\s*)?(\{.*"tool":.*\})(?:\s*```)?/s;
     const match = initialResponse.match(toolRegex);
     if (match) {
         try {
             const toolCall = JSON.parse(match[1]);
             if (options.onToken)
-                options.onToken(`[Executing ${toolCall.tool}...] `);
+                options.onToken(`\u001b[33m[Executing ${toolCall.tool}...]\u001b[0m\n`);
             const toolResult = await (0, toolRegistry_1.executeToolCall)(toolCall.tool, toolCall.args);
             fullPrompt += `\n${initialResponse}\n\nTool Output: ${toolResult}\n\nAssistant (Interpreting result):`;
             // Final Response

package/dist/index.js

@@ -1,3 +1,4 @@
+#!/usr/bin/env node
 "use strict";
 var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@react-frameui/loki-ai",
-  "version": "1.0.0",
+  "version": "1.0.2",
   "description": "LOKI - Local Omni Knowledge Interface. A powerful CLI AI assistant with tools, memory, WhatsApp integration, and more.",
   "keywords": [
     "ai",
@@ -71,4 +71,4 @@
   "engines": {
     "node": ">=18.0.0"
   }
-}
+}