@peebles-group/agentlib-js 1.0.2 → 1.0.4

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@peebles-group/agentlib-js",
-  "version": "1.0.2",
+  "version": "1.0.4",
   "description": "A minimal JavaScript library implementing concurrent async agents for illustrating multi-agent systems and other agentic design patterns including recursive ones purely through function calling loops.",
   "main": "index.js",
   "type": "module",
@@ -20,7 +20,7 @@
     "js-yaml": "^4.1.0",
     "mongodb": "^6.20.0",
     "openai": "^5.16.0",
-    "agentlib-js": "^1.0.0",
+    "@peebles-group/agentlib-js": "^1.0.0",
     "playwright": "^1.55.0",
     "prompt-sync": "^4.2.0",
     "sqlite": "^5.1.1",

package/src/Agent.js

@@ -3,7 +3,7 @@ import { defaultModel } from "./config.js";
 import { MCPManager } from "./mcp/MCPManager.js";
 
 export class Agent {
-  constructor(provider, apiKey, {model = defaultModel, tools = [], inputSchema = null, outputSchema = null, enableMCP = false} = {}) {
+  constructor(provider, apiKey, {model = defaultModel, tools = [], inputSchema = null, outputSchema = null, enableMCP = false, ...options} = {}) {
     this.llmService = new LLMService(provider, apiKey);
     this.model = model;
     this.nativeTools = tools;
@@ -11,6 +11,7 @@ export class Agent {
     this.outputSchema = outputSchema;
     this.mcpManager = enableMCP ? new MCPManager() : null;
     this.updateSystemPrompt();
+    this.options = options;
   }
 
   async addMCPServer(serverName, config) {
@@ -31,6 +32,34 @@ export class Agent {
     return result;
   }
 
+  addTool(tool) {
+    if (!tool || typeof tool !== 'object') {
+      throw new Error("Invalid tool: expected an object");
+    }
+    const { name, func } = tool;
+    if (typeof name !== 'string' || name.trim() === '') {
+      throw new Error("Invalid tool: missing valid 'name' (string)");
+    }
+    if (typeof func !== 'function') {
+      throw new Error("Invalid tool: missing 'func' (function)");
+    }
+
+    // Prevent name collisions across native and MCP tools
+    const nameExistsInNative = this.nativeTools.some(t => t && t.name === name);
+    const nameExistsInMCP = this.mcpManager ? this.mcpManager.getAllTools().some(t => t && t.name === name) : false;
+    if (nameExistsInNative || nameExistsInMCP) {
+      throw new Error(`Tool with name '${name}' already exists`);
+    }
+
+    if (typeof tool.description !== 'string') {
+      tool.description = '';
+    }
+
+    this.nativeTools.push(tool);
+    this.updateSystemPrompt();
+    return tool;
+  }
+
   getAllTools() {
     const mcpTools = this.mcpManager ? this.mcpManager.getAllTools() : [];
     return [...this.nativeTools, ...mcpTools];
@@ -63,27 +92,29 @@ export class Agent {
    */
   async run() {
     const allTools = this.getAllTools();
+    const executed = []
 
     // Step 1: send input to model
     let response = await this.llmService.chat(this.input, {
       model: this.model,
       outputSchema: this.outputSchema,
       tools: allTools,
+      ...this.options,
     });
 
     const { output, rawResponse } = response;
 
     // Step 2: Clean and add the response to input history
-    // Remove parsed_arguments (if it exists) from function calls before adding to history
-    const cleanedOutput = rawResponse.output.map(item => {
-      if (item.type === "function_call" && item.parsed_arguments) {
-        const { parsed_arguments, ...cleanItem } = item;
-        return cleanItem;
+    rawResponse.output.forEach(item => {
+      if (item.type === "function_call") {
+        // Remove parsed_arguments if it exists
+        const { parsed_arguments, ...rest } = item;
+        const cleanedItem = { ...rest, arguments: JSON.stringify(item.arguments) };
+        this.addInput(cleanedItem);
+      } else {
+        this.addInput(item);
       }
-      return item;
     });
-      
-    this.input = this.input.concat(cleanedOutput);
 
     // Step 3: collect all function calls
     const functionCalls = rawResponse.output.filter(item => item.type === "function_call");
@@ -91,12 +122,9 @@ export class Agent {
     if (functionCalls.length > 0) {
       for (const call of functionCalls) {
         let args;
-        try {
-          args = JSON.parse(call.arguments);
-        } catch (err) {
-          console.error("Failed to parse function call arguments:", call.arguments);
-          continue;
-        }
+        args = JSON.parse(call.arguments);
+        call.arguments = args
+        executed.push(call)
 
         const tool = allTools.find(t => t.name === call.name);
         if (!tool || !tool.func) {
@@ -113,7 +141,7 @@ export class Agent {
           output: JSON.stringify(result),
         });
       }
-
+      
       // Step 6: send updated input back to model for final response
       response = await this.llmService.chat(this.input, {
         tools: allTools,
@@ -121,6 +149,7 @@ export class Agent {
         outputSchema: this.outputSchema,
       });
     }
+    response.executed = executed;
     return response;
   }
 

package/src/config.js

@@ -1 +1 @@
-export const defaultModel = 'gpt-4o-mini';
+export const defaultModel = 'gpt-5';