@microsoft/teams.ai 0.2.7

package/dist/prompts/chat.d.mts

@@ -0,0 +1,146 @@
+import { Function, FunctionHandler } from '../function.mjs';
+import { IMemory } from '../memory.mjs';
+import { Message, ContentPart, ModelMessage } from '../message.mjs';
+import { IChatModel, TextChunkHandler } from '../models/chat.mjs';
+import { Schema } from '../schema.mjs';
+import { ITemplate } from '../template.mjs';
+import { IAiPlugin } from './plugin.mjs';
+import '@microsoft/teams.common';
+import '../citation.mjs';
+
+type ChatPromptOptions<TOptions extends Record<string, any> = Record<string, any>> = {
+    /**
+     * the name of the prompt
+     */
+    readonly name?: string;
+    /**
+     * the description of the prompt
+     */
+    readonly description?: string;
+    /**
+     * the model to send messages to
+     */
+    readonly model: IChatModel<TOptions>;
+    /**
+     * the defining characteristics/objective
+     * of the prompt. This is commonly used to provide a system prompt.
+     * If you supply the system prompt as part of the messages,
+     * you do not need to supply this option.
+     */
+    readonly instructions?: string | string[] | ITemplate;
+    /**
+     * the `role` of the initial message
+     */
+    readonly role?: 'system' | 'user';
+    /**
+     * the conversation history
+     */
+    readonly messages?: Message[] | IMemory;
+};
+type ChatPromptSendOptions<TOptions extends Record<string, any> = Record<string, any>> = {
+    /**
+     * the conversation history
+     */
+    readonly messages?: Message[] | IMemory;
+    /**
+     * the models request options
+     */
+    readonly request?: TOptions;
+    /**
+     * the callback to be called for each
+     * stream chunk
+     */
+    readonly onChunk?: TextChunkHandler;
+};
+/**
+ * a prompt that can interface with a
+ * chat model that provides utility like
+ * streaming and function calling
+ */
+interface IChatPrompt<TOptions extends Record<string, any> = Record<string, any>, TChatPromptPlugins extends readonly ChatPromptPlugin<string, any>[] = []> {
+    /**
+     * the prompt name
+     */
+    readonly name: string;
+    /**
+     * the prompt description
+     */
+    readonly description: string;
+    /**
+     * the chat history
+     */
+    readonly messages: IMemory;
+    /**
+     * the registered functions
+     */
+    readonly functions: Array<Function>;
+    /**
+     * the chat model
+     */
+    plugins: TChatPromptPlugins;
+    /**
+     * add another chat prompt as a
+     */
+    use(prompt: IChatPrompt): this;
+    use(name: string, prompt: IChatPrompt): this;
+    /**
+     * add a function that can be called
+     * by the model
+     */
+    function(name: string, description: string, handler: FunctionHandler): this;
+    function(name: string, description: string, parameters: Schema, handler: FunctionHandler): this;
+    usePlugin<TPluginName extends TChatPromptPlugins[number]['name']>(name: TPluginName, args: Extract<TChatPromptPlugins[number], {
+        name: TPluginName;
+    }>['onUsePlugin'] extends ((args: infer U) => void) | undefined ? U : never): this;
+    /**
+     * call a function
+     */
+    call<A extends Record<string, any>, R = any>(name: string, args?: A): Promise<R>;
+    /**
+     * send a message to the model and get a response
+     */
+    send(input: string | ContentPart[], options?: ChatPromptSendOptions<TOptions>): Promise<Pick<ModelMessage, 'content'> & Omit<ModelMessage, 'content'>>;
+}
+type ChatPromptPlugin<TPluginName extends string, TPluginUseArgs extends {}> = IAiPlugin<TPluginName, TPluginUseArgs, Parameters<IChatPrompt['send']>[0], ReturnType<IChatPrompt['send']>> & {
+    /**
+     * Optionally passed in to modify the functions array that
+     * is passed to the model
+     * @param functions
+     * @returns Functions
+     */
+    onBuildFunctions?: (functions: Function[]) => Function[] | Promise<Function[]>;
+};
+/**
+ * a prompt that can interface with a
+ * chat model that provides utility like
+ * streaming and function calling
+ */
+declare class ChatPrompt<TOptions extends Record<string, any> = Record<string, any>, TChatPromptPlugins extends readonly ChatPromptPlugin<string, any>[] = []> implements IChatPrompt<TOptions, TChatPromptPlugins> {
+    get name(): string;
+    protected readonly _name: string;
+    get description(): string;
+    protected readonly _description: string;
+    get messages(): IMemory;
+    protected readonly _messages: IMemory;
+    get functions(): Function[];
+    protected readonly _functions: Record<string, Function>;
+    get plugins(): TChatPromptPlugins;
+    protected readonly _plugins: TChatPromptPlugins;
+    protected readonly _role: 'system' | 'user';
+    protected readonly _template: ITemplate;
+    protected readonly _model: IChatModel<TOptions>;
+    constructor(options: ChatPromptOptions<TOptions>, plugins?: TChatPromptPlugins);
+    use(prompt: ChatPrompt): this;
+    use(name: string, prompt: ChatPrompt): this;
+    function(name: string, description: string, handler: FunctionHandler): this;
+    function(name: string, description: string, parameters: Schema, handler: FunctionHandler): this;
+    usePlugin<K extends TChatPromptPlugins[number]['name']>(name: K, args: Extract<TChatPromptPlugins[number], {
+        name: K;
+    }>['onUsePlugin'] extends ((args: infer U) => void) | undefined ? U : never): this;
+    call<A extends {
+        [key: string]: any;
+    }, R = any>(name: string, args?: A): Promise<R>;
+    send(input: string | ContentPart[], options?: ChatPromptSendOptions<TOptions>): Promise<ModelMessage>;
+}
+
+export { ChatPrompt, type ChatPromptOptions, type ChatPromptPlugin, type ChatPromptSendOptions, type IChatPrompt };

package/dist/prompts/chat.d.ts

