goblin-agents 3.2.0 → 3.3.0

package/contextCache.js

@@ -0,0 +1,4 @@
+const {Elf} = require('xcraft-core-goblin');
+const {ContextCache, ContextCacheLogic} = require('./lib/contextCache.js');
+
+exports.xcraftCommands = Elf.birth(ContextCache, ContextCacheLogic);

package/lib/contextCache.js

@@ -0,0 +1,69 @@
+// @ts-check
+const {Elf} = require('xcraft-core-goblin');
+const {record, string} = require('xcraft-core-stones');
+
+class ContextCacheShape {
+  id = string;
+  contexts = record(string, string);
+}
+
+class ContextCacheState extends Elf.Sculpt(ContextCacheShape) {}
+
+class ContextCacheLogic extends Elf.Spirit {
+  state = new ContextCacheState({
+    id: 'contextCache',
+    contexts: {},
+  });
+
+  append(contextId, markdown) {
+    const current = this.state.contexts[contextId] || '';
+    this.state.contexts[contextId] = current + markdown;
+  }
+
+  clear(contextId) {
+    delete this.state.contexts[contextId];
+  }
+}
+
+class ContextCache extends Elf.Alone {
+  logic = Elf.getLogic(ContextCacheLogic);
+  state = new ContextCacheState();
+
+  async init() {
+    return this;
+  }
+
+  async append(contextId, markdown) {
+    if (typeof contextId !== 'string') {
+      throw new TypeError('contextId must be a string');
+    }
+    if (typeof markdown !== 'string') {
+      throw new TypeError('markdown must be a string');
+    }
+
+    this.logic.append(contextId, markdown);
+  }
+
+  async read(contextId) {
+    if (typeof contextId !== 'string') {
+      throw new TypeError('contextId must be a string');
+    }
+
+    return this.state.contexts[contextId] || '';
+  }
+
+  async clear(contextId) {
+    if (typeof contextId !== 'string') {
+      throw new TypeError('contextId must be a string');
+    }
+
+    this.logic.clear(contextId);
+  }
+}
+
+module.exports = {
+  ContextCache,
+  ContextCacheLogic,
+  ContextCacheShape,
+  ContextCacheState,
+};

package/lib/llm/aiAgent.js

@@ -497,8 +497,8 @@ class AiAgent extends Elf {
     return cleanText(res);
   }
 
-  async reset(contextId, save = false) {
-    if (!contextId) {
+  async reset(contextId = null, save = false) {
+    if (contextId === null) {
       for (const contextId of Object.keys(this.state.messages)) {
         await this.reset(contextId);
       }
@@ -770,7 +770,7 @@ ${context}
 
   async callTools(contextId, message, userDesktopId) {
     const {tool_calls} = message;
-    if (!tool_calls || !userDesktopId) {
+    if (!tool_calls?.length || !userDesktopId) {
       return false;
     }
     this.logic.callTools(contextId, tool_calls);
@@ -780,20 +780,19 @@ ${context}
         function: {name},
       } = tool;
 
-      let result = '';
-      let toolArguments = {};
-      if (
-        tool.function.arguments &&
-        typeof tool.function.arguments === 'string'
-      ) {
-        toolArguments = JSON.parse(tool.function.arguments);
-      } else if (
-        tool.function.arguments &&
-        typeof tool.function.arguments === 'object'
-      ) {
-        toolArguments = tool.function.arguments;
-      }
       try {
+        let toolArguments = {};
+        if (
+          tool.function.arguments &&
+          typeof tool.function.arguments === 'string'
+        ) {
+          toolArguments = JSON.parse(tool.function.arguments);
+        } else if (
+          tool.function.arguments &&
+          typeof tool.function.arguments === 'object'
+        ) {
+          toolArguments = tool.function.arguments;
+        }
         const payload = {
           id: this.state.toolServiceId,
           desktopId: userDesktopId,
@@ -803,10 +802,18 @@ ${context}
         const toolNamespace = this.state.toolServiceId?.split('@', 1)[0];
         const toolCmd = `${toolNamespace}.${name}`;
         this.log.dbg(`🔧 Tool call: ${toolCmd}(${JSON.stringify(payload)})`);
-        result = await this.quest.cmd(toolCmd, payload);
-        await this.addToolMessage(contextId, id, name, result);
+        const result = await this.quest.cmd(toolCmd, payload);
+        const content =
+          typeof result === 'string' ? result : JSON.stringify(result ?? null);
+        await this.addToolMessage(contextId, id, name, content);
       } catch (err) {
         this.log.dbg(`🔧 Tool error : ${err.message}`);
+        await this.addToolMessage(
+          contextId,
+          id,
+          name,
+          JSON.stringify({status: 'error', message: err.message})
+        );
       }
     }
     return true;

package/lib/llm/utils.js

@@ -49,6 +49,9 @@ const getPDFImages = async (pdfPath, pagesToProcess) => {
 };
 
 const cleanHtmlContent = (html) => {
+  if (typeof html !== 'string') {
+    return '';
+  }
   const decodedHtml = HtmlEntities.decode(html);
   const textOnly = decodedHtml.replace(/<[^>]+>/g, '');
   const cleanedText = textOnly.replace(/\s+/g, ' ').trim();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "goblin-agents",
-  "version": "3.2.0",
+  "version": "3.3.0",
   "description": "AI Agents framework",
   "author": "Epsitec SA",
   "contributors": [