@atgs/tapeworm 0.1.4 → 0.1.6

package/dist/index.d.ts

@@ -6,3 +6,4 @@ export { default as ToolCall, ToolCallBuilder } from "./tool/toolCall";
 export { default as ToolSchema, ToolSchemaBuilder, Parameter, ParameterBuilder, } from "./tool/toolschema";
 export { default as OllamaModel } from "./model/OllamaModel";
 export { default as Message, MessageBuilder, MessageComponentType, MessageComponent, Content, Thinking, ToolResult, } from "./conversation/message";
+export { ToolName, ToolDescription, ToolParameter, ToolOutput, } from "./tool/toolDecorators";

package/dist/tapeworm.cjs.js

@@ -1 +1 @@
-"use strict";Object.defineProperty(exports,Symbol.toStringTag,{value:"Module"});class v{messages;manager;constructor(){this.messages=[],this.manager=new p}append(e){this.messages.push(e),this.messages=this.manager.compact(this.messages)}}class y{compact(e){throw new _("No implementation for conversation manager!")}configure(e){throw new _("No implementation for conversation manager!")}}class p extends y{compact(e){return e}configure(e){}}class _ extends Error{constructor(e){super(e),this.name="ConversationManagerNotImplemented"}}class h{role;content;constructor(e,t){this.role=e,this.content=t}static builder(){return new M}filter(e){return this.content.filter(t=>t.getMessageComponentType()==e)}}class M{_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 t of e)this._content.push(t);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(w.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 a={Content:"content",Thinking:"thinking",ToolCall:"toolcall",ToolResult:"toolresult"};class u{getMessageComponentType(){throw new Error("Message components that do not have a message component type are not allowed.")}}class w extends u{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 u{thought;constructor(e){super(),this.thought=e}getMessageComponentType(){return a.Thinking}get(){return this.thought}static of(e){return new this(e)}}class m extends u{id;toolName;toolResult;constructor(e,t,o){super(),this.id=e,this.toolName=t,this.toolResult=o}getMessageComponentType(){return a.ToolResult}static of(e,t){return new this(e.id,e.name,t)}}class b{async invoke(e){throw new T("The invoke function for this model was not correctly implemented.")}tokenLimit(){throw new T("The tokenLimit function for this model was not correctly implemented.")}}class N{messages;tools;constructor(e,t){this.messages=e,this.tools=t}static builder(){return new g}}class g{_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 N(this._messages,this._tools)}}class T extends Error{constructor(e){super(e),this.name="ModelNotImplementedError"}}class j extends Error{constructor(e){super(e),this.name="MessagesNotDefinedError"}}class f extends u{sequence;name;parameters;type;id;getMessageComponentType(){return a.ToolCall}constructor(e,t,o,s,i){super(),this.sequence=e,this.name=t,this.parameters=o,this.type=s,this.id=i}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 f(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,t,o,s,i){this.name=e,this.model=o,this.conversationManager=s,this.tools=t,this.callback=i,this.conversationManager.configure(o)}async invoke(e){this.conversation==null&&(this.conversation=new v,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 t=!1;for(;!t;){let o=await this._runQuery();this.callback(o),this.conversation.append(o),t=!0;const s=o.filter(a.ToolCall);if(s!=null&&s.length!=0){t=!1,s.sort((i,l)=>(i.sequence??0)<(l.sequence??0)?-1:1);for(let i of s)await this._runTool(i)}}}async _runQuery(){return await this.model.invoke(new g().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(m.of(e,new D("Agent does not have a tool with this name."))).build());return}let t=this.tools[this.toolNameToIndexMap[e.name]];try{let o=await t.execute(e.parameters);this.conversation.append(h.builder().role("tool").toolResult(m.of(e,o)).build())}catch(o){this.conversation.append(h.builder().role("tool").toolResult(m.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 p;_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(n){for(const e of n.filter(a.Thinking))console.log("\x1B[90m"+e.get()+"\x1B[0m");for(const e of n.filter(a.Content))console.log(e.get());for(const e of n.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 q{parameters;output;constructor(e,t){this.parameters=e,this.output=t}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 q(this._parameters,this._output)}}class E{name;description;type;required;constructor(e,t,o,s){this.name=e,this.description=t,this.type=o,this.required=s,this.assertValidType()}assertValidType(){}static builder(){return new P}}class P{_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 E(this._name,this._description,this._type,this._required)}}class L extends b{endpoint;model;options;constructor(e,t,o){super(),this.endpoint=e,this.model=t,this.options=o}async invoke(e){let t={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(t)});if(!o.ok)throw new Error(`HTTP error! status: ${o.status}`);let s=await o.json(),i=[];if(s.message.tool_calls)for(let l in s.message.tool_calls){const r=s.message.tool_calls[l];let c="function",S=r[c]?.name,B=r[c]?.index,O=r.id??void 0,I=typeof r[c]?.arguments=="string"?JSON.parse(r[c]?.arguments):r[c]?.arguments;i.push(f.builder().name(S).type(c).parameters(I).sequence(B).id(O).build())}return h.builder().toolCalls(i).role(s.message.role).content(s.message.content).thinking(s.message.thinking).build()}_formatTools(e){let t=[];for(let o of e.tools){let s={},i=[];for(let r of o.getToolSchema().parameters)s[r.name]={},s[r.name].type=r.type,s[r.name].description=r.description,r.required&&i.push(r.name);let l={type:"function",function:{name:o.getName(),description:o.getDescription(),parameters:{type:"object",properties:s,required:i}}};t.push(l)}return t}_formatMessages(e){let t=[];for(let o of e.messages){if(o.role=="assistant"||o.role=="system"||o.role=="user"){t.push(this._formatSingleMessage(o));continue}if(o.role=="tool"){for(const s of o.content)if(s.getMessageComponentType()==a.ToolResult){const i=s;t.push({role:o.role,name:i.toolName,content:JSON.stringify(i.toolResult)})}}}return t}_formatSingleMessage(e){let t={role:e.role},o,s,i;for(const l of e.content){if(l.getMessageComponentType()==a.Content){const r=l;o==null&&(o=""),o+=r.get()}if(l.getMessageComponentType()==a.Thinking){const r=l;i==null&&(i=""),i+=r.get()}if(l.getMessageComponentType()==a.ToolCall){const r=l;s==null&&(s=[]),s.push(this._formatToolCall(r))}}return o!=null&&(t.content=o),s!=null&&(t.tool_calls=s),i!=null&&(t.thinking=i),t}_formatToolCall(e){return{function:{name:e.name,arguments:e.parameters}}}}exports.Agent=k;exports.Content=w;exports.Conversation=v;exports.ConversationManager=y;exports.DefaultConversationManager=p;exports.Message=h;exports.MessageBuilder=M;exports.MessageComponent=u;exports.MessageComponentType=a;exports.Model=b;exports.ModelRequest=N;exports.ModelRequestBuilder=g;exports.OllamaModel=L;exports.Parameter=E;exports.ParameterBuilder=P;exports.Thinking=C;exports.Tool=F;exports.ToolCall=f;exports.ToolCallBuilder=x;exports.ToolResult=m;exports.ToolSchema=q;exports.ToolSchemaBuilder=R;
+"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;

package/dist/tapeworm.es.js

@@ -1,11 +1,11 @@
-class b {
+class x {
   messages;
   manager;
   /**
    * Initialize an empty conversation using the default manager.
    */
   constructor() {
-    this.messages = [], this.manager = new f();
+    this.messages = [], this.manager = new _();
   }
   /**
    * Append a message to the history and apply compaction rules.
@@ -15,14 +15,14 @@ class b {
     this.messages.push(e), this.messages = this.manager.compact(this.messages);
   }
 }
-class C {
+class P {
   /**
    * Compact or adjust a conversation according to a strategy.
    * @param conversation Full list of messages so far.
    * @returns Adjusted message list to keep.
    */
   compact(e) {
-    throw new p(
+    throw new g(
       "No implementation for conversation manager!"
     );
   }
@@ -31,12 +31,12 @@ class C {
    * @param model Model being used for this conversation.
    */
   configure(e) {
-    throw new p(
+    throw new g(
       "No implementation for conversation manager!"
     );
   }
 }
-class f extends C {
+class _ extends P {
   /**
    * Default strategy simply returns the conversation untouched.
    */
@@ -49,7 +49,7 @@ class f extends C {
   configure(e) {
   }
 }
-class p extends Error {
+class g extends Error {
   constructor(e) {
     super(e), this.name = "ConversationManagerNotImplemented";
   }
@@ -62,24 +62,24 @@ 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, t) {
-    this.role = e, this.content = t;
+  constructor(e, o) {
+    this.role = e, this.content = o;
   }
   /**
    * Create a new message builder for ergonomic construction.
    * @returns A MessageBuilder instance.
    */
   static builder() {
-    return new N();
+    return new k();
   }
   /**
    * Filters the message contents based on the message type.
    */
   filter(e) {
-    return this.content.filter((t) => t.getMessageComponentType() == e);
+    return this.content.filter((o) => o.getMessageComponentType() == e);
   }
 }
-class N {
+class k {
   _role;
   _content;
   /**
@@ -113,8 +113,8 @@ class N {
     if (e == null)
       return this;
     this.init();
-    for (const t of e)
-      this._content.push(t);
+    for (const o of e)
+      this._content.push(o);
     return this;
   }
   /**
@@ -131,7 +131,7 @@ class N {
    * @returns This builder for chaining.
    */
   thinking(e) {
-    return e == null ? this : (this.init(), this._content.push(k.of(e)), this);
+    return e == null ? this : (this.init(), this._content.push(C.of(e)), this);
   }
   /**
    * Attach content to the message if provided.
@@ -139,7 +139,7 @@ class N {
    * @returns This builder for chaining.
    */
   content(e) {
-    return e == null ? this : (this.init(), this._content.push(x.of(e)), this);
+    return e == null ? this : (this.init(), this._content.push(q.of(e)), this);
   }
   /**
    * Construct the Message with the accumulated values.
@@ -157,7 +157,7 @@ const l = {
   ToolCall: "toolcall",
   ToolResult: "toolresult"
 };
-class u {
+class c {
   /**
    * Get the type of this message component, as defined in MessageComponentType
    */
@@ -167,7 +167,7 @@ class u {
     );
   }
 }
-class x extends u {
+class q extends c {
   text;
   /**
    * Create a content component with the provided text.
@@ -199,7 +199,7 @@ class x extends u {
     return new this(e);
   }
 }
-class k extends u {
+class C extends c {
   thought;
   /**
    * Create a thinking component with the provided thought text.
@@ -231,7 +231,7 @@ class k extends u {
     return new this(e);
   }
 }
-class m extends u {
+class m extends c {
   id;
   toolName;
   toolResult;
@@ -241,8 +241,8 @@ class m extends u {
    * @param toolName Name of the tool that produced the result.
    * @param toolResult Output returned by the tool.
    */
-  constructor(e, t, o) {
-    super(), this.id = e, this.toolName = t, this.toolResult = o;
+  constructor(e, o, s) {
+    super(), this.id = e, this.toolName = o, this.toolResult = s;
   }
   /**
    * Get the type of this message component, as defined in MessageComponentType.
@@ -257,17 +257,17 @@ class m extends u {
    * @param toolResult Output to attach to the tool result component.
    * @returns a new ToolResult component mirroring the provided tool call.
    */
-  static of(e, t) {
-    return new this(e.id, e.name, t);
+  static of(e, o) {
+    return new this(e.id, e.name, o);
   }
 }
-class q {
+class O {
   /**
    * Execute the model against the provided request.
    * @param request Aggregated messages and tools to send to the model.
    */
   async invoke(e) {
-    throw new g(
+    throw new f(
       "The invoke function for this model was not correctly implemented."
     );
   }
@@ -275,7 +275,7 @@ class q {
    * Return the maximum token budget supported by this model.
    */
   tokenLimit() {
-    throw new g(
+    throw new f(
       "The tokenLimit function for this model was not correctly implemented."
     );
   }
@@ -288,18 +288,18 @@ class E {
    * @param messages Conversation history to send to the model.
    * @param tools Tool definitions available for function calling.
    */
-  constructor(e, t) {
-    this.messages = e, this.tools = t;
+  constructor(e, o) {
+    this.messages = e, this.tools = o;
   }
   /**
    * Create a builder for composing a model request.
    * @returns A ModelRequestBuilder instance.
    */
   static builder() {
-    return new _();
+    return new y();
   }
 }
-class _ {
+class y {
   _messages;
   _tools;
   /**
@@ -328,7 +328,7 @@ class _ {
     return new E(this._messages, this._tools);
   }
 }
-class g extends Error {
+class f extends Error {
   constructor(e) {
     super(e), this.name = "ModelNotImplementedError";
   }
@@ -338,7 +338,7 @@ class R extends Error {
     super(e), this.name = "MessagesNotDefinedError";
   }
 }
-class v extends u {
+class v extends c {
   sequence;
   name;
   parameters;
@@ -359,18 +359,18 @@ class v extends u {
    * @param type Type of tool call (e.g., function).
    * @param id Provider-generated identifier for correlating results.
    */
-  constructor(e, t, o, s, i) {
-    super(), this.sequence = e, this.name = t, this.parameters = o, this.type = s, this.id = i;
+  constructor(e, o, s, n, r) {
+    super(), this.sequence = e, this.name = o, this.parameters = s, this.type = n, this.id = r;
   }
   /**
    * Convenience factory for a ToolCallBuilder.
    * @returns A new ToolCallBuilder instance.
    */
   static builder() {
-    return new P();
+    return new S();
   }
 }
-class P {
+class S {
   _sequence;
   _name;
   _parameters;
@@ -435,7 +435,7 @@ class I extends Error {
     super(e), this.name = "ToolNotFoundError";
   }
 }
-class O {
+class j {
   name;
   systemPrompt;
   tools;
@@ -444,8 +444,8 @@ class O {
   conversationManager;
   toolNameToIndexMap;
   callback;
-  constructor(e, t, o, s, i) {
-    this.name = e, this.model = o, this.conversationManager = s, this.tools = t, this.callback = i, this.conversationManager.configure(o);
+  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);
   }
   /**
    * Run the full agent loop for a user query: seed the conversation, invoke the model,
@@ -453,22 +453,22 @@ class O {
    * @param query User-provided input to hand to the agent.
    */
   async invoke(e) {
-    this.conversation == null && (this.conversation = new b(), this.conversationManager != null && (this.conversation.manager = this.conversationManager), this.systemPrompt != null && this.conversation.append(
+    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()
     )), this.conversation.append(
       h.builder().role("user").content(e).build()
     );
-    let t = !1;
-    for (; !t; ) {
-      let o = await this._runQuery();
-      this.callback(o), this.conversation.append(o), t = !0;
-      const s = o.filter(
+    let o = !1;
+    for (; !o; ) {
+      let s = await this._runQuery();
+      this.callback(s), this.conversation.append(s), o = !0;
+      const n = s.filter(
         l.ToolCall
       );
-      if (s != null && s.length != 0) {
-        t = !1, s.sort((i, a) => (i.sequence ?? 0) < (a.sequence ?? 0) ? -1 : 1);
-        for (let i of s)
-          await this._runTool(i);
+      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);
       }
     }
   }
@@ -478,7 +478,7 @@ class O {
    */
   async _runQuery() {
     return await this.model.invoke(
-      new _().messages(this.conversation?.messages).tools(this.tools).build()
+      new y().messages(this.conversation?.messages).tools(this.tools).build()
     );
   }
   /**
@@ -500,15 +500,15 @@ class O {
       );
       return;
     }
-    let t = this.tools[this.toolNameToIndexMap[e.name]];
+    let o = this.tools[this.toolNameToIndexMap[e.name]];
     try {
-      let o = await t.execute(e.parameters);
+      let s = await o.execute(e.parameters);
       this.conversation.append(
-        h.builder().role("tool").toolResult(m.of(e, o)).build()
+        h.builder().role("tool").toolResult(m.of(e, s)).build()
       );
-    } catch (o) {
+    } catch (s) {
       this.conversation.append(
-        h.builder().role("tool").toolResult(m.of(e, JSON.stringify(o))).build()
+        h.builder().role("tool").toolResult(m.of(e, JSON.stringify(s))).build()
       );
     }
   }
@@ -523,18 +523,18 @@ class O {
     }
   }
   static builder() {
-    return new S();
+    return new B();
   }
 }
-class S {
+class B {
   _name;
   _systemPrompt;
   _tools;
   _model;
   _conversation;
-  _conversationManager = new f();
+  _conversationManager = new _();
   _toolNameToIndexMap;
-  _callback = (e) => j(e);
+  _callback = (e) => D(e);
   name(e) {
     return this._name = e, this;
   }
@@ -557,7 +557,7 @@ class S {
     return this._callback = e, this;
   }
   build() {
-    let e = new O(
+    let e = new j(
       this._name,
       this._tools,
       this._model,
@@ -567,15 +567,15 @@ class S {
     return this._conversation != null && (e.conversation = this._conversation), this._systemPrompt != null && (e.systemPrompt = this._systemPrompt), e;
   }
 }
-function j(n) {
-  for (const e of n.filter(l.Thinking))
+function D(t) {
+  for (const e of t.filter(l.Thinking))
     console.log("\x1B[90m" + e.get() + "\x1B[0m");
-  for (const e of n.filter(l.Content))
+  for (const e of t.filter(l.Content))
     console.log(e.get());
-  for (const e of n.filter(l.ToolCall))
+  for (const e of t.filter(l.ToolCall))
     console.log("\x1B[32mCalling Tool: " + e.name + "\x1B[0m");
 }
-class d extends Error {
+class p extends Error {
   constructor(e) {
     super(e), this.name = "ToolNotDefinedError";
   }
@@ -595,21 +595,21 @@ class F {
    * @returns A short, stable identifier string.
    */
   getName() {
-    throw new d("Tool name not defined.");
+    throw new p("Tool name not defined.");
   }
   /**
    * Human-readable description of what the tool does.
    * @returns Description string shown to the model.
    */
   getDescription() {
-    throw new d("Tool description not defined.");
+    throw new p("Tool description not defined.");
   }
   /**
    * Structured schema describing tool parameters and output.
    * @returns ToolSchema instance.
    */
   getToolSchema() {
-    throw new d("Tool parameter schema not defined.");
+    throw new p("Tool parameter schema not defined.");
   }
   // @ts-ignore
   /**
@@ -621,7 +621,7 @@ class F {
     return null;
   }
 }
-class B {
+class d {
   parameters;
   output;
   /**
@@ -629,18 +629,18 @@ class B {
    * @param parameters Ordered list of input parameters the tool accepts.
    * @param output Human-readable description of the tool output.
    */
-  constructor(e, t) {
-    this.parameters = e, this.output = t;
+  constructor(e, o) {
+    this.parameters = e, this.output = o;
   }
   /**
    * Return a tool schema builder, so you can properly construct tools instead of dealing with the constructor
    * @returns a tool schema builder.
    */
   static builder() {
-    return new D();
+    return new J();
   }
 }
-class D {
+class J {
   _parameters;
   _output;
   /**
@@ -664,10 +664,10 @@ class D {
    * @returns Constructed ToolSchema describing parameters and output.
    */
   build() {
-    return this._parameters == null && (this._parameters = []), new B(this._parameters, this._output);
+    return this._parameters == null && (this._parameters = []), new d(this._parameters, this._output);
   }
 }
-class J {
+class T {
   name;
   description;
   type;
@@ -679,8 +679,8 @@ class J {
    * @param type JSON-serializable type name (e.g., string, number).
    * @param required Whether the parameter must be provided.
    */
-  constructor(e, t, o, s) {
-    this.name = e, this.description = t, this.type = o, this.required = s, this.assertValidType();
+  constructor(e, o, s, n) {
+    this.name = e, this.description = o, this.type = s, this.required = n, this.assertValidType();
   }
   /**
    * Placeholder for validating supported parameter types.
@@ -733,7 +733,7 @@ class A {
    * @returns Constructed Parameter definition.
    */
   build() {
-    return this._required == null && (this._required = !1), new J(
+    return this._required == null && (this._required = !1), new T(
       this._name,
       this._description,
       this._type,
@@ -741,7 +741,7 @@ class A {
     );
   }
 }
-class L extends q {
+class L extends O {
   endpoint;
   model;
   options;
@@ -751,38 +751,38 @@ class L extends q {
    * @param model Model name/tag to use for inference.
    * @param options Additional Ollama options passed through to the API.
    */
-  constructor(e, t, o) {
-    super(), this.endpoint = e, this.model = t, this.options = o;
+  constructor(e, o, s) {
+    super(), this.endpoint = e, this.model = o, this.options = s;
   }
   /**
    * 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 t = {
+    let o = {
       model: this.model,
       messages: this._formatMessages(e),
       tools: this._formatTools(e),
       ...this.options
-    }, o = await fetch(this.endpoint + "/api/chat", {
+    }, s = await fetch(this.endpoint + "/api/chat", {
       method: "POST",
       headers: {
         "Content-Type": "application/json"
       },
-      body: JSON.stringify(t)
+      body: JSON.stringify(o)
     });
-    if (!o.ok)
-      throw new Error(`HTTP error! status: ${o.status}`);
-    let s = await o.json(), i = [];
-    if (s.message.tool_calls)
-      for (let a in s.message.tool_calls) {
-        const r = s.message.tool_calls[a];
-        let c = "function", T = r[c]?.name, y = r[c]?.index, w = r.id ?? void 0, M = typeof r[c]?.arguments == "string" ? JSON.parse(r[c]?.arguments) : r[c]?.arguments;
-        i.push(
-          v.builder().name(T).type(c).parameters(M).sequence(y).id(w).build()
+    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()
         );
       }
-    return h.builder().toolCalls(i).role(s.message.role).content(s.message.content).thinking(s.message.thinking).build();
+    return h.builder().toolCalls(r).role(n.message.role).content(n.message.content).thinking(n.message.thinking).build();
   }
   /**
    * Convert internal tool definitions into the JSON schema format expected by Ollama.
@@ -790,26 +790,26 @@ class L extends q {
    * @returns Array of tool schema objects formatted for the Ollama API.
    */
   _formatTools(e) {
-    let t = [];
-    for (let o of e.tools) {
-      let s = {}, i = [];
-      for (let r of o.getToolSchema().parameters)
-        s[r.name] = {}, s[r.name].type = r.type, s[r.name].description = r.description, r.required && i.push(r.name);
+    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 a = {
         type: "function",
         function: {
-          name: o.getName(),
-          description: o.getDescription(),
+          name: s.getName(),
+          description: s.getDescription(),
           parameters: {
             type: "object",
-            properties: s,
-            required: i
+            properties: n,
+            required: r
           }
         }
       };
-      t.push(a);
+      o.push(a);
     }
-    return t;
+    return o;
   }
   /**
    * Convert internal message objects into Ollama's chat message shape.
@@ -818,25 +818,25 @@ class L extends q {
    * @returns Array of serialized messages ready for Ollama.
    */
   _formatMessages(e) {
-    let t = [];
-    for (let o of e.messages) {
-      if (o.role == "assistant" || o.role == "system" || o.role == "user") {
-        t.push(this._formatSingleMessage(o));
+    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 (o.role == "tool") {
-        for (const s of o.content)
-          if (s.getMessageComponentType() == l.ToolResult) {
-            const i = s;
-            t.push({
-              role: o.role,
-              name: i.toolName,
-              content: JSON.stringify(i.toolResult)
+      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)
             });
           }
       }
     }
-    return t;
+    return o;
   }
   /**
    * Format everything sent by a user or LLM (no tools)
@@ -844,24 +844,24 @@ class L extends q {
    * @returns Serialized message payload.
    */
   _formatSingleMessage(e) {
-    let t = {
+    let o = {
       role: e.role
-    }, o, s, i;
+    }, s, n, r;
     for (const a of e.content) {
       if (a.getMessageComponentType() == l.Content) {
-        const r = a;
-        o == null && (o = ""), o += r.get();
+        const i = a;
+        s == null && (s = ""), s += i.get();
       }
       if (a.getMessageComponentType() == l.Thinking) {
-        const r = a;
-        i == null && (i = ""), i += r.get();
+        const i = a;
+        r == null && (r = ""), r += i.get();
       }
       if (a.getMessageComponentType() == l.ToolCall) {
-        const r = a;
-        s == null && (s = []), s.push(this._formatToolCall(r));
+        const i = a;
+        n == null && (n = []), n.push(this._formatToolCall(i));
       }
     }
-    return o != null && (t.content = o), s != null && (t.tool_calls = s), i != null && (t.thinking = i), t;
+    return s != null && (o.content = s), n != null && (o.tool_calls = n), r != null && (o.thinking = r), o;
   }
   /**
    * Convert a ToolCall into the structure Ollama expects.
@@ -877,27 +877,66 @@ class L extends q {
     };
   }
 }
+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 d(
+      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 d(
+      e.prototype.tapewormParams,
+      e.prototype.tapewormParamsOutput
+    );
+  };
+}
 export {
-  O as Agent,
-  x as Content,
-  b as Conversation,
-  C as ConversationManager,
-  f as DefaultConversationManager,
+  j as Agent,
+  q as Content,
+  x as Conversation,
+  P as ConversationManager,
+  _ as DefaultConversationManager,
   h as Message,
-  N as MessageBuilder,
-  u as MessageComponent,
+  k as MessageBuilder,
+  c as MessageComponent,
   l as MessageComponentType,
-  q as Model,
+  O as Model,
   E as ModelRequest,
-  _ as ModelRequestBuilder,
+  y as ModelRequestBuilder,
   L as OllamaModel,
-  J as Parameter,
+  T as Parameter,
   A as ParameterBuilder,
-  k as Thinking,
+  C as Thinking,
   F as Tool,
   v as ToolCall,
-  P as ToolCallBuilder,
+  S as ToolCallBuilder,
+  V as ToolDescription,
+  Q as ToolName,
+  K as ToolOutput,
+  H as ToolParameter,
   m as ToolResult,
-  B as ToolSchema,
-  D as ToolSchemaBuilder
+  d as ToolSchema,
+  J as ToolSchemaBuilder
 };

package/dist/tapeworm.umd.js

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

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@atgs/tapeworm",
-  "version": "0.1.4",
+  "version": "0.1.6",
   "type": "module",
   "main": "dist/tapeworm.cjs.js",
   "module": "dist/tapeworm.es.js",