@@ -0,0 +1,146 @@
+import { Function, FunctionHandler } from '../function.js';
+import { IMemory } from '../memory.js';
+import { Message, ContentPart, ModelMessage } from '../message.js';
+import { IChatModel, TextChunkHandler } from '../models/chat.js';
+import { Schema } from '../schema.js';
+import { ITemplate } from '../template.js';
+import { IAiPlugin } from './plugin.js';
+import '@microsoft/teams.common';
+import '../citation.js';
+
+type ChatPromptOptions<TOptions extends Record<string, any> = Record<string, any>> = {
+    /**
+     * the name of the prompt
+     */
+    readonly name?: string;
+    /**
+     * the description of the prompt
+     */
+    readonly description?: string;
+    /**
+     * the model to send messages to
+     */
+    readonly model: IChatModel<TOptions>;
+    /**
+     * the defining characteristics/objective
+     * of the prompt. This is commonly used to provide a system prompt.
+     * If you supply the system prompt as part of the messages,
+     * you do not need to supply this option.
+     */
+    readonly instructions?: string | string[] | ITemplate;
+    /**
+     * the `role` of the initial message
+     */
+    readonly role?: 'system' | 'user';
+    /**
+     * the conversation history
+     */
+    readonly messages?: Message[] | IMemory;
+};
+type ChatPromptSendOptions<TOptions extends Record<string, any> = Record<string, any>> = {
+    /**
+     * the conversation history
+     */
+    readonly messages?: Message[] | IMemory;
+    /**
+     * the models request options
+     */
+    readonly request?: TOptions;
+    /**
+     * the callback to be called for each
+     * stream chunk
+     */
+    readonly onChunk?: TextChunkHandler;
+};
+/**
+ * a prompt that can interface with a
+ * chat model that provides utility like
+ * streaming and function calling
+ */
+interface IChatPrompt<TOptions extends Record<string, any> = Record<string, any>, TChatPromptPlugins extends readonly ChatPromptPlugin<string, any>[] = []> {
+    /**
+     * the prompt name
+     */
+    readonly name: string;
+    /**
+     * the prompt description
+     */
+    readonly description: string;
+    /**
+     * the chat history
+     */
+    readonly messages: IMemory;
+    /**
+     * the registered functions
+     */
+    readonly functions: Array<Function>;
+    /**
+     * the chat model
+     */
+    plugins: TChatPromptPlugins;
+    /**
+     * add another chat prompt as a
+     */
+    use(prompt: IChatPrompt): this;
+    use(name: string, prompt: IChatPrompt): this;
+    /**
+     * add a function that can be called
+     * by the model
+     */
+    function(name: string, description: string, handler: FunctionHandler): this;
+    function(name: string, description: string, parameters: Schema, handler: FunctionHandler): this;
+    usePlugin<TPluginName extends TChatPromptPlugins[number]['name']>(name: TPluginName, args: Extract<TChatPromptPlugins[number], {
+        name: TPluginName;
+    }>['onUsePlugin'] extends ((args: infer U) => void) | undefined ? U : never): this;
+    /**
+     * call a function
+     */
+    call<A extends Record<string, any>, R = any>(name: string, args?: A): Promise<R>;
+    /**
+     * send a message to the model and get a response
+     */
+    send(input: string | ContentPart[], options?: ChatPromptSendOptions<TOptions>): Promise<Pick<ModelMessage, 'content'> & Omit<ModelMessage, 'content'>>;
+}
+type ChatPromptPlugin<TPluginName extends string, TPluginUseArgs extends {}> = IAiPlugin<TPluginName, TPluginUseArgs, Parameters<IChatPrompt['send']>[0], ReturnType<IChatPrompt['send']>> & {
+    /**
+     * Optionally passed in to modify the functions array that
+     * is passed to the model
+     * @param functions
+     * @returns Functions
+     */
+    onBuildFunctions?: (functions: Function[]) => Function[] | Promise<Function[]>;
+};
+/**
+ * a prompt that can interface with a
+ * chat model that provides utility like
+ * streaming and function calling
+ */
+declare class ChatPrompt<TOptions extends Record<string, any> = Record<string, any>, TChatPromptPlugins extends readonly ChatPromptPlugin<string, any>[] = []> implements IChatPrompt<TOptions, TChatPromptPlugins> {
+    get name(): string;
+    protected readonly _name: string;
+    get description(): string;
+    protected readonly _description: string;
+    get messages(): IMemory;
+    protected readonly _messages: IMemory;
+    get functions(): Function[];
+    protected readonly _functions: Record<string, Function>;
+    get plugins(): TChatPromptPlugins;
+    protected readonly _plugins: TChatPromptPlugins;
+    protected readonly _role: 'system' | 'user';
+    protected readonly _template: ITemplate;
+    protected readonly _model: IChatModel<TOptions>;
+    constructor(options: ChatPromptOptions<TOptions>, plugins?: TChatPromptPlugins);
+    use(prompt: ChatPrompt): this;
+    use(name: string, prompt: ChatPrompt): this;
+    function(name: string, description: string, handler: FunctionHandler): this;
+    function(name: string, description: string, parameters: Schema, handler: FunctionHandler): this;
+    usePlugin<K extends TChatPromptPlugins[number]['name']>(name: K, args: Extract<TChatPromptPlugins[number], {
+        name: K;
+    }>['onUsePlugin'] extends ((args: infer U) => void) | undefined ? U : never): this;
+    call<A extends {
+        [key: string]: any;
+    }, R = any>(name: string, args?: A): Promise<R>;
+    send(input: string | ContentPart[], options?: ChatPromptSendOptions<TOptions>): Promise<ModelMessage>;
+}
+
+export { ChatPrompt, type ChatPromptOptions, type ChatPromptPlugin, type ChatPromptSendOptions, type IChatPrompt };

package/dist/prompts/chat.js

@@ -0,0 +1,162 @@
+'use strict';
+
+var localMemory = require('../local-memory');
+var templates = require('../templates');
+
+class ChatPrompt {
+  get name() {
+    return this._name;
+  }
+  _name;
+  get description() {
+    return this._description;
+  }
+  _description;
+  get messages() {
+    return this._messages;
+  }
+  _messages;
+  get functions() {
+    return Object.values(this._functions);
+  }
+  _functions = {};
+  get plugins() {
+    return this._plugins;
+  }
+  _plugins;
+  _role;
+  _template;
+  _model;
+  constructor(options, plugins) {
+    this._name = options.name || "chat";
+    this._description = options.description || "an agent you can chat with";
+    this._role = options.role || "system";
+    this._model = options.model;
+    this._template = Array.isArray(options.instructions) ? new templates.StringTemplate(options.instructions.join("\n")) : typeof options.instructions !== "object" ? new templates.StringTemplate(options.instructions) : options.instructions;
+    this._messages = typeof options.messages === "object" && !Array.isArray(options.messages) ? options.messages : new localMemory.LocalMemory({ messages: options.messages || [] });
+    this._plugins = plugins || [];
+  }
+  use(...args) {
+    const prompt = args.length === 1 ? args[0] : args[1];
+    const name = args.length === 1 ? prompt.name : args[0];
+    this._functions[name] = {
+      name,
+      description: prompt.description,
+      parameters: {
+        type: "object",
+        properties: {
+          text: {
+            type: "string",
+            description: "the text to send to the assistant"
+          }
+        },
+        required: ["text"]
+      },
+      handler: ({ text }) => {
+        return prompt.send(text);
+      }
+    };
+    return this;
+  }
+  function(...args) {
+    const name = args[0];
+    const description = args[1];
+    const parameters = args.length === 3 ? null : args[2];
+    const handler = args[args.length - 1];
+    this._functions[name] = {
+      name,
+      description,
+      parameters: parameters || {},
+      handler
+    };
+    return this;
+  }
+  usePlugin(name, args) {
+    const plugin = this._plugins.find((p) => p.name === name);
+    if (!plugin) {
+      throw new Error(`Plugin "${name}" not found`);
+    }
+    if (plugin.onUsePlugin) {
+      plugin.onUsePlugin(args);
+    }
+    return this;
+  }
+  async call(name, args) {
+    const fn = this._functions[name];
+    if (!fn) {
+      throw new Error(`function "${name}" not found`);
+    }
+    return await fn.handler(args || {});
+  }
+  async send(input, options = {}) {
+    for (const plugin of this.plugins) {
+      if (plugin.onBeforeSend) {
+        input = await plugin.onBeforeSend(input);
+      }
+    }
+    const { onChunk } = options;
+    if (typeof input === "string") {
+      input = input.trim();
+    }
+    const messages = !Array.isArray(options.messages) ? options.messages || this._messages : new localMemory.LocalMemory({ messages: options.messages || [] });
+    let buffer = "";
+    let system = void 0;
+    const prompt = await this._template.render();
+    if (prompt) {
+      system = {
+        role: this._role,
+        content: prompt
+      };
+    }
+    let functions = Object.values(this._functions);
+    const pluginsWithOnBuildFunctions = this._plugins.filter(
+      (plugin) => plugin.onBuildFunctions != null
+    );
+    for (const plugin of pluginsWithOnBuildFunctions) {
+      functions = await plugin.onBuildFunctions(functions);
+    }
+    const fnMap = functions.reduce(
+      (acc, fn) => {
+        acc[fn.name] = fn;
+        return acc;
+      },
+      {}
+    );
+    const res = await this._model.send(
+      {
+        role: "user",
+        content: input
+      },
+      {
+        system,
+        messages,
+        request: options.request,
+        functions: fnMap,
+        onChunk: async (chunk) => {
+          if (!chunk || !onChunk) return;
+          buffer += chunk;
+          try {
+            await onChunk(buffer);
+            buffer = "";
+          } catch (err) {
+            return;
+          }
+        }
+      }
+    );
+    let output = {
+      ...res,
+      content: res.content || ""
+    };
+    for (const plugin of this.plugins) {
+      if (plugin.onAfterSend) {
+        output = await plugin.onAfterSend(output);
+      }
+    }
+    return output;
+  }
+}
+
+exports.ChatPrompt = ChatPrompt;
+//# sourceMappingURL=chat.js.map
+//# sourceMappingURL=chat.js.map

