@atgs/tapeworm 0.1.6 → 0.1.8

package/dist/tapeworm.es.js

@@ -54,7 +54,7 @@ class g extends Error {
     super(e), this.name = "ConversationManagerNotImplemented";
   }
 }
-class h {
+class u {
   role;
   content;
   /**
@@ -62,8 +62,8 @@ class h {
    * @param role Source of the message (e.g., user, assistant, tool).
    * @param content Ordered list of message components that make up the message body.
    */
-  constructor(e, o) {
-    this.role = e, this.content = o;
+  constructor(e, s) {
+    this.role = e, this.content = s;
   }
   /**
    * Create a new message builder for ergonomic construction.
@@ -76,7 +76,7 @@ class h {
    * Filters the message contents based on the message type.
    */
   filter(e) {
-    return this.content.filter((o) => o.getMessageComponentType() == e);
+    return this.content.filter((s) => s.getMessageComponentType() == e);
   }
 }
 class k {
@@ -113,8 +113,8 @@ class k {
     if (e == null)
       return this;
     this.init();
-    for (const o of e)
-      this._content.push(o);
+    for (const s of e)
+      this._content.push(s);
     return this;
   }
   /**
@@ -148,7 +148,7 @@ class k {
   build() {
     if (this._content == null || this._content.length == 0)
       throw new Error("Role-only messages are not supported by Tapeworm.");
-    return new h(this._role, this._content);
+    return new u(this._role, this._content);
   }
 }
 const l = {
@@ -241,8 +241,8 @@ class m extends c {
    * @param toolName Name of the tool that produced the result.
    * @param toolResult Output returned by the tool.
    */
-  constructor(e, o, s) {
-    super(), this.id = e, this.toolName = o, this.toolResult = s;
+  constructor(e, s, o) {
+    super(), this.id = e, this.toolName = s, this.toolResult = o;
   }
   /**
    * Get the type of this message component, as defined in MessageComponentType.
@@ -257,8 +257,8 @@ class m extends c {
    * @param toolResult Output to attach to the tool result component.
    * @returns a new ToolResult component mirroring the provided tool call.
    */
-  static of(e, o) {
-    return new this(e.id, e.name, o);
+  static of(e, s) {
+    return new this(e.id, e.name, s);
   }
 }
 class O {
@@ -288,8 +288,8 @@ class E {
    * @param messages Conversation history to send to the model.
    * @param tools Tool definitions available for function calling.
    */
-  constructor(e, o) {
-    this.messages = e, this.tools = o;
+  constructor(e, s) {
+    this.messages = e, this.tools = s;
   }
   /**
    * Create a builder for composing a model request.
@@ -359,8 +359,8 @@ class v extends c {
    * @param type Type of tool call (e.g., function).
    * @param id Provider-generated identifier for correlating results.
    */
-  constructor(e, o, s, n, r) {
-    super(), this.sequence = e, this.name = o, this.parameters = s, this.type = n, this.id = r;
+  constructor(e, s, o, r, n) {
+    super(), this.sequence = e, this.name = s, this.parameters = o, this.type = r, this.id = n;
   }
   /**
    * Convenience factory for a ToolCallBuilder.
@@ -436,39 +436,80 @@ class I extends Error {
   }
 }
 class j {
+  /**
+   * The name of the agent.
+   */
   name;
+  /**
+   * The system prompt (if any) for the agent.
+   */
   systemPrompt;
+  /**
+   * The tools this agent knows about and can run.
+   */
   tools;
+  /**
+   * The underlying model for the agent.
+   */
   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;
+  /**
+   * 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;
+  /**
+   * A cache for looking up tools in the Tools array.
+   * DO NOT MANUALLY SET THIS VALUE. AGENT AUTOMATICALLY TAKES CARE OF IT.
+   */
   toolNameToIndexMap;
+  /**
+   * The callback function, invoked for every response from the model.
+   * The default implementation just prints to std.out or your console.
+   */
   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);
+  /**
+   * 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(e, s, o, r, n) {
+    this.name = e, this.model = o, this.conversationManager = r, this.tools = s, this.callback = n, this.conversationManager.configure(o);
   }
   /**
    * 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.
    */
-  async invoke(e) {
+  async invoke(e, s = this.callback) {
     this.conversation == null && (this.conversation = new x(), this.conversationManager != null && (this.conversation.manager = this.conversationManager), this.systemPrompt != null && this.conversation.append(
-      h.builder().role("system").content(this.systemPrompt).build()
+      u.builder().role("system").content(this.systemPrompt).build()
     )), this.conversation.append(
-      h.builder().role("user").content(e).build()
+      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(
+      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((r, a) => (r.sequence ?? 0) < (a.sequence ?? 0) ? -1 : 1);
-        for (let r of n)
-          await this._runTool(r);
+        o = !1, n.sort((a, i) => (a.sequence ?? 0) < (i.sequence ?? 0) ? -1 : 1);
+        for (let a of n)
+          await this._runTool(a);
       }
     }
   }
@@ -489,7 +530,7 @@ class j {
   async _runTool(e) {
     if (this.generateToolNameToIndexMap(), !(e.name in this.toolNameToIndexMap)) {
       this.conversation.append(
-        h.builder().role("tool").toolResult(
+        u.builder().role("tool").toolResult(
           m.of(
             e,
             new I(
@@ -500,15 +541,15 @@ class j {
       );
       return;
     }
-    let o = this.tools[this.toolNameToIndexMap[e.name]];
+    let s = this.tools[this.toolNameToIndexMap[e.name]];
     try {
-      let s = await o.execute(e.parameters);
+      let o = await s.execute(e.parameters);
       this.conversation.append(
-        h.builder().role("tool").toolResult(m.of(e, s)).build()
+        u.builder().role("tool").toolResult(m.of(e, o)).build()
       );
-    } catch (s) {
+    } catch (o) {
       this.conversation.append(
-        h.builder().role("tool").toolResult(m.of(e, JSON.stringify(s))).build()
+        u.builder().role("tool").toolResult(m.of(e, JSON.stringify(o))).build()
       );
     }
   }
@@ -522,6 +563,9 @@ class j {
         this.toolNameToIndexMap[this.tools[e].getName()] = e;
     }
   }
+  /**
+   * Get a builder to build an Agent.
+   */
   static builder() {
     return new B();
   }
@@ -535,27 +579,51 @@ class B {
   _conversationManager = new _();
   _toolNameToIndexMap;
   _callback = (e) => D(e);
+  /**
+   * Set the name of the agent.
+   */
   name(e) {
     return this._name = e, this;
   }
+  /**
+   * Set the system prompt of the agent.
+   */
   systemPrompt(e) {
     return this._systemPrompt = e, this;
   }
+  /**
+   * Set the tools of the agent, if you already have an array of tools.
+   */
   tools(e) {
     return this._tools = e, this;
   }
+  /**
+   * Add a tool, 1 by 1, to the agent..
+   */
   addTool(e) {
     return this._tools == null && (this._tools = []), this._tools.push(e), this;
   }
+  /**
+   * Set the underlying model for the agent.
+   */
   model(e) {
     return this._model = e, this;
   }
+  /**
+   * Set the conversation manager for the agent.
+   */
   conversationManager(e) {
     return this._conversationManager = e, this;
   }
+  /**
+   * Set the callback for the agent.
+   */
   callback(e) {
     return this._callback = e, this;
   }
+  /**
+   * Build the agent.
+   */
   build() {
     let e = new j(
       this._name,
@@ -581,15 +649,6 @@ class p extends Error {
   }
 }
 class F {
-  name;
-  description;
-  tool_schema;
-  /**
-   * Initialize tool metadata from subclass implementations.
-   */
-  constructor() {
-    this.name = this.getName(), this.description = this.getDescription(), this.tool_schema = this.getToolSchema();
-  }
   /**
    * Name that uniquely identifies this tool.
    * @returns A short, stable identifier string.
@@ -629,8 +688,8 @@ class d {
    * @param parameters Ordered list of input parameters the tool accepts.
    * @param output Human-readable description of the tool output.
    */
-  constructor(e, o) {
-    this.parameters = e, this.output = o;
+  constructor(e, s) {
+    this.parameters = e, this.output = s;
   }
   /**
    * Return a tool schema builder, so you can properly construct tools instead of dealing with the constructor
@@ -679,8 +738,8 @@ class T {
    * @param type JSON-serializable type name (e.g., string, number).
    * @param required Whether the parameter must be provided.
    */
-  constructor(e, o, s, n) {
-    this.name = e, this.description = o, this.type = s, this.required = n, this.assertValidType();
+  constructor(e, s, o, r) {
+    this.name = e, this.description = s, this.type = o, this.required = r, this.assertValidType();
   }
   /**
    * Placeholder for validating supported parameter types.
@@ -751,38 +810,38 @@ class L extends O {
    * @param model Model name/tag to use for inference.
    * @param options Additional Ollama options passed through to the API.
    */
-  constructor(e, o, s) {
-    super(), this.endpoint = e, this.model = o, this.options = s;
+  constructor(e, s, o) {
+    super(), this.endpoint = e, this.model = s, this.options = o;
   }
   /**
    * Call the Ollama chat API with the provided conversation and tools.
    * Formats the request, performs the HTTP POST, and translates the response into a ModelResponse.
    */
   async invoke(e) {
-    let o = {
+    let s = {
       model: this.model,
       messages: this._formatMessages(e),
       tools: this._formatTools(e),
       ...this.options
-    }, s = await fetch(this.endpoint + "/api/chat", {
+    }, o = await fetch(this.endpoint + "/api/chat", {
       method: "POST",
       headers: {
         "Content-Type": "application/json"
       },
-      body: JSON.stringify(o)
+      body: JSON.stringify(s)
     });
-    if (!s.ok)
-      throw new Error(`HTTP error! status: ${s.status}`);
-    let n = await s.json(), r = [];
-    if (n.message.tool_calls)
-      for (let a in n.message.tool_calls) {
-        const i = n.message.tool_calls[a];
-        let u = "function", w = i[u]?.name, b = i[u]?.index, M = i.id ?? void 0, N = typeof i[u]?.arguments == "string" ? JSON.parse(i[u]?.arguments) : i[u]?.arguments;
-        r.push(
-          v.builder().name(w).type(u).parameters(N).sequence(b).id(M).build()
+    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", w = i[h]?.name, b = i[h]?.index, M = i.id ?? void 0, N = typeof i[h]?.arguments == "string" ? JSON.parse(i[h]?.arguments) : i[h]?.arguments;
+        n.push(
+          v.builder().name(w).type(h).parameters(N).sequence(b).id(M).build()
         );
       }
-    return h.builder().toolCalls(r).role(n.message.role).content(n.message.content).thinking(n.message.thinking).build();
+    return u.builder().toolCalls(n).role(r.message.role).content(r.message.content).thinking(r.message.thinking).build();
   }
   /**
    * Convert internal tool definitions into the JSON schema format expected by Ollama.
@@ -790,26 +849,26 @@ class L extends O {
    * @returns Array of tool schema objects formatted for the Ollama API.
    */
   _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 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: s.getName(),
-          description: s.getDescription(),
+          name: o.getName(),
+          description: o.getDescription(),
           parameters: {
             type: "object",
-            properties: n,
-            required: r
+            properties: r,
+            required: n
           }
         }
       };
-      o.push(a);
+      s.push(a);
     }
-    return o;
+    return s;
   }
   /**
    * Convert internal message objects into Ollama's chat message shape.
@@ -818,25 +877,25 @@ class L extends O {
    * @returns Array of serialized messages ready for Ollama.
    */
   _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));
+    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 (s.role == "tool") {
-        for (const n of s.content)
-          if (n.getMessageComponentType() == l.ToolResult) {
-            const r = n;
-            o.push({
-              role: s.role,
-              name: r.toolName,
-              content: JSON.stringify(r.toolResult)
+      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 o;
+    return s;
   }
   /**
    * Format everything sent by a user or LLM (no tools)
@@ -844,24 +903,24 @@ class L extends O {
    * @returns Serialized message payload.
    */
   _formatSingleMessage(e) {
-    let o = {
+    let s = {
       role: e.role
-    }, s, n, r;
+    }, o, r, n;
     for (const a of e.content) {
       if (a.getMessageComponentType() == l.Content) {
         const i = a;
-        s == null && (s = ""), s += i.get();
+        o == null && (o = ""), o += i.get();
       }
       if (a.getMessageComponentType() == l.Thinking) {
         const i = a;
-        r == null && (r = ""), r += i.get();
+        n == null && (n = ""), n += i.get();
       }
       if (a.getMessageComponentType() == l.ToolCall) {
         const i = a;
-        n == null && (n = []), n.push(this._formatToolCall(i));
+        r == null && (r = []), r.push(this._formatToolCall(i));
       }
     }
-    return s != null && (o.content = s), n != null && (o.tool_calls = n), r != null && (o.thinking = r), o;
+    return o != null && (s.content = o), r != null && (s.tool_calls = r), n != null && (s.thinking = n), s;
   }
   /**
    * Convert a ToolCall into the structure Ollama expects.
@@ -918,7 +977,7 @@ export {
   x as Conversation,
   P as ConversationManager,
   _ as DefaultConversationManager,
-  h as Message,
+  u as Message,
   k as MessageBuilder,
   c as MessageComponent,
   l as MessageComponentType,

package/dist/tapeworm.umd.js

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

package/dist/tool/tool.d.ts

@@ -1,16 +1,13 @@
+/**
+ * The Tool module contains the Tool class, which shows
+ * what properties are required by a class to be invocable by Tapeworm.
+ */
 import type ToolSchema from "./toolschema";
 /**
  * The Tool class represents a tool that a Tapeworm agent can use.
  *
  */
 export default class Tool {
-    private name;
-    private description;
-    private tool_schema;
-    /**
-     * Initialize tool metadata from subclass implementations.
-     */
-    constructor();
     /**
      * Name that uniquely identifies this tool.
      * @returns A short, stable identifier string.

package/dist/tool/toolCall.d.ts

@@ -1,3 +1,11 @@
+/**
+ * This module is dedicated to just the ToolCall MessageComponent.
+ *
+ * The reason is that it's the most complicated MessageComponent and deserves to have its own file,
+ * as this class contains more ways to instantiate than normal MessageComponents.
+ *
+ * @module
+ */
 import { MessageComponent, MessageComponentType } from "../conversation/message";
 /**
  * Represents a single tool invocation returned by a model, including the target

package/dist/tool/toolDecorators.d.ts

@@ -3,7 +3,7 @@
  *
  * They are different from the Babel decorators in @atgs/babel-plugin-tapeworm-decorator, but they achieve the same functionality.
  *
- *
+ * @module
  */
 export declare function ToolName(name: String): (toolClass: Function) => void;
 export declare function ToolDescription(description: String): (toolClass: Function) => void;

package/dist/tool/toolschema.d.ts

@@ -1,5 +1,17 @@
 /**
- * Tool schemas specify the input and output of a tool.
+ * This module contains everything related to tool schemas.
+ *
+ * Enclosed are the ToolSchema (and builder) and Parameter (and builder) classes.
+ *
+ * @module
+ */
+/**
+ * ToolSchemas define the input and output of a tool.
+ *
+ * They consist of a list of parameters and a description of the output of the tool.
+ *
+ * You can build these using ToolSchemaBuilder, but it can be more pragmatic to use
+ * the Typescript or Babel decorators instead, as they will automatically generate a ToolSchema for you.
  */
 export default class ToolSchema {
     parameters: Parameter[];

package/docs/.nojekyll

@@ -0,0 +1 @@
+TypeDoc added this file to prevent GitHub Pages from using Jekyll. You can turn off this behavior by setting the `githubPages` option to false.

package/docs/assets/hierarchy.js

@@ -0,0 +1 @@
+window.hierarchyData = "eJytk81ugzAQhN9lzw6NwSDwlVyjSlVvURRZsElQFrvCTi8R7145P6oTtQGqXiwDM/uNd80JOmOcBbkSGSsKxkW2ZtDhlrByjdEW5AlE5letWgQJpdGf2Fnlvy6VVjvsgMGh0TVIHucMjh2BhIqUtWhfqkC+CR+iHwpFe9cSsIsZJDhbz3zl2eUFg2rfUN2hBrlK03XPIE2DaAvcqiO5f0v4e72BoD2DogiCLdFatcPStB9Go3aj47QXY/RYYEqj+FwwzjPmWUke+67xubifqPtLqKtvuBechxfofd/oQ6N3k3k34whgnIdAY+gN7ZGmH/HbOgJ695ssTY30jNd6wea8RmfxlJkmQvg5JiKc4yuRatVIcCCOwv3gMZM8fuhtqegp0BlDG3cVRjfHEKnvvwBqqo+f"