@atgs/tapeworm 0.0.1 → 0.1.1

package/README.md

@@ -0,0 +1,120 @@
+![Tapeworm Logo](./tapeworm.svg)
+
+# Tapeworm
+
+Tapeworm is an in-browser and node agent framework.
+
+This is the root package for Tapeworm. You can consume other packages like @atgs/tapeworm_bedrock for AWS Bedrock support.
+
+It provides an object-oriented API to create agents that run either on Node or within the browser.
+
+**This project is currently in alpha and is under active development.**
+
+## Current Features
+
+- Base API Defined
+- Supports `function` tools
+- Supports Ollama models
+
+## Example
+
+```
+import {Agent, OllamaModel, Parameter, Tool, ToolSchema} from '../../dist/tapeworm.es.js';
+
+class SomeTool extends Tool {
+
+    // Get the name of the function
+    getName() {
+        return "AdditionTool";
+    }
+
+    // Tell models what this function does
+    getDescription() {
+        return "Adds two numbers together.";
+    }
+
+    // Define the tool schema
+    getToolSchema() {
+
+    }
+
+    // Actually run the code!
+    execute(input) {
+        let a = input.a;
+        let b = input.b;
+        return a + b;
+    }
+
+}
+
+const ollama = new Model();
+
+const agent = new Agent();
+agent.name = "agent";
+agent.tools = [new YourTool(), new YourOtherTool()]
+agent.system_prompt = "You are an agent "
+agent.model = ollama;
+
+await agent.invoke("What is 9 + 10"); // Most likely doesn't print 21.
+
+```
+
+## Roadmap
+
+Tapeworm seeks to be the most ergonomic agentic solution for Node and the browser.
+
+It has a long way to go before this project is there, but I believe we're off to a good start.
+
+Here are the main areas that will be addressed:
+
+### Better Agent Definitions
+
+Agents will have a dedicated builder, and I will attempt to make tool definitions a lot easier to write so that the `getName()`, `getDescription()`, and `getToolSchema()` can be generated from your JSDoc or with annotations.
+
+In the meantime, schemas are easy to define with builders. See the example calculator bot for an example.
+
+### More documentation and examples
+
+We will build JS Docs and expose them, and more examples.
+
+### Plug 'n Play Proxy Servers
+
+AI Proxy servers are lightweight wrappers around your inference provider. If you're writing a Node app, you'll likely not need one, but if you're writing a browser application and want tools executing in the browser, they're likely necessary for security unless you're writing a prototype, because AI proxy servers allow you to keep your API keys private and meter consumption.
+
+A production-ready proxy server targeting Node will be available from this repository. It will offer these features:
+
+- Very simple configuration
+- Portability/modularity in your existing node applications
+- Ability to integrate into Redis or other databases for per-IP/per-customer metering
+
+### Providers
+
+Tapeworm has only a built-in implementation for Ollama models at the moment.
+
+Soon, this will change. I will be targeting newer providers in this order:
+
+- Your own proxy server
+- AWS Bedrock
+- Vertex AI
+- OpenAI API
+- HuggingFace API
+
+### Streaming support with obfuscation
+
+Tapeworm is synchronous at the moment. The Agent and Model interfaces will be expanded to support streaming. Further, they will be implemented in such a way to prevent side channel attacks.
+
+### Conversation Managers
+
+Summarizing and RAG-based conversation managers will be implemented to prevent token overflow errors.
+
+### Orchestration
+
+Swarms, teams, and graphs will become first-class features of Tapeworm, along with the tools they require to run efficiently.
+
+### Built-in tools
+
+More built-in tools will be added. Would love to get a code interpreter for JS working.
+
+## Any suggestions?
+
+Feel free to leave an issue.

package/dist/agent/agent.d.ts

