@samrahimi/smol-js 0.1.0 → 0.2.0

package/dist/index.d.mts

@@ -767,6 +767,65 @@ declare class UserInputTool extends Tool {
 }
 declare const finalAnswerTool: FinalAnswerTool;
 
+/**
+ * AgentTool - Wraps a CodeAgent as a Tool for use by other agents
+ *
+ * This enables nested/hierarchical agent architectures where a "manager" agent
+ * can delegate tasks to specialized "worker" agents.
+ *
+ * Based on the ManagedAgent pattern from Python smolagents.
+ */
+
+interface AgentToolConfig {
+    /**
+     * The agent to wrap as a tool
+     */
+    agent: Agent;
+    /**
+     * Name for the tool (defaults to agent's name or "managed_agent")
+     */
+    name?: string;
+    /**
+     * Description of what this agent does (used in parent agent's prompt)
+     */
+    description?: string;
+    /**
+     * Additional context to provide to the agent with each task
+     */
+    additionalContext?: string;
+    /**
+     * Whether to provide the full result object or just the output
+     * @default false
+     */
+    returnFullResult?: boolean;
+}
+declare class AgentTool extends Tool {
+    readonly name: string;
+    readonly description: string;
+    readonly inputs: ToolInputs;
+    readonly outputType = "string";
+    private agent;
+    private additionalContext?;
+    private returnFullResult;
+    constructor(config: AgentToolConfig);
+    /**
+     * Generate a default description based on the agent's configuration
+     */
+    private generateDescription;
+    /**
+     * Execute the agent with the given task
+     */
+    execute(args: Record<string, unknown>): Promise<unknown>;
+    /**
+     * Override toCodePrompt to provide a cleaner signature for nested agents
+     */
+    toCodePrompt(): string;
+}
+/**
+ * Helper function to quickly wrap an agent as a tool
+ */
+declare function agentAsTool(agent: Agent, options?: Omit<AgentToolConfig, 'agent'>): AgentTool;
+
 /**
  * System prompts for CodeAgent
  *
@@ -790,4 +849,4 @@ declare const FINAL_ANSWER_PROMPT = "Based on the steps you've taken so far, pro
  */
 declare function getErrorRecoveryPrompt(error: string): string;
 
-export { type ActionOutput, type ActionStep, Agent, type AgentConfig, type AgentConfig$1 as AgentConfigType, AgentLogger, AgentMemory, type ChatMessage, CodeAgent, type CodeAgentConfig, type CodeExecutionOutput, type ExecutorConfig, FINAL_ANSWER_PROMPT, type FinalAnswerStep, FinalAnswerTool, type GenerateOptions, LocalExecutor, LogLevel, type MemoryStep, type MessageRole, Model, type ModelConfig, OpenAIModel, type OpenAIModelConfig, type PromptVariables, type RunResult, type StreamEvent, type SystemPromptStep, type TaskStep, type Timing, type TokenUsage, Tool, type ToolCall, type ToolInput, type ToolInputType, type ToolInputs, UserInputTool, createTool, finalAnswerTool, generateSystemPrompt, getErrorRecoveryPrompt };
+export { type ActionOutput, type ActionStep, Agent, type AgentConfig, type AgentConfig$1 as AgentConfigType, AgentLogger, AgentMemory, AgentTool, type AgentToolConfig, type ChatMessage, CodeAgent, type CodeAgentConfig, type CodeExecutionOutput, type ExecutorConfig, FINAL_ANSWER_PROMPT, type FinalAnswerStep, FinalAnswerTool, type GenerateOptions, LocalExecutor, LogLevel, type MemoryStep, type MessageRole, Model, type ModelConfig, OpenAIModel, type OpenAIModelConfig, type PromptVariables, type RunResult, type StreamEvent, type SystemPromptStep, type TaskStep, type Timing, type TokenUsage, Tool, type ToolCall, type ToolInput, type ToolInputType, type ToolInputs, UserInputTool, agentAsTool, createTool, finalAnswerTool, generateSystemPrompt, getErrorRecoveryPrompt };

package/dist/index.d.ts

@@ -767,6 +767,65 @@ declare class UserInputTool extends Tool {
 }
 declare const finalAnswerTool: FinalAnswerTool;
 
+/**
+ * AgentTool - Wraps a CodeAgent as a Tool for use by other agents
+ *
+ * This enables nested/hierarchical agent architectures where a "manager" agent
+ * can delegate tasks to specialized "worker" agents.
+ *
+ * Based on the ManagedAgent pattern from Python smolagents.
+ */
+
+interface AgentToolConfig {
+    /**
+     * The agent to wrap as a tool
+     */
+    agent: Agent;
+    /**
+     * Name for the tool (defaults to agent's name or "managed_agent")
+     */
+    name?: string;
+    /**
+     * Description of what this agent does (used in parent agent's prompt)
+     */
+    description?: string;
+    /**
+     * Additional context to provide to the agent with each task
+     */
+    additionalContext?: string;
+    /**
+     * Whether to provide the full result object or just the output
+     * @default false
+     */
+    returnFullResult?: boolean;
+}
+declare class AgentTool extends Tool {
+    readonly name: string;
+    readonly description: string;
+    readonly inputs: ToolInputs;
+    readonly outputType = "string";
+    private agent;
+    private additionalContext?;
+    private returnFullResult;
+    constructor(config: AgentToolConfig);
+    /**
+     * Generate a default description based on the agent's configuration
+     */
+    private generateDescription;
+    /**
+     * Execute the agent with the given task
+     */
+    execute(args: Record<string, unknown>): Promise<unknown>;
+    /**
+     * Override toCodePrompt to provide a cleaner signature for nested agents
+     */
+    toCodePrompt(): string;
+}
+/**
+ * Helper function to quickly wrap an agent as a tool
+ */
+declare function agentAsTool(agent: Agent, options?: Omit<AgentToolConfig, 'agent'>): AgentTool;
+
 /**
  * System prompts for CodeAgent
  *
@@ -790,4 +849,4 @@ declare const FINAL_ANSWER_PROMPT = "Based on the steps you've taken so far, pro
  */
 declare function getErrorRecoveryPrompt(error: string): string;
 