package/dist/prompts/chat.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/prompts/chat.ts"],"names":["StringTemplate","LocalMemory"],"mappings":";;;;;AA0JO,MAAM,UAIb,CAAA;AAAA,EACE,IAAI,IAAO,GAAA;AACT,IAAA,OAAO,IAAK,CAAA,KAAA;AAAA;AACd,EACmB,KAAA;AAAA,EAEnB,IAAI,WAAc,GAAA;AAChB,IAAA,OAAO,IAAK,CAAA,YAAA;AAAA;AACd,EACmB,YAAA;AAAA,EAEnB,IAAI,QAAW,GAAA;AACb,IAAA,OAAO,IAAK,CAAA,SAAA;AAAA;AACd,EACmB,SAAA;AAAA,EAEnB,IAAI,SAAY,GAAA;AACd,IAAO,OAAA,MAAA,CAAO,MAAO,CAAA,IAAA,CAAK,UAAU,CAAA;AAAA;AACtC,EACmB,aAAuC,EAAC;AAAA,EAE3D,IAAI,OAAU,GAAA;AACZ,IAAA,OAAO,IAAK,CAAA,QAAA;AAAA;AACd,EACmB,QAAA;AAAA,EAEA,KAAA;AAAA,EACA,SAAA;AAAA,EACA,MAAA;AAAA,EAEnB,WAAA,CAAY,SAAsC,OAA8B,EAAA;AAC9E,IAAK,IAAA,CAAA,KAAA,GAAQ,QAAQ,IAAQ,IAAA,MAAA;AAC7B,IAAK,IAAA,CAAA,YAAA,GAAe,QAAQ,WAAe,IAAA,4BAAA;AAC3C,IAAK,IAAA,CAAA,KAAA,GAAQ,QAAQ,IAAQ,IAAA,QAAA;AAC7B,IAAA,IAAA,CAAK,SAAS,OAAQ,CAAA,KAAA;AACtB,IAAK,IAAA,CAAA,SAAA,GAAY,MAAM,OAAQ,CAAA,OAAA,CAAQ,YAAY,CAC/C,GAAA,IAAIA,wBAAe,CAAA,OAAA,CAAQ,YAAa,CAAA,IAAA,CAAK,IAAI,CAAC,CAAA,GAClD,OAAO,OAAA,CAAQ,YAAiB,KAAA,QAAA,GAC9B,IAAIA,wBAAe,CAAA,OAAA,CAAQ,YAAY,CAAA,GACvC,OAAQ,CAAA,YAAA;AAEd,IAAK,IAAA,CAAA,SAAA,GACH,OAAO,OAAQ,CAAA,QAAA,KAAa,YAAY,CAAC,KAAA,CAAM,QAAQ,OAAQ,CAAA,QAAQ,IACnE,OAAQ,CAAA,QAAA,GACR,IAAIC,uBAAY,CAAA,EAAE,UAAU,OAAQ,CAAA,QAAA,IAAY,EAAC,EAAG,CAAA;AAE1D,IAAK,IAAA,CAAA,QAAA,GAAW,WAAY,EAAC;AAAA;AAC/B,EAIA,OAAO,IAAa,EAAA;AAClB,IAAM,MAAA,MAAA,GAAqB,KAAK,MAAW,KAAA,CAAA,GAAI,KAAK,CAAC,CAAA,GAAI,KAAK,CAAC,CAAA;AAC/D,IAAA,MAAM,OAAe,IAAK,CAAA,MAAA,KAAW,IAAI,MAAO,CAAA,IAAA,GAAO,KAAK,CAAC,CAAA;AAC7D,IAAK,IAAA,CAAA,UAAA,CAAW,IAAI,CAAI,GAAA;AAAA,MACtB,IAAA;AAAA,MACA,aAAa,MAAO,CAAA,WAAA;AAAA,MACpB,UAAY,EAAA;AAAA,QACV,IAAM,EAAA,QAAA;AAAA,QACN,UAAY,EAAA;AAAA,UACV,IAAM,EAAA;AAAA,YACJ,IAAM,EAAA,QAAA;AAAA,YACN,WAAa,EAAA;AAAA;AACf,SACF;AAAA,QACA,QAAA,EAAU,CAAC,MAAM;AAAA,OACnB;AAAA,MACA,OAAS,EAAA,CAAC,EAAE,IAAA,EAA6B,KAAA;AACvC,QAAO,OAAA,MAAA,CAAO,KAAK,IAAI,CAAA;AAAA;AACzB,KACF;AAEA,IAAO,OAAA,IAAA;AAAA;AACT,EAIA,YAAY,IAAa,EAAA;AACvB,IAAM,MAAA,IAAA,GAAe,KAAK,CAAC,CAAA;AAC3B,IAAM,MAAA,WAAA,GAAsB,KAAK,CAAC,CAAA;AAClC,IAAA,MAAM,aAA4B,IAAK,CAAA,MAAA,KAAW,CAAI,GAAA,IAAA,GAAO,KAAK,CAAC,CAAA;AACnE,IAAA,MAAM,OAA2B,GAAA,IAAA,CAAK,IAAK,CAAA,MAAA,GAAS,CAAC,CAAA;AACrD,IAAK,IAAA,CAAA,UAAA,CAAW,IAAI,CAAI,GAAA;AAAA,MACtB,IAAA;AAAA,MACA,WAAA;AAAA,MACA,UAAA,EAAY,cAAc,EAAC;AAAA,MAC3B;AAAA,KACF;AAEA,IAAO,OAAA,IAAA;AAAA;AACT,EAEA,SAAA,CACE,MACA,IAKM,EAAA;AACN,IAAM,MAAA,MAAA,GAAS,KAAK,QAAS,CAAA,IAAA,CAAK,CAAC,CAAM,KAAA,CAAA,CAAE,SAAS,IAAI,CAAA;AACxD,IAAA,IAAI,CAAC,MAAQ,EAAA;AACX,MAAA,MAAM,IAAI,KAAA,CAAM,CAAW,QAAA,EAAA,IAAI,CAAa,WAAA,CAAA,CAAA;AAAA;AAG9C,IAAA,IAAI,OAAO,WAAa,EAAA;AACtB,MAAA,MAAA,CAAO,YAAY,IAAI,CAAA;AAAA;AAGzB,IAAO,OAAA,IAAA;AAAA;AACT,EAEA,MAAM,IAAgD,CAAA,IAAA,EAAc,IAAsB,EAAA;AACxF,IAAM,MAAA,EAAA,GAAK,IAAK,CAAA,UAAA,CAAW,IAAI,CAAA;AAE/B,IAAA,IAAI,CAAC,EAAI,EAAA;AACP,MAAA,MAAM,IAAI,KAAA,CAAM,CAAa,UAAA,EAAA,IAAI,CAAa,WAAA,CAAA,CAAA;AAAA;AAGhD,IAAA,OAAO,MAAM,EAAA,CAAG,OAAQ,CAAA,IAAA,IAAQ,EAAE,CAAA;AAAA;AACpC,EAEA,MAAM,IAAA,CAAK,KAA+B,EAAA,OAAA,GAA2C,EAAI,EAAA;AACvF,IAAW,KAAA,MAAA,MAAA,IAAU,KAAK,OAAS,EAAA;AACjC,MAAA,IAAI,OAAO,YAAc,EAAA;AACvB,QAAQ,KAAA,GAAA,MAAM,MAAO,CAAA,YAAA,CAAa,KAAK,CAAA;AAAA;AACzC;AAGF,IAAM,MAAA,EAAE,SAAY,GAAA,OAAA;AAEpB,IAAI,IAAA,OAAO,UAAU,QAAU,EAAA;AAC7B,MAAA,KAAA,GAAQ,MAAM,IAAK,EAAA;AAAA;AAGrB,IAAA,MAAM,WAAW,CAAC,KAAA,CAAM,QAAQ,OAAQ,CAAA,QAAQ,IAC5C,OAAQ,CAAA,QAAA,IAAY,KAAK,SACzB,GAAA,IAAIA,wBAAY,EAAE,QAAA,EAAU,QAAQ,QAAY,IAAA,IAAI,CAAA;AAExD,IAAA,IAAI,MAAS,GAAA,EAAA;AACb,IAAA,IAAI,MAAkD,GAAA,MAAA;AACtD,IAAA,MAAM,MAAS,GAAA,MAAM,IAAK,CAAA,SAAA,CAAU,MAAO,EAAA;AAE3C,IAAA,IAAI,MAAQ,EAAA;AACV,MAAS,MAAA,GAAA;AAAA,QACP,MAAM,IAAK,CAAA,KAAA;AAAA,QACX,OAAS,EAAA;AAAA,OACX;AAAA;AAGF,IAAA,IAAI,SAAY,GAAA,MAAA,CAAO,MAAO,CAAA,IAAA,CAAK,UAAU,CAAA;AAC7C,IAAM,MAAA,2BAAA,GAA8B,KAAK,QAAS,CAAA,MAAA;AAAA,MAChD,CAAC,MACC,KAAA,MAAA,CAAO,gBAAoB,IAAA;AAAA,KAC/B;AACA,IAAA,KAAA,MAAW,UAAU,2BAA6B,EAAA;AAChD,MAAY,SAAA,GAAA,MAAM,MAAO,CAAA,gBAAA,CAAiB,SAAS,CAAA;AAAA;AAGrD,IAAA,MAAM,QAAQ,SAAU,CAAA,MAAA;AAAA,MACtB,CAAC,KAAK,EAAO,KAAA;AACX,QAAI,GAAA,CAAA,EAAA,CAAG,IAAI,CAAI,GAAA,EAAA;AACf,QAAO,OAAA,GAAA;AAAA,OACT;AAAA,MACA;AAAC,KACH;AAEA,IAAM,MAAA,GAAA,GAAM,MAAM,IAAA,CAAK,MAAO,CAAA,IAAA;AAAA,MAC5B;AAAA,QACE,IAAM,EAAA,MAAA;AAAA,QACN,OAAS,EAAA;AAAA,OACX;AAAA,MACA;AAAA,QACE,MAAA;AAAA,QACA,QAAA;AAAA,QACA,SAAS,OAAQ,CAAA,OAAA;AAAA,QACjB,SAAW,EAAA,KAAA;AAAA,QACX,OAAA,EAAS,OAAO,KAAU,KAAA;AACxB,UAAI,IAAA,CAAC,KAAS,IAAA,CAAC,OAAS,EAAA;AACxB,UAAU,MAAA,IAAA,KAAA;AAEV,UAAI,IAAA;AACF,YAAA,MAAM,QAAQ,MAAM,CAAA;AACpB,YAAS,MAAA,GAAA,EAAA;AAAA,mBACF,GAAK,EAAA;AACZ,YAAA;AAAA;AACF;AACF;AACF,KACF;AAEA,IAAA,IAAI,MAAuD,GAAA;AAAA,MACzD,GAAG,GAAA;AAAA,MACH,OAAA,EAAS,IAAI,OAAW,IAAA;AAAA,KAC1B;AACA,IAAW,KAAA,MAAA,MAAA,IAAU,KAAK,OAAS,EAAA;AACjC,MAAA,IAAI,OAAO,WAAa,EAAA;AACtB,QAAS,MAAA,GAAA,MAAM,MAAO,CAAA,WAAA,CAAY,MAAM,CAAA;AAAA;AAC1C;AAGF,IAAO,OAAA,MAAA;AAAA;AAEX","file":"chat.js","sourcesContent":["import { Function, FunctionHandler } from '../function';\nimport { LocalMemory } from '../local-memory';\nimport { IMemory } from '../memory';\nimport { ContentPart, Message, ModelMessage, SystemMessage, UserMessage } from '../message';\nimport { IChatModel, TextChunkHandler } from '../models';\nimport { Schema } from '../schema';\nimport { ITemplate } from '../template';\nimport { StringTemplate } from '../templates';\nimport { WithRequired } from '../utils/types';\nimport { IAiPlugin } from './plugin';\n\nexport type ChatPromptOptions<TOptions extends Record<string, any> = Record<string, any>> = {\n  /**\n   * the name of the prompt\n   */\n  readonly name?: string;\n\n  /**\n   * the description of the prompt\n   */\n  readonly description?: string;\n\n  /**\n   * the model to send messages to\n   */\n  readonly model: IChatModel<TOptions>;\n\n  /**\n   * the defining characteristics/objective\n   * of the prompt. This is commonly used to provide a system prompt.\n   * If you supply the system prompt as part of the messages,\n   * you do not need to supply this option.\n   */\n  readonly instructions?: string | string[] | ITemplate;\n\n  /**\n   * the `role` of the initial message\n   */\n  readonly role?: 'system' | 'user';\n\n  /**\n   * the conversation history\n   */\n  readonly messages?: Message[] | IMemory;\n};\n\nexport type ChatPromptSendOptions<TOptions extends Record<string, any> = Record<string, any>> = {\n  /**\n   * the conversation history\n   */\n  readonly messages?: Message[] | IMemory;\n\n  /**\n   * the models request options\n   */\n  readonly request?: TOptions;\n\n  /**\n   * the callback to be called for each\n   * stream chunk\n   */\n  readonly onChunk?: TextChunkHandler;\n};\n\n/**\n * a prompt that can interface with a\n * chat model that provides utility like\n * streaming and function calling\n */\nexport interface IChatPrompt<\n  TOptions extends Record<string, any> = Record<string, any>,\n  TChatPromptPlugins extends readonly ChatPromptPlugin<string, any>[] = [],\n> {\n  /**\n   * the prompt name\n   */\n  readonly name: string;\n\n  /**\n   * the prompt description\n   */\n  readonly description: string;\n\n  /**\n   * the chat history\n   */\n  readonly messages: IMemory;\n\n  /**\n   * the registered functions\n   */\n  readonly functions: Array<Function>;\n\n  /**\n   * the chat model\n   */\n  plugins: TChatPromptPlugins;\n\n  /**\n   * add another chat prompt as a\n   */\n  use(prompt: IChatPrompt): this;\n  use(name: string, prompt: IChatPrompt): this;\n\n  /**\n   * add a function that can be called\n   * by the model\n   */\n  function(name: string, description: string, handler: FunctionHandler): this;\n  function(name: string, description: string, parameters: Schema, handler: FunctionHandler): this;\n\n  usePlugin<TPluginName extends TChatPromptPlugins[number]['name']>(\n    name: TPluginName,\n    args: Extract<TChatPromptPlugins[number], { name: TPluginName }>['onUsePlugin'] extends\n      | ((args: infer U) => void)\n      | undefined\n      ? U\n      : never\n  ): this;\n\n  /**\n   * call a function\n   */\n  call<A extends Record<string, any>, R = any>(name: string, args?: A): Promise<R>;\n\n  /**\n   * send a message to the model and get a response\n   */\n  send(\n    input: string | ContentPart[],\n    options?: ChatPromptSendOptions<TOptions>\n  ): Promise<Pick<ModelMessage, 'content'> & Omit<ModelMessage, 'content'>>;\n}\n\nexport type ChatPromptPlugin<TPluginName extends string, TPluginUseArgs extends {}> = IAiPlugin<\n  TPluginName,\n  TPluginUseArgs,\n  Parameters<IChatPrompt['send']>[0],\n  ReturnType<IChatPrompt['send']>\n> & {\n  /**\n   * Optionally passed in to modify the functions array that\n   * is passed to the model\n   * @param functions\n   * @returns Functions\n   */\n  onBuildFunctions?: (functions: Function[]) => Function[] | Promise<Function[]>;\n};\n\n/**\n * a prompt that can interface with a\n * chat model that provides utility like\n * streaming and function calling\n */\nexport class ChatPrompt<\n  TOptions extends Record<string, any> = Record<string, any>,\n  TChatPromptPlugins extends readonly ChatPromptPlugin<string, any>[] = [],\n> implements IChatPrompt<TOptions, TChatPromptPlugins>\n{\n  get name() {\n    return this._name;\n  }\n  protected readonly _name: string;\n\n  get description() {\n    return this._description;\n  }\n  protected readonly _description: string;\n\n  get messages() {\n    return this._messages;\n  }\n  protected readonly _messages: IMemory;\n\n  get functions() {\n    return Object.values(this._functions);\n  }\n  protected readonly _functions: Record<string, Function> = {};\n\n  get plugins() {\n    return this._plugins;\n  }\n  protected readonly _plugins: TChatPromptPlugins;\n\n  protected readonly _role: 'system' | 'user';\n  protected readonly _template: ITemplate;\n  protected readonly _model: IChatModel<TOptions>;\n\n  constructor(options: ChatPromptOptions<TOptions>, plugins?: TChatPromptPlugins) {\n    this._name = options.name || 'chat';\n    this._description = options.description || 'an agent you can chat with';\n    this._role = options.role || 'system';\n    this._model = options.model;\n    this._template = Array.isArray(options.instructions)\n      ? new StringTemplate(options.instructions.join('\\n'))\n      : typeof options.instructions !== 'object'\n        ? new StringTemplate(options.instructions)\n        : options.instructions;\n\n    this._messages =\n      typeof options.messages === 'object' && !Array.isArray(options.messages)\n        ? options.messages\n        : new LocalMemory({ messages: options.messages || [] });\n\n    this._plugins = plugins || ([] as unknown as TChatPromptPlugins);\n  }\n\n  use(prompt: ChatPrompt): this;\n  use(name: string, prompt: ChatPrompt): this;\n  use(...args: any[]) {\n    const prompt: ChatPrompt = args.length === 1 ? args[0] : args[1];\n    const name: string = args.length === 1 ? prompt.name : args[0];\n    this._functions[name] = {\n      name,\n      description: prompt.description,\n      parameters: {\n        type: 'object',\n        properties: {\n          text: {\n            type: 'string',\n            description: 'the text to send to the assistant',\n          },\n        },\n        required: ['text'],\n      },\n      handler: ({ text }: { text: string }) => {\n        return prompt.send(text);\n      },\n    };\n\n    return this;\n  }\n\n  function(name: string, description: string, handler: FunctionHandler): this;\n  function(name: string, description: string, parameters: Schema, handler: FunctionHandler): this;\n  function(...args: any[]) {\n    const name: string = args[0];\n    const description: string = args[1];\n    const parameters: Schema | null = args.length === 3 ? null : args[2];\n    const handler: FunctionHandler = args[args.length - 1];\n    this._functions[name] = {\n      name,\n      description,\n      parameters: parameters || {},\n      handler,\n    };\n\n    return this;\n  }\n\n  usePlugin<K extends TChatPromptPlugins[number]['name']>(\n    name: K,\n    args: Extract<TChatPromptPlugins[number], { name: K }>['onUsePlugin'] extends\n      | ((args: infer U) => void)\n      | undefined\n      ? U\n      : never\n  ): this {\n    const plugin = this._plugins.find((p) => p.name === name);\n    if (!plugin) {\n      throw new Error(`Plugin \"${name}\" not found`);\n    }\n\n    if (plugin.onUsePlugin) {\n      plugin.onUsePlugin(args);\n    }\n\n    return this;\n  }\n\n  async call<A extends { [key: string]: any }, R = any>(name: string, args?: A): Promise<R> {\n    const fn = this._functions[name];\n\n    if (!fn) {\n      throw new Error(`function \"${name}\" not found`);\n    }\n\n    return await fn.handler(args || {});\n  }\n\n  async send(input: string | ContentPart[], options: ChatPromptSendOptions<TOptions> = {}) {\n    for (const plugin of this.plugins) {\n      if (plugin.onBeforeSend) {\n        input = await plugin.onBeforeSend(input);\n      }\n    }\n\n    const { onChunk } = options;\n\n    if (typeof input === 'string') {\n      input = input.trim();\n    }\n\n    const messages = !Array.isArray(options.messages)\n      ? options.messages || this._messages\n      : new LocalMemory({ messages: options.messages || [] });\n\n    let buffer = '';\n    let system: SystemMessage | UserMessage | undefined = undefined;\n    const prompt = await this._template.render();\n\n    if (prompt) {\n      system = {\n        role: this._role,\n        content: prompt,\n      };\n    }\n\n    let functions = Object.values(this._functions);\n    const pluginsWithOnBuildFunctions = this._plugins.filter(\n      (plugin): plugin is WithRequired<TChatPromptPlugins[number], 'onBuildFunctions'> =>\n        plugin.onBuildFunctions != null\n    );\n    for (const plugin of pluginsWithOnBuildFunctions) {\n      functions = await plugin.onBuildFunctions(functions);\n    }\n\n    const fnMap = functions.reduce(\n      (acc, fn) => {\n        acc[fn.name] = fn;\n        return acc;\n      },\n      {} as Record<string, Function>\n    );\n\n    const res = await this._model.send(\n      {\n        role: 'user',\n        content: input,\n      },\n      {\n        system,\n        messages,\n        request: options.request,\n        functions: fnMap,\n        onChunk: async (chunk) => {\n          if (!chunk || !onChunk) return;\n          buffer += chunk;\n\n          try {\n            await onChunk(buffer);\n            buffer = '';\n          } catch (err) {\n            return;\n          }\n        },\n      }\n    );\n\n    let output: Awaited<ReturnType<typeof this._model.send>> = {\n      ...res,\n      content: res.content || '',\n    };\n    for (const plugin of this.plugins) {\n      if (plugin.onAfterSend) {\n        output = await plugin.onAfterSend(output);\n      }\n    }\n\n    return output;\n  }\n}\n"]}