@@ -0,0 +1,61 @@
+import Conversation, { ConversationManager } from "../conversation/conversation";
+import Message from "../conversation/message";
+import { type Model } from "../model/model";
+import type Tool from "../tool/tool";
+import type ToolCall from "../tool/toolCall";
+/**
+ * Coordinates a model, its tools, and the running conversation to fulfill user queries.
+ */
+export default class Agent {
+    name: string;
+    systemPrompt?: string;
+    tools: Tool[];
+    model: Model;
+    conversation: Conversation;
+    conversationManager?: ConversationManager;
+    toolNameToIndexMap: any | undefined;
+    callback: (m: Message) => void;
+    constructor(name: string, tools: Tool[], model: Model, conversationManager: ConversationManager, callback: (m: Message) => void);
+    /**
+     * Run the full agent loop for a user query: seed the conversation, invoke the model,
+     * and execute any returned tool calls until completion.
+     * @param query User-provided input to hand to the agent.
+     */
+    invoke(query: string): Promise<void>;
+    /**
+     * Ask the backing model for the next response given the current conversation state.
+     * @returns Parsed model response including content, thinking, and tool calls.
+     */
+    _runQuery(): Promise<Message>;
+    /**
+     * Execute a single tool call and append the result (or error) back to the conversation.
+     * @param toolCall Tool invocation details returned by the model.
+     * @returns Tool execution output, if any.
+     */
+    _runTool(toolCall: ToolCall): Promise<any>;
+    /**
+     * Build a lookup from tool name to index for efficient resolution of tool calls.
+     */
+    generateToolNameToIndexMap(): void;
+    static builder(): AgentBuilder;
+}
+declare class AgentBuilder {
+    _name: string;
+    _systemPrompt?: string;
+    _tools: Tool[];
+    _model: Model;
+    _conversation: Conversation | undefined;
+    _conversationManager: ConversationManager;
+    _toolNameToIndexMap: any | undefined;
+    _callback: (m: Message) => void;
+    name(name: string): AgentBuilder;
+    systemPrompt(systemPrompt: string | undefined): AgentBuilder;
+    tools(tools: Tool[]): AgentBuilder;
+    addTool(tool: Tool): AgentBuilder;
+    model(model: Model): AgentBuilder;
+    conversationManager(mgr: ConversationManager): AgentBuilder;
+    callback(callback: (m: Message) => void): AgentBuilder;
+    build(): Agent;
+}
+export declare function defaultCallback(m: Message): void;
+export {};

package/dist/conversation/conversation.d.ts

@@ -0,0 +1,47 @@
+import type { Message } from "..";
+import type { Model } from "../model/model";
+/**
+ * Maintains ordered chat messages and delegates to a conversation manager for compaction.
+ */
+export default class Conversation {
+    messages: Message[];
+    manager: ConversationManager;
+    /**
+     * Initialize an empty conversation using the default manager.
+     */
+    constructor();
+    /**
+     * Append a message to the history and apply compaction rules.
+     * @param message Message to store.
+     */
+    append(message: Message): void;
+}
+/**
+ * Strategy interface for pruning or transforming a conversation history.
+ */
+export declare class ConversationManager {
+    /**
+     * Compact or adjust a conversation according to a strategy.
+     * @param conversation Full list of messages so far.
+     * @returns Adjusted message list to keep.
+     */
+    compact(_conversation: Message[]): Message[];
+    /**
+     * Allow the manager to consider model-specific constraints (e.g., token limits).
+     * @param model Model being used for this conversation.
+     */
+    configure(_model: Model): void;
+}
+/**
+ * Default conversation manager that leaves all messages intact.
+ */
+export declare class DefaultConversationManager extends ConversationManager {
+    /**
+     * Default strategy simply returns the conversation untouched.
+     */
+    compact(conversation: Message[]): Message[];
+    /**
+     * Default manager ignores model configuration because it does no compaction.
+     */
+    configure(_model: Model): void;
+}

package/dist/conversation/message.d.ts

