@open1s/ezbos 1.0.5 → 1.1.1

package/dist/tool.js

@@ -30,9 +30,6 @@ export class ToolBuilder {
     handle(callback) {
         const isAsync = callback.constructor.name === 'AsyncFunction' ||
             callback.toString().startsWith('async ');
-        if (isAsync) {
-            throw new Error(`Tool "${this._name}" uses an async callback, but jsbos does not support async tool callbacks. Please use a synchronous callback instead.`);
-        }
         const properties = {};
         for (const [key, spec] of Object.entries(this._params)) {
             properties[key] = { type: spec.type };
@@ -46,13 +43,10 @@ export class ToolBuilder {
             properties,
             required: this._required.length > 0 ? this._required : Object.keys(this._params)
         };
-        const wrappedCallback = (rawArgs) => {
+        const wrappedCallback = async (rawArgs) => {
             try {
                 const args = typeof rawArgs === 'string' ? JSON.parse(rawArgs) : rawArgs;
-                const result = callback(args);
-                if (result instanceof Promise) {
-                    throw new Error('Async tool callbacks are not supported. Use a sync callback instead.');
-                }
+                const result = await callback(args);
                 if (isErrorResult(result))
                     return 'Error: ' + (result.error || 'Unknown error');
                 if (result === undefined)

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@open1s/ezbos",
-  "version": "1.0.5",
+  "version": "1.1.1",
   "description": "Simple BrainOS - Easy wrapper for @open1s/jsbos",
   "type": "module",
   "main": "./dist/index.js",
@@ -38,7 +38,7 @@
   ],
   "license": "MIT",
   "dependencies": {
-    "@open1s/jsbos": "^2.2.4"
+    "@open1s/jsbos": "^2.2.6"
   },
   "devDependencies": {
     "@modelcontextprotocol/sdk": "^1.29.0",

package/src/__tests__/agent-stop.test.ts

@@ -0,0 +1,42 @@
+import { describe, it } from 'node:test';
+import assert from 'node:assert';
+import { BrainOS } from '../index.js';
+
+describe('AgentBuilder.stop()', () => {
+  it('should call stop on inner agent without error', async () => {
+    const brain = new BrainOS();
+    await brain.start();
+
+    const builder = brain.agent('test-stop');
+    const agent = await builder.start();
+
+    builder.stop();
+
+    await agent.close();
+    await brain.stop();
+  });
+
+  it('should handle stop when no inner agent', async () => {
+    const brain = new BrainOS();
+    await brain.start();
+
+    const builder = brain.agent('test-stop-empty');
+    builder.stop();
+    builder.stop();
+
+    await brain.stop();
+  });
+
+  it('should stop after start and allow close', async () => {
+    const brain = new BrainOS();
+    await brain.start();
+
+    const builder = brain.agent('test-stop-stream');
+    const agent = await builder.start();
+
+    builder.stop();
+    await agent.close();
+
+    await brain.stop();
+  });
+});

package/src/agent.ts

@@ -403,7 +403,12 @@ export class Agent {
     this._inner.resetPerfMetrics();
   }
 
+  async stop(): Promise<void> {
+    this._inner.stop();
+  }
+
   async close(): Promise<void> {
+    this._inner.stop();
     this._inner.close();
   }
 }

package/src/tool.ts

@@ -57,9 +57,6 @@ export class ToolBuilder {
   handle(callback: (args: Record<string, any>) => any): InternalToolDef {
     const isAsync = callback.constructor.name === 'AsyncFunction' ||
       callback.toString().startsWith('async ');
-    if (isAsync) {
-      throw new Error(`Tool "${this._name}" uses an async callback, but jsbos does not support async tool callbacks. Please use a synchronous callback instead.`);
-    }
 
     const properties: Record<string, any> = {};
     for (const [key, spec] of Object.entries(this._params)) {
@@ -74,14 +71,10 @@ export class ToolBuilder {
       required: this._required.length > 0 ? this._required : Object.keys(this._params)
     };
 
-    const wrappedCallback = (rawArgs: any): string => {
+    const wrappedCallback = async (rawArgs: any): Promise<string> => {
       try {
         const args = typeof rawArgs === 'string' ? JSON.parse(rawArgs) : rawArgs;
-        const result = callback(args);
-
-        if (result instanceof Promise) {
-          throw new Error('Async tool callbacks are not supported. Use a sync callback instead.');
-        }
+        const result = await callback(args);
 
         if (isErrorResult(result)) return 'Error: ' + (result.error || 'Unknown error');
         if (result === undefined) return '';