@microsoft/teams.ai 0.2.12 → 0.2.13

package/dist/models/index.d.mts

@@ -1,9 +1,9 @@
 import { IAudioModel } from './audio.mjs';
 export { AudioToTextParams, TextToAudioParams } from './audio.mjs';
-import { IImageModel } from './image.mjs';
-export { TextToImageParams } from './image.mjs';
 import { IChatModel } from './chat.mjs';
 export { ChatSendOptions, TextChunkHandler } from './chat.mjs';
+import { IImageModel } from './image.mjs';
+export { TextToImageParams } from './image.mjs';
 import '../function.mjs';
 import '../schema.mjs';
 import '../memory.mjs';

package/dist/models/index.d.ts

@@ -1,9 +1,9 @@
 import { IAudioModel } from './audio.js';
 export { AudioToTextParams, TextToAudioParams } from './audio.js';
-import { IImageModel } from './image.js';
-export { TextToImageParams } from './image.js';
 import { IChatModel } from './chat.js';
 export { ChatSendOptions, TextChunkHandler } from './chat.js';
+import { IImageModel } from './image.js';
+export { TextToImageParams } from './image.js';
 import '../function.js';
 import '../schema.js';
 import '../memory.js';

package/dist/prompts/chat.js.map

@@ -1 +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"]}
+{"version":3,"sources":["../../src/prompts/chat.ts"],"names":["StringTemplate","LocalMemory"],"mappings":";;;;;AA2JO,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';\r\nimport { LocalMemory } from '../local-memory';\r\nimport { IMemory } from '../memory';\r\nimport { ContentPart, Message, ModelMessage, SystemMessage, UserMessage } from '../message';\r\nimport { IChatModel, TextChunkHandler } from '../models';\r\nimport { Schema } from '../schema';\r\nimport { ITemplate } from '../template';\r\nimport { StringTemplate } from '../templates';\r\nimport { WithRequired } from '../utils/types';\r\n\r\nimport { IAiPlugin } from './plugin';\r\n\r\nexport type ChatPromptOptions<TOptions extends Record<string, any> = Record<string, any>> = {\r\n  /**\r\n   * the name of the prompt\r\n   */\r\n  readonly name?: string;\r\n\r\n  /**\r\n   * the description of the prompt\r\n   */\r\n  readonly description?: string;\r\n\r\n  /**\r\n   * the model to send messages to\r\n   */\r\n  readonly model: IChatModel<TOptions>;\r\n\r\n  /**\r\n   * the defining characteristics/objective\r\n   * of the prompt. This is commonly used to provide a system prompt.\r\n   * If you supply the system prompt as part of the messages,\r\n   * you do not need to supply this option.\r\n   */\r\n  readonly instructions?: string | string[] | ITemplate;\r\n\r\n  /**\r\n   * the `role` of the initial message\r\n   */\r\n  readonly role?: 'system' | 'user';\r\n\r\n  /**\r\n   * the conversation history\r\n   */\r\n  readonly messages?: Message[] | IMemory;\r\n};\r\n\r\nexport type ChatPromptSendOptions<TOptions extends Record<string, any> = Record<string, any>> = {\r\n  /**\r\n   * the conversation history\r\n   */\r\n  readonly messages?: Message[] | IMemory;\r\n\r\n  /**\r\n   * the models request options\r\n   */\r\n  readonly request?: TOptions;\r\n\r\n  /**\r\n   * the callback to be called for each\r\n   * stream chunk\r\n   */\r\n  readonly onChunk?: TextChunkHandler;\r\n};\r\n\r\n/**\r\n * a prompt that can interface with a\r\n * chat model that provides utility like\r\n * streaming and function calling\r\n */\r\nexport interface IChatPrompt<\r\n  TOptions extends Record<string, any> = Record<string, any>,\r\n  TChatPromptPlugins extends readonly ChatPromptPlugin<string, any>[] = [],\r\n> {\r\n  /**\r\n   * the prompt name\r\n   */\r\n  readonly name: string;\r\n\r\n  /**\r\n   * the prompt description\r\n   */\r\n  readonly description: string;\r\n\r\n  /**\r\n   * the chat history\r\n   */\r\n  readonly messages: IMemory;\r\n\r\n  /**\r\n   * the registered functions\r\n   */\r\n  readonly functions: Array<Function>;\r\n\r\n  /**\r\n   * the chat model\r\n   */\r\n  plugins: TChatPromptPlugins;\r\n\r\n  /**\r\n   * add another chat prompt as a\r\n   */\r\n  use(prompt: IChatPrompt): this;\r\n  use(name: string, prompt: IChatPrompt): this;\r\n\r\n  /**\r\n   * add a function that can be called\r\n   * by the model\r\n   */\r\n  function(name: string, description: string, handler: FunctionHandler): this;\r\n  function(name: string, description: string, parameters: Schema, handler: FunctionHandler): this;\r\n\r\n  usePlugin<TPluginName extends TChatPromptPlugins[number]['name']>(\r\n    name: TPluginName,\r\n    args: Extract<TChatPromptPlugins[number], { name: TPluginName }>['onUsePlugin'] extends\r\n      | ((args: infer U) => void)\r\n      | undefined\r\n      ? U\r\n      : never\r\n  ): this;\r\n\r\n  /**\r\n   * call a function\r\n   */\r\n  call<A extends Record<string, any>, R = any>(name: string, args?: A): Promise<R>;\r\n\r\n  /**\r\n   * send a message to the model and get a response\r\n   */\r\n  send(\r\n    input: string | ContentPart[],\r\n    options?: ChatPromptSendOptions<TOptions>\r\n  ): Promise<Pick<ModelMessage, 'content'> & Omit<ModelMessage, 'content'>>;\r\n}\r\n\r\nexport type ChatPromptPlugin<TPluginName extends string, TPluginUseArgs extends {}> = IAiPlugin<\r\n  TPluginName,\r\n  TPluginUseArgs,\r\n  Parameters<IChatPrompt['send']>[0],\r\n  ReturnType<IChatPrompt['send']>\r\n> & {\r\n  /**\r\n   * Optionally passed in to modify the functions array that\r\n   * is passed to the model\r\n   * @param functions\r\n   * @returns Functions\r\n   */\r\n  onBuildFunctions?: (functions: Function[]) => Function[] | Promise<Function[]>;\r\n};\r\n\r\n/**\r\n * a prompt that can interface with a\r\n * chat model that provides utility like\r\n * streaming and function calling\r\n */\r\nexport class ChatPrompt<\r\n  TOptions extends Record<string, any> = Record<string, any>,\r\n  TChatPromptPlugins extends readonly ChatPromptPlugin<string, any>[] = [],\r\n> implements IChatPrompt<TOptions, TChatPromptPlugins>\r\n{\r\n  get name() {\r\n    return this._name;\r\n  }\r\n  protected readonly _name: string;\r\n\r\n  get description() {\r\n    return this._description;\r\n  }\r\n  protected readonly _description: string;\r\n\r\n  get messages() {\r\n    return this._messages;\r\n  }\r\n  protected readonly _messages: IMemory;\r\n\r\n  get functions() {\r\n    return Object.values(this._functions);\r\n  }\r\n  protected readonly _functions: Record<string, Function> = {};\r\n\r\n  get plugins() {\r\n    return this._plugins;\r\n  }\r\n  protected readonly _plugins: TChatPromptPlugins;\r\n\r\n  protected readonly _role: 'system' | 'user';\r\n  protected readonly _template: ITemplate;\r\n  protected readonly _model: IChatModel<TOptions>;\r\n\r\n  constructor(options: ChatPromptOptions<TOptions>, plugins?: TChatPromptPlugins) {\r\n    this._name = options.name || 'chat';\r\n    this._description = options.description || 'an agent you can chat with';\r\n    this._role = options.role || 'system';\r\n    this._model = options.model;\r\n    this._template = Array.isArray(options.instructions)\r\n      ? new StringTemplate(options.instructions.join('\\n'))\r\n      : typeof options.instructions !== 'object'\r\n        ? new StringTemplate(options.instructions)\r\n        : options.instructions;\r\n\r\n    this._messages =\r\n      typeof options.messages === 'object' && !Array.isArray(options.messages)\r\n        ? options.messages\r\n        : new LocalMemory({ messages: options.messages || [] });\r\n\r\n    this._plugins = plugins || ([] as unknown as TChatPromptPlugins);\r\n  }\r\n\r\n  use(prompt: ChatPrompt): this;\r\n  use(name: string, prompt: ChatPrompt): this;\r\n  use(...args: any[]) {\r\n    const prompt: ChatPrompt = args.length === 1 ? args[0] : args[1];\r\n    const name: string = args.length === 1 ? prompt.name : args[0];\r\n    this._functions[name] = {\r\n      name,\r\n      description: prompt.description,\r\n      parameters: {\r\n        type: 'object',\r\n        properties: {\r\n          text: {\r\n            type: 'string',\r\n            description: 'the text to send to the assistant',\r\n          },\r\n        },\r\n        required: ['text'],\r\n      },\r\n      handler: ({ text }: { text: string }) => {\r\n        return prompt.send(text);\r\n      },\r\n    };\r\n\r\n    return this;\r\n  }\r\n\r\n  function(name: string, description: string, handler: FunctionHandler): this;\r\n  function(name: string, description: string, parameters: Schema, handler: FunctionHandler): this;\r\n  function(...args: any[]) {\r\n    const name: string = args[0];\r\n    const description: string = args[1];\r\n    const parameters: Schema | null = args.length === 3 ? null : args[2];\r\n    const handler: FunctionHandler = args[args.length - 1];\r\n    this._functions[name] = {\r\n      name,\r\n      description,\r\n      parameters: parameters || {},\r\n      handler,\r\n    };\r\n\r\n    return this;\r\n  }\r\n\r\n  usePlugin<K extends TChatPromptPlugins[number]['name']>(\r\n    name: K,\r\n    args: Extract<TChatPromptPlugins[number], { name: K }>['onUsePlugin'] extends\r\n      | ((args: infer U) => void)\r\n      | undefined\r\n      ? U\r\n      : never\r\n  ): this {\r\n    const plugin = this._plugins.find((p) => p.name === name);\r\n    if (!plugin) {\r\n      throw new Error(`Plugin \"${name}\" not found`);\r\n    }\r\n\r\n    if (plugin.onUsePlugin) {\r\n      plugin.onUsePlugin(args);\r\n    }\r\n\r\n    return this;\r\n  }\r\n\r\n  async call<A extends { [key: string]: any }, R = any>(name: string, args?: A): Promise<R> {\r\n    const fn = this._functions[name];\r\n\r\n    if (!fn) {\r\n      throw new Error(`function \"${name}\" not found`);\r\n    }\r\n\r\n    return await fn.handler(args || {});\r\n  }\r\n\r\n  async send(input: string | ContentPart[], options: ChatPromptSendOptions<TOptions> = {}) {\r\n    for (const plugin of this.plugins) {\r\n      if (plugin.onBeforeSend) {\r\n        input = await plugin.onBeforeSend(input);\r\n      }\r\n    }\r\n\r\n    const { onChunk } = options;\r\n\r\n    if (typeof input === 'string') {\r\n      input = input.trim();\r\n    }\r\n\r\n    const messages = !Array.isArray(options.messages)\r\n      ? options.messages || this._messages\r\n      : new LocalMemory({ messages: options.messages || [] });\r\n\r\n    let buffer = '';\r\n    let system: SystemMessage | UserMessage | undefined = undefined;\r\n    const prompt = await this._template.render();\r\n\r\n    if (prompt) {\r\n      system = {\r\n        role: this._role,\r\n        content: prompt,\r\n      };\r\n    }\r\n\r\n    let functions = Object.values(this._functions);\r\n    const pluginsWithOnBuildFunctions = this._plugins.filter(\r\n      (plugin): plugin is WithRequired<TChatPromptPlugins[number], 'onBuildFunctions'> =>\r\n        plugin.onBuildFunctions != null\r\n    );\r\n    for (const plugin of pluginsWithOnBuildFunctions) {\r\n      functions = await plugin.onBuildFunctions(functions);\r\n    }\r\n\r\n    const fnMap = functions.reduce(\r\n      (acc, fn) => {\r\n        acc[fn.name] = fn;\r\n        return acc;\r\n      },\r\n      {} as Record<string, Function>\r\n    );\r\n\r\n    const res = await this._model.send(\r\n      {\r\n        role: 'user',\r\n        content: input,\r\n      },\r\n      {\r\n        system,\r\n        messages,\r\n        request: options.request,\r\n        functions: fnMap,\r\n        onChunk: async (chunk) => {\r\n          if (!chunk || !onChunk) return;\r\n          buffer += chunk;\r\n\r\n          try {\r\n            await onChunk(buffer);\r\n            buffer = '';\r\n          } catch (err) {\r\n            return;\r\n          }\r\n        },\r\n      }\r\n    );\r\n\r\n    let output: Awaited<ReturnType<typeof this._model.send>> = {\r\n      ...res,\r\n      content: res.content || '',\r\n    };\r\n    for (const plugin of this.plugins) {\r\n      if (plugin.onAfterSend) {\r\n        output = await plugin.onAfterSend(output);\r\n      }\r\n    }\r\n\r\n    return output;\r\n  }\r\n}\r\n"]}