@@ -0,0 +1,189 @@
+import type ToolCall from "../tool/toolCall";
+/**
+ * Messages are the units of a Conversation
+ * They consist of a series of MessageComponents, several of which are already defined by Tapeworm
+ * and can be extended to be used for other use cases.
+ */
+export default class Message {
+    role: string;
+    content: MessageComponent[];
+    /**
+     * Wrap a set of message components with an associated sender role.
+     * @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(role: string, content: MessageComponent[]);
+    /**
+     * Create a new message builder for ergonomic construction.
+     * @returns A MessageBuilder instance.
+     */
+    static builder(): MessageBuilder;
+    /**
+     * Filters the message contents based on the message type.
+     */
+    filter(type: MessageComponentType): MessageComponent[];
+}
+/**
+ * Tapeworm prefers builders rather than constructors.
+ *
+ * This builder provides ergonomic methods to construct a message w/ its message components.
+ *
+ */
+export declare class MessageBuilder {
+    private _role;
+    private _content;
+    /**
+     * Set the role associated with the message being built.
+     * @param role Source of the message (e.g., user, assistant, tool).
+     * @returns This builder for chaining.
+     */
+    role(role: string): MessageBuilder;
+    /**
+     * Ensure the content array is initialized before appending components.
+     */
+    init(): void;
+    /**
+     * Attach a tool call component if provided.
+     * @param toolCall Tool call to append to the message.
+     * @returns This builder for chaining.
+     */
+    toolCall(toolCall: ToolCall | undefined): MessageBuilder;
+    /**
+     * Attach SEVERAL tool call component if provided.
+     * @param toolCall Tool call to append to the message.
+     * @returns This builder for chaining.
+     */
+    toolCalls(toolCalls: ToolCall[] | undefined): MessageBuilder;
+    /**
+     * Attach a tool result component if provided.
+     * @param toolResult Result of a previously executed tool call.
+     * @returns This builder for chaining.
+     */
+    toolResult(toolResult: ToolResult | undefined): MessageBuilder;
+    /**
+     * Attach a thinking component if provided.
+     * @param thinking Reflection or chain-of-thought text.
+     * @returns This builder for chaining.
+     */
+    thinking(thinking: string | undefined): MessageBuilder;
+    /**
+     * Attach content to the message if provided.
+     * @param content Assistant or user text content.
+     * @returns This builder for chaining.
+     */
+    content(content: string | undefined): MessageBuilder;
+    /**
+     * Construct the Message with the accumulated values.
+     * @returns Finalized Message instance.
+     */
+    build(): Message;
+}
+/**
+ * These are the official MessageComponent types supported by Tapeworm.
+ *
+ * Tapeworm guarantees that these can all be processed by officially supported models.
+ */
+export declare const MessageComponentType: {
+    Content: string;
+    Thinking: string;
+    ToolCall: string;
+    ToolResult: string;
+};
+export type MessageComponentType = (typeof MessageComponentType)[keyof typeof MessageComponentType];
+/**
+ * This is the base class of a message component.
+ * It must override the getMessageComponentType function.
+ */
+export declare class MessageComponent {
+    /**
+     * Get the type of this message component, as defined in MessageComponentType
+     */
+    getMessageComponentType(): MessageComponentType;
+}
+/**
+ * A Content message block, representing text from the LLM.
+ * It contains only content.
+ * It must override the getMessageComponentType function.
+ */
+export declare class Content extends MessageComponent {
+    text: string;
+    /**
+     * Create a content component with the provided text.
+     * @param text Body of the content block.
+     */
+    constructor(text: string);
+    /**
+     * Get the type of this message component, as defined in MessageComponentType.
+     * This returns MessageComponentType.Content.
+     */
+    getMessageComponentType(): MessageComponentType;
+    /**
+     * Get the content of this MessageComponent.
+     * @returns the string contained in this component.
+     */
+    get(): string;
+    /**
+     * Shorthand for the constructor
+     * @param text - the text this content block represents.
+     * @returns a new Content messagecomponent with the text.
+     */
+    static of(text: string): Content;
+}
+/**
+ * A Thinking message block, representing a thought from the LLM.
+ * It contains only a single thought, much like my brain as I'm writing this.
+ * It must override the getMessageComponentType function.
+ */
+export declare class Thinking extends MessageComponent {
+    thought: string;
+    /**
+     * Create a thinking component with the provided thought text.
+     * @param thought Chain-of-thought or reasoning string.
+     */
+    constructor(thought: string);
+    /**
+     * Get the type of this message component, as defined in MessageComponentType.
+     * This returns MessageComponentType.Content.
+     */
+    getMessageComponentType(): MessageComponentType;
+    /**
+     * Get the content of this MessageComponent.
+     * @returns the string contained in this component.
+     */
+    get(): string;
+    /**
+     * Shorthand for the constructor
+     * @param thought - the text this content block represents.
+     * @returns a new Thinking message component with the text.
+     */
+    static of(thought: string): Thinking;
+}
+/**
+ * A Thinking message block, representing a thought from the LLM.
+ * It contains only a single thought, much like my brain as I'm writing this.
+ * It must override the getMessageComponentType function.
+ */
+export declare class ToolResult extends MessageComponent {
+    id: string;
+    toolName: string;
+    toolResult: any;
+    /**
+     * Create a tool result component for a completed tool call.
+     * @param id Identifier of the originating tool call.
+     * @param toolName Name of the tool that produced the result.
+     * @param toolResult Output returned by the tool.
+     */
+    constructor(id: string, toolName: string, toolResult: any);
+    /**
+     * Get the type of this message component, as defined in MessageComponentType.
+     * This returns MessageComponentType.ToolResult.
+     */
+    getMessageComponentType(): MessageComponentType;
+    /**
+     * Shorthand for the constructor
+     * @param toolCall Original tool call that produced this result.
+     * @param toolResult Output to attach to the tool result component.
+     * @returns a new ToolResult component mirroring the provided tool call.
+     */
+    static of(toolCall: ToolCall, toolResult: any): ToolResult;
+}