package/dist/prompts/chat.mjs

@@ -0,0 +1,160 @@
+import { LocalMemory } from '../local-memory';
+import { StringTemplate } from '../templates';
+
+class ChatPrompt {
+  get name() {
+    return this._name;
+  }
+  _name;
+  get description() {
+    return this._description;
+  }
+  _description;
+  get messages() {
+    return this._messages;
+  }
+  _messages;
+  get functions() {
+    return Object.values(this._functions);
+  }
+  _functions = {};
+  get plugins() {
+    return this._plugins;
+  }
+  _plugins;
+  _role;
+  _template;
+  _model;
+  constructor(options, plugins) {
+    this._name = options.name || "chat";
+    this._description = options.description || "an agent you can chat with";
+    this._role = options.role || "system";
+    this._model = options.model;
+    this._template = Array.isArray(options.instructions) ? new StringTemplate(options.instructions.join("\n")) : typeof options.instructions !== "object" ? new StringTemplate(options.instructions) : options.instructions;
+    this._messages = typeof options.messages === "object" && !Array.isArray(options.messages) ? options.messages : new LocalMemory({ messages: options.messages || [] });
+    this._plugins = plugins || [];
+  }
+  use(...args) {
+    const prompt = args.length === 1 ? args[0] : args[1];
+    const name = args.length === 1 ? prompt.name : args[0];
+    this._functions[name] = {
+      name,
+      description: prompt.description,
+      parameters: {
+        type: "object",
+        properties: {
+          text: {
+            type: "string",
+            description: "the text to send to the assistant"
+          }
+        },
+        required: ["text"]
+      },
+      handler: ({ text }) => {
+        return prompt.send(text);
+      }
+    };
+    return this;
+  }
+  function(...args) {
+    const name = args[0];
+    const description = args[1];
+    const parameters = args.length === 3 ? null : args[2];
+    const handler = args[args.length - 1];
+    this._functions[name] = {
+      name,
+      description,
+      parameters: parameters || {},
+      handler
+    };
+    return this;
+  }
+  usePlugin(name, args) {
+    const plugin = this._plugins.find((p) => p.name === name);
+    if (!plugin) {
+      throw new Error(`Plugin "${name}" not found`);
+    }
+    if (plugin.onUsePlugin) {
+      plugin.onUsePlugin(args);
+    }
+    return this;
+  }
+  async call(name, args) {
+    const fn = this._functions[name];
+    if (!fn) {
+      throw new Error(`function "${name}" not found`);
+    }
+    return await fn.handler(args || {});
+  }
+  async send(input, options = {}) {
+    for (const plugin of this.plugins) {
+      if (plugin.onBeforeSend) {
+        input = await plugin.onBeforeSend(input);
+      }
+    }
+    const { onChunk } = options;
+    if (typeof input === "string") {
+      input = input.trim();
+    }
+    const messages = !Array.isArray(options.messages) ? options.messages || this._messages : new LocalMemory({ messages: options.messages || [] });
+    let buffer = "";
+    let system = void 0;
+    const prompt = await this._template.render();
+    if (prompt) {
+      system = {
+        role: this._role,
+        content: prompt
+      };
+    }
+    let functions = Object.values(this._functions);
+    const pluginsWithOnBuildFunctions = this._plugins.filter(
+      (plugin) => plugin.onBuildFunctions != null
+    );
+    for (const plugin of pluginsWithOnBuildFunctions) {
+      functions = await plugin.onBuildFunctions(functions);
+    }
+    const fnMap = functions.reduce(
+      (acc, fn) => {
+        acc[fn.name] = fn;
+        return acc;
+      },
+      {}
+    );
+    const res = await this._model.send(
+      {
+        role: "user",
+        content: input
+      },
+      {
+        system,
+        messages,
+        request: options.request,
+        functions: fnMap,
+        onChunk: async (chunk) => {
+          if (!chunk || !onChunk) return;
+          buffer += chunk;
+          try {
+            await onChunk(buffer);
+            buffer = "";
+          } catch (err) {
+            return;
+          }
+        }
+      }
+    );
+    let output = {
+      ...res,
+      content: res.content || ""
+    };
+    for (const plugin of this.plugins) {
+      if (plugin.onAfterSend) {
+        output = await plugin.onAfterSend(output);
+      }
+    }
+    return output;
+  }
+}
+
+export { ChatPrompt };
+//# sourceMappingURL=chat.mjs.map
+//# sourceMappingURL=chat.mjs.map

