@atgs/tapeworm 0.1.6 → 0.1.8

package/README.md

@@ -1,13 +1,11 @@
 ![Tapeworm Logo](https://raw.githubusercontent.com/andygrace227/tapeworm/main/tapeworm.svg)
 
-# Tapeworm
+# Tapeworm Core
 
-<h3>In-browser and Node agent framework.</h3>
+<h3>In-browser and Node Agent Framework.</h3>
 
 [![npm version](https://img.shields.io/npm/v/@atgs/tapeworm.svg?style=flat-square)](https://www.npmjs.org/package/@atgs/tapeworm.svg)
-
 [![npm downloads](https://img.shields.io/npm/dm/@atgs/tapeworm?style=flat-square)](https://npm-stat.com/charts.html?package=@atgs/tapeworm)
-
 [repo link](https://github.com/andygrace227/tapeworm)
 
 This is the root package for Tapeworm. You can consume other packages like @atgs/tapeworm_bedrock for AWS Bedrock support.
@@ -18,13 +16,22 @@ It provides an object-oriented API to create agents that run either on Node or w
 
 ## Current Features
 
-- Base API Defined.
 - Supports `function` tools.
 - Supports Ollama models.
-- Has a Babel plugin to make tool creation really easy.
+- Has a Babel plugin to make tool creation easy.
+- Has TS decorators to make tool creation really easy, too!
+- Supports browser and Node.
+
+## Tapeworm's Tenets
+
+- **Be the most ergonomic agentic solution for Node and the browser.** Each commit should make it easier to develop and deploy agentic AI solutions.
+- **Be as model-agnostic as possible.** Use your own machine, AWS, Google, a literal potato... we don't care.
+- **Keep things light.** We already waste so much water and energy with AI. The overhead from Tapeworm should be kept to a minimum when possible.
 
 ## Examples
 
+### How do I define a tool?
+
 #### With the babel plugin (@atgs/@atgs/babel-plugin-tapeworm-decorator) (recommended, super concise)
 
 ```js
@@ -50,22 +57,37 @@ class AdditionTool extends Tool {
     return a + b;
   }
 }
+```
 
-const ollama = new OllamaModel("http://localhost:11434", "gpt-oss:20b", {
-  stream: false,
-});
-
-const agent = Agent.builder()
-  .name("calculatorAgent")
-  .tools([new AdditionTool()])
-  .systemPrompt("You are an agent that runs math operations.")
-  .model(ollama)
-  .build();
-
-await agent.invoke("What is 9 + 10?");
+#### With Typescript Decorators
+
+```ts
+@ToolName("AdditionTool")
+@ToolDescription("Adds two numbers together.")
+@ToolParameter({
+  name: "a",
+  description: "The first number to add",
+  required: true,
+  type: "number",
+})
+@ToolParameter({
+  name: "b",
+  description: "The second number to add",
+  required: true,
+  type: "number",
+})
+@ToolOutput("The sum of inputs a and b")
+class AdditionTool extends Tool {
+  execute(input: any) {
+    let a = +input.a;
+    let b = +input.b;
+    console.log("Adding " + a + " and " + b + ": " + (a + b));
+    return a + b;
+  }
+}
 ```
 
-#### Without the plugin (still pretty readable.)
+#### Without the plugin (still pretty readable)
 
 ```js
 import {
@@ -77,18 +99,10 @@ import {
 } from "../../dist/tapeworm.es.js";
 
 class AdditionTool extends Tool {
-  /**
-   * Unique name used to reference this tool.
-   * @returns Tool identifier string.
-   */
   getName() {
     return "AdditionTool";
   }
 
-  /**
-   * Short description provided to the model.
-   * @returns Human-readable explanation of the tool.
-   */
   getDescription() {
     return "Adds two numbers together.";
   }
@@ -117,9 +131,11 @@ class AdditionTool extends Tool {
     return a + b;
   }
 }
+```
 
-// Because this is a test file, we are going to run this locally using Ollama
+#### Then calling the agent:
 
+```js
 const ollama = new OllamaModel("http://localhost:11434", "gpt-oss:20b", {
   stream: false,
 });
@@ -131,15 +147,9 @@ const agent = Agent.builder()
   .model(ollama)
   .build();
 
-await agent.invoke("What is 9 + 10");
+await agent.invoke("What is 9 + 10?");
 ```
 
-## Tapeworm's Tenets
-
-- **Be the most ergonomic agentic solution for Node and the browser.** Each commit should make it easier to develop and deploy agentic AI solutions.
-- **Be as model-agnostic as possible.** Use your own machine, AWS, Google, a literal potato... we don't care.
-- **Keep things light.** We already waste so much water and energy with AI. The overhead from Tapeworm should be kept to a minimum when possible.
-
 ## Roadmap
 
 Tapeworm seeks to be the most ergonomic agentic solution for Node and the browser.

package/dist/agent/agent.d.ts

@@ -1,27 +1,90 @@
+/**
+ * The Agent module contains all things related to Agents, including:
+ * - the Agent class
+ * - the AgentBuilder class
+ * - the default callback handler for an Agent.
+ *
+ * This module contains most of the business logic for running the main agent loop.
+ *
+ * @module
+ */
 import Conversation, { ConversationManager } from "../conversation/conversation";
 import Message from "../conversation/message";
 import { type Model } from "../model/model";
 import type Tool from "../tool/tool";
 import type ToolCall from "../tool/toolCall";
 /**
- * Coordinates a model, its tools, and the running conversation to fulfill user queries.
+ * Agents combine a model with your tools.
+ *
+ * To build an agent, use the AgentBuilder class:
+ *
+ * ```ts
+ * const agent = Agent.builder()
+ *  .name("yourAgent")
+ *  .tools([new YourTool()])
+ *  .systemPrompt("You are an agent.")
+ *  .model(yourModel)
+ *  .build();
+ * ```
+ *
  */
 export default class Agent {
+    /**
+     * The name of the agent.
+     */
     name: string;
+    /**
+     * The system prompt (if any) for the agent.
+     */
     systemPrompt?: string;
+    /**
+     * The tools this agent knows about and can run.
+     */
     tools: Tool[];
+    /**
+     * The underlying model for the agent.
+     */
     model: Model;
+    /**
+     * The conversation for the agent.
+     * You do not normally need to set this parameter unless you are building a use case
+     * where you have a persistent session with an agent.
+     */
     conversation: Conversation;
+    /**
+     * The conversation manager for the agent.
+     * Normally, a converstation manager that does nothing will be used.
+     * When SummarizingConversationManager or RAGConversationManager is implemented, these will help you avoid
+     * token overflow.
+     */
     conversationManager?: ConversationManager;
-    toolNameToIndexMap: any | undefined;
+    /**
+     * A cache for looking up tools in the Tools array.
+     * DO NOT MANUALLY SET THIS VALUE. AGENT AUTOMATICALLY TAKES CARE OF IT.
+     */
+    private toolNameToIndexMap;
+    /**
+     * The callback function, invoked for every response from the model.
+     * The default implementation just prints to std.out or your console.
+     */
     callback: (m: Message) => void;
+    /**
+     * This is the constructor for an Agent.
+     *
+     * i IMPLORE YOU not to use this method. You CAN, but you should be using the Builder instead
+     * for the best results, as this method WILL change with future updates.
+     *
+     * Builder will be a lot more stable and will always be backwards compatible.
+     */
     constructor(name: string, tools: Tool[], model: Model, conversationManager: ConversationManager, callback: (m: Message) => void);
     /**
      * Run the full agent loop for a user query: seed the conversation, invoke the model,
      * and execute any returned tool calls until completion.
+     *   *
      * @param query User-provided input to hand to the agent.
+     * @param callback A function that handles the messages coming from the agent.
      */
-    invoke(query: string): Promise<void>;
+    invoke(query: string, callback?: (m: Message) => void): Promise<void>;
     /**
      * Ask the backing model for the next response given the current conversation state.
      * @returns Parsed model response including content, thinking, and tool calls.
@@ -37,9 +100,12 @@ export default class Agent {
      * Build a lookup from tool name to index for efficient resolution of tool calls.
      */
     generateToolNameToIndexMap(): void;
+    /**
+     * Get a builder to build an Agent.
+     */
     static builder(): AgentBuilder;
 }
-declare class AgentBuilder {
+export declare class AgentBuilder {
     _name: string;
     _systemPrompt?: string;
     _tools: Tool[];
@@ -48,14 +114,45 @@ declare class AgentBuilder {
     _conversationManager: ConversationManager;
     _toolNameToIndexMap: any | undefined;
     _callback: (m: Message) => void;
+    /**
+     * Set the name of the agent.
+     */
     name(name: string): AgentBuilder;
+    /**
+     * Set the system prompt of the agent.
+     */
     systemPrompt(systemPrompt: string | undefined): AgentBuilder;
+    /**
+     * Set the tools of the agent, if you already have an array of tools.
+     */
     tools(tools: Tool[]): AgentBuilder;
+    /**
+     * Add a tool, 1 by 1, to the agent..
+     */
     addTool(tool: Tool): AgentBuilder;
+    /**
+     * Set the underlying model for the agent.
+     */
     model(model: Model): AgentBuilder;
+    /**
+     * Set the conversation manager for the agent.
+     */
     conversationManager(mgr: ConversationManager): AgentBuilder;
+    /**
+     * Set the callback for the agent.
+     */
     callback(callback: (m: Message) => void): AgentBuilder;
+    /**
+     * Build the agent.
+     */
     build(): Agent;
 }
+/**
+ * This function is Tapeworm's default callback function.
+ * If you do not specify your own callback function, then this will be called and all
+ * it will do is log output to the screen.
+ *
+ * You can override this per invocation or on the Agent itself to do other things
+ * with the LLM output.
+ */
 export declare function defaultCallback(m: Message): void;
-export {};

package/dist/conversation/conversation.d.ts

@@ -1,7 +1,20 @@
+/**
+ * The Conversation module contains the Conversation class and the ConversationManager prototypes.
+ *
+ * The Conversation is a wrapper around both a manager and an array of messages.
+ *
+ * The Conversation Manager is an object that can perform operations on only the conversation.
+ *
+ * While no Conversation Managers have been implemented yet, they will allow resilience to overflowing context windows.
+ *
+ * @module
+ */
 import type { Message } from "..";
 import type { Model } from "../model/model";
 /**
- * Maintains ordered chat messages and delegates to a conversation manager for compaction.
+ * A Conversation holds an array of messages and its associated manager.
+ *
+ * Conversations are (mostly) immutable, until the manager steps in and compacts the conversation.
  */
 export default class Conversation {
     messages: Message[];
@@ -17,7 +30,7 @@ export default class Conversation {
     append(message: Message): void;
 }
 /**
- * Strategy interface for pruning or transforming a conversation history.
+ * Interface for compacting a Conversation. Future implementations can summarize or do RAG-based compaction
  */
 export declare class ConversationManager {
     /**

package/dist/conversation/message.d.ts

@@ -1,3 +1,15 @@
+/**
+ * The Messages module contains all the shapes that make up a single Message.
+ * - The actual Message class, which is a role + an array of MessageComponents
+ *   - And a builder, too, which you should use 99.999% of the time!
+ * - An enum called MessageComponentType
+ * - The MessageComponents
+ *   - **Content** - normal text from the LLM
+ *   - **Thinking** - chain-of-thought reasoning from the LLM
+ *   - **ToolCall** is not technically contained here, but it is a message component type
+ *   - **ToolResult** - encodes the result of a tool call.
+ * @module
+ */
 import type ToolCall from "../tool/toolCall";
 /**
  * Messages are the units of a Conversation

package/dist/index.d.ts

@@ -1,3 +1,10 @@
+/**
+ * The index module gathers the exports from Tapeworm.
+ *
+ * This file should be modified when new classes are introduced to Tapeworm
+ *
+ * @module
+ */
 export { default as Agent } from "./agent/agent";
 export { default as Conversation, ConversationManager, DefaultConversationManager, } from "./conversation/conversation";
 export { Model, ModelRequest, ModelRequestBuilder } from "./model/model";

package/dist/model/OllamaModel.d.ts

@@ -1,8 +1,18 @@
+/**
+ * Tapeworm's first mostly-working model is OllamaModel,
+ * which routes model requests to an Ollama endpoint on your local machine.
+ *
+ * Other Models will be housed in separate NPM packages to allow better portability
+ * and to leverage the existing NPM packages (for example, the AWS SDK in the case of Bedrock)
+ * for those models.
+ *
+ * @module
+ */
 import { Message } from "..";
 import ToolCall from "../tool/toolCall";
 import { Model, ModelRequest } from "./model";
 /**
- * Model adapter that translates between Tapeworm message/tool shapes and the Ollama chat API.
+ * The Ollama Model class. This translates between Tapeworm message/tool shapes and the Ollama chat API.
  */
 export default class OllamaModel extends Model {
     endpoint: string;

package/dist/model/model.d.ts

@@ -1,3 +1,13 @@
+/**
+ * The Model module lays out the interface for a Tapeworm model.
+ *
+ * It contains the base Model class, which doesn't do anything other than throw errors and specify an interface,
+ * and also the ModelRequest and ModelRequestBuilder classes, which show what kind of shape a model expects to
+ * recieve upon invocation.
+ *
+ *
+ * @module
+ */
 import type Message from "../conversation/message";
 import type Tool from "../tool/tool";
 /**

package/dist/tapeworm.cjs.js

@@ -1 +1 @@
-"use strict";Object.defineProperty(exports,Symbol.toStringTag,{value:"Module"});class w{messages;manager;constructor(){this.messages=[],this.manager=new g}append(e){this.messages.push(e),this.messages=this.manager.compact(this.messages)}}class M{compact(e){throw new y("No implementation for conversation manager!")}configure(e){throw new y("No implementation for conversation manager!")}}class g extends M{compact(e){return e}configure(e){}}class y extends Error{constructor(e){super(e),this.name="ConversationManagerNotImplemented"}}class u{role;content;constructor(e,o){this.role=e,this.content=o}static builder(){return new b}filter(e){return this.content.filter(o=>o.getMessageComponentType()==e)}}class b{_role;_content;role(e){return this._role=e,this}init(){this._content==null&&(this._content=[])}toolCall(e){return e==null?this:(this.init(),this._content.push(e),this)}toolCalls(e){if(e==null)return this;this.init();for(const o of e)this._content.push(o);return this}toolResult(e){return e==null?this:(this.init(),this._content.push(e),this)}thinking(e){return e==null?this:(this.init(),this._content.push(C.of(e)),this)}content(e){return e==null?this:(this.init(),this._content.push(N.of(e)),this)}build(){if(this._content==null||this._content.length==0)throw new Error("Role-only messages are not supported by Tapeworm.");return new u(this._role,this._content)}}const a={Content:"content",Thinking:"thinking",ToolCall:"toolcall",ToolResult:"toolresult"};class m{getMessageComponentType(){throw new Error("Message components that do not have a message component type are not allowed.")}}class N extends m{text;constructor(e){super(),this.text=e}getMessageComponentType(){return a.Content}get(){return this.text}static of(e){return new this(e)}}class C extends m{thought;constructor(e){super(),this.thought=e}getMessageComponentType(){return a.Thinking}get(){return this.thought}static of(e){return new this(e)}}class c extends m{id;toolName;toolResult;constructor(e,o,s){super(),this.id=e,this.toolName=o,this.toolResult=s}getMessageComponentType(){return a.ToolResult}static of(e,o){return new this(e.id,e.name,o)}}class P{async invoke(e){throw new v("The invoke function for this model was not correctly implemented.")}tokenLimit(){throw new v("The tokenLimit function for this model was not correctly implemented.")}}class q{messages;tools;constructor(e,o){this.messages=e,this.tools=o}static builder(){return new f}}class f{_messages;_tools;messages(e){return this._messages=e,this}tools(e){return this._tools=e,this}build(){if(this._tools==null&&(this._tools=[]),this._messages==null)throw new j("Requests to the model should include content.");return new q(this._messages,this._tools)}}class v extends Error{constructor(e){super(e),this.name="ModelNotImplementedError"}}class j extends Error{constructor(e){super(e),this.name="MessagesNotDefinedError"}}class _ extends m{sequence;name;parameters;type;id;getMessageComponentType(){return a.ToolCall}constructor(e,o,s,n,r){super(),this.sequence=e,this.name=o,this.parameters=s,this.type=n,this.id=r}static builder(){return new x}}class x{_sequence;_name;_parameters;_type;_id;sequence(e){return this._sequence=e,this}name(e){return this._name=e,this}parameters(e){return this._parameters=e,this}type(e){return this._type=e,this}id(e){return e!=null&&(this._id=e),this}build(){return this._sequence==null&&(this._sequence=0),this._id==null&&(this._id=(Math.random()+1).toString(36).slice(2,7)),new _(this._sequence,this._name,this._parameters,this._type,this._id)}}class D extends Error{constructor(e){super(e),this.name="ToolNotFoundError"}}class k{name;systemPrompt;tools;model;conversation;conversationManager;toolNameToIndexMap;callback;constructor(e,o,s,n,r){this.name=e,this.model=s,this.conversationManager=n,this.tools=o,this.callback=r,this.conversationManager.configure(s)}async invoke(e){this.conversation==null&&(this.conversation=new w,this.conversationManager!=null&&(this.conversation.manager=this.conversationManager),this.systemPrompt!=null&&this.conversation.append(u.builder().role("system").content(this.systemPrompt).build())),this.conversation.append(u.builder().role("user").content(e).build());let o=!1;for(;!o;){let s=await this._runQuery();this.callback(s),this.conversation.append(s),o=!0;const n=s.filter(a.ToolCall);if(n!=null&&n.length!=0){o=!1,n.sort((r,l)=>(r.sequence??0)<(l.sequence??0)?-1:1);for(let r of n)await this._runTool(r)}}}async _runQuery(){return await this.model.invoke(new f().messages(this.conversation?.messages).tools(this.tools).build())}async _runTool(e){if(this.generateToolNameToIndexMap(),!(e.name in this.toolNameToIndexMap)){this.conversation.append(u.builder().role("tool").toolResult(c.of(e,new D("Agent does not have a tool with this name."))).build());return}let o=this.tools[this.toolNameToIndexMap[e.name]];try{let s=await o.execute(e.parameters);this.conversation.append(u.builder().role("tool").toolResult(c.of(e,s)).build())}catch(s){this.conversation.append(u.builder().role("tool").toolResult(c.of(e,JSON.stringify(s))).build())}}generateToolNameToIndexMap(){if(this.toolNameToIndexMap==null){this.toolNameToIndexMap={};for(let e=0;e<this.tools.length;e++)this.toolNameToIndexMap[this.tools[e].getName()]=e}}static builder(){return new A}}class A{_name;_systemPrompt;_tools;_model;_conversation;_conversationManager=new g;_toolNameToIndexMap;_callback=e=>J(e);name(e){return this._name=e,this}systemPrompt(e){return this._systemPrompt=e,this}tools(e){return this._tools=e,this}addTool(e){return this._tools==null&&(this._tools=[]),this._tools.push(e),this}model(e){return this._model=e,this}conversationManager(e){return this._conversationManager=e,this}callback(e){return this._callback=e,this}build(){let e=new k(this._name,this._tools,this._model,this._conversationManager,this._callback);return this._conversation!=null&&(e.conversation=this._conversation),this._systemPrompt!=null&&(e.systemPrompt=this._systemPrompt),e}}function J(t){for(const e of t.filter(a.Thinking))console.log("\x1B[90m"+e.get()+"\x1B[0m");for(const e of t.filter(a.Content))console.log(e.get());for(const e of t.filter(a.ToolCall))console.log("\x1B[32mCalling Tool: "+e.name+"\x1B[0m")}class d extends Error{constructor(e){super(e),this.name="ToolNotDefinedError"}}class F{name;description;tool_schema;constructor(){this.name=this.getName(),this.description=this.getDescription(),this.tool_schema=this.getToolSchema()}getName(){throw new d("Tool name not defined.")}getDescription(){throw new d("Tool description not defined.")}getToolSchema(){throw new d("Tool parameter schema not defined.")}execute(e){return null}}class p{parameters;output;constructor(e,o){this.parameters=e,this.output=o}static builder(){return new O}}class O{_parameters;_output;addParameter(e){return this._parameters==null&&(this._parameters=[]),this._parameters.push(e),this}output(e){return this._output=e,this}build(){return this._parameters==null&&(this._parameters=[]),new p(this._parameters,this._output)}}class T{name;description;type;required;constructor(e,o,s,n){this.name=e,this.description=o,this.type=s,this.required=n,this.assertValidType()}assertValidType(){}static builder(){return new S}}class S{_name;_description;_type;_required;name(e){return this._name=e,this}description(e){return this._description=e,this}type(e){return this._type=e,this}required(e){return this._required=e,this}build(){return this._required==null&&(this._required=!1),new T(this._name,this._description,this._type,this._required)}}class L extends P{endpoint;model;options;constructor(e,o,s){super(),this.endpoint=e,this.model=o,this.options=s}async invoke(e){let o={model:this.model,messages:this._formatMessages(e),tools:this._formatTools(e),...this.options},s=await fetch(this.endpoint+"/api/chat",{method:"POST",headers:{"Content-Type":"application/json"},body:JSON.stringify(o)});if(!s.ok)throw new Error(`HTTP error! status: ${s.status}`);let n=await s.json(),r=[];if(n.message.tool_calls)for(let l in n.message.tool_calls){const i=n.message.tool_calls[l];let h="function",R=i[h]?.name,E=i[h]?.index,B=i.id??void 0,I=typeof i[h]?.arguments=="string"?JSON.parse(i[h]?.arguments):i[h]?.arguments;r.push(_.builder().name(R).type(h).parameters(I).sequence(E).id(B).build())}return u.builder().toolCalls(r).role(n.message.role).content(n.message.content).thinking(n.message.thinking).build()}_formatTools(e){let o=[];for(let s of e.tools){let n={},r=[];for(let i of s.getToolSchema().parameters)n[i.name]={},n[i.name].type=i.type,n[i.name].description=i.description,i.required&&r.push(i.name);let l={type:"function",function:{name:s.getName(),description:s.getDescription(),parameters:{type:"object",properties:n,required:r}}};o.push(l)}return o}_formatMessages(e){let o=[];for(let s of e.messages){if(s.role=="assistant"||s.role=="system"||s.role=="user"){o.push(this._formatSingleMessage(s));continue}if(s.role=="tool"){for(const n of s.content)if(n.getMessageComponentType()==a.ToolResult){const r=n;o.push({role:s.role,name:r.toolName,content:JSON.stringify(r.toolResult)})}}}return o}_formatSingleMessage(e){let o={role:e.role},s,n,r;for(const l of e.content){if(l.getMessageComponentType()==a.Content){const i=l;s==null&&(s=""),s+=i.get()}if(l.getMessageComponentType()==a.Thinking){const i=l;r==null&&(r=""),r+=i.get()}if(l.getMessageComponentType()==a.ToolCall){const i=l;n==null&&(n=[]),n.push(this._formatToolCall(i))}}return s!=null&&(o.content=s),n!=null&&(o.tool_calls=n),r!=null&&(o.thinking=r),o}_formatToolCall(e){return{function:{name:e.name,arguments:e.parameters}}}}function Q(t){return e=>{e.prototype.getName=()=>t}}function V(t){return e=>{Object.defineProperty(e.prototype,"getDescription",{value:function(){return t},enumerable:!0,configurable:!0,writable:!0})}}function H(t){return e=>{e.prototype.tapewormParams==null&&(e.prototype.tapewormParams=[]),e.prototype.tapewormParamsOutput==null&&(e.prototype.tapewormParamsOutput=""),e.prototype.tapewormParams.push(T.builder().name(t.name).description(t.description).required(t.required).type(t.type).build()),e.prototype.getToolSchema=()=>new p(e.prototype.tapewormParams,e.prototype.tapewormParamsOutput)}}function K(t){return e=>{e.prototype.tapewormParams==null&&(e.prototype.tapewormParams=[]),e.prototype.tapewormParamsOutput==null&&(e.prototype.tapewormParamsOutput=""),e.prototype.tapewormParamsOutput=t,e.prototype.getToolSchema=()=>new p(e.prototype.tapewormParams,e.prototype.tapewormParamsOutput)}}exports.Agent=k;exports.Content=N;exports.Conversation=w;exports.ConversationManager=M;exports.DefaultConversationManager=g;exports.Message=u;exports.MessageBuilder=b;exports.MessageComponent=m;exports.MessageComponentType=a;exports.Model=P;exports.ModelRequest=q;exports.ModelRequestBuilder=f;exports.OllamaModel=L;exports.Parameter=T;exports.ParameterBuilder=S;exports.Thinking=C;exports.Tool=F;exports.ToolCall=_;exports.ToolCallBuilder=x;exports.ToolDescription=V;exports.ToolName=Q;exports.ToolOutput=K;exports.ToolParameter=H;exports.ToolResult=c;exports.ToolSchema=p;exports.ToolSchemaBuilder=O;
+"use strict";Object.defineProperty(exports,Symbol.toStringTag,{value:"Module"});class w{messages;manager;constructor(){this.messages=[],this.manager=new g}append(e){this.messages.push(e),this.messages=this.manager.compact(this.messages)}}class M{compact(e){throw new T("No implementation for conversation manager!")}configure(e){throw new T("No implementation for conversation manager!")}}class g extends M{compact(e){return e}configure(e){}}class T extends Error{constructor(e){super(e),this.name="ConversationManagerNotImplemented"}}class u{role;content;constructor(e,s){this.role=e,this.content=s}static builder(){return new b}filter(e){return this.content.filter(s=>s.getMessageComponentType()==e)}}class b{_role;_content;role(e){return this._role=e,this}init(){this._content==null&&(this._content=[])}toolCall(e){return e==null?this:(this.init(),this._content.push(e),this)}toolCalls(e){if(e==null)return this;this.init();for(const s of e)this._content.push(s);return this}toolResult(e){return e==null?this:(this.init(),this._content.push(e),this)}thinking(e){return e==null?this:(this.init(),this._content.push(N.of(e)),this)}content(e){return e==null?this:(this.init(),this._content.push(C.of(e)),this)}build(){if(this._content==null||this._content.length==0)throw new Error("Role-only messages are not supported by Tapeworm.");return new u(this._role,this._content)}}const l={Content:"content",Thinking:"thinking",ToolCall:"toolcall",ToolResult:"toolresult"};class m{getMessageComponentType(){throw new Error("Message components that do not have a message component type are not allowed.")}}class C extends m{text;constructor(e){super(),this.text=e}getMessageComponentType(){return l.Content}get(){return this.text}static of(e){return new this(e)}}class N extends m{thought;constructor(e){super(),this.thought=e}getMessageComponentType(){return l.Thinking}get(){return this.thought}static of(e){return new this(e)}}class c extends m{id;toolName;toolResult;constructor(e,s,o){super(),this.id=e,this.toolName=s,this.toolResult=o}getMessageComponentType(){return l.ToolResult}static of(e,s){return new this(e.id,e.name,s)}}class P{async invoke(e){throw new v("The invoke function for this model was not correctly implemented.")}tokenLimit(){throw new v("The tokenLimit function for this model was not correctly implemented.")}}class q{messages;tools;constructor(e,s){this.messages=e,this.tools=s}static builder(){return new f}}class f{_messages;_tools;messages(e){return this._messages=e,this}tools(e){return this._tools=e,this}build(){if(this._tools==null&&(this._tools=[]),this._messages==null)throw new j("Requests to the model should include content.");return new q(this._messages,this._tools)}}class v extends Error{constructor(e){super(e),this.name="ModelNotImplementedError"}}class j extends Error{constructor(e){super(e),this.name="MessagesNotDefinedError"}}class _ extends m{sequence;name;parameters;type;id;getMessageComponentType(){return l.ToolCall}constructor(e,s,o,r,n){super(),this.sequence=e,this.name=s,this.parameters=o,this.type=r,this.id=n}static builder(){return new x}}class x{_sequence;_name;_parameters;_type;_id;sequence(e){return this._sequence=e,this}name(e){return this._name=e,this}parameters(e){return this._parameters=e,this}type(e){return this._type=e,this}id(e){return e!=null&&(this._id=e),this}build(){return this._sequence==null&&(this._sequence=0),this._id==null&&(this._id=(Math.random()+1).toString(36).slice(2,7)),new _(this._sequence,this._name,this._parameters,this._type,this._id)}}class D extends Error{constructor(e){super(e),this.name="ToolNotFoundError"}}class k{name;systemPrompt;tools;model;conversation;conversationManager;toolNameToIndexMap;callback;constructor(e,s,o,r,n){this.name=e,this.model=o,this.conversationManager=r,this.tools=s,this.callback=n,this.conversationManager.configure(o)}async invoke(e,s=this.callback){this.conversation==null&&(this.conversation=new w,this.conversationManager!=null&&(this.conversation.manager=this.conversationManager),this.systemPrompt!=null&&this.conversation.append(u.builder().role("system").content(this.systemPrompt).build())),this.conversation.append(u.builder().role("user").content(e).build());let o=!1;for(;!o;){let r=await this._runQuery();s(r),this.conversation.append(r),o=!0;const n=r.filter(l.ToolCall);if(n!=null&&n.length!=0){o=!1,n.sort((a,i)=>(a.sequence??0)<(i.sequence??0)?-1:1);for(let a of n)await this._runTool(a)}}}async _runQuery(){return await this.model.invoke(new f().messages(this.conversation?.messages).tools(this.tools).build())}async _runTool(e){if(this.generateToolNameToIndexMap(),!(e.name in this.toolNameToIndexMap)){this.conversation.append(u.builder().role("tool").toolResult(c.of(e,new D("Agent does not have a tool with this name."))).build());return}let s=this.tools[this.toolNameToIndexMap[e.name]];try{let o=await s.execute(e.parameters);this.conversation.append(u.builder().role("tool").toolResult(c.of(e,o)).build())}catch(o){this.conversation.append(u.builder().role("tool").toolResult(c.of(e,JSON.stringify(o))).build())}}generateToolNameToIndexMap(){if(this.toolNameToIndexMap==null){this.toolNameToIndexMap={};for(let e=0;e<this.tools.length;e++)this.toolNameToIndexMap[this.tools[e].getName()]=e}}static builder(){return new A}}class A{_name;_systemPrompt;_tools;_model;_conversation;_conversationManager=new g;_toolNameToIndexMap;_callback=e=>J(e);name(e){return this._name=e,this}systemPrompt(e){return this._systemPrompt=e,this}tools(e){return this._tools=e,this}addTool(e){return this._tools==null&&(this._tools=[]),this._tools.push(e),this}model(e){return this._model=e,this}conversationManager(e){return this._conversationManager=e,this}callback(e){return this._callback=e,this}build(){let e=new k(this._name,this._tools,this._model,this._conversationManager,this._callback);return this._conversation!=null&&(e.conversation=this._conversation),this._systemPrompt!=null&&(e.systemPrompt=this._systemPrompt),e}}function J(t){for(const e of t.filter(l.Thinking))console.log("\x1B[90m"+e.get()+"\x1B[0m");for(const e of t.filter(l.Content))console.log(e.get());for(const e of t.filter(l.ToolCall))console.log("\x1B[32mCalling Tool: "+e.name+"\x1B[0m")}class d extends Error{constructor(e){super(e),this.name="ToolNotDefinedError"}}class F{getName(){throw new d("Tool name not defined.")}getDescription(){throw new d("Tool description not defined.")}getToolSchema(){throw new d("Tool parameter schema not defined.")}execute(e){return null}}class p{parameters;output;constructor(e,s){this.parameters=e,this.output=s}static builder(){return new O}}class O{_parameters;_output;addParameter(e){return this._parameters==null&&(this._parameters=[]),this._parameters.push(e),this}output(e){return this._output=e,this}build(){return this._parameters==null&&(this._parameters=[]),new p(this._parameters,this._output)}}class y{name;description;type;required;constructor(e,s,o,r){this.name=e,this.description=s,this.type=o,this.required=r,this.assertValidType()}assertValidType(){}static builder(){return new R}}class R{_name;_description;_type;_required;name(e){return this._name=e,this}description(e){return this._description=e,this}type(e){return this._type=e,this}required(e){return this._required=e,this}build(){return this._required==null&&(this._required=!1),new y(this._name,this._description,this._type,this._required)}}class L extends P{endpoint;model;options;constructor(e,s,o){super(),this.endpoint=e,this.model=s,this.options=o}async invoke(e){let s={model:this.model,messages:this._formatMessages(e),tools:this._formatTools(e),...this.options},o=await fetch(this.endpoint+"/api/chat",{method:"POST",headers:{"Content-Type":"application/json"},body:JSON.stringify(s)});if(!o.ok)throw new Error(`HTTP error! status: ${o.status}`);let r=await o.json(),n=[];if(r.message.tool_calls)for(let a in r.message.tool_calls){const i=r.message.tool_calls[a];let h="function",S=i[h]?.name,E=i[h]?.index,B=i.id??void 0,I=typeof i[h]?.arguments=="string"?JSON.parse(i[h]?.arguments):i[h]?.arguments;n.push(_.builder().name(S).type(h).parameters(I).sequence(E).id(B).build())}return u.builder().toolCalls(n).role(r.message.role).content(r.message.content).thinking(r.message.thinking).build()}_formatTools(e){let s=[];for(let o of e.tools){let r={},n=[];for(let i of o.getToolSchema().parameters)r[i.name]={},r[i.name].type=i.type,r[i.name].description=i.description,i.required&&n.push(i.name);let a={type:"function",function:{name:o.getName(),description:o.getDescription(),parameters:{type:"object",properties:r,required:n}}};s.push(a)}return s}_formatMessages(e){let s=[];for(let o of e.messages){if(o.role=="assistant"||o.role=="system"||o.role=="user"){s.push(this._formatSingleMessage(o));continue}if(o.role=="tool"){for(const r of o.content)if(r.getMessageComponentType()==l.ToolResult){const n=r;s.push({role:o.role,name:n.toolName,content:JSON.stringify(n.toolResult)})}}}return s}_formatSingleMessage(e){let s={role:e.role},o,r,n;for(const a of e.content){if(a.getMessageComponentType()==l.Content){const i=a;o==null&&(o=""),o+=i.get()}if(a.getMessageComponentType()==l.Thinking){const i=a;n==null&&(n=""),n+=i.get()}if(a.getMessageComponentType()==l.ToolCall){const i=a;r==null&&(r=[]),r.push(this._formatToolCall(i))}}return o!=null&&(s.content=o),r!=null&&(s.tool_calls=r),n!=null&&(s.thinking=n),s}_formatToolCall(e){return{function:{name:e.name,arguments:e.parameters}}}}function Q(t){return e=>{e.prototype.getName=()=>t}}function V(t){return e=>{Object.defineProperty(e.prototype,"getDescription",{value:function(){return t},enumerable:!0,configurable:!0,writable:!0})}}function H(t){return e=>{e.prototype.tapewormParams==null&&(e.prototype.tapewormParams=[]),e.prototype.tapewormParamsOutput==null&&(e.prototype.tapewormParamsOutput=""),e.prototype.tapewormParams.push(y.builder().name(t.name).description(t.description).required(t.required).type(t.type).build()),e.prototype.getToolSchema=()=>new p(e.prototype.tapewormParams,e.prototype.tapewormParamsOutput)}}function K(t){return e=>{e.prototype.tapewormParams==null&&(e.prototype.tapewormParams=[]),e.prototype.tapewormParamsOutput==null&&(e.prototype.tapewormParamsOutput=""),e.prototype.tapewormParamsOutput=t,e.prototype.getToolSchema=()=>new p(e.prototype.tapewormParams,e.prototype.tapewormParamsOutput)}}exports.Agent=k;exports.Content=C;exports.Conversation=w;exports.ConversationManager=M;exports.DefaultConversationManager=g;exports.Message=u;exports.MessageBuilder=b;exports.MessageComponent=m;exports.MessageComponentType=l;exports.Model=P;exports.ModelRequest=q;exports.ModelRequestBuilder=f;exports.OllamaModel=L;exports.Parameter=y;exports.ParameterBuilder=R;exports.Thinking=N;exports.Tool=F;exports.ToolCall=_;exports.ToolCallBuilder=x;exports.ToolDescription=V;exports.ToolName=Q;exports.ToolOutput=K;exports.ToolParameter=H;exports.ToolResult=c;exports.ToolSchema=p;exports.ToolSchemaBuilder=O;