package/dist/index.d.ts

@@ -0,0 +1,8 @@
+export { default as Agent } from "./agent/agent";
+export { default as Conversation, ConversationManager, DefaultConversationManager, } from "./conversation/conversation";
+export { Model, ModelRequest, ModelRequestBuilder } from "./model/model";
+export { default as Tool } from "./tool/tool";
+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";

package/dist/model/OllamaModel.d.ts

@@ -0,0 +1,48 @@
+import { Message } from "..";
+import ToolCall from "../tool/toolCall";
+import { Model, ModelRequest } from "./model";
+/**
+ * Model adapter that translates between Tapeworm message/tool shapes and the Ollama chat API.
+ */
+export default class OllamaModel extends Model {
+    endpoint: string;
+    model: string;
+    options?: any;
+    /**
+     * Create a model wrapper for an Ollama chat endpoint.
+     * @param endpoint Base URL of the Ollama server (e.g., http://localhost:11434).
+     * @param model Model name/tag to use for inference.
+     * @param options Additional Ollama options passed through to the API.
+     */
+    constructor(endpoint: string, model: string, options: any);
+    /**
+     * 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.
+     */
+    invoke(request: ModelRequest): Promise<Message>;
+    /**
+     * Convert internal tool definitions into the JSON schema format expected by Ollama.
+     * @param request Model request containing declared tools.
+     * @returns Array of tool schema objects formatted for the Ollama API.
+     */
+    _formatTools(request: ModelRequest): any;
+    /**
+     * Convert internal message objects into Ollama's chat message shape.
+     * Passes through assistant/system/user roles and maps tool role into the expected structure.
+     * @param request Model request containing the message history.
+     * @returns Array of serialized messages ready for Ollama.
+     */
+    _formatMessages(request: ModelRequest): any;
+    /**
+     * Format everything sent by a user or LLM (no tools)
+     * @param message Message to serialize for Ollama.
+     * @returns Serialized message payload.
+     */
+    _formatSingleMessage(message: Message): any;
+    /**
+     * Convert a ToolCall into the structure Ollama expects.
+     * @param toolCall ToolCall to serialize.
+     * @returns Minimal representation containing function name and arguments.
+     */
+    _formatToolCall(toolCall: ToolCall): any;
+}

package/dist/model/model.d.ts