package/dist/prompts/chat.mjs.map

@@ -1 +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"]}
+{"version":3,"sources":["../../src/prompts/chat.ts"],"names":[],"mappings":";;;AA2JO,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';\r\nimport { LocalMemory } from '../local-memory';\r\nimport { IMemory } from '../memory';\r\nimport { ContentPart, Message, ModelMessage, SystemMessage, UserMessage } from '../message';\r\nimport { IChatModel, TextChunkHandler } from '../models';\r\nimport { Schema } from '../schema';\r\nimport { ITemplate } from '../template';\r\nimport { StringTemplate } from '../templates';\r\nimport { WithRequired } from '../utils/types';\r\n\r\nimport { IAiPlugin } from './plugin';\r\n\r\nexport type ChatPromptOptions<TOptions extends Record<string, any> = Record<string, any>> = {\r\n  /**\r\n   * the name of the prompt\r\n   */\r\n  readonly name?: string;\r\n\r\n  /**\r\n   * the description of the prompt\r\n   */\r\n  readonly description?: string;\r\n\r\n  /**\r\n   * the model to send messages to\r\n   */\r\n  readonly model: IChatModel<TOptions>;\r\n\r\n  /**\r\n   * the defining characteristics/objective\r\n   * of the prompt. This is commonly used to provide a system prompt.\r\n   * If you supply the system prompt as part of the messages,\r\n   * you do not need to supply this option.\r\n   */\r\n  readonly instructions?: string | string[] | ITemplate;\r\n\r\n  /**\r\n   * the `role` of the initial message\r\n   */\r\n  readonly role?: 'system' | 'user';\r\n\r\n  /**\r\n   * the conversation history\r\n   */\r\n  readonly messages?: Message[] | IMemory;\r\n};\r\n\r\nexport type ChatPromptSendOptions<TOptions extends Record<string, any> = Record<string, any>> = {\r\n  /**\r\n   * the conversation history\r\n   */\r\n  readonly messages?: Message[] | IMemory;\r\n\r\n  /**\r\n   * the models request options\r\n   */\r\n  readonly request?: TOptions;\r\n\r\n  /**\r\n   * the callback to be called for each\r\n   * stream chunk\r\n   */\r\n  readonly onChunk?: TextChunkHandler;\r\n};\r\n\r\n/**\r\n * a prompt that can interface with a\r\n * chat model that provides utility like\r\n * streaming and function calling\r\n */\r\nexport interface IChatPrompt<\r\n  TOptions extends Record<string, any> = Record<string, any>,\r\n  TChatPromptPlugins extends readonly ChatPromptPlugin<string, any>[] = [],\r\n> {\r\n  /**\r\n   * the prompt name\r\n   */\r\n  readonly name: string;\r\n\r\n  /**\r\n   * the prompt description\r\n   */\r\n  readonly description: string;\r\n\r\n  /**\r\n   * the chat history\r\n   */\r\n  readonly messages: IMemory;\r\n\r\n  /**\r\n   * the registered functions\r\n   */\r\n  readonly functions: Array<Function>;\r\n\r\n  /**\r\n   * the chat model\r\n   */\r\n  plugins: TChatPromptPlugins;\r\n\r\n  /**\r\n   * add another chat prompt as a\r\n   */\r\n  use(prompt: IChatPrompt): this;\r\n  use(name: string, prompt: IChatPrompt): this;\r\n\r\n  /**\r\n   * add a function that can be called\r\n   * by the model\r\n   */\r\n  function(name: string, description: string, handler: FunctionHandler): this;\r\n  function(name: string, description: string, parameters: Schema, handler: FunctionHandler): this;\r\n\r\n  usePlugin<TPluginName extends TChatPromptPlugins[number]['name']>(\r\n    name: TPluginName,\r\n    args: Extract<TChatPromptPlugins[number], { name: TPluginName }>['onUsePlugin'] extends\r\n      | ((args: infer U) => void)\r\n      | undefined\r\n      ? U\r\n      : never\r\n  ): this;\r\n\r\n  /**\r\n   * call a function\r\n   */\r\n  call<A extends Record<string, any>, R = any>(name: string, args?: A): Promise<R>;\r\n\r\n  /**\r\n   * send a message to the model and get a response\r\n   */\r\n  send(\r\n    input: string | ContentPart[],\r\n    options?: ChatPromptSendOptions<TOptions>\r\n  ): Promise<Pick<ModelMessage, 'content'> & Omit<ModelMessage, 'content'>>;\r\n}\r\n\r\nexport type ChatPromptPlugin<TPluginName extends string, TPluginUseArgs extends {}> = IAiPlugin<\r\n  TPluginName,\r\n  TPluginUseArgs,\r\n  Parameters<IChatPrompt['send']>[0],\r\n  ReturnType<IChatPrompt['send']>\r\n> & {\r\n  /**\r\n   * Optionally passed in to modify the functions array that\r\n   * is passed to the model\r\n   * @param functions\r\n   * @returns Functions\r\n   */\r\n  onBuildFunctions?: (functions: Function[]) => Function[] | Promise<Function[]>;\r\n};\r\n\r\n/**\r\n * a prompt that can interface with a\r\n * chat model that provides utility like\r\n * streaming and function calling\r\n */\r\nexport class ChatPrompt<\r\n  TOptions extends Record<string, any> = Record<string, any>,\r\n  TChatPromptPlugins extends readonly ChatPromptPlugin<string, any>[] = [],\r\n> implements IChatPrompt<TOptions, TChatPromptPlugins>\r\n{\r\n  get name() {\r\n    return this._name;\r\n  }\r\n  protected readonly _name: string;\r\n\r\n  get description() {\r\n    return this._description;\r\n  }\r\n  protected readonly _description: string;\r\n\r\n  get messages() {\r\n    return this._messages;\r\n  }\r\n  protected readonly _messages: IMemory;\r\n\r\n  get functions() {\r\n    return Object.values(this._functions);\r\n  }\r\n  protected readonly _functions: Record<string, Function> = {};\r\n\r\n  get plugins() {\r\n    return this._plugins;\r\n  }\r\n  protected readonly _plugins: TChatPromptPlugins;\r\n\r\n  protected readonly _role: 'system' | 'user';\r\n  protected readonly _template: ITemplate;\r\n  protected readonly _model: IChatModel<TOptions>;\r\n\r\n  constructor(options: ChatPromptOptions<TOptions>, plugins?: TChatPromptPlugins) {\r\n    this._name = options.name || 'chat';\r\n    this._description = options.description || 'an agent you can chat with';\r\n    this._role = options.role || 'system';\r\n    this._model = options.model;\r\n    this._template = Array.isArray(options.instructions)\r\n      ? new StringTemplate(options.instructions.join('\\n'))\r\n      : typeof options.instructions !== 'object'\r\n        ? new StringTemplate(options.instructions)\r\n        : options.instructions;\r\n\r\n    this._messages =\r\n      typeof options.messages === 'object' && !Array.isArray(options.messages)\r\n        ? options.messages\r\n        : new LocalMemory({ messages: options.messages || [] });\r\n\r\n    this._plugins = plugins || ([] as unknown as TChatPromptPlugins);\r\n  }\r\n\r\n  use(prompt: ChatPrompt): this;\r\n  use(name: string, prompt: ChatPrompt): this;\r\n  use(...args: any[]) {\r\n    const prompt: ChatPrompt = args.length === 1 ? args[0] : args[1];\r\n    const name: string = args.length === 1 ? prompt.name : args[0];\r\n    this._functions[name] = {\r\n      name,\r\n      description: prompt.description,\r\n      parameters: {\r\n        type: 'object',\r\n        properties: {\r\n          text: {\r\n            type: 'string',\r\n            description: 'the text to send to the assistant',\r\n          },\r\n        },\r\n        required: ['text'],\r\n      },\r\n      handler: ({ text }: { text: string }) => {\r\n        return prompt.send(text);\r\n      },\r\n    };\r\n\r\n    return this;\r\n  }\r\n\r\n  function(name: string, description: string, handler: FunctionHandler): this;\r\n  function(name: string, description: string, parameters: Schema, handler: FunctionHandler): this;\r\n  function(...args: any[]) {\r\n    const name: string = args[0];\r\n    const description: string = args[1];\r\n    const parameters: Schema | null = args.length === 3 ? null : args[2];\r\n    const handler: FunctionHandler = args[args.length - 1];\r\n    this._functions[name] = {\r\n      name,\r\n      description,\r\n      parameters: parameters || {},\r\n      handler,\r\n    };\r\n\r\n    return this;\r\n  }\r\n\r\n  usePlugin<K extends TChatPromptPlugins[number]['name']>(\r\n    name: K,\r\n    args: Extract<TChatPromptPlugins[number], { name: K }>['onUsePlugin'] extends\r\n      | ((args: infer U) => void)\r\n      | undefined\r\n      ? U\r\n      : never\r\n  ): this {\r\n    const plugin = this._plugins.find((p) => p.name === name);\r\n    if (!plugin) {\r\n      throw new Error(`Plugin \"${name}\" not found`);\r\n    }\r\n\r\n    if (plugin.onUsePlugin) {\r\n      plugin.onUsePlugin(args);\r\n    }\r\n\r\n    return this;\r\n  }\r\n\r\n  async call<A extends { [key: string]: any }, R = any>(name: string, args?: A): Promise<R> {\r\n    const fn = this._functions[name];\r\n\r\n    if (!fn) {\r\n      throw new Error(`function \"${name}\" not found`);\r\n    }\r\n\r\n    return await fn.handler(args || {});\r\n  }\r\n\r\n  async send(input: string | ContentPart[], options: ChatPromptSendOptions<TOptions> = {}) {\r\n    for (const plugin of this.plugins) {\r\n      if (plugin.onBeforeSend) {\r\n        input = await plugin.onBeforeSend(input);\r\n      }\r\n    }\r\n\r\n    const { onChunk } = options;\r\n\r\n    if (typeof input === 'string') {\r\n      input = input.trim();\r\n    }\r\n\r\n    const messages = !Array.isArray(options.messages)\r\n      ? options.messages || this._messages\r\n      : new LocalMemory({ messages: options.messages || [] });\r\n\r\n    let buffer = '';\r\n    let system: SystemMessage | UserMessage | undefined = undefined;\r\n    const prompt = await this._template.render();\r\n\r\n    if (prompt) {\r\n      system = {\r\n        role: this._role,\r\n        content: prompt,\r\n      };\r\n    }\r\n\r\n    let functions = Object.values(this._functions);\r\n    const pluginsWithOnBuildFunctions = this._plugins.filter(\r\n      (plugin): plugin is WithRequired<TChatPromptPlugins[number], 'onBuildFunctions'> =>\r\n        plugin.onBuildFunctions != null\r\n    );\r\n    for (const plugin of pluginsWithOnBuildFunctions) {\r\n      functions = await plugin.onBuildFunctions(functions);\r\n    }\r\n\r\n    const fnMap = functions.reduce(\r\n      (acc, fn) => {\r\n        acc[fn.name] = fn;\r\n        return acc;\r\n      },\r\n      {} as Record<string, Function>\r\n    );\r\n\r\n    const res = await this._model.send(\r\n      {\r\n        role: 'user',\r\n        content: input,\r\n      },\r\n      {\r\n        system,\r\n        messages,\r\n        request: options.request,\r\n        functions: fnMap,\r\n        onChunk: async (chunk) => {\r\n          if (!chunk || !onChunk) return;\r\n          buffer += chunk;\r\n\r\n          try {\r\n            await onChunk(buffer);\r\n            buffer = '';\r\n          } catch (err) {\r\n            return;\r\n          }\r\n        },\r\n      }\r\n    );\r\n\r\n    let output: Awaited<ReturnType<typeof this._model.send>> = {\r\n      ...res,\r\n      content: res.content || '',\r\n    };\r\n    for (const plugin of this.plugins) {\r\n      if (plugin.onAfterSend) {\r\n        output = await plugin.onAfterSend(output);\r\n      }\r\n    }\r\n\r\n    return output;\r\n  }\r\n}\r\n"]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@microsoft/teams.ai",
-  "version": "0.2.12",
+  "version": "0.2.13",
   "license": "MIT",
   "main": "./dist/index.js",
   "module": "./dist/index.mjs",
@@ -31,7 +31,8 @@
     "lint": "npx eslint",
     "lint:fix": "npx eslint --fix",
     "build": "npx tsup",
-    "test": "npx jest"
+    "test": "npx jest",
+    "fmt": "npx prettier --write \"**/*.{js,ts,tsx,md,json}\""
   },
   "peerDependencies": {
     "@microsoft/teams.common": "*"