package/dist/prompts/chat.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/prompts/chat.ts"],"names":[],"mappings":";;;AA0JO,MAAM,UAIb,CAAA;AAAA,EACE,IAAI,IAAO,GAAA;AACT,IAAA,OAAO,IAAK,CAAA,KAAA;AAAA;AACd,EACmB,KAAA;AAAA,EAEnB,IAAI,WAAc,GAAA;AAChB,IAAA,OAAO,IAAK,CAAA,YAAA;AAAA;AACd,EACmB,YAAA;AAAA,EAEnB,IAAI,QAAW,GAAA;AACb,IAAA,OAAO,IAAK,CAAA,SAAA;AAAA;AACd,EACmB,SAAA;AAAA,EAEnB,IAAI,SAAY,GAAA;AACd,IAAO,OAAA,MAAA,CAAO,MAAO,CAAA,IAAA,CAAK,UAAU,CAAA;AAAA;AACtC,EACmB,aAAuC,EAAC;AAAA,EAE3D,IAAI,OAAU,GAAA;AACZ,IAAA,OAAO,IAAK,CAAA,QAAA;AAAA;AACd,EACmB,QAAA;AAAA,EAEA,KAAA;AAAA,EACA,SAAA;AAAA,EACA,MAAA;AAAA,EAEnB,WAAA,CAAY,SAAsC,OAA8B,EAAA;AAC9E,IAAK,IAAA,CAAA,KAAA,GAAQ,QAAQ,IAAQ,IAAA,MAAA;AAC7B,IAAK,IAAA,CAAA,YAAA,GAAe,QAAQ,WAAe,IAAA,4BAAA;AAC3C,IAAK,IAAA,CAAA,KAAA,GAAQ,QAAQ,IAAQ,IAAA,QAAA;AAC7B,IAAA,IAAA,CAAK,SAAS,OAAQ,CAAA,KAAA;AACtB,IAAK,IAAA,CAAA,SAAA,GAAY,MAAM,OAAQ,CAAA,OAAA,CAAQ,YAAY,CAC/C,GAAA,IAAI,cAAe,CAAA,OAAA,CAAQ,YAAa,CAAA,IAAA,CAAK,IAAI,CAAC,CAAA,GAClD,OAAO,OAAA,CAAQ,YAAiB,KAAA,QAAA,GAC9B,IAAI,cAAe,CAAA,OAAA,CAAQ,YAAY,CAAA,GACvC,OAAQ,CAAA,YAAA;AAEd,IAAK,IAAA,CAAA,SAAA,GACH,OAAO,OAAQ,CAAA,QAAA,KAAa,YAAY,CAAC,KAAA,CAAM,QAAQ,OAAQ,CAAA,QAAQ,IACnE,OAAQ,CAAA,QAAA,GACR,IAAI,WAAY,CAAA,EAAE,UAAU,OAAQ,CAAA,QAAA,IAAY,EAAC,EAAG,CAAA;AAE1D,IAAK,IAAA,CAAA,QAAA,GAAW,WAAY,EAAC;AAAA;AAC/B,EAIA,OAAO,IAAa,EAAA;AAClB,IAAM,MAAA,MAAA,GAAqB,KAAK,MAAW,KAAA,CAAA,GAAI,KAAK,CAAC,CAAA,GAAI,KAAK,CAAC,CAAA;AAC/D,IAAA,MAAM,OAAe,IAAK,CAAA,MAAA,KAAW,IAAI,MAAO,CAAA,IAAA,GAAO,KAAK,CAAC,CAAA;AAC7D,IAAK,IAAA,CAAA,UAAA,CAAW,IAAI,CAAI,GAAA;AAAA,MACtB,IAAA;AAAA,MACA,aAAa,MAAO,CAAA,WAAA;AAAA,MACpB,UAAY,EAAA;AAAA,QACV,IAAM,EAAA,QAAA;AAAA,QACN,UAAY,EAAA;AAAA,UACV,IAAM,EAAA;AAAA,YACJ,IAAM,EAAA,QAAA;AAAA,YACN,WAAa,EAAA;AAAA;AACf,SACF;AAAA,QACA,QAAA,EAAU,CAAC,MAAM;AAAA,OACnB;AAAA,MACA,OAAS,EAAA,CAAC,EAAE,IAAA,EAA6B,KAAA;AACvC,QAAO,OAAA,MAAA,CAAO,KAAK,IAAI,CAAA;AAAA;AACzB,KACF;AAEA,IAAO,OAAA,IAAA;AAAA;AACT,EAIA,YAAY,IAAa,EAAA;AACvB,IAAM,MAAA,IAAA,GAAe,KAAK,CAAC,CAAA;AAC3B,IAAM,MAAA,WAAA,GAAsB,KAAK,CAAC,CAAA;AAClC,IAAA,MAAM,aAA4B,IAAK,CAAA,MAAA,KAAW,CAAI,GAAA,IAAA,GAAO,KAAK,CAAC,CAAA;AACnE,IAAA,MAAM,OAA2B,GAAA,IAAA,CAAK,IAAK,CAAA,MAAA,GAAS,CAAC,CAAA;AACrD,IAAK,IAAA,CAAA,UAAA,CAAW,IAAI,CAAI,GAAA;AAAA,MACtB,IAAA;AAAA,MACA,WAAA;AAAA,MACA,UAAA,EAAY,cAAc,EAAC;AAAA,MAC3B;AAAA,KACF;AAEA,IAAO,OAAA,IAAA;AAAA;AACT,EAEA,SAAA,CACE,MACA,IAKM,EAAA;AACN,IAAM,MAAA,MAAA,GAAS,KAAK,QAAS,CAAA,IAAA,CAAK,CAAC,CAAM,KAAA,CAAA,CAAE,SAAS,IAAI,CAAA;AACxD,IAAA,IAAI,CAAC,MAAQ,EAAA;AACX,MAAA,MAAM,IAAI,KAAA,CAAM,CAAW,QAAA,EAAA,IAAI,CAAa,WAAA,CAAA,CAAA;AAAA;AAG9C,IAAA,IAAI,OAAO,WAAa,EAAA;AACtB,MAAA,MAAA,CAAO,YAAY,IAAI,CAAA;AAAA;AAGzB,IAAO,OAAA,IAAA;AAAA;AACT,EAEA,MAAM,IAAgD,CAAA,IAAA,EAAc,IAAsB,EAAA;AACxF,IAAM,MAAA,EAAA,GAAK,IAAK,CAAA,UAAA,CAAW,IAAI,CAAA;AAE/B,IAAA,IAAI,CAAC,EAAI,EAAA;AACP,MAAA,MAAM,IAAI,KAAA,CAAM,CAAa,UAAA,EAAA,IAAI,CAAa,WAAA,CAAA,CAAA;AAAA;AAGhD,IAAA,OAAO,MAAM,EAAA,CAAG,OAAQ,CAAA,IAAA,IAAQ,EAAE,CAAA;AAAA;AACpC,EAEA,MAAM,IAAA,CAAK,KAA+B,EAAA,OAAA,GAA2C,EAAI,EAAA;AACvF,IAAW,KAAA,MAAA,MAAA,IAAU,KAAK,OAAS,EAAA;AACjC,MAAA,IAAI,OAAO,YAAc,EAAA;AACvB,QAAQ,KAAA,GAAA,MAAM,MAAO,CAAA,YAAA,CAAa,KAAK,CAAA;AAAA;AACzC;AAGF,IAAM,MAAA,EAAE,SAAY,GAAA,OAAA;AAEpB,IAAI,IAAA,OAAO,UAAU,QAAU,EAAA;AAC7B,MAAA,KAAA,GAAQ,MAAM,IAAK,EAAA;AAAA;AAGrB,IAAA,MAAM,WAAW,CAAC,KAAA,CAAM,QAAQ,OAAQ,CAAA,QAAQ,IAC5C,OAAQ,CAAA,QAAA,IAAY,KAAK,SACzB,GAAA,IAAI,YAAY,EAAE,QAAA,EAAU,QAAQ,QAAY,IAAA,IAAI,CAAA;AAExD,IAAA,IAAI,MAAS,GAAA,EAAA;AACb,IAAA,IAAI,MAAkD,GAAA,MAAA;AACtD,IAAA,MAAM,MAAS,GAAA,MAAM,IAAK,CAAA,SAAA,CAAU,MAAO,EAAA;AAE3C,IAAA,IAAI,MAAQ,EAAA;AACV,MAAS,MAAA,GAAA;AAAA,QACP,MAAM,IAAK,CAAA,KAAA;AAAA,QACX,OAAS,EAAA;AAAA,OACX;AAAA;AAGF,IAAA,IAAI,SAAY,GAAA,MAAA,CAAO,MAAO,CAAA,IAAA,CAAK,UAAU,CAAA;AAC7C,IAAM,MAAA,2BAAA,GAA8B,KAAK,QAAS,CAAA,MAAA;AAAA,MAChD,CAAC,MACC,KAAA,MAAA,CAAO,gBAAoB,IAAA;AAAA,KAC/B;AACA,IAAA,KAAA,MAAW,UAAU,2BAA6B,EAAA;AAChD,MAAY,SAAA,GAAA,MAAM,MAAO,CAAA,gBAAA,CAAiB,SAAS,CAAA;AAAA;AAGrD,IAAA,MAAM,QAAQ,SAAU,CAAA,MAAA;AAAA,MACtB,CAAC,KAAK,EAAO,KAAA;AACX,QAAI,GAAA,CAAA,EAAA,CAAG,IAAI,CAAI,GAAA,EAAA;AACf,QAAO,OAAA,GAAA;AAAA,OACT;AAAA,MACA;AAAC,KACH;AAEA,IAAM,MAAA,GAAA,GAAM,MAAM,IAAA,CAAK,MAAO,CAAA,IAAA;AAAA,MAC5B;AAAA,QACE,IAAM,EAAA,MAAA;AAAA,QACN,OAAS,EAAA;AAAA,OACX;AAAA,MACA;AAAA,QACE,MAAA;AAAA,QACA,QAAA;AAAA,QACA,SAAS,OAAQ,CAAA,OAAA;AAAA,QACjB,SAAW,EAAA,KAAA;AAAA,QACX,OAAA,EAAS,OAAO,KAAU,KAAA;AACxB,UAAI,IAAA,CAAC,KAAS,IAAA,CAAC,OAAS,EAAA;AACxB,UAAU,MAAA,IAAA,KAAA;AAEV,UAAI,IAAA;AACF,YAAA,MAAM,QAAQ,MAAM,CAAA;AACpB,YAAS,MAAA,GAAA,EAAA;AAAA,mBACF,GAAK,EAAA;AACZ,YAAA;AAAA;AACF;AACF;AACF,KACF;AAEA,IAAA,IAAI,MAAuD,GAAA;AAAA,MACzD,GAAG,GAAA;AAAA,MACH,OAAA,EAAS,IAAI,OAAW,IAAA;AAAA,KAC1B;AACA,IAAW,KAAA,MAAA,MAAA,IAAU,KAAK,OAAS,EAAA;AACjC,MAAA,IAAI,OAAO,WAAa,EAAA;AACtB,QAAS,MAAA,GAAA,MAAM,MAAO,CAAA,WAAA,CAAY,MAAM,CAAA;AAAA;AAC1C;AAGF,IAAO,OAAA,MAAA;AAAA;AAEX","file":"chat.mjs","sourcesContent":["import { Function, FunctionHandler } from '../function';\nimport { LocalMemory } from '../local-memory';\nimport { IMemory } from '../memory';\nimport { ContentPart, Message, ModelMessage, SystemMessage, UserMessage } from '../message';\nimport { IChatModel, TextChunkHandler } from '../models';\nimport { Schema } from '../schema';\nimport { ITemplate } from '../template';\nimport { StringTemplate } from '../templates';\nimport { WithRequired } from '../utils/types';\nimport { IAiPlugin } from './plugin';\n\nexport type ChatPromptOptions<TOptions extends Record<string, any> = Record<string, any>> = {\n  /**\n   * the name of the prompt\n   */\n  readonly name?: string;\n\n  /**\n   * the description of the prompt\n   */\n  readonly description?: string;\n\n  /**\n   * the model to send messages to\n   */\n  readonly model: IChatModel<TOptions>;\n\n  /**\n   * the defining characteristics/objective\n   * of the prompt. This is commonly used to provide a system prompt.\n   * If you supply the system prompt as part of the messages,\n   * you do not need to supply this option.\n   */\n  readonly instructions?: string | string[] | ITemplate;\n\n  /**\n   * the `role` of the initial message\n   */\n  readonly role?: 'system' | 'user';\n\n  /**\n   * the conversation history\n   */\n  readonly messages?: Message[] | IMemory;\n};\n\nexport type ChatPromptSendOptions<TOptions extends Record<string, any> = Record<string, any>> = {\n  /**\n   * the conversation history\n   */\n  readonly messages?: Message[] | IMemory;\n\n  /**\n   * the models request options\n   */\n  readonly request?: TOptions;\n\n  /**\n   * the callback to be called for each\n   * stream chunk\n   */\n  readonly onChunk?: TextChunkHandler;\n};\n\n/**\n * a prompt that can interface with a\n * chat model that provides utility like\n * streaming and function calling\n */\nexport interface IChatPrompt<\n  TOptions extends Record<string, any> = Record<string, any>,\n  TChatPromptPlugins extends readonly ChatPromptPlugin<string, any>[] = [],\n> {\n  /**\n   * the prompt name\n   */\n  readonly name: string;\n\n  /**\n   * the prompt description\n   */\n  readonly description: string;\n\n  /**\n   * the chat history\n   */\n  readonly messages: IMemory;\n\n  /**\n   * the registered functions\n   */\n  readonly functions: Array<Function>;\n\n  /**\n   * the chat model\n   */\n  plugins: TChatPromptPlugins;\n\n  /**\n   * add another chat prompt as a\n   */\n  use(prompt: IChatPrompt): this;\n  use(name: string, prompt: IChatPrompt): this;\n\n  /**\n   * add a function that can be called\n   * by the model\n   */\n  function(name: string, description: string, handler: FunctionHandler): this;\n  function(name: string, description: string, parameters: Schema, handler: FunctionHandler): this;\n\n  usePlugin<TPluginName extends TChatPromptPlugins[number]['name']>(\n    name: TPluginName,\n    args: Extract<TChatPromptPlugins[number], { name: TPluginName }>['onUsePlugin'] extends\n      | ((args: infer U) => void)\n      | undefined\n      ? U\n      : never\n  ): this;\n\n  /**\n   * call a function\n   */\n  call<A extends Record<string, any>, R = any>(name: string, args?: A): Promise<R>;\n\n  /**\n   * send a message to the model and get a response\n   */\n  send(\n    input: string | ContentPart[],\n    options?: ChatPromptSendOptions<TOptions>\n  ): Promise<Pick<ModelMessage, 'content'> & Omit<ModelMessage, 'content'>>;\n}\n\nexport type ChatPromptPlugin<TPluginName extends string, TPluginUseArgs extends {}> = IAiPlugin<\n  TPluginName,\n  TPluginUseArgs,\n  Parameters<IChatPrompt['send']>[0],\n  ReturnType<IChatPrompt['send']>\n> & {\n  /**\n   * Optionally passed in to modify the functions array that\n   * is passed to the model\n   * @param functions\n   * @returns Functions\n   */\n  onBuildFunctions?: (functions: Function[]) => Function[] | Promise<Function[]>;\n};\n\n/**\n * a prompt that can interface with a\n * chat model that provides utility like\n * streaming and function calling\n */\nexport class ChatPrompt<\n  TOptions extends Record<string, any> = Record<string, any>,\n  TChatPromptPlugins extends readonly ChatPromptPlugin<string, any>[] = [],\n> implements IChatPrompt<TOptions, TChatPromptPlugins>\n{\n  get name() {\n    return this._name;\n  }\n  protected readonly _name: string;\n\n  get description() {\n    return this._description;\n  }\n  protected readonly _description: string;\n\n  get messages() {\n    return this._messages;\n  }\n  protected readonly _messages: IMemory;\n\n  get functions() {\n    return Object.values(this._functions);\n  }\n  protected readonly _functions: Record<string, Function> = {};\n\n  get plugins() {\n    return this._plugins;\n  }\n  protected readonly _plugins: TChatPromptPlugins;\n\n  protected readonly _role: 'system' | 'user';\n  protected readonly _template: ITemplate;\n  protected readonly _model: IChatModel<TOptions>;\n\n  constructor(options: ChatPromptOptions<TOptions>, plugins?: TChatPromptPlugins) {\n    this._name = options.name || 'chat';\n    this._description = options.description || 'an agent you can chat with';\n    this._role = options.role || 'system';\n    this._model = options.model;\n    this._template = Array.isArray(options.instructions)\n      ? new StringTemplate(options.instructions.join('\\n'))\n      : typeof options.instructions !== 'object'\n        ? new StringTemplate(options.instructions)\n        : options.instructions;\n\n    this._messages =\n      typeof options.messages === 'object' && !Array.isArray(options.messages)\n        ? options.messages\n        : new LocalMemory({ messages: options.messages || [] });\n\n    this._plugins = plugins || ([] as unknown as TChatPromptPlugins);\n  }\n\n  use(prompt: ChatPrompt): this;\n  use(name: string, prompt: ChatPrompt): this;\n  use(...args: any[]) {\n    const prompt: ChatPrompt = args.length === 1 ? args[0] : args[1];\n    const name: string = args.length === 1 ? prompt.name : args[0];\n    this._functions[name] = {\n      name,\n      description: prompt.description,\n      parameters: {\n        type: 'object',\n        properties: {\n          text: {\n            type: 'string',\n            description: 'the text to send to the assistant',\n          },\n        },\n        required: ['text'],\n      },\n      handler: ({ text }: { text: string }) => {\n        return prompt.send(text);\n      },\n    };\n\n    return this;\n  }\n\n  function(name: string, description: string, handler: FunctionHandler): this;\n  function(name: string, description: string, parameters: Schema, handler: FunctionHandler): this;\n  function(...args: any[]) {\n    const name: string = args[0];\n    const description: string = args[1];\n    const parameters: Schema | null = args.length === 3 ? null : args[2];\n    const handler: FunctionHandler = args[args.length - 1];\n    this._functions[name] = {\n      name,\n      description,\n      parameters: parameters || {},\n      handler,\n    };\n\n    return this;\n  }\n\n  usePlugin<K extends TChatPromptPlugins[number]['name']>(\n    name: K,\n    args: Extract<TChatPromptPlugins[number], { name: K }>['onUsePlugin'] extends\n      | ((args: infer U) => void)\n      | undefined\n      ? U\n      : never\n  ): this {\n    const plugin = this._plugins.find((p) => p.name === name);\n    if (!plugin) {\n      throw new Error(`Plugin \"${name}\" not found`);\n    }\n\n    if (plugin.onUsePlugin) {\n      plugin.onUsePlugin(args);\n    }\n\n    return this;\n  }\n\n  async call<A extends { [key: string]: any }, R = any>(name: string, args?: A): Promise<R> {\n    const fn = this._functions[name];\n\n    if (!fn) {\n      throw new Error(`function \"${name}\" not found`);\n    }\n\n    return await fn.handler(args || {});\n  }\n\n  async send(input: string | ContentPart[], options: ChatPromptSendOptions<TOptions> = {}) {\n    for (const plugin of this.plugins) {\n      if (plugin.onBeforeSend) {\n        input = await plugin.onBeforeSend(input);\n      }\n    }\n\n    const { onChunk } = options;\n\n    if (typeof input === 'string') {\n      input = input.trim();\n    }\n\n    const messages = !Array.isArray(options.messages)\n      ? options.messages || this._messages\n      : new LocalMemory({ messages: options.messages || [] });\n\n    let buffer = '';\n    let system: SystemMessage | UserMessage | undefined = undefined;\n    const prompt = await this._template.render();\n\n    if (prompt) {\n      system = {\n        role: this._role,\n        content: prompt,\n      };\n    }\n\n    let functions = Object.values(this._functions);\n    const pluginsWithOnBuildFunctions = this._plugins.filter(\n      (plugin): plugin is WithRequired<TChatPromptPlugins[number], 'onBuildFunctions'> =>\n        plugin.onBuildFunctions != null\n    );\n    for (const plugin of pluginsWithOnBuildFunctions) {\n      functions = await plugin.onBuildFunctions(functions);\n    }\n\n    const fnMap = functions.reduce(\n      (acc, fn) => {\n        acc[fn.name] = fn;\n        return acc;\n      },\n      {} as Record<string, Function>\n    );\n\n    const res = await this._model.send(\n      {\n        role: 'user',\n        content: input,\n      },\n      {\n        system,\n        messages,\n        request: options.request,\n        functions: fnMap,\n        onChunk: async (chunk) => {\n          if (!chunk || !onChunk) return;\n          buffer += chunk;\n\n          try {\n            await onChunk(buffer);\n            buffer = '';\n          } catch (err) {\n            return;\n          }\n        },\n      }\n    );\n\n    let output: Awaited<ReturnType<typeof this._model.send>> = {\n      ...res,\n      content: res.content || '',\n    };\n    for (const plugin of this.plugins) {\n      if (plugin.onAfterSend) {\n        output = await plugin.onAfterSend(output);\n      }\n    }\n\n    return output;\n  }\n}\n"]}