@@ -0,0 +1,56 @@
+import type Message from "../conversation/message";
+import type Tool from "../tool/tool";
+/**
+ * A Model represents the LLM backend for the agent.
+ */
+export declare class Model {
+    /**
+     * Execute the model against the provided request.
+     * @param request Aggregated messages and tools to send to the model.
+     */
+    invoke(_request: ModelRequest): Promise<Message>;
+    /**
+     * Return the maximum token budget supported by this model.
+     */
+    tokenLimit(): void;
+}
+/**
+ * Container for the messages and tools required to invoke a model.
+ */
+export declare class ModelRequest {
+    messages: Message[];
+    tools: Tool[];
+    /**
+     * Create a new model request envelope.
+     * @param messages Conversation history to send to the model.
+     * @param tools Tool definitions available for function calling.
+     */
+    constructor(messages: Message[], tools: Tool[]);
+    /**
+     * Create a builder for composing a model request.
+     * @returns A ModelRequestBuilder instance.
+     */
+    static builder(): ModelRequestBuilder;
+}
+/**
+ * Fluent builder for constructing ModelRequest instances.
+ */
+export declare class ModelRequestBuilder {
+    private _messages;
+    private _tools;
+    /**
+     * Supply the conversation messages to include.
+     * @returns This builder for chaining.
+     */
+    messages(messages: Message[]): ModelRequestBuilder;
+    /**
+     * Supply the tool list for this request.
+     * @returns This builder for chaining.
+     */
+    tools(tools: Tool[]): ModelRequestBuilder;
+    /**
+     * Build the ModelRequest, defaulting tools to an empty array.
+     * @returns Constructed ModelRequest.
+     */
+    build(): ModelRequest;
+}

package/dist/tapeworm.cjs.js