-export { type ActionOutput, type ActionStep, Agent, type AgentConfig, type AgentConfig$1 as AgentConfigType, AgentLogger, AgentMemory, type ChatMessage, CodeAgent, type CodeAgentConfig, type CodeExecutionOutput, type ExecutorConfig, FINAL_ANSWER_PROMPT, type FinalAnswerStep, FinalAnswerTool, type GenerateOptions, LocalExecutor, LogLevel, type MemoryStep, type MessageRole, Model, type ModelConfig, OpenAIModel, type OpenAIModelConfig, type PromptVariables, type RunResult, type StreamEvent, type SystemPromptStep, type TaskStep, type Timing, type TokenUsage, Tool, type ToolCall, type ToolInput, type ToolInputType, type ToolInputs, UserInputTool, createTool, finalAnswerTool, generateSystemPrompt, getErrorRecoveryPrompt };
+export { type ActionOutput, type ActionStep, Agent, type AgentConfig, type AgentConfig$1 as AgentConfigType, AgentLogger, AgentMemory, AgentTool, type AgentToolConfig, type ChatMessage, CodeAgent, type CodeAgentConfig, type CodeExecutionOutput, type ExecutorConfig, FINAL_ANSWER_PROMPT, type FinalAnswerStep, FinalAnswerTool, type GenerateOptions, LocalExecutor, LogLevel, type MemoryStep, type MessageRole, Model, type ModelConfig, OpenAIModel, type OpenAIModelConfig, type PromptVariables, type RunResult, type StreamEvent, type SystemPromptStep, type TaskStep, type Timing, type TokenUsage, Tool, type ToolCall, type ToolInput, type ToolInputType, type ToolInputs, UserInputTool, agentAsTool, createTool, finalAnswerTool, generateSystemPrompt, getErrorRecoveryPrompt };

package/dist/index.js

@@ -33,6 +33,7 @@ __export(index_exports, {
   Agent: () => Agent,
   AgentLogger: () => AgentLogger,
   AgentMemory: () => AgentMemory,
+  AgentTool: () => AgentTool,
   CodeAgent: () => CodeAgent,
   FINAL_ANSWER_PROMPT: () => FINAL_ANSWER_PROMPT,
   FinalAnswerTool: () => FinalAnswerTool,
@@ -42,6 +43,7 @@ __export(index_exports, {
   OpenAIModel: () => OpenAIModel,
   Tool: () => Tool,
   UserInputTool: () => UserInputTool,
+  agentAsTool: () => agentAsTool,
   createTool: () => createTool,
   finalAnswerTool: () => finalAnswerTool,
   generateSystemPrompt: () => generateSystemPrompt,
@@ -1749,11 +1751,87 @@ var OpenAIModel = class extends Model {
     });
   }
 };
+
+// src/tools/AgentTool.ts
+var AgentTool = class extends Tool {
+  name;
+  description;
+  inputs = {
+    task: {
+      type: "string",
+      description: "The task or question to delegate to this agent",
+      required: true
+    }
+  };
+  outputType = "string";
+  agent;
+  additionalContext;
+  returnFullResult;
+  constructor(config) {
+    super();
+    this.agent = config.agent;
+    this.name = config.name ?? "managed_agent";
+    this.additionalContext = config.additionalContext;
+    this.returnFullResult = config.returnFullResult ?? false;
+    this.description = config.description ?? this.generateDescription();
+  }
+  /**
+   * Generate a default description based on the agent's configuration
+   */
+  generateDescription() {
+    return `Delegates a task to a specialized agent.
+This agent can help with complex sub-tasks that require multiple steps.
+Pass a clear, specific task description and the agent will work autonomously to solve it.
+Returns the agent's final answer as a string.`;
+  }
+  /**
+   * Execute the agent with the given task
+   */
+  async execute(args) {
+    let task = args.task;
+    if (this.additionalContext) {
+      task = `${this.additionalContext}
+
+Task: ${task}`;
+    }
+    const result = await this.agent.run(task, true);
+    if (this.returnFullResult) {
+      return {
+        output: result.output,
+        steps: result.steps.length,
+        duration: result.duration
+      };
+    }
+    const output = result.output;
+    if (typeof output === "string") {
+      return output;
+    }
+    return JSON.stringify(output, null, 2);
+  }
+  /**
+   * Override toCodePrompt to provide a cleaner signature for nested agents
+   */
+  toCodePrompt() {
+    return `
+/**
+ * ${this.description}
+ *
+ * @param task - The task or question to delegate to this specialized agent
+ * @returns The agent's answer as a string
+ */
+async function ${this.name}(task: string): Promise<string> { ... }
+`.trim();
+  }
+};
+function agentAsTool(agent, options) {
+  return new AgentTool({ agent, ...options });
+}
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   Agent,
   AgentLogger,
   AgentMemory,
+  AgentTool,
   CodeAgent,
   FINAL_ANSWER_PROMPT,
   FinalAnswerTool,
@@ -1763,6 +1841,7 @@ var OpenAIModel = class extends Model {
   OpenAIModel,
   Tool,
   UserInputTool,
+  agentAsTool,
   createTool,
   finalAnswerTool,
   generateSystemPrompt,