@@ -1 +1 @@
-"use strict";Object.defineProperty(exports,Symbol.toStringTag,{value:"Module"});class _{messages;manager;constructor(){this.messages=[],this.manager=new T}append(e){this.messages.push(e),this.messages=this.manager.compact(this.messages)}}class a{role;content;toolCalls;toolName;thinking;constructor(e,s,t,o,r){this.role=e,this.content=s,this.toolCalls=t,this.toolName=o,this.thinking=r}setRole(e){this.role=e}setContent(e){this.content=e}setToolCalls(e){this.toolCalls=e}static builder(){return new f}}class f{_role;_content;_toolCalls;_toolName;_thinking;role(e){return this._role=e,this}content(e){return this._content=e,this}toolCalls(e){return this._toolCalls=e,this}toolName(e){return this._toolName=e,this}thinking(e){return this._thinking=e,this}build(){return new a(this._role,this._content,this._toolCalls,this._toolName,this._thinking)}}class y{compact(e){throw new p("No implementation for conversation manager!")}configure(e){throw new p("No implementation for conversation manager!")}}class T extends y{compact(e){return e}configure(e){}}class p extends Error{constructor(e){super(e),this.name="ConversationManagerNotImplemented"}}class v{async invoke(e){throw new g("The invoke function for this model was not correctly implemented.")}tokenLimit(){throw new g("The tokenLimit function for this model was not correctly implemented.")}}class w{messages;tools;constructor(e,s){this.messages=e,this.tools=s}static builder(){return new u}}class u{_messages;_tools;messages(e){return this._messages=e,this}tools(e){return this._tools=e,this}build(){return this._tools==null&&(this._tools=[]),new w(this._messages,this._tools)}}class b{toolCalls;role;content;thinking;constructor(e,s,t,o){this.toolCalls=e,this.role=s,this.content=t,this.thinking=o}static builder(){return new m}}class m{_toolCalls;_role;_content;_thinking;toolCalls(e){return this._toolCalls=e,this}role(e){return this._role=e,this}content(e){return this._content=e,this}thinking(e){return this._thinking=e,this}build(){return new b(this._toolCalls,this._role,this._content,this._thinking)}}class g extends Error{constructor(e){super(e),this.name="ModelNotImplementedError"}}class B{name;system_prompt;tools;model;conversation;conversationManager;toolNameToIndexMap;async invoke(e){this.conversation==null&&(this.conversation=new _,this.conversationManager!=null&&(this.conversation.manager=this.conversationManager),this.conversation.append(a.builder().role("system").content(this.system_prompt).build())),this.conversation.append(a.builder().role("user").content(e).build());let s=!1;for(;!s;){let t=await this._runQuery();if(this.conversation.append(a.builder().role(t.role??"assistant").content(t.content??"").thinking(t.thinking??"").build()),console.log("Assistant thinking: "+t.thinking),console.log("Assistant reply: "+t.content),s=!0,t.toolCalls!=null&&t.toolCalls.length!=0){s=!1,t.toolCalls.sort((o,r)=>(o.sequence??0)<(r.sequence??0)?-1:1);for(let o in t.toolCalls)this._runTool(t.toolCalls[o])}}}async _runQuery(){return await this.model.invoke(new u().messages(this.conversation?.messages).tools(this.tools).build())}async _runTool(e){if(console.log("Calling tool: "+JSON.stringify(e)),this.generateToolNameToIndexMap(),!(e.name in this.toolNameToIndexMap)){this.conversation.append(a.builder().role("tool").toolName(e.name).content(JSON.stringify({error:"tool does not exist"})).build());return}let s=this.tools[this.toolNameToIndexMap[e.name]];try{let t=s.execute(e.parameters);this.conversation.append(a.builder().role("tool").toolName(e.name).content(JSON.stringify(t)).build())}catch(t){this.conversation.append(a.builder().role("tool").toolName(e.name).content(JSON.stringify(t)).build())}}generateToolNameToIndexMap(){if(this.toolNameToIndexMap==null){this.toolNameToIndexMap={};for(let e=0;e<this.tools.length;e++)this.toolNameToIndexMap[this.tools[e].getName()]=e}}}class c extends Error{constructor(e){super(e),this.name="ToolNotDefinedError"}}class I{name;description;tool_schema;constructor(){this.name=this.getName(),this.description=this.getDescription(),this.tool_schema=this.getToolSchema()}getName(){throw new c("Tool name not defined.")}getDescription(){throw new c("Tool description not defined.")}getToolSchema(){throw new c("Tool parameter schema not defined.")}execute(e){return null}}class d{sequence;name;parameters;type;constructor(e,s,t,o){this.sequence=e,this.name=s,this.parameters=t,this.type=o}static builder(){return new M}}class M{_sequence;_name;_parameters;_type;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}build(){return this._sequence==null&&(this._sequence=0),new d(this._sequence,this._name,this._parameters,this._type)}}class N{parameters;output;constructor(e,s){this.parameters=e,this.output=s}static builder(){return new C}}class C{_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 N(this._parameters,this._output)}}class q{name;description;type;required;constructor(e,s,t,o){this.name=e,this.description=s,this.type=t,this.required=o,this.assertValidType()}assertValidType(){}static builder(){return new k}}class k{_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 q(this._name,this._description,this._type,this._required)}}class j extends v{endpoint;model;options;constructor(e,s,t){super(),this.endpoint=e,this.model=s,this.options=t}async invoke(e){let s={model:this.model,messages:this._formatMessages(e),tools:this._formatTools(e),...this.options},t=await fetch(this.endpoint+"/api/chat",{method:"POST",headers:{"Content-Type":"application/json"},body:JSON.stringify(s)});if(!t.ok)throw new Error(`HTTP error! status: ${t.status}`);let o=await t.json(),r=[];if(o.message.tool_calls)for(let h in o.message.tool_calls){const i=o.message.tool_calls[h];let l="function",x=i[l]?.name,O=i[l]?.index,S=i[l]?.arguments;r.push(d.builder().name(x).type(l).parameters(S).sequence(O).build())}return new m().toolCalls(r).role(o.message.role).content(o.message.content).thinking(o.message.thinking).build()}_formatTools(e){let s=[],t=[];for(let o of e.tools){let r={};for(let i of o.getToolSchema().parameters)r[i.name]={},r[i.name].type=i.name,r[i.name].description=i.description,i.required&&t.push(i.name);let h={type:"function",function:{name:o.getName(),description:o.getDescription(),parameters:{type:"object",properties:r,required:t}}};s.push(h)}return s}_formatMessages(e){let s=[];for(let t of e.messages){if(t.role=="assistant"||t.role=="system"||t.role=="user"){s.push(t);continue}t.role=="tool"&&s.push({role:"tool",content:t.content,tool_name:t.toolName})}return s}}exports.Agent=B;exports.Conversation=_;exports.ConversationManager=y;exports.DefaultConversationManager=T;exports.Message=a;exports.MessageBuilder=f;exports.Model=v;exports.ModelRequest=w;exports.ModelRequestBuilder=u;exports.ModelResponse=b;exports.ModelResponseBuilder=m;exports.OllamaModel=j;exports.Parameter=q;exports.ParameterBuilder=k;exports.Tool=I;exports.ToolCall=d;exports.ToolCallBuilder=M;exports.ToolSchema=N;exports.ToolSchemaBuilder=C;
+"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;