beeai-framework 0.1.28 → 0.1.30

package/dist/tools/database/qdrant.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../src/tools/database/qdrant.ts"],"names":["QdrantAction","QdrantDatabaseTool","Tool","name","description","inputSchema","z","object","action","nativeEnum","describe","collectionName","string","optional","vector","array","number","vectors","topK","coerce","int","default","filter","record","any","payload","ids","or","emitter","Emitter","root","child","namespace","creator","validateInput","schema","input","ToolInputValidationError","register","client","QdrantClient","options","connection","_run","_options","collections","listCollections","JSONToolOutput","ToolError","collectionInfo","getCollectionInfo","searchResults","search","insertResults","insert","deleteResults","delete","response","getCollections","map","collection","error","getCollection","points","index","id","uuidv4","upsert","searchParams","query","limit","with_payload"],"mappings":";;;;;;;;;;;;;;;;;;;;;;AA6BO,IAAKA,YAAAA,6BAAAA,aAAAA,EAAAA;;;;;;AAAAA,EAAAA,OAAAA,aAAAA;;AAQL,MAAMC,2BAA2BC,aAAAA,CAAAA;AAAAA,EAAAA;;;EAItCC,IAAAA,GAAO,oBAAA;EAEPC,WAAAA,GAAc,CAAA;UACR,iBAAA,CAAA;UACA,mBAAA,CAAA;UACA,QAAA,CAAA;UACA,QAAA,CAAA;UACA,QAAA,CAAA,kCAAA,CAAA;EAENC,WAAAA,GAAc;AACZ,IAAA,OAAOC,MAAEC,MAAAA,CAAO;AACdC,MAAAA,MAAAA,EAAQF,KAAAA,CACLG,UAAAA,CAAWT,YAAAA,CAAAA,CACXU,SACC,CAAA,uBAAA,EAAwB,iBAAA,CAAA,wBAAA,EAAuD,mBAAA,gDAA8E,QAAA,CAAA,2BAAA,EAAiD,QAAA,CAAA,0BAAA,EAAgD,QAAA,CAAA,iBAAA,CAAwC,CAAA;AAE1SC,MAAAA,cAAAA,EAAgBL,KAAAA,CACbM,MAAAA,EAAM,CACNC,QAAAA,GACAH,QAAAA,CACC,CAAA,kDAAA,EAAmD,mBAAA,CAAA,EAAA,EAAmC,QAAA,CAAA,EAAA,EAAwB,QAAA,CAAA,MAAA,EAA4B,QAAA,CAAA,CAAuB,CAAA;MAErKI,MAAAA,EAAQR,KAAAA,CACLS,KAAAA,CAAMT,KAAAA,CAAEU,MAAAA,EAAM,CAAA,CACdH,QAAAA,EAAQ,CACRH,QAAAA,CAAS,CAAA,uCAAA,EAAwC,QAAA,CAAA,CAAA,CAAwB,CAAA;AAC5EO,MAAAA,OAAAA,EAASX,KAAAA,CACNS,KAAAA,CAAMT,KAAAA,CAAES,KAAAA,CAAMT,MAAEU,MAAAA,EAAM,CAAA,CAAA,CACtBH,QAAAA,EAAQ,CACRH,QAAAA,CAAS,CAAA,oCAAA,EAAqC,QAAA,CAAA,CAAA,CAAwB,CAAA;AACzEQ,MAAAA,IAAAA,EAAMZ,KAAAA,CAAEa,MAAAA,CACLH,MAAAA,EAAM,CACNI,GAAAA,EAAG,CACHC,OAAAA,CAAQ,EAAA,CAAA,CACRR,QAAAA,EAAQ,CACRH,QAAAA,CAAS,CAAA,0CAAA,CAA4C,CAAA;AACxDY,MAAAA,MAAAA,EAAQhB,KAAAA,CACLiB,MAAAA,CAAOjB,KAAAA,CAAEM,MAAAA,IAAUN,KAAAA,CAAEkB,GAAAA,EAAG,CAAA,CACxBX,QAAAA,EAAQ,CACRH,QAAAA,CAAS,CAAA,oBAAA,EAAqB,QAAA,CAAA,CAAA,CAAwB,CAAA;AACzDe,MAAAA,OAAAA,EAASnB,KAAAA,CACNS,KAAAA,CAAMT,KAAAA,CAAEiB,MAAAA,CAAOjB,MAAEM,MAAAA,EAAM,EAAIN,KAAAA,CAAEkB,GAAAA,EAAG,CAAA,CAAA,CAChCX,QAAAA,EAAQ,CACRH,SAAS,CAAA,0CAAA,CAA4C,CAAA;AACxDgB,MAAAA,GAAAA,EAAKpB,KAAAA,CACFS,KAAAA,CAAMT,KAAAA,CAAEM,MAAAA,GAASe,EAAAA,CAAGrB,KAAAA,CAAEU,MAAAA,EAAM,CAAA,CAAA,CAC5BH,QAAAA,EAAQ,CACRH,SAAS,CAAA,iCAAA,CAAmC;KACjD,CAAA;AACF;EAEgBkB,OAAAA,GACdC,mBAAAA,CAAQC,KAAKC,KAAAA,CAAM;IACjBC,SAAAA,EAAW;AAAC,MAAA,MAAA;AAAQ,MAAA,UAAA;AAAY,MAAA;;IAChCC,OAAAA,EAAS;GACX,CAAA;AAEQC,EAAAA,aAAAA,CACRC,QACAC,KAAAA,EACkC;AAClC,IAAA,KAAA,CAAMF,aAAAA,CAAcC,QAAQC,KAAAA,CAAAA;AAC5B,IAAA,IAAIA,KAAAA,CAAM5B,MAAAA,IAAM,iBAAA,IAAoC,CAAC4B,MAAMzB,cAAAA,EAAgB;AACzE,MAAA,MAAM,IAAI0B,iCAAAA,CACR,CAAA,gCAAA,EAAiC,mBAAA,CAAA,EAAA,EAAmC,QAAA,CAAA,EAAA,EAAwB,QAAA,CAAA,MAAA,EAA4B,QAAA,CAAA,SAAA,CAAgC,CAAA;AAE5J;AACA,IAAA,IAAID,KAAAA,CAAM5B,WAAM,QAAA,KAA6B,CAAC4B,MAAMzB,cAAAA,IAAkB,CAACyB,MAAMtB,MAAAA,CAAAA,EAAS;AACpF,MAAA,MAAM,IAAIuB,iCAAAA,CAAyB,CAAA,uBAAA,EAAwB,QAAA,CAAA,QAAA,CAA+B,CAAA;AAC5F;AACA,IAAA,IAAID,KAAAA,CAAM5B,WAAM,QAAA,KAA6B,CAAC4B,MAAMzB,cAAAA,IAAkB,CAACyB,MAAMnB,OAAAA,CAAAA,EAAU;AACrF,MAAA,MAAM,IAAIoB,iCAAAA,CAAyB,CAAA,yBAAA,EAA0B,QAAA,CAAA,QAAA,CAA+B,CAAA;AAC9F;AACF;EAEA;AACE,IAAA,IAAA,CAAKC,QAAAA,EAAQ;AACf;AAEA,EAAA,MACgBC,MAAAA,GAAgC;AAC9C,IAAA,OAAO,IAAIC,yBAAAA,CAAa,IAAA,CAAKC,OAAAA,CAAQC,UAAU,CAAA;AACjD;EAEA,MAAgBC,IAAAA,CACdP,OACAQ,QAAAA,EAC8B;AAC9B,IAAA,QAAQR,MAAM5B,MAAAA;AACZ,MAAA,KAAA,iBAAA,EAAmC;AACjC,QAAA,MAAMqC,WAAAA,GAAc,MAAM,IAAA,CAAKC,eAAAA,EAAe;AAC9C,QAAA,OAAO,IAAIC,wBAAeF,WAAAA,CAAAA;AAC5B;AAEA,MAAA,KAAA,mBAAA,EAAqC;AACnC,QAAA,IAAI,CAACT,MAAMzB,cAAAA,EAAgB;AACzB,UAAA,MAAM,IAAIqC,mBAAU,mEAAA,CAAA;AACtB;AACA,QAAA,MAAMC,cAAAA,GAAiB,MAAM,IAAA,CAAKC,iBAAAA,CAAkBd,MAAMzB,cAAc,CAAA;AACxE,QAAA,OAAO,IAAIoC,wBAAeE,cAAAA,CAAAA;AAC5B;AAEA,MAAA,KAAA,QAAA,EAA0B;AACxB,QAAA,IAAI,CAACb,KAAAA,CAAMzB,cAAAA,IAAkB,CAACyB,MAAMtB,MAAAA,EAAQ;AAC1C,UAAA,MAAM,IAAIkC,mBAAU,oEAAA,CAAA;AACtB;AACA,QAAA,MAAMG,aAAAA,GAAgB,MAAM,IAAA,CAAKC,MAAAA,CAAOhB,KAAAA,CAAAA;AACxC,QAAA,OAAO,IAAIW,wBAAeI,aAAAA,CAAAA;AAC5B;AAEA,MAAA,KAAA,QAAA,EAA0B;AACxB,QAAA,IAAI,CAACf,KAAAA,CAAMzB,cAAAA,IAAkB,CAACyB,MAAMnB,OAAAA,EAAS;AAC3C,UAAA,MAAM,IAAI+B,mBACR,qEAAA,CAAA;AAEJ;AACA,QAAA,MAAMK,aAAAA,GAAgB,MAAM,IAAA,CAAKC,MAAAA,CAAOlB,KAAAA,CAAAA;AACxC,QAAA,OAAO,IAAIW,wBAAeM,aAAAA,CAAAA;AAC5B;AAEA,MAAA,KAAA,QAAA,EAA0B;AACxB,QAAA,IAAI,CAACjB,KAAAA,CAAMzB,cAAAA,IAAkB,CAACyB,MAAMV,GAAAA,EAAK;AACvC,UAAA,MAAM,IAAIsB,mBAAU,+DAAA,CAAA;AACtB;AACA,QAAA,MAAMO,aAAAA,GAAgB,MAAM,IAAA,CAAKC,MAAAA,CAAOpB,KAAAA,CAAAA;AACxC,QAAA,OAAO,IAAIW,wBAAeQ,aAAAA,CAAAA;AAC5B;MAEA,SAAS;AACP,QAAA,MAAM,IAAIP,kBAAAA,CAAU,CAAA,0BAAA,EAA6BZ,KAAAA,CAAM5B,MAAM,CAAA,CAAE,CAAA;AACjE;AACF;AACF;AAEA,EAAA,MAAgBsC,eAAAA,GAAqC;AACnD,IAAA,IAAI;AACF,MAAA,MAAMP,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAAA,EAAM;AAChC,MAAA,MAAMkB,QAAAA,GAAW,MAAMlB,MAAAA,CAAOmB,cAAAA,EAAc;AAE5C,MAAA,OAAOD,SAASZ,WAAAA,CAAYc,GAAAA,CAAI,CAACC,UAAAA,KAAeA,WAAWzD,IAAI,CAAA;AACjE,KAAA,CAAA,OAAS0D,KAAAA,EAAO;AACd,MAAA,MAAM,IAAIb,kBAAAA,CAAU,CAAA,wCAAA,EAA2Ca,KAAAA,CAAAA,CAAO,CAAA;AACxE;AACF;AAEA,EAAA,MAAgBX,kBAAkBvC,cAAAA,EAAsC;AACtE,IAAA,MAAM4B,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAAA,EAAM;AAChC,IAAA,MAAMkB,QAAAA,GAAW,MAAMlB,MAAAA,CAAOuB,aAAAA,CAAcnD,cAAAA,CAAAA;AAC5C,IAAA,OAAO8C,QAAAA;AACT;AAEA,EAAA,MAAgBH,OAAOlB,KAAAA,EAAsC;AAC3D,IAAA,MAAMG,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAAA,EAAM;AAEhC,IAAA,MAAMwB,SAAS3B,KAAAA,CAAMnB,OAAAA,CAAS0C,GAAAA,CAAI,CAAC7C,QAAQkD,KAAAA,MAAW;AACpDC,MAAAA,EAAAA,EAAI7B,KAAAA,EAAOV,GAAAA,GAAMsC,KAAAA,CAAAA,IAAUE,OAAAA,EAAAA;AAC3BpD,MAAAA,MAAAA;AACAW,MAAAA,OAAAA,EAASW,KAAAA,EAAOX,OAAAA,GAAUuC,KAAAA,CAAAA,IAAU;KACtC,CAAA,CAAA;AAEA,IAAA,MAAMP,QAAAA,GAAW,MAAMlB,MAAAA,CAAO4B,MAAAA,CAAO/B,MAAMzB,cAAAA,EAA0B;AACnEoD,MAAAA;KACF,CAAA;AAEA,IAAA,OAAON,QAAAA;AACT;AAEA,EAAA,MAAgBL,OAAOhB,KAAAA,EAAsC;AAC3D,IAAA,MAAMG,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAAA,EAAM;AAEhC,IAAA,MAAM6B,YAAAA,GAAoB;AACxBC,MAAAA,KAAAA,EAAOjC,KAAAA,CAAMtB,MAAAA;AACbwD,MAAAA,KAAAA,EAAOlC,MAAMlB,IAAAA,IAAQ,EAAA;MACrBqD,YAAAA,EAAc,IAAA;AACdjD,MAAAA,MAAAA,EAAQc,KAAAA,CAAMd;AAChB,KAAA;AAEA,IAAA,MAAMmC,WAAW,MAAMlB,MAAAA,CAAO8B,KAAAA,CAAMjC,KAAAA,CAAMzB,gBAA0ByD,YAAAA,CAAAA;AACpE,IAAA,OAAOX,QAAAA;AACT;AAEA,EAAA,MAAgBD,OAAOpB,KAAAA,EAAsC;AAC3D,IAAA,MAAMG,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAAA,EAAM;AAEhC,IAAA,MAAMkB,QAAAA,GAAW,MAAMlB,MAAAA,CAAOiB,MAAAA,CAAOpB,MAAMzB,cAAAA,EAA0B;AACnEoD,MAAAA,MAAAA,EAAQ3B,KAAAA,CAAMV;KAChB,CAAA;AAEA,IAAA,OAAO+B,QAAAA;AACT;AACF","file":"qdrant.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport {\n  Tool,\n  ToolInput,\n  ToolError,\n  BaseToolOptions,\n  BaseToolRunOptions,\n  JSONToolOutput,\n  ToolInputValidationError,\n  ToolEmitter,\n} from \"@/tools/base.js\";\nimport { Cache } from \"@/cache/decoratorCache.js\";\nimport { AnyToolSchemaLike } from \"@/internals/helpers/schema.js\";\nimport { QdrantClient, QdrantClientParams } from \"@qdrant/js-client-rest\";\nimport { v4 as uuidv4 } from \"uuid\";\n\nimport { z } from \"zod\";\nimport { Emitter } from \"@/emitter/emitter.js\";\n\nexport interface QdrantToolOptions extends BaseToolOptions, QdrantClientParams {\n  connection: QdrantClientParams;\n}\n\nexport type QdrantSearchToolResult = any;\n\nexport enum QdrantAction {\n  ListCollections = \"ListCollections\",\n  GetCollectionInfo = \"GetCollectionInfo\",\n  Search = \"Search\",\n  Insert = \"Insert\",\n  Delete = \"Delete\",\n}\n\nexport class QdrantDatabaseTool extends Tool<\n  JSONToolOutput<QdrantSearchToolResult>,\n  QdrantToolOptions\n> {\n  name = \"QdrantDatabaseTool\";\n\n  description = `Can query data from a Qdrant vector database. IMPORTANT: strictly follow this order of actions:\n     1. ${QdrantAction.ListCollections} - List all the Qdrant collections\n     2. ${QdrantAction.GetCollectionInfo} - Get information about a Qdrant collection\n     3. ${QdrantAction.Insert} - Insert data into a Qdrant collection\n     3. ${QdrantAction.Search} - Perform search on a Qdrant collection\n     4. ${QdrantAction.Delete} - Delete from a Qdrant collection`;\n\n  inputSchema() {\n    return z.object({\n      action: z\n        .nativeEnum(QdrantAction)\n        .describe(\n          `The action to perform. ${QdrantAction.ListCollections} lists all collections, ${QdrantAction.GetCollectionInfo} fetches details for a specified collection, ${QdrantAction.Search} executes a vector search, ${QdrantAction.Insert} inserts new vectors, and ${QdrantAction.Delete} removes vectors.`,\n        ),\n      collectionName: z\n        .string()\n        .optional()\n        .describe(\n          `The name of the collection to query, required for ${QdrantAction.GetCollectionInfo}, ${QdrantAction.Search}, ${QdrantAction.Insert}, and ${QdrantAction.Delete}`,\n        ),\n      vector: z\n        .array(z.number())\n        .optional()\n        .describe(`The vector to search for, required for ${QdrantAction.Search}.`),\n      vectors: z\n        .array(z.array(z.number()))\n        .optional()\n        .describe(`The vectors to insert, required for ${QdrantAction.Insert}.`),\n      topK: z.coerce\n        .number()\n        .int()\n        .default(10)\n        .optional()\n        .describe(`The number of nearest neighbors to return.`),\n      filter: z\n        .record(z.string(), z.any())\n        .optional()\n        .describe(`Optional filter for ${QdrantAction.Search}.`),\n      payload: z\n        .array(z.record(z.string(), z.any()))\n        .optional()\n        .describe(`Additional payload to insert with vectors.`),\n      ids: z\n        .array(z.string().or(z.number()))\n        .optional()\n        .describe(`Array of IDs to delete or insert.`),\n    });\n  }\n\n  public readonly emitter: ToolEmitter<ToolInput<this>, JSONToolOutput<QdrantSearchToolResult>> =\n    Emitter.root.child({\n      namespace: [\"tool\", \"database\", \"qdrant\"],\n      creator: this,\n    });\n\n  protected validateInput(\n    schema: AnyToolSchemaLike,\n    input: unknown,\n  ): asserts input is ToolInput<this> {\n    super.validateInput(schema, input);\n    if (input.action != QdrantAction.ListCollections && !input.collectionName) {\n      throw new ToolInputValidationError(\n        `Collection name is required for ${QdrantAction.GetCollectionInfo}, ${QdrantAction.Search}, ${QdrantAction.Insert}, and ${QdrantAction.Delete} actions.`,\n      );\n    }\n    if (input.action === QdrantAction.Search && (!input.collectionName || !input.vector)) {\n      throw new ToolInputValidationError(`Vector is required for ${QdrantAction.Search} action.`);\n    }\n    if (input.action === QdrantAction.Insert && (!input.collectionName || !input.vectors)) {\n      throw new ToolInputValidationError(`Vectors are required for ${QdrantAction.Insert} action.`);\n    }\n  }\n\n  static {\n    this.register();\n  }\n\n  @Cache()\n  protected async client(): Promise<QdrantClient> {\n    return new QdrantClient(this.options.connection);\n  }\n\n  protected async _run(\n    input: ToolInput<this>,\n    _options: Partial<BaseToolRunOptions>,\n  ): Promise<JSONToolOutput<any>> {\n    switch (input.action) {\n      case QdrantAction.ListCollections: {\n        const collections = await this.listCollections();\n        return new JSONToolOutput(collections);\n      }\n\n      case QdrantAction.GetCollectionInfo: {\n        if (!input.collectionName) {\n          throw new ToolError(\"A collection name is required for Qdrant GetCollectionInfo action\");\n        }\n        const collectionInfo = await this.getCollectionInfo(input.collectionName);\n        return new JSONToolOutput(collectionInfo);\n      }\n\n      case QdrantAction.Search: {\n        if (!input.collectionName || !input.vector) {\n          throw new ToolError(\"A collection name and vector are required for Qdrant Search action\");\n        }\n        const searchResults = await this.search(input);\n        return new JSONToolOutput(searchResults);\n      }\n\n      case QdrantAction.Insert: {\n        if (!input.collectionName || !input.vectors) {\n          throw new ToolError(\n            \"A collection name and vectors are required for Qdrant Insert action\",\n          );\n        }\n        const insertResults = await this.insert(input);\n        return new JSONToolOutput(insertResults);\n      }\n\n      case QdrantAction.Delete: {\n        if (!input.collectionName || !input.ids) {\n          throw new ToolError(\"Collection name and ids are required for Qdrant Delete action\");\n        }\n        const deleteResults = await this.delete(input);\n        return new JSONToolOutput(deleteResults);\n      }\n\n      default: {\n        throw new ToolError(`Invalid action specified: ${input.action}`);\n      }\n    }\n  }\n\n  protected async listCollections(): Promise<string[]> {\n    try {\n      const client = await this.client();\n      const response = await client.getCollections();\n\n      return response.collections.map((collection) => collection.name);\n    } catch (error) {\n      throw new ToolError(`Failed to list collections from Qdrant: ${error}`);\n    }\n  }\n\n  protected async getCollectionInfo(collectionName: string): Promise<any> {\n    const client = await this.client();\n    const response = await client.getCollection(collectionName);\n    return response;\n  }\n\n  protected async insert(input: ToolInput<this>): Promise<any> {\n    const client = await this.client();\n\n    const points = input.vectors!.map((vector, index) => ({\n      id: input?.ids?.[index] ?? uuidv4(),\n      vector: vector,\n      payload: input?.payload?.[index] || {},\n    }));\n\n    const response = await client.upsert(input.collectionName as string, {\n      points: points,\n    });\n\n    return response;\n  }\n\n  protected async search(input: ToolInput<this>): Promise<any> {\n    const client = await this.client();\n\n    const searchParams: any = {\n      query: input.vector,\n      limit: input.topK || 10,\n      with_payload: true,\n      filter: input.filter,\n    };\n\n    const response = await client.query(input.collectionName as string, searchParams);\n    return response;\n  }\n\n  protected async delete(input: ToolInput<this>): Promise<any> {\n    const client = await this.client();\n\n    const response = await client.delete(input.collectionName as string, {\n      points: input.ids!,\n    });\n\n    return response;\n  }\n}\n"]}
+{"version":3,"sources":["../../../src/tools/database/qdrant.ts"],"names":["QdrantAction","QdrantDatabaseTool","Tool","name","description","inputSchema","z","object","action","nativeEnum","describe","collectionName","string","optional","vector","array","number","vectors","topK","coerce","int","default","filter","record","any","payload","ids","or","emitter","Emitter","root","child","namespace","creator","validateInput","schema","input","ToolInputValidationError","register","client","QdrantClient","options","connection","_run","_options","collections","listCollections","JSONToolOutput","ToolError","collectionInfo","getCollectionInfo","searchResults","search","insertResults","insert","deleteResults","delete","response","getCollections","map","collection","error","getCollection","points","index","id","uuidv4","upsert","searchParams","query","limit","with_payload"],"mappings":";;;;;;;;;;;;;;;;;;;;;;AA6BO,IAAKA,YAAAA,6BAAAA,aAAAA,EAAAA;;;;;;AAAAA,EAAAA,OAAAA,aAAAA;;AAQL,MAAMC,2BAA2BC,aAAAA,CAAAA;AAAAA,EAAAA;;;EAItCC,IAAAA,GAAO,oBAAA;EAEPC,WAAAA,GAAc,CAAA;UACR,iBAAA,CAAA;UACA,mBAAA,CAAA;UACA,QAAA,CAAA;UACA,QAAA,CAAA;UACA,QAAA,CAAA,kCAAA,CAAA;EAENC,WAAAA,GAAc;AACZ,IAAA,OAAOC,MAAEC,MAAAA,CAAO;AACdC,MAAAA,MAAAA,EAAQF,KAAAA,CACLG,UAAAA,CAAWT,YAAAA,CAAAA,CACXU,SACC,CAAA,uBAAA,EAAwB,iBAAA,CAAA,wBAAA,EAAuD,mBAAA,gDAA8E,QAAA,CAAA,2BAAA,EAAiD,QAAA,CAAA,0BAAA,EAAgD,QAAA,CAAA,iBAAA,CAAwC,CAAA;AAE1SC,MAAAA,cAAAA,EAAgBL,KAAAA,CACbM,MAAAA,EAAM,CACNC,QAAAA,GACAH,QAAAA,CACC,CAAA,kDAAA,EAAmD,mBAAA,CAAA,EAAA,EAAmC,QAAA,CAAA,EAAA,EAAwB,QAAA,CAAA,MAAA,EAA4B,QAAA,CAAA,CAAuB,CAAA;MAErKI,MAAAA,EAAQR,KAAAA,CACLS,KAAAA,CAAMT,KAAAA,CAAEU,MAAAA,EAAM,CAAA,CACdH,QAAAA,EAAQ,CACRH,QAAAA,CAAS,CAAA,uCAAA,EAAwC,QAAA,CAAA,CAAA,CAAwB,CAAA;AAC5EO,MAAAA,OAAAA,EAASX,KAAAA,CACNS,KAAAA,CAAMT,KAAAA,CAAES,KAAAA,CAAMT,MAAEU,MAAAA,EAAM,CAAA,CAAA,CACtBH,QAAAA,EAAQ,CACRH,QAAAA,CAAS,CAAA,oCAAA,EAAqC,QAAA,CAAA,CAAA,CAAwB,CAAA;AACzEQ,MAAAA,IAAAA,EAAMZ,KAAAA,CAAEa,MAAAA,CACLH,MAAAA,EAAM,CACNI,GAAAA,EAAG,CACHC,OAAAA,CAAQ,EAAA,CAAA,CACRR,QAAAA,EAAQ,CACRH,QAAAA,CAAS,CAAA,0CAAA,CAA4C,CAAA;AACxDY,MAAAA,MAAAA,EAAQhB,KAAAA,CACLiB,MAAAA,CAAOjB,KAAAA,CAAEM,MAAAA,IAAUN,KAAAA,CAAEkB,GAAAA,EAAG,CAAA,CACxBX,QAAAA,EAAQ,CACRH,QAAAA,CAAS,CAAA,oBAAA,EAAqB,QAAA,CAAA,CAAA,CAAwB,CAAA;AACzDe,MAAAA,OAAAA,EAASnB,KAAAA,CACNS,KAAAA,CAAMT,KAAAA,CAAEiB,MAAAA,CAAOjB,MAAEM,MAAAA,EAAM,EAAIN,KAAAA,CAAEkB,GAAAA,EAAG,CAAA,CAAA,CAChCX,QAAAA,EAAQ,CACRH,SAAS,CAAA,0CAAA,CAA4C,CAAA;AACxDgB,MAAAA,GAAAA,EAAKpB,KAAAA,CACFS,KAAAA,CAAMT,KAAAA,CAAEM,MAAAA,GAASe,EAAAA,CAAGrB,KAAAA,CAAEU,MAAAA,EAAM,CAAA,CAAA,CAC5BH,QAAAA,EAAQ,CACRH,SAAS,CAAA,iCAAA,CAAmC;KACjD,CAAA;AACF,EAAA;EAEgBkB,OAAAA,GACdC,mBAAAA,CAAQC,KAAKC,KAAAA,CAAM;IACjBC,SAAAA,EAAW;AAAC,MAAA,MAAA;AAAQ,MAAA,UAAA;AAAY,MAAA;;IAChCC,OAAAA,EAAS;GACX,CAAA;AAEQC,EAAAA,aAAAA,CACRC,QACAC,KAAAA,EACkC;AAClC,IAAA,KAAA,CAAMF,aAAAA,CAAcC,QAAQC,KAAAA,CAAAA;AAC5B,IAAA,IAAIA,KAAAA,CAAM5B,MAAAA,IAAM,iBAAA,IAAoC,CAAC4B,MAAMzB,cAAAA,EAAgB;AACzE,MAAA,MAAM,IAAI0B,iCAAAA,CACR,CAAA,gCAAA,EAAiC,mBAAA,CAAA,EAAA,EAAmC,QAAA,CAAA,EAAA,EAAwB,QAAA,CAAA,MAAA,EAA4B,QAAA,CAAA,SAAA,CAAgC,CAAA;AAE5J,IAAA;AACA,IAAA,IAAID,KAAAA,CAAM5B,WAAM,QAAA,KAA6B,CAAC4B,MAAMzB,cAAAA,IAAkB,CAACyB,MAAMtB,MAAAA,CAAAA,EAAS;AACpF,MAAA,MAAM,IAAIuB,iCAAAA,CAAyB,CAAA,uBAAA,EAAwB,QAAA,CAAA,QAAA,CAA+B,CAAA;AAC5F,IAAA;AACA,IAAA,IAAID,KAAAA,CAAM5B,WAAM,QAAA,KAA6B,CAAC4B,MAAMzB,cAAAA,IAAkB,CAACyB,MAAMnB,OAAAA,CAAAA,EAAU;AACrF,MAAA,MAAM,IAAIoB,iCAAAA,CAAyB,CAAA,yBAAA,EAA0B,QAAA,CAAA,QAAA,CAA+B,CAAA;AAC9F,IAAA;AACF,EAAA;EAEA;AACE,IAAA,IAAA,CAAKC,QAAAA,EAAQ;AACf;AAEA,EAAA,MACgBC,MAAAA,GAAgC;AAC9C,IAAA,OAAO,IAAIC,yBAAAA,CAAa,IAAA,CAAKC,OAAAA,CAAQC,UAAU,CAAA;AACjD,EAAA;EAEA,MAAgBC,IAAAA,CACdP,OACAQ,QAAAA,EAC8B;AAC9B,IAAA,QAAQR,MAAM5B,MAAAA;AACZ,MAAA,KAAA,iBAAA,EAAmC;AACjC,QAAA,MAAMqC,WAAAA,GAAc,MAAM,IAAA,CAAKC,eAAAA,EAAe;AAC9C,QAAA,OAAO,IAAIC,wBAAeF,WAAAA,CAAAA;AAC5B,MAAA;AAEA,MAAA,KAAA,mBAAA,EAAqC;AACnC,QAAA,IAAI,CAACT,MAAMzB,cAAAA,EAAgB;AACzB,UAAA,MAAM,IAAIqC,mBAAU,mEAAA,CAAA;AACtB,QAAA;AACA,QAAA,MAAMC,cAAAA,GAAiB,MAAM,IAAA,CAAKC,iBAAAA,CAAkBd,MAAMzB,cAAc,CAAA;AACxE,QAAA,OAAO,IAAIoC,wBAAeE,cAAAA,CAAAA;AAC5B,MAAA;AAEA,MAAA,KAAA,QAAA,EAA0B;AACxB,QAAA,IAAI,CAACb,KAAAA,CAAMzB,cAAAA,IAAkB,CAACyB,MAAMtB,MAAAA,EAAQ;AAC1C,UAAA,MAAM,IAAIkC,mBAAU,oEAAA,CAAA;AACtB,QAAA;AACA,QAAA,MAAMG,aAAAA,GAAgB,MAAM,IAAA,CAAKC,MAAAA,CAAOhB,KAAAA,CAAAA;AACxC,QAAA,OAAO,IAAIW,wBAAeI,aAAAA,CAAAA;AAC5B,MAAA;AAEA,MAAA,KAAA,QAAA,EAA0B;AACxB,QAAA,IAAI,CAACf,KAAAA,CAAMzB,cAAAA,IAAkB,CAACyB,MAAMnB,OAAAA,EAAS;AAC3C,UAAA,MAAM,IAAI+B,mBACR,qEAAA,CAAA;AAEJ,QAAA;AACA,QAAA,MAAMK,aAAAA,GAAgB,MAAM,IAAA,CAAKC,MAAAA,CAAOlB,KAAAA,CAAAA;AACxC,QAAA,OAAO,IAAIW,wBAAeM,aAAAA,CAAAA;AAC5B,MAAA;AAEA,MAAA,KAAA,QAAA,EAA0B;AACxB,QAAA,IAAI,CAACjB,KAAAA,CAAMzB,cAAAA,IAAkB,CAACyB,MAAMV,GAAAA,EAAK;AACvC,UAAA,MAAM,IAAIsB,mBAAU,+DAAA,CAAA;AACtB,QAAA;AACA,QAAA,MAAMO,aAAAA,GAAgB,MAAM,IAAA,CAAKC,MAAAA,CAAOpB,KAAAA,CAAAA;AACxC,QAAA,OAAO,IAAIW,wBAAeQ,aAAAA,CAAAA;AAC5B,MAAA;MAEA,SAAS;AACP,QAAA,MAAM,IAAIP,kBAAAA,CAAU,CAAA,0BAAA,EAA6BZ,KAAAA,CAAM5B,MAAM,CAAA,CAAE,CAAA;AACjE,MAAA;AACF;AACF,EAAA;AAEA,EAAA,MAAgBsC,eAAAA,GAAqC;AACnD,IAAA,IAAI;AACF,MAAA,MAAMP,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAAA,EAAM;AAChC,MAAA,MAAMkB,QAAAA,GAAW,MAAMlB,MAAAA,CAAOmB,cAAAA,EAAc;AAE5C,MAAA,OAAOD,SAASZ,WAAAA,CAAYc,GAAAA,CAAI,CAACC,UAAAA,KAAeA,WAAWzD,IAAI,CAAA;AACjE,IAAA,CAAA,CAAA,OAAS0D,KAAAA,EAAO;AACd,MAAA,MAAM,IAAIb,kBAAAA,CAAU,CAAA,wCAAA,EAA2Ca,KAAAA,CAAAA,CAAO,CAAA;AACxE,IAAA;AACF,EAAA;AAEA,EAAA,MAAgBX,kBAAkBvC,cAAAA,EAAsC;AACtE,IAAA,MAAM4B,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAAA,EAAM;AAChC,IAAA,MAAMkB,QAAAA,GAAW,MAAMlB,MAAAA,CAAOuB,aAAAA,CAAcnD,cAAAA,CAAAA;AAC5C,IAAA,OAAO8C,QAAAA;AACT,EAAA;AAEA,EAAA,MAAgBH,OAAOlB,KAAAA,EAAsC;AAC3D,IAAA,MAAMG,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAAA,EAAM;AAEhC,IAAA,MAAMwB,SAAS3B,KAAAA,CAAMnB,OAAAA,CAAS0C,GAAAA,CAAI,CAAC7C,QAAQkD,KAAAA,MAAW;AACpDC,MAAAA,EAAAA,EAAI7B,KAAAA,EAAOV,GAAAA,GAAMsC,KAAAA,CAAAA,IAAUE,OAAAA,EAAAA;AAC3BpD,MAAAA,MAAAA;AACAW,MAAAA,OAAAA,EAASW,KAAAA,EAAOX,OAAAA,GAAUuC,KAAAA,CAAAA,IAAU;KACtC,CAAA,CAAA;AAEA,IAAA,MAAMP,QAAAA,GAAW,MAAMlB,MAAAA,CAAO4B,MAAAA,CAAO/B,MAAMzB,cAAAA,EAA0B;AACnEoD,MAAAA;KACF,CAAA;AAEA,IAAA,OAAON,QAAAA;AACT,EAAA;AAEA,EAAA,MAAgBL,OAAOhB,KAAAA,EAAsC;AAC3D,IAAA,MAAMG,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAAA,EAAM;AAEhC,IAAA,MAAM6B,YAAAA,GAAoB;AACxBC,MAAAA,KAAAA,EAAOjC,KAAAA,CAAMtB,MAAAA;AACbwD,MAAAA,KAAAA,EAAOlC,MAAMlB,IAAAA,IAAQ,EAAA;MACrBqD,YAAAA,EAAc,IAAA;AACdjD,MAAAA,MAAAA,EAAQc,KAAAA,CAAMd;AAChB,KAAA;AAEA,IAAA,MAAMmC,WAAW,MAAMlB,MAAAA,CAAO8B,KAAAA,CAAMjC,KAAAA,CAAMzB,gBAA0ByD,YAAAA,CAAAA;AACpE,IAAA,OAAOX,QAAAA;AACT,EAAA;AAEA,EAAA,MAAgBD,OAAOpB,KAAAA,EAAsC;AAC3D,IAAA,MAAMG,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAAA,EAAM;AAEhC,IAAA,MAAMkB,QAAAA,GAAW,MAAMlB,MAAAA,CAAOiB,MAAAA,CAAOpB,MAAMzB,cAAAA,EAA0B;AACnEoD,MAAAA,MAAAA,EAAQ3B,KAAAA,CAAMV;KAChB,CAAA;AAEA,IAAA,OAAO+B,QAAAA;AACT,EAAA;AACF","file":"qdrant.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport {\n  Tool,\n  ToolInput,\n  ToolError,\n  BaseToolOptions,\n  BaseToolRunOptions,\n  JSONToolOutput,\n  ToolInputValidationError,\n  ToolEmitter,\n} from \"@/tools/base.js\";\nimport { Cache } from \"@/cache/decoratorCache.js\";\nimport { AnyToolSchemaLike } from \"@/internals/helpers/schema.js\";\nimport { QdrantClient, QdrantClientParams } from \"@qdrant/js-client-rest\";\nimport { v4 as uuidv4 } from \"uuid\";\n\nimport { z } from \"zod\";\nimport { Emitter } from \"@/emitter/emitter.js\";\n\nexport interface QdrantToolOptions extends BaseToolOptions, QdrantClientParams {\n  connection: QdrantClientParams;\n}\n\nexport type QdrantSearchToolResult = any;\n\nexport enum QdrantAction {\n  ListCollections = \"ListCollections\",\n  GetCollectionInfo = \"GetCollectionInfo\",\n  Search = \"Search\",\n  Insert = \"Insert\",\n  Delete = \"Delete\",\n}\n\nexport class QdrantDatabaseTool extends Tool<\n  JSONToolOutput<QdrantSearchToolResult>,\n  QdrantToolOptions\n> {\n  name = \"QdrantDatabaseTool\";\n\n  description = `Can query data from a Qdrant vector database. IMPORTANT: strictly follow this order of actions:\n     1. ${QdrantAction.ListCollections} - List all the Qdrant collections\n     2. ${QdrantAction.GetCollectionInfo} - Get information about a Qdrant collection\n     3. ${QdrantAction.Insert} - Insert data into a Qdrant collection\n     3. ${QdrantAction.Search} - Perform search on a Qdrant collection\n     4. ${QdrantAction.Delete} - Delete from a Qdrant collection`;\n\n  inputSchema() {\n    return z.object({\n      action: z\n        .nativeEnum(QdrantAction)\n        .describe(\n          `The action to perform. ${QdrantAction.ListCollections} lists all collections, ${QdrantAction.GetCollectionInfo} fetches details for a specified collection, ${QdrantAction.Search} executes a vector search, ${QdrantAction.Insert} inserts new vectors, and ${QdrantAction.Delete} removes vectors.`,\n        ),\n      collectionName: z\n        .string()\n        .optional()\n        .describe(\n          `The name of the collection to query, required for ${QdrantAction.GetCollectionInfo}, ${QdrantAction.Search}, ${QdrantAction.Insert}, and ${QdrantAction.Delete}`,\n        ),\n      vector: z\n        .array(z.number())\n        .optional()\n        .describe(`The vector to search for, required for ${QdrantAction.Search}.`),\n      vectors: z\n        .array(z.array(z.number()))\n        .optional()\n        .describe(`The vectors to insert, required for ${QdrantAction.Insert}.`),\n      topK: z.coerce\n        .number()\n        .int()\n        .default(10)\n        .optional()\n        .describe(`The number of nearest neighbors to return.`),\n      filter: z\n        .record(z.string(), z.any())\n        .optional()\n        .describe(`Optional filter for ${QdrantAction.Search}.`),\n      payload: z\n        .array(z.record(z.string(), z.any()))\n        .optional()\n        .describe(`Additional payload to insert with vectors.`),\n      ids: z\n        .array(z.string().or(z.number()))\n        .optional()\n        .describe(`Array of IDs to delete or insert.`),\n    });\n  }\n\n  public readonly emitter: ToolEmitter<ToolInput<this>, JSONToolOutput<QdrantSearchToolResult>> =\n    Emitter.root.child({\n      namespace: [\"tool\", \"database\", \"qdrant\"],\n      creator: this,\n    });\n\n  protected validateInput(\n    schema: AnyToolSchemaLike,\n    input: unknown,\n  ): asserts input is ToolInput<this> {\n    super.validateInput(schema, input);\n    if (input.action != QdrantAction.ListCollections && !input.collectionName) {\n      throw new ToolInputValidationError(\n        `Collection name is required for ${QdrantAction.GetCollectionInfo}, ${QdrantAction.Search}, ${QdrantAction.Insert}, and ${QdrantAction.Delete} actions.`,\n      );\n    }\n    if (input.action === QdrantAction.Search && (!input.collectionName || !input.vector)) {\n      throw new ToolInputValidationError(`Vector is required for ${QdrantAction.Search} action.`);\n    }\n    if (input.action === QdrantAction.Insert && (!input.collectionName || !input.vectors)) {\n      throw new ToolInputValidationError(`Vectors are required for ${QdrantAction.Insert} action.`);\n    }\n  }\n\n  static {\n    this.register();\n  }\n\n  @Cache()\n  protected async client(): Promise<QdrantClient> {\n    return new QdrantClient(this.options.connection);\n  }\n\n  protected async _run(\n    input: ToolInput<this>,\n    _options: Partial<BaseToolRunOptions>,\n  ): Promise<JSONToolOutput<any>> {\n    switch (input.action) {\n      case QdrantAction.ListCollections: {\n        const collections = await this.listCollections();\n        return new JSONToolOutput(collections);\n      }\n\n      case QdrantAction.GetCollectionInfo: {\n        if (!input.collectionName) {\n          throw new ToolError(\"A collection name is required for Qdrant GetCollectionInfo action\");\n        }\n        const collectionInfo = await this.getCollectionInfo(input.collectionName);\n        return new JSONToolOutput(collectionInfo);\n      }\n\n      case QdrantAction.Search: {\n        if (!input.collectionName || !input.vector) {\n          throw new ToolError(\"A collection name and vector are required for Qdrant Search action\");\n        }\n        const searchResults = await this.search(input);\n        return new JSONToolOutput(searchResults);\n      }\n\n      case QdrantAction.Insert: {\n        if (!input.collectionName || !input.vectors) {\n          throw new ToolError(\n            \"A collection name and vectors are required for Qdrant Insert action\",\n          );\n        }\n        const insertResults = await this.insert(input);\n        return new JSONToolOutput(insertResults);\n      }\n\n      case QdrantAction.Delete: {\n        if (!input.collectionName || !input.ids) {\n          throw new ToolError(\"Collection name and ids are required for Qdrant Delete action\");\n        }\n        const deleteResults = await this.delete(input);\n        return new JSONToolOutput(deleteResults);\n      }\n\n      default: {\n        throw new ToolError(`Invalid action specified: ${input.action}`);\n      }\n    }\n  }\n\n  protected async listCollections(): Promise<string[]> {\n    try {\n      const client = await this.client();\n      const response = await client.getCollections();\n\n      return response.collections.map((collection) => collection.name);\n    } catch (error) {\n      throw new ToolError(`Failed to list collections from Qdrant: ${error}`);\n    }\n  }\n\n  protected async getCollectionInfo(collectionName: string): Promise<any> {\n    const client = await this.client();\n    const response = await client.getCollection(collectionName);\n    return response;\n  }\n\n  protected async insert(input: ToolInput<this>): Promise<any> {\n    const client = await this.client();\n\n    const points = input.vectors!.map((vector, index) => ({\n      id: input?.ids?.[index] ?? uuidv4(),\n      vector: vector,\n      payload: input?.payload?.[index] || {},\n    }));\n\n    const response = await client.upsert(input.collectionName as string, {\n      points: points,\n    });\n\n    return response;\n  }\n\n  protected async search(input: ToolInput<this>): Promise<any> {\n    const client = await this.client();\n\n    const searchParams: any = {\n      query: input.vector,\n      limit: input.topK || 10,\n      with_payload: true,\n      filter: input.filter,\n    };\n\n    const response = await client.query(input.collectionName as string, searchParams);\n    return response;\n  }\n\n  protected async delete(input: ToolInput<this>): Promise<any> {\n    const client = await this.client();\n\n    const response = await client.delete(input.collectionName as string, {\n      points: input.ids!,\n    });\n\n    return response;\n  }\n}\n"]}

package/dist/tools/database/qdrant.d.cts

@@ -1,10 +1,10 @@
-import { BaseToolOptions, Tool, JSONToolOutput, ToolEmitter, ToolInput, BaseToolRunOptions } from '../base.cjs';
+import { Tool, JSONToolOutput, BaseToolOptions, ToolEmitter, ToolInput, BaseToolRunOptions } from '../base.cjs';
 import { AnyToolSchemaLike } from '../../internals/helpers/schema.cjs';
 import { QdrantClientParams, QdrantClient } from '@qdrant/js-client-rest';
 import { z } from 'zod';
 import 'ajv';
 import '../../context.cjs';
-import '../../emitter-UkIkNrek.cjs';
+import '../../emitter-D4OcelZ9.cjs';
 import '../../internals/types.cjs';
 import '../../internals/helpers/guards.cjs';
 import '../../internals/serializable.cjs';

package/dist/tools/database/qdrant.d.ts

@@ -1,10 +1,10 @@
-import { BaseToolOptions, Tool, JSONToolOutput, ToolEmitter, ToolInput, BaseToolRunOptions } from '../base.js';
+import { Tool, JSONToolOutput, BaseToolOptions, ToolEmitter, ToolInput, BaseToolRunOptions } from '../base.js';
 import { AnyToolSchemaLike } from '../../internals/helpers/schema.js';
 import { QdrantClientParams, QdrantClient } from '@qdrant/js-client-rest';
 import { z } from 'zod';
 import 'ajv';
 import '../../context.js';
-import '../../emitter-CL56qab4.js';
+import '../../emitter-kHxkUxco.js';
 import '../../internals/types.js';
 import '../../internals/helpers/guards.js';
 import '../../internals/serializable.js';

package/dist/tools/database/qdrant.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../src/tools/database/qdrant.ts"],"names":["QdrantAction","QdrantDatabaseTool","Tool","name","description","inputSchema","z","object","action","nativeEnum","describe","collectionName","string","optional","vector","array","number","vectors","topK","coerce","int","default","filter","record","any","payload","ids","or","emitter","Emitter","root","child","namespace","creator","validateInput","schema","input","ToolInputValidationError","register","client","QdrantClient","options","connection","_run","_options","collections","listCollections","JSONToolOutput","ToolError","collectionInfo","getCollectionInfo","searchResults","search","insertResults","insert","deleteResults","delete","response","getCollections","map","collection","error","getCollection","points","index","id","uuidv4","upsert","searchParams","query","limit","with_payload"],"mappings":";;;;;;;;;;;;;;;;;;;;AA6BO,IAAKA,YAAAA,6BAAAA,aAAAA,EAAAA;;;;;;AAAAA,EAAAA,OAAAA,aAAAA;;AAQL,MAAMC,2BAA2BC,IAAAA,CAAAA;AAAAA,EAAAA;;;EAItCC,IAAAA,GAAO,oBAAA;EAEPC,WAAAA,GAAc,CAAA;UACR,iBAAA,CAAA;UACA,mBAAA,CAAA;UACA,QAAA,CAAA;UACA,QAAA,CAAA;UACA,QAAA,CAAA,kCAAA,CAAA;EAENC,WAAAA,GAAc;AACZ,IAAA,OAAOC,EAAEC,MAAAA,CAAO;AACdC,MAAAA,MAAAA,EAAQF,CAAAA,CACLG,UAAAA,CAAWT,YAAAA,CAAAA,CACXU,SACC,CAAA,uBAAA,EAAwB,iBAAA,CAAA,wBAAA,EAAuD,mBAAA,gDAA8E,QAAA,CAAA,2BAAA,EAAiD,QAAA,CAAA,0BAAA,EAAgD,QAAA,CAAA,iBAAA,CAAwC,CAAA;AAE1SC,MAAAA,cAAAA,EAAgBL,CAAAA,CACbM,MAAAA,EAAM,CACNC,QAAAA,GACAH,QAAAA,CACC,CAAA,kDAAA,EAAmD,mBAAA,CAAA,EAAA,EAAmC,QAAA,CAAA,EAAA,EAAwB,QAAA,CAAA,MAAA,EAA4B,QAAA,CAAA,CAAuB,CAAA;MAErKI,MAAAA,EAAQR,CAAAA,CACLS,KAAAA,CAAMT,CAAAA,CAAEU,MAAAA,EAAM,CAAA,CACdH,QAAAA,EAAQ,CACRH,QAAAA,CAAS,CAAA,uCAAA,EAAwC,QAAA,CAAA,CAAA,CAAwB,CAAA;AAC5EO,MAAAA,OAAAA,EAASX,CAAAA,CACNS,KAAAA,CAAMT,CAAAA,CAAES,KAAAA,CAAMT,EAAEU,MAAAA,EAAM,CAAA,CAAA,CACtBH,QAAAA,EAAQ,CACRH,QAAAA,CAAS,CAAA,oCAAA,EAAqC,QAAA,CAAA,CAAA,CAAwB,CAAA;AACzEQ,MAAAA,IAAAA,EAAMZ,CAAAA,CAAEa,MAAAA,CACLH,MAAAA,EAAM,CACNI,GAAAA,EAAG,CACHC,OAAAA,CAAQ,EAAA,CAAA,CACRR,QAAAA,EAAQ,CACRH,QAAAA,CAAS,CAAA,0CAAA,CAA4C,CAAA;AACxDY,MAAAA,MAAAA,EAAQhB,CAAAA,CACLiB,MAAAA,CAAOjB,CAAAA,CAAEM,MAAAA,IAAUN,CAAAA,CAAEkB,GAAAA,EAAG,CAAA,CACxBX,QAAAA,EAAQ,CACRH,QAAAA,CAAS,CAAA,oBAAA,EAAqB,QAAA,CAAA,CAAA,CAAwB,CAAA;AACzDe,MAAAA,OAAAA,EAASnB,CAAAA,CACNS,KAAAA,CAAMT,CAAAA,CAAEiB,MAAAA,CAAOjB,EAAEM,MAAAA,EAAM,EAAIN,CAAAA,CAAEkB,GAAAA,EAAG,CAAA,CAAA,CAChCX,QAAAA,EAAQ,CACRH,SAAS,CAAA,0CAAA,CAA4C,CAAA;AACxDgB,MAAAA,GAAAA,EAAKpB,CAAAA,CACFS,KAAAA,CAAMT,CAAAA,CAAEM,MAAAA,GAASe,EAAAA,CAAGrB,CAAAA,CAAEU,MAAAA,EAAM,CAAA,CAAA,CAC5BH,QAAAA,EAAQ,CACRH,SAAS,CAAA,iCAAA,CAAmC;KACjD,CAAA;AACF;EAEgBkB,OAAAA,GACdC,OAAAA,CAAQC,KAAKC,KAAAA,CAAM;IACjBC,SAAAA,EAAW;AAAC,MAAA,MAAA;AAAQ,MAAA,UAAA;AAAY,MAAA;;IAChCC,OAAAA,EAAS;GACX,CAAA;AAEQC,EAAAA,aAAAA,CACRC,QACAC,KAAAA,EACkC;AAClC,IAAA,KAAA,CAAMF,aAAAA,CAAcC,QAAQC,KAAAA,CAAAA;AAC5B,IAAA,IAAIA,KAAAA,CAAM5B,MAAAA,IAAM,iBAAA,IAAoC,CAAC4B,MAAMzB,cAAAA,EAAgB;AACzE,MAAA,MAAM,IAAI0B,wBAAAA,CACR,CAAA,gCAAA,EAAiC,mBAAA,CAAA,EAAA,EAAmC,QAAA,CAAA,EAAA,EAAwB,QAAA,CAAA,MAAA,EAA4B,QAAA,CAAA,SAAA,CAAgC,CAAA;AAE5J;AACA,IAAA,IAAID,KAAAA,CAAM5B,WAAM,QAAA,KAA6B,CAAC4B,MAAMzB,cAAAA,IAAkB,CAACyB,MAAMtB,MAAAA,CAAAA,EAAS;AACpF,MAAA,MAAM,IAAIuB,wBAAAA,CAAyB,CAAA,uBAAA,EAAwB,QAAA,CAAA,QAAA,CAA+B,CAAA;AAC5F;AACA,IAAA,IAAID,KAAAA,CAAM5B,WAAM,QAAA,KAA6B,CAAC4B,MAAMzB,cAAAA,IAAkB,CAACyB,MAAMnB,OAAAA,CAAAA,EAAU;AACrF,MAAA,MAAM,IAAIoB,wBAAAA,CAAyB,CAAA,yBAAA,EAA0B,QAAA,CAAA,QAAA,CAA+B,CAAA;AAC9F;AACF;EAEA;AACE,IAAA,IAAA,CAAKC,QAAAA,EAAQ;AACf;AAEA,EAAA,MACgBC,MAAAA,GAAgC;AAC9C,IAAA,OAAO,IAAIC,YAAAA,CAAa,IAAA,CAAKC,OAAAA,CAAQC,UAAU,CAAA;AACjD;EAEA,MAAgBC,IAAAA,CACdP,OACAQ,QAAAA,EAC8B;AAC9B,IAAA,QAAQR,MAAM5B,MAAAA;AACZ,MAAA,KAAA,iBAAA,EAAmC;AACjC,QAAA,MAAMqC,WAAAA,GAAc,MAAM,IAAA,CAAKC,eAAAA,EAAe;AAC9C,QAAA,OAAO,IAAIC,eAAeF,WAAAA,CAAAA;AAC5B;AAEA,MAAA,KAAA,mBAAA,EAAqC;AACnC,QAAA,IAAI,CAACT,MAAMzB,cAAAA,EAAgB;AACzB,UAAA,MAAM,IAAIqC,UAAU,mEAAA,CAAA;AACtB;AACA,QAAA,MAAMC,cAAAA,GAAiB,MAAM,IAAA,CAAKC,iBAAAA,CAAkBd,MAAMzB,cAAc,CAAA;AACxE,QAAA,OAAO,IAAIoC,eAAeE,cAAAA,CAAAA;AAC5B;AAEA,MAAA,KAAA,QAAA,EAA0B;AACxB,QAAA,IAAI,CAACb,KAAAA,CAAMzB,cAAAA,IAAkB,CAACyB,MAAMtB,MAAAA,EAAQ;AAC1C,UAAA,MAAM,IAAIkC,UAAU,oEAAA,CAAA;AACtB;AACA,QAAA,MAAMG,aAAAA,GAAgB,MAAM,IAAA,CAAKC,MAAAA,CAAOhB,KAAAA,CAAAA;AACxC,QAAA,OAAO,IAAIW,eAAeI,aAAAA,CAAAA;AAC5B;AAEA,MAAA,KAAA,QAAA,EAA0B;AACxB,QAAA,IAAI,CAACf,KAAAA,CAAMzB,cAAAA,IAAkB,CAACyB,MAAMnB,OAAAA,EAAS;AAC3C,UAAA,MAAM,IAAI+B,UACR,qEAAA,CAAA;AAEJ;AACA,QAAA,MAAMK,aAAAA,GAAgB,MAAM,IAAA,CAAKC,MAAAA,CAAOlB,KAAAA,CAAAA;AACxC,QAAA,OAAO,IAAIW,eAAeM,aAAAA,CAAAA;AAC5B;AAEA,MAAA,KAAA,QAAA,EAA0B;AACxB,QAAA,IAAI,CAACjB,KAAAA,CAAMzB,cAAAA,IAAkB,CAACyB,MAAMV,GAAAA,EAAK;AACvC,UAAA,MAAM,IAAIsB,UAAU,+DAAA,CAAA;AACtB;AACA,QAAA,MAAMO,aAAAA,GAAgB,MAAM,IAAA,CAAKC,MAAAA,CAAOpB,KAAAA,CAAAA;AACxC,QAAA,OAAO,IAAIW,eAAeQ,aAAAA,CAAAA;AAC5B;MAEA,SAAS;AACP,QAAA,MAAM,IAAIP,SAAAA,CAAU,CAAA,0BAAA,EAA6BZ,KAAAA,CAAM5B,MAAM,CAAA,CAAE,CAAA;AACjE;AACF;AACF;AAEA,EAAA,MAAgBsC,eAAAA,GAAqC;AACnD,IAAA,IAAI;AACF,MAAA,MAAMP,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAAA,EAAM;AAChC,MAAA,MAAMkB,QAAAA,GAAW,MAAMlB,MAAAA,CAAOmB,cAAAA,EAAc;AAE5C,MAAA,OAAOD,SAASZ,WAAAA,CAAYc,GAAAA,CAAI,CAACC,UAAAA,KAAeA,WAAWzD,IAAI,CAAA;AACjE,KAAA,CAAA,OAAS0D,KAAAA,EAAO;AACd,MAAA,MAAM,IAAIb,SAAAA,CAAU,CAAA,wCAAA,EAA2Ca,KAAAA,CAAAA,CAAO,CAAA;AACxE;AACF;AAEA,EAAA,MAAgBX,kBAAkBvC,cAAAA,EAAsC;AACtE,IAAA,MAAM4B,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAAA,EAAM;AAChC,IAAA,MAAMkB,QAAAA,GAAW,MAAMlB,MAAAA,CAAOuB,aAAAA,CAAcnD,cAAAA,CAAAA;AAC5C,IAAA,OAAO8C,QAAAA;AACT;AAEA,EAAA,MAAgBH,OAAOlB,KAAAA,EAAsC;AAC3D,IAAA,MAAMG,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAAA,EAAM;AAEhC,IAAA,MAAMwB,SAAS3B,KAAAA,CAAMnB,OAAAA,CAAS0C,GAAAA,CAAI,CAAC7C,QAAQkD,KAAAA,MAAW;AACpDC,MAAAA,EAAAA,EAAI7B,KAAAA,EAAOV,GAAAA,GAAMsC,KAAAA,CAAAA,IAAUE,EAAAA,EAAAA;AAC3BpD,MAAAA,MAAAA;AACAW,MAAAA,OAAAA,EAASW,KAAAA,EAAOX,OAAAA,GAAUuC,KAAAA,CAAAA,IAAU;KACtC,CAAA,CAAA;AAEA,IAAA,MAAMP,QAAAA,GAAW,MAAMlB,MAAAA,CAAO4B,MAAAA,CAAO/B,MAAMzB,cAAAA,EAA0B;AACnEoD,MAAAA;KACF,CAAA;AAEA,IAAA,OAAON,QAAAA;AACT;AAEA,EAAA,MAAgBL,OAAOhB,KAAAA,EAAsC;AAC3D,IAAA,MAAMG,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAAA,EAAM;AAEhC,IAAA,MAAM6B,YAAAA,GAAoB;AACxBC,MAAAA,KAAAA,EAAOjC,KAAAA,CAAMtB,MAAAA;AACbwD,MAAAA,KAAAA,EAAOlC,MAAMlB,IAAAA,IAAQ,EAAA;MACrBqD,YAAAA,EAAc,IAAA;AACdjD,MAAAA,MAAAA,EAAQc,KAAAA,CAAMd;AAChB,KAAA;AAEA,IAAA,MAAMmC,WAAW,MAAMlB,MAAAA,CAAO8B,KAAAA,CAAMjC,KAAAA,CAAMzB,gBAA0ByD,YAAAA,CAAAA;AACpE,IAAA,OAAOX,QAAAA;AACT;AAEA,EAAA,MAAgBD,OAAOpB,KAAAA,EAAsC;AAC3D,IAAA,MAAMG,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAAA,EAAM;AAEhC,IAAA,MAAMkB,QAAAA,GAAW,MAAMlB,MAAAA,CAAOiB,MAAAA,CAAOpB,MAAMzB,cAAAA,EAA0B;AACnEoD,MAAAA,MAAAA,EAAQ3B,KAAAA,CAAMV;KAChB,CAAA;AAEA,IAAA,OAAO+B,QAAAA;AACT;AACF","file":"qdrant.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport {\n  Tool,\n  ToolInput,\n  ToolError,\n  BaseToolOptions,\n  BaseToolRunOptions,\n  JSONToolOutput,\n  ToolInputValidationError,\n  ToolEmitter,\n} from \"@/tools/base.js\";\nimport { Cache } from \"@/cache/decoratorCache.js\";\nimport { AnyToolSchemaLike } from \"@/internals/helpers/schema.js\";\nimport { QdrantClient, QdrantClientParams } from \"@qdrant/js-client-rest\";\nimport { v4 as uuidv4 } from \"uuid\";\n\nimport { z } from \"zod\";\nimport { Emitter } from \"@/emitter/emitter.js\";\n\nexport interface QdrantToolOptions extends BaseToolOptions, QdrantClientParams {\n  connection: QdrantClientParams;\n}\n\nexport type QdrantSearchToolResult = any;\n\nexport enum QdrantAction {\n  ListCollections = \"ListCollections\",\n  GetCollectionInfo = \"GetCollectionInfo\",\n  Search = \"Search\",\n  Insert = \"Insert\",\n  Delete = \"Delete\",\n}\n\nexport class QdrantDatabaseTool extends Tool<\n  JSONToolOutput<QdrantSearchToolResult>,\n  QdrantToolOptions\n> {\n  name = \"QdrantDatabaseTool\";\n\n  description = `Can query data from a Qdrant vector database. IMPORTANT: strictly follow this order of actions:\n     1. ${QdrantAction.ListCollections} - List all the Qdrant collections\n     2. ${QdrantAction.GetCollectionInfo} - Get information about a Qdrant collection\n     3. ${QdrantAction.Insert} - Insert data into a Qdrant collection\n     3. ${QdrantAction.Search} - Perform search on a Qdrant collection\n     4. ${QdrantAction.Delete} - Delete from a Qdrant collection`;\n\n  inputSchema() {\n    return z.object({\n      action: z\n        .nativeEnum(QdrantAction)\n        .describe(\n          `The action to perform. ${QdrantAction.ListCollections} lists all collections, ${QdrantAction.GetCollectionInfo} fetches details for a specified collection, ${QdrantAction.Search} executes a vector search, ${QdrantAction.Insert} inserts new vectors, and ${QdrantAction.Delete} removes vectors.`,\n        ),\n      collectionName: z\n        .string()\n        .optional()\n        .describe(\n          `The name of the collection to query, required for ${QdrantAction.GetCollectionInfo}, ${QdrantAction.Search}, ${QdrantAction.Insert}, and ${QdrantAction.Delete}`,\n        ),\n      vector: z\n        .array(z.number())\n        .optional()\n        .describe(`The vector to search for, required for ${QdrantAction.Search}.`),\n      vectors: z\n        .array(z.array(z.number()))\n        .optional()\n        .describe(`The vectors to insert, required for ${QdrantAction.Insert}.`),\n      topK: z.coerce\n        .number()\n        .int()\n        .default(10)\n        .optional()\n        .describe(`The number of nearest neighbors to return.`),\n      filter: z\n        .record(z.string(), z.any())\n        .optional()\n        .describe(`Optional filter for ${QdrantAction.Search}.`),\n      payload: z\n        .array(z.record(z.string(), z.any()))\n        .optional()\n        .describe(`Additional payload to insert with vectors.`),\n      ids: z\n        .array(z.string().or(z.number()))\n        .optional()\n        .describe(`Array of IDs to delete or insert.`),\n    });\n  }\n\n  public readonly emitter: ToolEmitter<ToolInput<this>, JSONToolOutput<QdrantSearchToolResult>> =\n    Emitter.root.child({\n      namespace: [\"tool\", \"database\", \"qdrant\"],\n      creator: this,\n    });\n\n  protected validateInput(\n    schema: AnyToolSchemaLike,\n    input: unknown,\n  ): asserts input is ToolInput<this> {\n    super.validateInput(schema, input);\n    if (input.action != QdrantAction.ListCollections && !input.collectionName) {\n      throw new ToolInputValidationError(\n        `Collection name is required for ${QdrantAction.GetCollectionInfo}, ${QdrantAction.Search}, ${QdrantAction.Insert}, and ${QdrantAction.Delete} actions.`,\n      );\n    }\n    if (input.action === QdrantAction.Search && (!input.collectionName || !input.vector)) {\n      throw new ToolInputValidationError(`Vector is required for ${QdrantAction.Search} action.`);\n    }\n    if (input.action === QdrantAction.Insert && (!input.collectionName || !input.vectors)) {\n      throw new ToolInputValidationError(`Vectors are required for ${QdrantAction.Insert} action.`);\n    }\n  }\n\n  static {\n    this.register();\n  }\n\n  @Cache()\n  protected async client(): Promise<QdrantClient> {\n    return new QdrantClient(this.options.connection);\n  }\n\n  protected async _run(\n    input: ToolInput<this>,\n    _options: Partial<BaseToolRunOptions>,\n  ): Promise<JSONToolOutput<any>> {\n    switch (input.action) {\n      case QdrantAction.ListCollections: {\n        const collections = await this.listCollections();\n        return new JSONToolOutput(collections);\n      }\n\n      case QdrantAction.GetCollectionInfo: {\n        if (!input.collectionName) {\n          throw new ToolError(\"A collection name is required for Qdrant GetCollectionInfo action\");\n        }\n        const collectionInfo = await this.getCollectionInfo(input.collectionName);\n        return new JSONToolOutput(collectionInfo);\n      }\n\n      case QdrantAction.Search: {\n        if (!input.collectionName || !input.vector) {\n          throw new ToolError(\"A collection name and vector are required for Qdrant Search action\");\n        }\n        const searchResults = await this.search(input);\n        return new JSONToolOutput(searchResults);\n      }\n\n      case QdrantAction.Insert: {\n        if (!input.collectionName || !input.vectors) {\n          throw new ToolError(\n            \"A collection name and vectors are required for Qdrant Insert action\",\n          );\n        }\n        const insertResults = await this.insert(input);\n        return new JSONToolOutput(insertResults);\n      }\n\n      case QdrantAction.Delete: {\n        if (!input.collectionName || !input.ids) {\n          throw new ToolError(\"Collection name and ids are required for Qdrant Delete action\");\n        }\n        const deleteResults = await this.delete(input);\n        return new JSONToolOutput(deleteResults);\n      }\n\n      default: {\n        throw new ToolError(`Invalid action specified: ${input.action}`);\n      }\n    }\n  }\n\n  protected async listCollections(): Promise<string[]> {\n    try {\n      const client = await this.client();\n      const response = await client.getCollections();\n\n      return response.collections.map((collection) => collection.name);\n    } catch (error) {\n      throw new ToolError(`Failed to list collections from Qdrant: ${error}`);\n    }\n  }\n\n  protected async getCollectionInfo(collectionName: string): Promise<any> {\n    const client = await this.client();\n    const response = await client.getCollection(collectionName);\n    return response;\n  }\n\n  protected async insert(input: ToolInput<this>): Promise<any> {\n    const client = await this.client();\n\n    const points = input.vectors!.map((vector, index) => ({\n      id: input?.ids?.[index] ?? uuidv4(),\n      vector: vector,\n      payload: input?.payload?.[index] || {},\n    }));\n\n    const response = await client.upsert(input.collectionName as string, {\n      points: points,\n    });\n\n    return response;\n  }\n\n  protected async search(input: ToolInput<this>): Promise<any> {\n    const client = await this.client();\n\n    const searchParams: any = {\n      query: input.vector,\n      limit: input.topK || 10,\n      with_payload: true,\n      filter: input.filter,\n    };\n\n    const response = await client.query(input.collectionName as string, searchParams);\n    return response;\n  }\n\n  protected async delete(input: ToolInput<this>): Promise<any> {\n    const client = await this.client();\n\n    const response = await client.delete(input.collectionName as string, {\n      points: input.ids!,\n    });\n\n    return response;\n  }\n}\n"]}
+{"version":3,"sources":["../../../src/tools/database/qdrant.ts"],"names":["QdrantAction","QdrantDatabaseTool","Tool","name","description","inputSchema","z","object","action","nativeEnum","describe","collectionName","string","optional","vector","array","number","vectors","topK","coerce","int","default","filter","record","any","payload","ids","or","emitter","Emitter","root","child","namespace","creator","validateInput","schema","input","ToolInputValidationError","register","client","QdrantClient","options","connection","_run","_options","collections","listCollections","JSONToolOutput","ToolError","collectionInfo","getCollectionInfo","searchResults","search","insertResults","insert","deleteResults","delete","response","getCollections","map","collection","error","getCollection","points","index","id","uuidv4","upsert","searchParams","query","limit","with_payload"],"mappings":";;;;;;;;;;;;;;;;;;;;AA6BO,IAAKA,YAAAA,6BAAAA,aAAAA,EAAAA;;;;;;AAAAA,EAAAA,OAAAA,aAAAA;;AAQL,MAAMC,2BAA2BC,IAAAA,CAAAA;AAAAA,EAAAA;;;EAItCC,IAAAA,GAAO,oBAAA;EAEPC,WAAAA,GAAc,CAAA;UACR,iBAAA,CAAA;UACA,mBAAA,CAAA;UACA,QAAA,CAAA;UACA,QAAA,CAAA;UACA,QAAA,CAAA,kCAAA,CAAA;EAENC,WAAAA,GAAc;AACZ,IAAA,OAAOC,EAAEC,MAAAA,CAAO;AACdC,MAAAA,MAAAA,EAAQF,CAAAA,CACLG,UAAAA,CAAWT,YAAAA,CAAAA,CACXU,SACC,CAAA,uBAAA,EAAwB,iBAAA,CAAA,wBAAA,EAAuD,mBAAA,gDAA8E,QAAA,CAAA,2BAAA,EAAiD,QAAA,CAAA,0BAAA,EAAgD,QAAA,CAAA,iBAAA,CAAwC,CAAA;AAE1SC,MAAAA,cAAAA,EAAgBL,CAAAA,CACbM,MAAAA,EAAM,CACNC,QAAAA,GACAH,QAAAA,CACC,CAAA,kDAAA,EAAmD,mBAAA,CAAA,EAAA,EAAmC,QAAA,CAAA,EAAA,EAAwB,QAAA,CAAA,MAAA,EAA4B,QAAA,CAAA,CAAuB,CAAA;MAErKI,MAAAA,EAAQR,CAAAA,CACLS,KAAAA,CAAMT,CAAAA,CAAEU,MAAAA,EAAM,CAAA,CACdH,QAAAA,EAAQ,CACRH,QAAAA,CAAS,CAAA,uCAAA,EAAwC,QAAA,CAAA,CAAA,CAAwB,CAAA;AAC5EO,MAAAA,OAAAA,EAASX,CAAAA,CACNS,KAAAA,CAAMT,CAAAA,CAAES,KAAAA,CAAMT,EAAEU,MAAAA,EAAM,CAAA,CAAA,CACtBH,QAAAA,EAAQ,CACRH,QAAAA,CAAS,CAAA,oCAAA,EAAqC,QAAA,CAAA,CAAA,CAAwB,CAAA;AACzEQ,MAAAA,IAAAA,EAAMZ,CAAAA,CAAEa,MAAAA,CACLH,MAAAA,EAAM,CACNI,GAAAA,EAAG,CACHC,OAAAA,CAAQ,EAAA,CAAA,CACRR,QAAAA,EAAQ,CACRH,QAAAA,CAAS,CAAA,0CAAA,CAA4C,CAAA;AACxDY,MAAAA,MAAAA,EAAQhB,CAAAA,CACLiB,MAAAA,CAAOjB,CAAAA,CAAEM,MAAAA,IAAUN,CAAAA,CAAEkB,GAAAA,EAAG,CAAA,CACxBX,QAAAA,EAAQ,CACRH,QAAAA,CAAS,CAAA,oBAAA,EAAqB,QAAA,CAAA,CAAA,CAAwB,CAAA;AACzDe,MAAAA,OAAAA,EAASnB,CAAAA,CACNS,KAAAA,CAAMT,CAAAA,CAAEiB,MAAAA,CAAOjB,EAAEM,MAAAA,EAAM,EAAIN,CAAAA,CAAEkB,GAAAA,EAAG,CAAA,CAAA,CAChCX,QAAAA,EAAQ,CACRH,SAAS,CAAA,0CAAA,CAA4C,CAAA;AACxDgB,MAAAA,GAAAA,EAAKpB,CAAAA,CACFS,KAAAA,CAAMT,CAAAA,CAAEM,MAAAA,GAASe,EAAAA,CAAGrB,CAAAA,CAAEU,MAAAA,EAAM,CAAA,CAAA,CAC5BH,QAAAA,EAAQ,CACRH,SAAS,CAAA,iCAAA,CAAmC;KACjD,CAAA;AACF,EAAA;EAEgBkB,OAAAA,GACdC,OAAAA,CAAQC,KAAKC,KAAAA,CAAM;IACjBC,SAAAA,EAAW;AAAC,MAAA,MAAA;AAAQ,MAAA,UAAA;AAAY,MAAA;;IAChCC,OAAAA,EAAS;GACX,CAAA;AAEQC,EAAAA,aAAAA,CACRC,QACAC,KAAAA,EACkC;AAClC,IAAA,KAAA,CAAMF,aAAAA,CAAcC,QAAQC,KAAAA,CAAAA;AAC5B,IAAA,IAAIA,KAAAA,CAAM5B,MAAAA,IAAM,iBAAA,IAAoC,CAAC4B,MAAMzB,cAAAA,EAAgB;AACzE,MAAA,MAAM,IAAI0B,wBAAAA,CACR,CAAA,gCAAA,EAAiC,mBAAA,CAAA,EAAA,EAAmC,QAAA,CAAA,EAAA,EAAwB,QAAA,CAAA,MAAA,EAA4B,QAAA,CAAA,SAAA,CAAgC,CAAA;AAE5J,IAAA;AACA,IAAA,IAAID,KAAAA,CAAM5B,WAAM,QAAA,KAA6B,CAAC4B,MAAMzB,cAAAA,IAAkB,CAACyB,MAAMtB,MAAAA,CAAAA,EAAS;AACpF,MAAA,MAAM,IAAIuB,wBAAAA,CAAyB,CAAA,uBAAA,EAAwB,QAAA,CAAA,QAAA,CAA+B,CAAA;AAC5F,IAAA;AACA,IAAA,IAAID,KAAAA,CAAM5B,WAAM,QAAA,KAA6B,CAAC4B,MAAMzB,cAAAA,IAAkB,CAACyB,MAAMnB,OAAAA,CAAAA,EAAU;AACrF,MAAA,MAAM,IAAIoB,wBAAAA,CAAyB,CAAA,yBAAA,EAA0B,QAAA,CAAA,QAAA,CAA+B,CAAA;AAC9F,IAAA;AACF,EAAA;EAEA;AACE,IAAA,IAAA,CAAKC,QAAAA,EAAQ;AACf;AAEA,EAAA,MACgBC,MAAAA,GAAgC;AAC9C,IAAA,OAAO,IAAIC,YAAAA,CAAa,IAAA,CAAKC,OAAAA,CAAQC,UAAU,CAAA;AACjD,EAAA;EAEA,MAAgBC,IAAAA,CACdP,OACAQ,QAAAA,EAC8B;AAC9B,IAAA,QAAQR,MAAM5B,MAAAA;AACZ,MAAA,KAAA,iBAAA,EAAmC;AACjC,QAAA,MAAMqC,WAAAA,GAAc,MAAM,IAAA,CAAKC,eAAAA,EAAe;AAC9C,QAAA,OAAO,IAAIC,eAAeF,WAAAA,CAAAA;AAC5B,MAAA;AAEA,MAAA,KAAA,mBAAA,EAAqC;AACnC,QAAA,IAAI,CAACT,MAAMzB,cAAAA,EAAgB;AACzB,UAAA,MAAM,IAAIqC,UAAU,mEAAA,CAAA;AACtB,QAAA;AACA,QAAA,MAAMC,cAAAA,GAAiB,MAAM,IAAA,CAAKC,iBAAAA,CAAkBd,MAAMzB,cAAc,CAAA;AACxE,QAAA,OAAO,IAAIoC,eAAeE,cAAAA,CAAAA;AAC5B,MAAA;AAEA,MAAA,KAAA,QAAA,EAA0B;AACxB,QAAA,IAAI,CAACb,KAAAA,CAAMzB,cAAAA,IAAkB,CAACyB,MAAMtB,MAAAA,EAAQ;AAC1C,UAAA,MAAM,IAAIkC,UAAU,oEAAA,CAAA;AACtB,QAAA;AACA,QAAA,MAAMG,aAAAA,GAAgB,MAAM,IAAA,CAAKC,MAAAA,CAAOhB,KAAAA,CAAAA;AACxC,QAAA,OAAO,IAAIW,eAAeI,aAAAA,CAAAA;AAC5B,MAAA;AAEA,MAAA,KAAA,QAAA,EAA0B;AACxB,QAAA,IAAI,CAACf,KAAAA,CAAMzB,cAAAA,IAAkB,CAACyB,MAAMnB,OAAAA,EAAS;AAC3C,UAAA,MAAM,IAAI+B,UACR,qEAAA,CAAA;AAEJ,QAAA;AACA,QAAA,MAAMK,aAAAA,GAAgB,MAAM,IAAA,CAAKC,MAAAA,CAAOlB,KAAAA,CAAAA;AACxC,QAAA,OAAO,IAAIW,eAAeM,aAAAA,CAAAA;AAC5B,MAAA;AAEA,MAAA,KAAA,QAAA,EAA0B;AACxB,QAAA,IAAI,CAACjB,KAAAA,CAAMzB,cAAAA,IAAkB,CAACyB,MAAMV,GAAAA,EAAK;AACvC,UAAA,MAAM,IAAIsB,UAAU,+DAAA,CAAA;AACtB,QAAA;AACA,QAAA,MAAMO,aAAAA,GAAgB,MAAM,IAAA,CAAKC,MAAAA,CAAOpB,KAAAA,CAAAA;AACxC,QAAA,OAAO,IAAIW,eAAeQ,aAAAA,CAAAA;AAC5B,MAAA;MAEA,SAAS;AACP,QAAA,MAAM,IAAIP,SAAAA,CAAU,CAAA,0BAAA,EAA6BZ,KAAAA,CAAM5B,MAAM,CAAA,CAAE,CAAA;AACjE,MAAA;AACF;AACF,EAAA;AAEA,EAAA,MAAgBsC,eAAAA,GAAqC;AACnD,IAAA,IAAI;AACF,MAAA,MAAMP,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAAA,EAAM;AAChC,MAAA,MAAMkB,QAAAA,GAAW,MAAMlB,MAAAA,CAAOmB,cAAAA,EAAc;AAE5C,MAAA,OAAOD,SAASZ,WAAAA,CAAYc,GAAAA,CAAI,CAACC,UAAAA,KAAeA,WAAWzD,IAAI,CAAA;AACjE,IAAA,CAAA,CAAA,OAAS0D,KAAAA,EAAO;AACd,MAAA,MAAM,IAAIb,SAAAA,CAAU,CAAA,wCAAA,EAA2Ca,KAAAA,CAAAA,CAAO,CAAA;AACxE,IAAA;AACF,EAAA;AAEA,EAAA,MAAgBX,kBAAkBvC,cAAAA,EAAsC;AACtE,IAAA,MAAM4B,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAAA,EAAM;AAChC,IAAA,MAAMkB,QAAAA,GAAW,MAAMlB,MAAAA,CAAOuB,aAAAA,CAAcnD,cAAAA,CAAAA;AAC5C,IAAA,OAAO8C,QAAAA;AACT,EAAA;AAEA,EAAA,MAAgBH,OAAOlB,KAAAA,EAAsC;AAC3D,IAAA,MAAMG,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAAA,EAAM;AAEhC,IAAA,MAAMwB,SAAS3B,KAAAA,CAAMnB,OAAAA,CAAS0C,GAAAA,CAAI,CAAC7C,QAAQkD,KAAAA,MAAW;AACpDC,MAAAA,EAAAA,EAAI7B,KAAAA,EAAOV,GAAAA,GAAMsC,KAAAA,CAAAA,IAAUE,EAAAA,EAAAA;AAC3BpD,MAAAA,MAAAA;AACAW,MAAAA,OAAAA,EAASW,KAAAA,EAAOX,OAAAA,GAAUuC,KAAAA,CAAAA,IAAU;KACtC,CAAA,CAAA;AAEA,IAAA,MAAMP,QAAAA,GAAW,MAAMlB,MAAAA,CAAO4B,MAAAA,CAAO/B,MAAMzB,cAAAA,EAA0B;AACnEoD,MAAAA;KACF,CAAA;AAEA,IAAA,OAAON,QAAAA;AACT,EAAA;AAEA,EAAA,MAAgBL,OAAOhB,KAAAA,EAAsC;AAC3D,IAAA,MAAMG,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAAA,EAAM;AAEhC,IAAA,MAAM6B,YAAAA,GAAoB;AACxBC,MAAAA,KAAAA,EAAOjC,KAAAA,CAAMtB,MAAAA;AACbwD,MAAAA,KAAAA,EAAOlC,MAAMlB,IAAAA,IAAQ,EAAA;MACrBqD,YAAAA,EAAc,IAAA;AACdjD,MAAAA,MAAAA,EAAQc,KAAAA,CAAMd;AAChB,KAAA;AAEA,IAAA,MAAMmC,WAAW,MAAMlB,MAAAA,CAAO8B,KAAAA,CAAMjC,KAAAA,CAAMzB,gBAA0ByD,YAAAA,CAAAA;AACpE,IAAA,OAAOX,QAAAA;AACT,EAAA;AAEA,EAAA,MAAgBD,OAAOpB,KAAAA,EAAsC;AAC3D,IAAA,MAAMG,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAAA,EAAM;AAEhC,IAAA,MAAMkB,QAAAA,GAAW,MAAMlB,MAAAA,CAAOiB,MAAAA,CAAOpB,MAAMzB,cAAAA,EAA0B;AACnEoD,MAAAA,MAAAA,EAAQ3B,KAAAA,CAAMV;KAChB,CAAA;AAEA,IAAA,OAAO+B,QAAAA;AACT,EAAA;AACF","file":"qdrant.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport {\n  Tool,\n  ToolInput,\n  ToolError,\n  BaseToolOptions,\n  BaseToolRunOptions,\n  JSONToolOutput,\n  ToolInputValidationError,\n  ToolEmitter,\n} from \"@/tools/base.js\";\nimport { Cache } from \"@/cache/decoratorCache.js\";\nimport { AnyToolSchemaLike } from \"@/internals/helpers/schema.js\";\nimport { QdrantClient, QdrantClientParams } from \"@qdrant/js-client-rest\";\nimport { v4 as uuidv4 } from \"uuid\";\n\nimport { z } from \"zod\";\nimport { Emitter } from \"@/emitter/emitter.js\";\n\nexport interface QdrantToolOptions extends BaseToolOptions, QdrantClientParams {\n  connection: QdrantClientParams;\n}\n\nexport type QdrantSearchToolResult = any;\n\nexport enum QdrantAction {\n  ListCollections = \"ListCollections\",\n  GetCollectionInfo = \"GetCollectionInfo\",\n  Search = \"Search\",\n  Insert = \"Insert\",\n  Delete = \"Delete\",\n}\n\nexport class QdrantDatabaseTool extends Tool<\n  JSONToolOutput<QdrantSearchToolResult>,\n  QdrantToolOptions\n> {\n  name = \"QdrantDatabaseTool\";\n\n  description = `Can query data from a Qdrant vector database. IMPORTANT: strictly follow this order of actions:\n     1. ${QdrantAction.ListCollections} - List all the Qdrant collections\n     2. ${QdrantAction.GetCollectionInfo} - Get information about a Qdrant collection\n     3. ${QdrantAction.Insert} - Insert data into a Qdrant collection\n     3. ${QdrantAction.Search} - Perform search on a Qdrant collection\n     4. ${QdrantAction.Delete} - Delete from a Qdrant collection`;\n\n  inputSchema() {\n    return z.object({\n      action: z\n        .nativeEnum(QdrantAction)\n        .describe(\n          `The action to perform. ${QdrantAction.ListCollections} lists all collections, ${QdrantAction.GetCollectionInfo} fetches details for a specified collection, ${QdrantAction.Search} executes a vector search, ${QdrantAction.Insert} inserts new vectors, and ${QdrantAction.Delete} removes vectors.`,\n        ),\n      collectionName: z\n        .string()\n        .optional()\n        .describe(\n          `The name of the collection to query, required for ${QdrantAction.GetCollectionInfo}, ${QdrantAction.Search}, ${QdrantAction.Insert}, and ${QdrantAction.Delete}`,\n        ),\n      vector: z\n        .array(z.number())\n        .optional()\n        .describe(`The vector to search for, required for ${QdrantAction.Search}.`),\n      vectors: z\n        .array(z.array(z.number()))\n        .optional()\n        .describe(`The vectors to insert, required for ${QdrantAction.Insert}.`),\n      topK: z.coerce\n        .number()\n        .int()\n        .default(10)\n        .optional()\n        .describe(`The number of nearest neighbors to return.`),\n      filter: z\n        .record(z.string(), z.any())\n        .optional()\n        .describe(`Optional filter for ${QdrantAction.Search}.`),\n      payload: z\n        .array(z.record(z.string(), z.any()))\n        .optional()\n        .describe(`Additional payload to insert with vectors.`),\n      ids: z\n        .array(z.string().or(z.number()))\n        .optional()\n        .describe(`Array of IDs to delete or insert.`),\n    });\n  }\n\n  public readonly emitter: ToolEmitter<ToolInput<this>, JSONToolOutput<QdrantSearchToolResult>> =\n    Emitter.root.child({\n      namespace: [\"tool\", \"database\", \"qdrant\"],\n      creator: this,\n    });\n\n  protected validateInput(\n    schema: AnyToolSchemaLike,\n    input: unknown,\n  ): asserts input is ToolInput<this> {\n    super.validateInput(schema, input);\n    if (input.action != QdrantAction.ListCollections && !input.collectionName) {\n      throw new ToolInputValidationError(\n        `Collection name is required for ${QdrantAction.GetCollectionInfo}, ${QdrantAction.Search}, ${QdrantAction.Insert}, and ${QdrantAction.Delete} actions.`,\n      );\n    }\n    if (input.action === QdrantAction.Search && (!input.collectionName || !input.vector)) {\n      throw new ToolInputValidationError(`Vector is required for ${QdrantAction.Search} action.`);\n    }\n    if (input.action === QdrantAction.Insert && (!input.collectionName || !input.vectors)) {\n      throw new ToolInputValidationError(`Vectors are required for ${QdrantAction.Insert} action.`);\n    }\n  }\n\n  static {\n    this.register();\n  }\n\n  @Cache()\n  protected async client(): Promise<QdrantClient> {\n    return new QdrantClient(this.options.connection);\n  }\n\n  protected async _run(\n    input: ToolInput<this>,\n    _options: Partial<BaseToolRunOptions>,\n  ): Promise<JSONToolOutput<any>> {\n    switch (input.action) {\n      case QdrantAction.ListCollections: {\n        const collections = await this.listCollections();\n        return new JSONToolOutput(collections);\n      }\n\n      case QdrantAction.GetCollectionInfo: {\n        if (!input.collectionName) {\n          throw new ToolError(\"A collection name is required for Qdrant GetCollectionInfo action\");\n        }\n        const collectionInfo = await this.getCollectionInfo(input.collectionName);\n        return new JSONToolOutput(collectionInfo);\n      }\n\n      case QdrantAction.Search: {\n        if (!input.collectionName || !input.vector) {\n          throw new ToolError(\"A collection name and vector are required for Qdrant Search action\");\n        }\n        const searchResults = await this.search(input);\n        return new JSONToolOutput(searchResults);\n      }\n\n      case QdrantAction.Insert: {\n        if (!input.collectionName || !input.vectors) {\n          throw new ToolError(\n            \"A collection name and vectors are required for Qdrant Insert action\",\n          );\n        }\n        const insertResults = await this.insert(input);\n        return new JSONToolOutput(insertResults);\n      }\n\n      case QdrantAction.Delete: {\n        if (!input.collectionName || !input.ids) {\n          throw new ToolError(\"Collection name and ids are required for Qdrant Delete action\");\n        }\n        const deleteResults = await this.delete(input);\n        return new JSONToolOutput(deleteResults);\n      }\n\n      default: {\n        throw new ToolError(`Invalid action specified: ${input.action}`);\n      }\n    }\n  }\n\n  protected async listCollections(): Promise<string[]> {\n    try {\n      const client = await this.client();\n      const response = await client.getCollections();\n\n      return response.collections.map((collection) => collection.name);\n    } catch (error) {\n      throw new ToolError(`Failed to list collections from Qdrant: ${error}`);\n    }\n  }\n\n  protected async getCollectionInfo(collectionName: string): Promise<any> {\n    const client = await this.client();\n    const response = await client.getCollection(collectionName);\n    return response;\n  }\n\n  protected async insert(input: ToolInput<this>): Promise<any> {\n    const client = await this.client();\n\n    const points = input.vectors!.map((vector, index) => ({\n      id: input?.ids?.[index] ?? uuidv4(),\n      vector: vector,\n      payload: input?.payload?.[index] || {},\n    }));\n\n    const response = await client.upsert(input.collectionName as string, {\n      points: points,\n    });\n\n    return response;\n  }\n\n  protected async search(input: ToolInput<this>): Promise<any> {\n    const client = await this.client();\n\n    const searchParams: any = {\n      query: input.vector,\n      limit: input.topK || 10,\n      with_payload: true,\n      filter: input.filter,\n    };\n\n    const response = await client.query(input.collectionName as string, searchParams);\n    return response;\n  }\n\n  protected async delete(input: ToolInput<this>): Promise<any> {\n    const client = await this.client();\n\n    const response = await client.delete(input.collectionName as string, {\n      points: input.ids!,\n    });\n\n    return response;\n  }\n}\n"]}

package/dist/tools/database/sql.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../src/tools/database/sql.ts"],"names":["SQLToolAction","GetMetadata","Query","SQLToolOutput","JSONToolOutput","SQLTool","Tool","name","description","inputSchema","z","object","action","nativeEnum","describe","query","string","optional","emitter","Emitter","root","child","namespace","creator","options","connection","dialect","ValidationError","message","propertyName","schema","provider","storage","validateInput","input","ToolInputValidationError","register","sequelize","Sequelize","authenticate","error","ToolError","isRetryable","isFatal","_run","_options","metadata","getMetadata","executeQuery","isReadOnlyQuery","success","results","Array","isArray","length","schemaHint","errorMessage","normalizedQuery","trim","toUpperCase","startsWith","destroy","cache","Cache","getInstance","entry","get","clear","data","close"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;AA8BO,MAAMA,aAAAA,GAAgB;EAC3BC,WAAAA,EAAa,cAAA;EACbC,KAAAA,EAAO;AACT;AAEO,MAAMC,sBAAsBC,uBAAAA,CAAAA;AAAAA,EAAAA;;;AAAqB;AAEjD,MAAMC,gBAAgBC,aAAAA,CAAAA;AAAAA,EAAAA;;;EAC3BC,IAAAA,GAAO,SAAA;EAEPC,WAAAA,GAAc,CAAA;AACRR,MAAAA,EAAAA,aAAAA,CAAcC,WAAW,CAAA;AACzBD,MAAAA,EAAAA,aAAAA,CAAcE,KAAK,CAAA,kCAAA,CAAA;EAEzBO,WAAAA,GAAc;AACZ,IAAA,OAAOC,MAAEC,MAAAA,CAAO;MACdC,MAAAA,EAAQF,KAAAA,CACLG,UAAAA,CAAWb,aAAAA,CAAAA,CACXc,QAAAA,CACC,CAAA,uBAAA,EAA0Bd,aAAAA,CAAcC,WAAW,CAAA,gCAAA,EAAmCD,aAAAA,CAAcE,KAAK,CAAA,sBAAA,CAAwB,CAAA;MAErIa,KAAAA,EAAOL,KAAAA,CACJM,QAAM,CACNC,QAAAA,GACAH,QAAAA,CAAS,CAAA,2CAAA,EAA8Cd,aAAAA,CAAcE,KAAK,CAAA,OAAA,CAAS;KACxF,CAAA;AACF;EAEgBgB,OAAAA,GAAuDC,mBAAAA,CAAQC,KAAKC,KAAAA,CAAM;IACxFC,SAAAA,EAAW;AAAC,MAAA,MAAA;AAAQ,MAAA,UAAA;AAAY,MAAA;;IAChCC,OAAAA,EAAS;GACX,CAAA;AAEA,EAAA,WAAA,CAAmBC,OAAAA,EAAsB;AACvC,IAAA,KAAA,CAAMA,OAAAA,CAAAA;AACN,IAAA,IAAI,CAACA,OAAAA,CAAQC,UAAAA,CAAWC,OAAAA,EAAS;AAC/B,MAAA,MAAM,IAAIC,mBAAAA,CAAgB;AACxB,QAAA;UACEC,OAAAA,EAAS,sBAAA;UACTC,YAAAA,EAAc;AAChB;AACD,OAAA,CAAA;AACH;AACA,IAAA,IACE,CAACL,QAAQC,UAAAA,CAAWK,MAAAA,KACnBN,QAAQO,QAAAA,KAAa,QAAA,IAAYP,OAAAA,CAAQO,QAAAA,KAAa,KAAA,CAAA,EACvD;AACA,MAAA,MAAM,IAAIJ,mBAAAA,CAAgB;AACxB,QAAA;UACEC,OAAAA,EAAS,CAAA,yBAAA,EAA4BJ,QAAQO,QAAQ,CAAA,CAAA;UACrDF,YAAAA,EAAc;AAChB;AACD,OAAA,CAAA;AACH;AACA,IAAA,IAAI,CAACL,OAAAA,CAAQC,UAAAA,CAAWO,OAAAA,IAAWR,OAAAA,CAAQO,aAAa,QAAA,EAAU;AAChE,MAAA,MAAM,IAAIJ,mBAAAA,CAAgB;AACxB,QAAA;UACEC,OAAAA,EAAS,CAAA,yBAAA,EAA4BJ,QAAQO,QAAQ,CAAA,CAAA;UACrDF,YAAAA,EAAc;AAChB;AACD,OAAA,CAAA;AACH;AACF;AAEUI,EAAAA,aAAAA,CACRH,QACAI,KAAAA,EACkC;AAClC,IAAA,KAAA,CAAMD,aAAAA,CAAcH,QAAQI,KAAAA,CAAAA;AAC5B,IAAA,IAAIA,MAAMtB,MAAAA,KAAWZ,aAAAA,CAAcE,KAAAA,IAAS,CAACgC,MAAMnB,KAAAA,EAAO;AACxD,MAAA,MAAM,IAAIoB,iCAAAA,CACR,CAAA,0BAAA,EAA6BnC,aAAAA,CAAcE,KAAK,CAAA,QAAA,CAAU,CAAA;AAE9D;AACF;EAEA;AACE,IAAA,IAAA,CAAKkC,QAAAA,EAAQ;AACf;AAEA,EAAA,MACgBX,UAAAA,GAAiC;AAC/C,IAAA,IAAI;AACF,MAAA,MAAMY,WAAAA,GAAY,IAAIC,mBAAAA,CAAU,IAAA,CAAKd,QAAQC,UAAU,CAAA;AAEvD,MAAA,MAAMY,YAAUE,YAAAA,EAAY;AAC5B,MAAA,OAAOF,WAAAA;AACT,KAAA,CAAA,OAASG,KAAAA,EAAO;AACd,MAAA,MAAM,IAAIC,kBAAAA,CAAU,CAAA,+BAAA,EAAkCD,KAAAA,CAAAA,CAAAA,EAAS,EAAA,EAAI;QACjEE,WAAAA,EAAa,KAAA;QACbC,OAAAA,EAAS;OACX,CAAA;AACF;AACF;EAEA,MAAgBC,IAAAA,CACdV,OACAW,QAAAA,EACwB;AACxB,IAAA,MAAM,EAAEd,QAAAA,EAAUN,UAAAA,EAAU,GAAK,IAAA,CAAKD,OAAAA;AACtC,IAAA,MAAM,EAAEM,QAAM,GAAKL,UAAAA;AAEnB,IAAA,IAAIS,KAAAA,CAAMtB,MAAAA,KAAWZ,aAAAA,CAAcC,WAAAA,EAAa;AAC9C,MAAA,MAAMoC,SAAAA,GAAY,MAAM,IAAA,CAAKZ,UAAAA,EAAU;AACvC,MAAA,MAAMqB,QAAAA,GAAW,MAAMC,wBAAAA,CAAYV,SAAAA,EAAWN,UAAUD,MAAAA,CAAAA;AACxD,MAAA,OAAO,IAAI3B,cAAc2C,QAAAA,CAAAA;AAC3B;AAEA,IAAA,IAAIZ,KAAAA,CAAMtB,MAAAA,KAAWZ,aAAAA,CAAcE,KAAAA,EAAO;AACxC,MAAA,OAAO,MAAM,IAAA,CAAK8C,YAAAA,CAAad,KAAAA,CAAMnB,KAAAA,EAAQgB,UAAUD,MAAAA,CAAAA;AACzD;AAEA,IAAA,MAAM,IAAIW,kBAAAA,CAAU,CAAA,0BAAA,EAA6BP,KAAAA,CAAMtB,MAAM,CAAA,CAAE,CAAA;AACjE;EAEA,MAAgBoC,YAAAA,CACdjC,KAAAA,EACAgB,QAAAA,EACAD,MAAAA,EACwB;AACxB,IAAA,IAAI,CAAC,IAAA,CAAKmB,eAAAA,CAAgBlC,KAAAA,CAAAA,EAAQ;AAChC,MAAA,OAAO,IAAIX,uBAAAA,CAAe;QACxB8C,OAAAA,EAAS,KAAA;QACTV,KAAAA,EAAO;OACT,CAAA;AACF;AAEA,IAAA,IAAI;AACF,MAAA,MAAMH,SAAAA,GAAY,MAAM,IAAA,CAAKZ,UAAAA,EAAU;AACvC,MAAA,MAAM,CAAC0B,OAAAA,CAAAA,GAAW,MAAMd,SAAAA,CAAUtB,MAAMA,KAAAA,CAAAA;AACxC,MAAA,IAAIqC,MAAMC,OAAAA,CAAQF,OAAAA,CAAAA,IAAYA,OAAAA,CAAQG,SAAS,CAAA,EAAG;AAChD,QAAA,OAAO,IAAIlD,uBAAAA,CAAe;UAAE8C,OAAAA,EAAS,IAAA;AAAMC,UAAAA;SAAQ,CAAA;AACrD;AAEA,MAAA,OAAO,IAAI/C,uBAAAA,CAAe;QACxB8C,OAAAA,EAAS,KAAA;QACTtB,OAAAA,EAAS,CAAA,gBAAA;OACX,CAAA;AACF,KAAA,CAAA,OAASY,KAAAA,EAAO;AACd,MAAA,MAAMe,aAAazB,MAAAA,GACf,CAAA,4DAAA,EAA+DA,MAAAA,CAAAA,4BAAAA,EAAqCA,MAAAA,CAAAA,WAAAA,CAAAA,GACpG,EAAA;AACJ,MAAA,MAAM0B,YAAAA,GAAe,sEAAsEzB,QAAAA,CAAAA;QACzFwB,UAAAA;kCAC0BxC,KAAAA,CAAAA,qBAAAA,EAA6ByB,MAAMZ,OAAO,CAAA,CAAA,CAAA;AACtE,MAAA,MAAM,IAAIa,mBAAUe,YAAAA,CAAAA;AACtB;AACF;AAEQP,EAAAA,eAAAA,CAAgBlC,KAAAA,EAAwB;AAC9C,IAAA,MAAM0C,eAAAA,GAAkB1C,KAAAA,CAAM2C,IAAAA,EAAI,CAAGC,WAAAA,EAAW;AAChD,IAAA,OACEF,eAAAA,CAAgBG,UAAAA,CAAW,QAAA,CAAA,IAC3BH,eAAAA,CAAgBG,WAAW,MAAA,CAAA,IAC3BH,eAAAA,CAAgBG,UAAAA,CAAW,MAAA,CAAA;AAE/B;AAEA,EAAA,MAAaC,OAAAA,GAAyB;AAEpC,IAAA,MAAMC,KAAAA,GAAQC,wBAAAA,CAAMC,WAAAA,CAAY,IAAA,EAAM,YAAA,CAAA;AACtC,IAAA,MAAMC,KAAAA,GAAQH,MAAMI,GAAAA,EAAG;AAEvB,IAAA,IAAID,KAAAA,EAAO;AACTH,MAAAA,KAAAA,CAAMK,KAAAA,EAAK;AAEX,MAAA,IAAI;AACF,QAAA,MAAMF,KAAAA,CAAMG,KAAKC,KAAAA,EAAK;AACxB,OAAA,CAAA,OAAS7B,KAAAA,EAAO;AACd,QAAA,MAAM,IAAIC,mBAAU,CAAA,uCAAA,CAAA,EAA2C;AAACD,UAAAA;AAAM,SAAA,CAAA;AACxE;AACF;AACF;AACF","file":"sql.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport {\n  Tool,\n  ToolInput,\n  ToolError,\n  BaseToolOptions,\n  BaseToolRunOptions,\n  JSONToolOutput,\n  ToolInputValidationError,\n  ToolEmitter,\n} from \"@/tools/base.js\";\nimport { z } from \"zod\";\nimport { Sequelize, Options } from \"sequelize\";\nimport { Provider, getMetadata } from \"@/tools/database/metadata.js\";\nimport { Cache } from \"@/cache/decoratorCache.js\";\nimport { ValidationError } from \"ajv\";\nimport { AnyToolSchemaLike } from \"@/internals/helpers/schema.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\n\ninterface ToolOptions extends BaseToolOptions {\n  provider: Provider;\n  connection: Options;\n}\n\ntype ToolRunOptions = BaseToolRunOptions;\n\nexport const SQLToolAction = {\n  GetMetadata: \"GET_METADATA\",\n  Query: \"QUERY\",\n} as const;\n\nexport class SQLToolOutput extends JSONToolOutput<any> {}\n\nexport class SQLTool extends Tool<SQLToolOutput, ToolOptions, ToolRunOptions> {\n  name = \"SQLTool\";\n\n  description = `Converts natural language to SQL query and executes it. IMPORTANT: strictly follow this order of actions:\n   1. ${SQLToolAction.GetMetadata} - get database tables structure (metadata)\n   2. ${SQLToolAction.Query} - execute the generated SQL query`;\n\n  inputSchema() {\n    return z.object({\n      action: z\n        .nativeEnum(SQLToolAction)\n        .describe(\n          `The action to perform. ${SQLToolAction.GetMetadata} get database tables structure, ${SQLToolAction.Query} execute the SQL query`,\n        ),\n      query: z\n        .string()\n        .optional()\n        .describe(`The SQL query to be executed, required for ${SQLToolAction.Query} action`),\n    });\n  }\n\n  public readonly emitter: ToolEmitter<ToolInput<this>, SQLToolOutput> = Emitter.root.child({\n    namespace: [\"tool\", \"database\", \"sql\"],\n    creator: this,\n  });\n\n  public constructor(options: ToolOptions) {\n    super(options);\n    if (!options.connection.dialect) {\n      throw new ValidationError([\n        {\n          message: \"Property is required\",\n          propertyName: \"connection.dialect\",\n        },\n      ]);\n    }\n    if (\n      !options.connection.schema &&\n      (options.provider === \"oracle\" || options.provider === \"db2\")\n    ) {\n      throw new ValidationError([\n        {\n          message: `Property is required for ${options.provider}`,\n          propertyName: \"connection.schema\",\n        },\n      ]);\n    }\n    if (!options.connection.storage && options.provider === \"sqlite\") {\n      throw new ValidationError([\n        {\n          message: `Property is required for ${options.provider}`,\n          propertyName: \"connection.storage\",\n        },\n      ]);\n    }\n  }\n\n  protected validateInput(\n    schema: AnyToolSchemaLike,\n    input: unknown,\n  ): asserts input is ToolInput<this> {\n    super.validateInput(schema, input);\n    if (input.action === SQLToolAction.Query && !input.query) {\n      throw new ToolInputValidationError(\n        `SQL Query is required for ${SQLToolAction.Query} action.`,\n      );\n    }\n  }\n\n  static {\n    this.register();\n  }\n\n  @Cache()\n  protected async connection(): Promise<Sequelize> {\n    try {\n      const sequelize = new Sequelize(this.options.connection);\n\n      await sequelize.authenticate();\n      return sequelize;\n    } catch (error) {\n      throw new ToolError(`Unable to connect to database: ${error}`, [], {\n        isRetryable: false,\n        isFatal: true,\n      });\n    }\n  }\n\n  protected async _run(\n    input: ToolInput<this>,\n    _options: Partial<ToolRunOptions>,\n  ): Promise<SQLToolOutput> {\n    const { provider, connection } = this.options;\n    const { schema } = connection;\n\n    if (input.action === SQLToolAction.GetMetadata) {\n      const sequelize = await this.connection();\n      const metadata = await getMetadata(sequelize, provider, schema);\n      return new SQLToolOutput(metadata);\n    }\n\n    if (input.action === SQLToolAction.Query) {\n      return await this.executeQuery(input.query!, provider, schema);\n    }\n\n    throw new ToolError(`Invalid action specified: ${input.action}`);\n  }\n\n  protected async executeQuery(\n    query: string,\n    provider: Provider,\n    schema: string | undefined,\n  ): Promise<SQLToolOutput> {\n    if (!this.isReadOnlyQuery(query)) {\n      return new JSONToolOutput({\n        success: false,\n        error: \"Invalid query. Only SELECT queries are allowed.\",\n      });\n    }\n\n    try {\n      const sequelize = await this.connection();\n      const [results] = await sequelize.query(query);\n      if (Array.isArray(results) && results.length > 0) {\n        return new JSONToolOutput({ success: true, results });\n      }\n\n      return new JSONToolOutput({\n        success: false,\n        message: `No rows selected`,\n      });\n    } catch (error) {\n      const schemaHint = schema\n        ? `Fully qualify the table names by appending the schema name \"${schema}\" as a prefix, for example: ${schema}.table_name`\n        : \"\";\n      const errorMessage = `Generate a correct query that retrieves data using the appropriate ${provider} dialect.\n      ${schemaHint}\n      The original request was: ${query}, and the error was: ${error.message}.`;\n      throw new ToolError(errorMessage);\n    }\n  }\n\n  private isReadOnlyQuery(query: string): boolean {\n    const normalizedQuery = query.trim().toUpperCase();\n    return (\n      normalizedQuery.startsWith(\"SELECT\") ||\n      normalizedQuery.startsWith(\"SHOW\") ||\n      normalizedQuery.startsWith(\"DESC\")\n    );\n  }\n\n  public async destroy(): Promise<void> {\n    // @ts-expect-error protected property\n    const cache = Cache.getInstance(this, \"connection\");\n    const entry = cache.get();\n\n    if (entry) {\n      cache.clear();\n\n      try {\n        await entry.data.close();\n      } catch (error) {\n        throw new ToolError(`Failed to close the database connection`, [error]);\n      }\n    }\n  }\n}\n"]}
+{"version":3,"sources":["../../../src/tools/database/sql.ts"],"names":["SQLToolAction","GetMetadata","Query","SQLToolOutput","JSONToolOutput","SQLTool","Tool","name","description","inputSchema","z","object","action","nativeEnum","describe","query","string","optional","emitter","Emitter","root","child","namespace","creator","options","connection","dialect","ValidationError","message","propertyName","schema","provider","storage","validateInput","input","ToolInputValidationError","register","sequelize","Sequelize","authenticate","error","ToolError","isRetryable","isFatal","_run","_options","metadata","getMetadata","executeQuery","isReadOnlyQuery","success","results","Array","isArray","length","schemaHint","errorMessage","normalizedQuery","trim","toUpperCase","startsWith","destroy","cache","Cache","getInstance","entry","get","clear","data","close"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;AA8BO,MAAMA,aAAAA,GAAgB;EAC3BC,WAAAA,EAAa,cAAA;EACbC,KAAAA,EAAO;AACT;AAEO,MAAMC,sBAAsBC,uBAAAA,CAAAA;AAAAA,EAAAA;;;AAAqB;AAEjD,MAAMC,gBAAgBC,aAAAA,CAAAA;AAAAA,EAAAA;;;EAC3BC,IAAAA,GAAO,SAAA;EAEPC,WAAAA,GAAc,CAAA;AACRR,MAAAA,EAAAA,aAAAA,CAAcC,WAAW,CAAA;AACzBD,MAAAA,EAAAA,aAAAA,CAAcE,KAAK,CAAA,kCAAA,CAAA;EAEzBO,WAAAA,GAAc;AACZ,IAAA,OAAOC,MAAEC,MAAAA,CAAO;MACdC,MAAAA,EAAQF,KAAAA,CACLG,UAAAA,CAAWb,aAAAA,CAAAA,CACXc,QAAAA,CACC,CAAA,uBAAA,EAA0Bd,aAAAA,CAAcC,WAAW,CAAA,gCAAA,EAAmCD,aAAAA,CAAcE,KAAK,CAAA,sBAAA,CAAwB,CAAA;MAErIa,KAAAA,EAAOL,KAAAA,CACJM,QAAM,CACNC,QAAAA,GACAH,QAAAA,CAAS,CAAA,2CAAA,EAA8Cd,aAAAA,CAAcE,KAAK,CAAA,OAAA,CAAS;KACxF,CAAA;AACF,EAAA;EAEgBgB,OAAAA,GAAuDC,mBAAAA,CAAQC,KAAKC,KAAAA,CAAM;IACxFC,SAAAA,EAAW;AAAC,MAAA,MAAA;AAAQ,MAAA,UAAA;AAAY,MAAA;;IAChCC,OAAAA,EAAS;GACX,CAAA;AAEA,EAAA,WAAA,CAAmBC,OAAAA,EAAsB;AACvC,IAAA,KAAA,CAAMA,OAAAA,CAAAA;AACN,IAAA,IAAI,CAACA,OAAAA,CAAQC,UAAAA,CAAWC,OAAAA,EAAS;AAC/B,MAAA,MAAM,IAAIC,mBAAAA,CAAgB;AACxB,QAAA;UACEC,OAAAA,EAAS,sBAAA;UACTC,YAAAA,EAAc;AAChB;AACD,OAAA,CAAA;AACH,IAAA;AACA,IAAA,IACE,CAACL,QAAQC,UAAAA,CAAWK,MAAAA,KACnBN,QAAQO,QAAAA,KAAa,QAAA,IAAYP,OAAAA,CAAQO,QAAAA,KAAa,KAAA,CAAA,EACvD;AACA,MAAA,MAAM,IAAIJ,mBAAAA,CAAgB;AACxB,QAAA;UACEC,OAAAA,EAAS,CAAA,yBAAA,EAA4BJ,QAAQO,QAAQ,CAAA,CAAA;UACrDF,YAAAA,EAAc;AAChB;AACD,OAAA,CAAA;AACH,IAAA;AACA,IAAA,IAAI,CAACL,OAAAA,CAAQC,UAAAA,CAAWO,OAAAA,IAAWR,OAAAA,CAAQO,aAAa,QAAA,EAAU;AAChE,MAAA,MAAM,IAAIJ,mBAAAA,CAAgB;AACxB,QAAA;UACEC,OAAAA,EAAS,CAAA,yBAAA,EAA4BJ,QAAQO,QAAQ,CAAA,CAAA;UACrDF,YAAAA,EAAc;AAChB;AACD,OAAA,CAAA;AACH,IAAA;AACF,EAAA;AAEUI,EAAAA,aAAAA,CACRH,QACAI,KAAAA,EACkC;AAClC,IAAA,KAAA,CAAMD,aAAAA,CAAcH,QAAQI,KAAAA,CAAAA;AAC5B,IAAA,IAAIA,MAAMtB,MAAAA,KAAWZ,aAAAA,CAAcE,KAAAA,IAAS,CAACgC,MAAMnB,KAAAA,EAAO;AACxD,MAAA,MAAM,IAAIoB,iCAAAA,CACR,CAAA,0BAAA,EAA6BnC,aAAAA,CAAcE,KAAK,CAAA,QAAA,CAAU,CAAA;AAE9D,IAAA;AACF,EAAA;EAEA;AACE,IAAA,IAAA,CAAKkC,QAAAA,EAAQ;AACf;AAEA,EAAA,MACgBX,UAAAA,GAAiC;AAC/C,IAAA,IAAI;AACF,MAAA,MAAMY,WAAAA,GAAY,IAAIC,mBAAAA,CAAU,IAAA,CAAKd,QAAQC,UAAU,CAAA;AAEvD,MAAA,MAAMY,YAAUE,YAAAA,EAAY;AAC5B,MAAA,OAAOF,WAAAA;AACT,IAAA,CAAA,CAAA,OAASG,KAAAA,EAAO;AACd,MAAA,MAAM,IAAIC,kBAAAA,CAAU,CAAA,+BAAA,EAAkCD,KAAAA,CAAAA,CAAAA,EAAS,EAAA,EAAI;QACjEE,WAAAA,EAAa,KAAA;QACbC,OAAAA,EAAS;OACX,CAAA;AACF,IAAA;AACF,EAAA;EAEA,MAAgBC,IAAAA,CACdV,OACAW,QAAAA,EACwB;AACxB,IAAA,MAAM,EAAEd,QAAAA,EAAUN,UAAAA,EAAU,GAAK,IAAA,CAAKD,OAAAA;AACtC,IAAA,MAAM,EAAEM,QAAM,GAAKL,UAAAA;AAEnB,IAAA,IAAIS,KAAAA,CAAMtB,MAAAA,KAAWZ,aAAAA,CAAcC,WAAAA,EAAa;AAC9C,MAAA,MAAMoC,SAAAA,GAAY,MAAM,IAAA,CAAKZ,UAAAA,EAAU;AACvC,MAAA,MAAMqB,QAAAA,GAAW,MAAMC,wBAAAA,CAAYV,SAAAA,EAAWN,UAAUD,MAAAA,CAAAA;AACxD,MAAA,OAAO,IAAI3B,cAAc2C,QAAAA,CAAAA;AAC3B,IAAA;AAEA,IAAA,IAAIZ,KAAAA,CAAMtB,MAAAA,KAAWZ,aAAAA,CAAcE,KAAAA,EAAO;AACxC,MAAA,OAAO,MAAM,IAAA,CAAK8C,YAAAA,CAAad,KAAAA,CAAMnB,KAAAA,EAAQgB,UAAUD,MAAAA,CAAAA;AACzD,IAAA;AAEA,IAAA,MAAM,IAAIW,kBAAAA,CAAU,CAAA,0BAAA,EAA6BP,KAAAA,CAAMtB,MAAM,CAAA,CAAE,CAAA;AACjE,EAAA;EAEA,MAAgBoC,YAAAA,CACdjC,KAAAA,EACAgB,QAAAA,EACAD,MAAAA,EACwB;AACxB,IAAA,IAAI,CAAC,IAAA,CAAKmB,eAAAA,CAAgBlC,KAAAA,CAAAA,EAAQ;AAChC,MAAA,OAAO,IAAIX,uBAAAA,CAAe;QACxB8C,OAAAA,EAAS,KAAA;QACTV,KAAAA,EAAO;OACT,CAAA;AACF,IAAA;AAEA,IAAA,IAAI;AACF,MAAA,MAAMH,SAAAA,GAAY,MAAM,IAAA,CAAKZ,UAAAA,EAAU;AACvC,MAAA,MAAM,CAAC0B,OAAAA,CAAAA,GAAW,MAAMd,SAAAA,CAAUtB,MAAMA,KAAAA,CAAAA;AACxC,MAAA,IAAIqC,MAAMC,OAAAA,CAAQF,OAAAA,CAAAA,IAAYA,OAAAA,CAAQG,SAAS,CAAA,EAAG;AAChD,QAAA,OAAO,IAAIlD,uBAAAA,CAAe;UAAE8C,OAAAA,EAAS,IAAA;AAAMC,UAAAA;SAAQ,CAAA;AACrD,MAAA;AAEA,MAAA,OAAO,IAAI/C,uBAAAA,CAAe;QACxB8C,OAAAA,EAAS,KAAA;QACTtB,OAAAA,EAAS,CAAA,gBAAA;OACX,CAAA;AACF,IAAA,CAAA,CAAA,OAASY,KAAAA,EAAO;AACd,MAAA,MAAMe,aAAazB,MAAAA,GACf,CAAA,4DAAA,EAA+DA,MAAAA,CAAAA,4BAAAA,EAAqCA,MAAAA,CAAAA,WAAAA,CAAAA,GACpG,EAAA;AACJ,MAAA,MAAM0B,YAAAA,GAAe,sEAAsEzB,QAAAA,CAAAA;QACzFwB,UAAAA;kCAC0BxC,KAAAA,CAAAA,qBAAAA,EAA6ByB,MAAMZ,OAAO,CAAA,CAAA,CAAA;AACtE,MAAA,MAAM,IAAIa,mBAAUe,YAAAA,CAAAA;AACtB,IAAA;AACF,EAAA;AAEQP,EAAAA,eAAAA,CAAgBlC,KAAAA,EAAwB;AAC9C,IAAA,MAAM0C,eAAAA,GAAkB1C,KAAAA,CAAM2C,IAAAA,EAAI,CAAGC,WAAAA,EAAW;AAChD,IAAA,OACEF,eAAAA,CAAgBG,UAAAA,CAAW,QAAA,CAAA,IAC3BH,eAAAA,CAAgBG,WAAW,MAAA,CAAA,IAC3BH,eAAAA,CAAgBG,UAAAA,CAAW,MAAA,CAAA;AAE/B,EAAA;AAEA,EAAA,MAAaC,OAAAA,GAAyB;AAEpC,IAAA,MAAMC,KAAAA,GAAQC,wBAAAA,CAAMC,WAAAA,CAAY,IAAA,EAAM,YAAA,CAAA;AACtC,IAAA,MAAMC,KAAAA,GAAQH,MAAMI,GAAAA,EAAG;AAEvB,IAAA,IAAID,KAAAA,EAAO;AACTH,MAAAA,KAAAA,CAAMK,KAAAA,EAAK;AAEX,MAAA,IAAI;AACF,QAAA,MAAMF,KAAAA,CAAMG,KAAKC,KAAAA,EAAK;AACxB,MAAA,CAAA,CAAA,OAAS7B,KAAAA,EAAO;AACd,QAAA,MAAM,IAAIC,mBAAU,CAAA,uCAAA,CAAA,EAA2C;AAACD,UAAAA;AAAM,SAAA,CAAA;AACxE,MAAA;AACF,IAAA;AACF,EAAA;AACF","file":"sql.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport {\n  Tool,\n  ToolInput,\n  ToolError,\n  BaseToolOptions,\n  BaseToolRunOptions,\n  JSONToolOutput,\n  ToolInputValidationError,\n  ToolEmitter,\n} from \"@/tools/base.js\";\nimport { z } from \"zod\";\nimport { Sequelize, Options } from \"sequelize\";\nimport { Provider, getMetadata } from \"@/tools/database/metadata.js\";\nimport { Cache } from \"@/cache/decoratorCache.js\";\nimport { ValidationError } from \"ajv\";\nimport { AnyToolSchemaLike } from \"@/internals/helpers/schema.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\n\ninterface ToolOptions extends BaseToolOptions {\n  provider: Provider;\n  connection: Options;\n}\n\ntype ToolRunOptions = BaseToolRunOptions;\n\nexport const SQLToolAction = {\n  GetMetadata: \"GET_METADATA\",\n  Query: \"QUERY\",\n} as const;\n\nexport class SQLToolOutput extends JSONToolOutput<any> {}\n\nexport class SQLTool extends Tool<SQLToolOutput, ToolOptions, ToolRunOptions> {\n  name = \"SQLTool\";\n\n  description = `Converts natural language to SQL query and executes it. IMPORTANT: strictly follow this order of actions:\n   1. ${SQLToolAction.GetMetadata} - get database tables structure (metadata)\n   2. ${SQLToolAction.Query} - execute the generated SQL query`;\n\n  inputSchema() {\n    return z.object({\n      action: z\n        .nativeEnum(SQLToolAction)\n        .describe(\n          `The action to perform. ${SQLToolAction.GetMetadata} get database tables structure, ${SQLToolAction.Query} execute the SQL query`,\n        ),\n      query: z\n        .string()\n        .optional()\n        .describe(`The SQL query to be executed, required for ${SQLToolAction.Query} action`),\n    });\n  }\n\n  public readonly emitter: ToolEmitter<ToolInput<this>, SQLToolOutput> = Emitter.root.child({\n    namespace: [\"tool\", \"database\", \"sql\"],\n    creator: this,\n  });\n\n  public constructor(options: ToolOptions) {\n    super(options);\n    if (!options.connection.dialect) {\n      throw new ValidationError([\n        {\n          message: \"Property is required\",\n          propertyName: \"connection.dialect\",\n        },\n      ]);\n    }\n    if (\n      !options.connection.schema &&\n      (options.provider === \"oracle\" || options.provider === \"db2\")\n    ) {\n      throw new ValidationError([\n        {\n          message: `Property is required for ${options.provider}`,\n          propertyName: \"connection.schema\",\n        },\n      ]);\n    }\n    if (!options.connection.storage && options.provider === \"sqlite\") {\n      throw new ValidationError([\n        {\n          message: `Property is required for ${options.provider}`,\n          propertyName: \"connection.storage\",\n        },\n      ]);\n    }\n  }\n\n  protected validateInput(\n    schema: AnyToolSchemaLike,\n    input: unknown,\n  ): asserts input is ToolInput<this> {\n    super.validateInput(schema, input);\n    if (input.action === SQLToolAction.Query && !input.query) {\n      throw new ToolInputValidationError(\n        `SQL Query is required for ${SQLToolAction.Query} action.`,\n      );\n    }\n  }\n\n  static {\n    this.register();\n  }\n\n  @Cache()\n  protected async connection(): Promise<Sequelize> {\n    try {\n      const sequelize = new Sequelize(this.options.connection);\n\n      await sequelize.authenticate();\n      return sequelize;\n    } catch (error) {\n      throw new ToolError(`Unable to connect to database: ${error}`, [], {\n        isRetryable: false,\n        isFatal: true,\n      });\n    }\n  }\n\n  protected async _run(\n    input: ToolInput<this>,\n    _options: Partial<ToolRunOptions>,\n  ): Promise<SQLToolOutput> {\n    const { provider, connection } = this.options;\n    const { schema } = connection;\n\n    if (input.action === SQLToolAction.GetMetadata) {\n      const sequelize = await this.connection();\n      const metadata = await getMetadata(sequelize, provider, schema);\n      return new SQLToolOutput(metadata);\n    }\n\n    if (input.action === SQLToolAction.Query) {\n      return await this.executeQuery(input.query!, provider, schema);\n    }\n\n    throw new ToolError(`Invalid action specified: ${input.action}`);\n  }\n\n  protected async executeQuery(\n    query: string,\n    provider: Provider,\n    schema: string | undefined,\n  ): Promise<SQLToolOutput> {\n    if (!this.isReadOnlyQuery(query)) {\n      return new JSONToolOutput({\n        success: false,\n        error: \"Invalid query. Only SELECT queries are allowed.\",\n      });\n    }\n\n    try {\n      const sequelize = await this.connection();\n      const [results] = await sequelize.query(query);\n      if (Array.isArray(results) && results.length > 0) {\n        return new JSONToolOutput({ success: true, results });\n      }\n\n      return new JSONToolOutput({\n        success: false,\n        message: `No rows selected`,\n      });\n    } catch (error) {\n      const schemaHint = schema\n        ? `Fully qualify the table names by appending the schema name \"${schema}\" as a prefix, for example: ${schema}.table_name`\n        : \"\";\n      const errorMessage = `Generate a correct query that retrieves data using the appropriate ${provider} dialect.\n      ${schemaHint}\n      The original request was: ${query}, and the error was: ${error.message}.`;\n      throw new ToolError(errorMessage);\n    }\n  }\n\n  private isReadOnlyQuery(query: string): boolean {\n    const normalizedQuery = query.trim().toUpperCase();\n    return (\n      normalizedQuery.startsWith(\"SELECT\") ||\n      normalizedQuery.startsWith(\"SHOW\") ||\n      normalizedQuery.startsWith(\"DESC\")\n    );\n  }\n\n  public async destroy(): Promise<void> {\n    // @ts-expect-error protected property\n    const cache = Cache.getInstance(this, \"connection\");\n    const entry = cache.get();\n\n    if (entry) {\n      cache.clear();\n\n      try {\n        await entry.data.close();\n      } catch (error) {\n        throw new ToolError(`Failed to close the database connection`, [error]);\n      }\n    }\n  }\n}\n"]}

package/dist/tools/database/sql.d.cts

@@ -1,11 +1,11 @@
-import { JSONToolOutput, Tool, BaseToolOptions, BaseToolRunOptions, ToolEmitter, ToolInput } from '../base.cjs';
+import { Tool, JSONToolOutput, BaseToolOptions, BaseToolRunOptions, ToolEmitter, ToolInput } from '../base.cjs';
 import { z } from 'zod';
 import { Options, Sequelize } from 'sequelize';
 import { Provider } from './metadata.cjs';
 import { AnyToolSchemaLike } from '../../internals/helpers/schema.cjs';
 import 'ajv';
 import '../../context.cjs';
-import '../../emitter-UkIkNrek.cjs';
+import '../../emitter-D4OcelZ9.cjs';
 import '../../internals/types.cjs';
 import '../../internals/helpers/guards.cjs';
 import '../../internals/serializable.cjs';

package/dist/tools/database/sql.d.ts

@@ -1,11 +1,11 @@
-import { JSONToolOutput, Tool, BaseToolOptions, BaseToolRunOptions, ToolEmitter, ToolInput } from '../base.js';
+import { Tool, JSONToolOutput, BaseToolOptions, BaseToolRunOptions, ToolEmitter, ToolInput } from '../base.js';
 import { z } from 'zod';
 import { Options, Sequelize } from 'sequelize';
 import { Provider } from './metadata.js';
 import { AnyToolSchemaLike } from '../../internals/helpers/schema.js';
 import 'ajv';
 import '../../context.js';
-import '../../emitter-CL56qab4.js';
+import '../../emitter-kHxkUxco.js';
 import '../../internals/types.js';
 import '../../internals/helpers/guards.js';
 import '../../internals/serializable.js';

package/dist/tools/database/sql.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../src/tools/database/sql.ts"],"names":["SQLToolAction","GetMetadata","Query","SQLToolOutput","JSONToolOutput","SQLTool","Tool","name","description","inputSchema","z","object","action","nativeEnum","describe","query","string","optional","emitter","Emitter","root","child","namespace","creator","options","connection","dialect","ValidationError","message","propertyName","schema","provider","storage","validateInput","input","ToolInputValidationError","register","sequelize","Sequelize","authenticate","error","ToolError","isRetryable","isFatal","_run","_options","metadata","getMetadata","executeQuery","isReadOnlyQuery","success","results","Array","isArray","length","schemaHint","errorMessage","normalizedQuery","trim","toUpperCase","startsWith","destroy","cache","Cache","getInstance","entry","get","clear","data","close"],"mappings":";;;;;;;;;;;;;;;;;;;;;AA8BO,MAAMA,aAAAA,GAAgB;EAC3BC,WAAAA,EAAa,cAAA;EACbC,KAAAA,EAAO;AACT;AAEO,MAAMC,sBAAsBC,cAAAA,CAAAA;AAAAA,EAAAA;;;AAAqB;AAEjD,MAAMC,gBAAgBC,IAAAA,CAAAA;AAAAA,EAAAA;;;EAC3BC,IAAAA,GAAO,SAAA;EAEPC,WAAAA,GAAc,CAAA;AACRR,MAAAA,EAAAA,aAAAA,CAAcC,WAAW,CAAA;AACzBD,MAAAA,EAAAA,aAAAA,CAAcE,KAAK,CAAA,kCAAA,CAAA;EAEzBO,WAAAA,GAAc;AACZ,IAAA,OAAOC,EAAEC,MAAAA,CAAO;MACdC,MAAAA,EAAQF,CAAAA,CACLG,UAAAA,CAAWb,aAAAA,CAAAA,CACXc,QAAAA,CACC,CAAA,uBAAA,EAA0Bd,aAAAA,CAAcC,WAAW,CAAA,gCAAA,EAAmCD,aAAAA,CAAcE,KAAK,CAAA,sBAAA,CAAwB,CAAA;MAErIa,KAAAA,EAAOL,CAAAA,CACJM,QAAM,CACNC,QAAAA,GACAH,QAAAA,CAAS,CAAA,2CAAA,EAA8Cd,aAAAA,CAAcE,KAAK,CAAA,OAAA,CAAS;KACxF,CAAA;AACF;EAEgBgB,OAAAA,GAAuDC,OAAAA,CAAQC,KAAKC,KAAAA,CAAM;IACxFC,SAAAA,EAAW;AAAC,MAAA,MAAA;AAAQ,MAAA,UAAA;AAAY,MAAA;;IAChCC,OAAAA,EAAS;GACX,CAAA;AAEA,EAAA,WAAA,CAAmBC,OAAAA,EAAsB;AACvC,IAAA,KAAA,CAAMA,OAAAA,CAAAA;AACN,IAAA,IAAI,CAACA,OAAAA,CAAQC,UAAAA,CAAWC,OAAAA,EAAS;AAC/B,MAAA,MAAM,IAAIC,eAAAA,CAAgB;AACxB,QAAA;UACEC,OAAAA,EAAS,sBAAA;UACTC,YAAAA,EAAc;AAChB;AACD,OAAA,CAAA;AACH;AACA,IAAA,IACE,CAACL,QAAQC,UAAAA,CAAWK,MAAAA,KACnBN,QAAQO,QAAAA,KAAa,QAAA,IAAYP,OAAAA,CAAQO,QAAAA,KAAa,KAAA,CAAA,EACvD;AACA,MAAA,MAAM,IAAIJ,eAAAA,CAAgB;AACxB,QAAA;UACEC,OAAAA,EAAS,CAAA,yBAAA,EAA4BJ,QAAQO,QAAQ,CAAA,CAAA;UACrDF,YAAAA,EAAc;AAChB;AACD,OAAA,CAAA;AACH;AACA,IAAA,IAAI,CAACL,OAAAA,CAAQC,UAAAA,CAAWO,OAAAA,IAAWR,OAAAA,CAAQO,aAAa,QAAA,EAAU;AAChE,MAAA,MAAM,IAAIJ,eAAAA,CAAgB;AACxB,QAAA;UACEC,OAAAA,EAAS,CAAA,yBAAA,EAA4BJ,QAAQO,QAAQ,CAAA,CAAA;UACrDF,YAAAA,EAAc;AAChB;AACD,OAAA,CAAA;AACH;AACF;AAEUI,EAAAA,aAAAA,CACRH,QACAI,KAAAA,EACkC;AAClC,IAAA,KAAA,CAAMD,aAAAA,CAAcH,QAAQI,KAAAA,CAAAA;AAC5B,IAAA,IAAIA,MAAMtB,MAAAA,KAAWZ,aAAAA,CAAcE,KAAAA,IAAS,CAACgC,MAAMnB,KAAAA,EAAO;AACxD,MAAA,MAAM,IAAIoB,wBAAAA,CACR,CAAA,0BAAA,EAA6BnC,aAAAA,CAAcE,KAAK,CAAA,QAAA,CAAU,CAAA;AAE9D;AACF;EAEA;AACE,IAAA,IAAA,CAAKkC,QAAAA,EAAQ;AACf;AAEA,EAAA,MACgBX,UAAAA,GAAiC;AAC/C,IAAA,IAAI;AACF,MAAA,MAAMY,SAAAA,GAAY,IAAIC,SAAAA,CAAU,IAAA,CAAKd,QAAQC,UAAU,CAAA;AAEvD,MAAA,MAAMY,UAAUE,YAAAA,EAAY;AAC5B,MAAA,OAAOF,SAAAA;AACT,KAAA,CAAA,OAASG,KAAAA,EAAO;AACd,MAAA,MAAM,IAAIC,SAAAA,CAAU,CAAA,+BAAA,EAAkCD,KAAAA,CAAAA,CAAAA,EAAS,EAAA,EAAI;QACjEE,WAAAA,EAAa,KAAA;QACbC,OAAAA,EAAS;OACX,CAAA;AACF;AACF;EAEA,MAAgBC,IAAAA,CACdV,OACAW,QAAAA,EACwB;AACxB,IAAA,MAAM,EAAEd,QAAAA,EAAUN,UAAAA,EAAU,GAAK,IAAA,CAAKD,OAAAA;AACtC,IAAA,MAAM,EAAEM,QAAM,GAAKL,UAAAA;AAEnB,IAAA,IAAIS,KAAAA,CAAMtB,MAAAA,KAAWZ,aAAAA,CAAcC,WAAAA,EAAa;AAC9C,MAAA,MAAMoC,SAAAA,GAAY,MAAM,IAAA,CAAKZ,UAAAA,EAAU;AACvC,MAAA,MAAMqB,QAAAA,GAAW,MAAMC,WAAAA,CAAYV,SAAAA,EAAWN,UAAUD,MAAAA,CAAAA;AACxD,MAAA,OAAO,IAAI3B,cAAc2C,QAAAA,CAAAA;AAC3B;AAEA,IAAA,IAAIZ,KAAAA,CAAMtB,MAAAA,KAAWZ,aAAAA,CAAcE,KAAAA,EAAO;AACxC,MAAA,OAAO,MAAM,IAAA,CAAK8C,YAAAA,CAAad,KAAAA,CAAMnB,KAAAA,EAAQgB,UAAUD,MAAAA,CAAAA;AACzD;AAEA,IAAA,MAAM,IAAIW,SAAAA,CAAU,CAAA,0BAAA,EAA6BP,KAAAA,CAAMtB,MAAM,CAAA,CAAE,CAAA;AACjE;EAEA,MAAgBoC,YAAAA,CACdjC,KAAAA,EACAgB,QAAAA,EACAD,MAAAA,EACwB;AACxB,IAAA,IAAI,CAAC,IAAA,CAAKmB,eAAAA,CAAgBlC,KAAAA,CAAAA,EAAQ;AAChC,MAAA,OAAO,IAAIX,cAAAA,CAAe;QACxB8C,OAAAA,EAAS,KAAA;QACTV,KAAAA,EAAO;OACT,CAAA;AACF;AAEA,IAAA,IAAI;AACF,MAAA,MAAMH,SAAAA,GAAY,MAAM,IAAA,CAAKZ,UAAAA,EAAU;AACvC,MAAA,MAAM,CAAC0B,OAAAA,CAAAA,GAAW,MAAMd,SAAAA,CAAUtB,MAAMA,KAAAA,CAAAA;AACxC,MAAA,IAAIqC,MAAMC,OAAAA,CAAQF,OAAAA,CAAAA,IAAYA,OAAAA,CAAQG,SAAS,CAAA,EAAG;AAChD,QAAA,OAAO,IAAIlD,cAAAA,CAAe;UAAE8C,OAAAA,EAAS,IAAA;AAAMC,UAAAA;SAAQ,CAAA;AACrD;AAEA,MAAA,OAAO,IAAI/C,cAAAA,CAAe;QACxB8C,OAAAA,EAAS,KAAA;QACTtB,OAAAA,EAAS,CAAA,gBAAA;OACX,CAAA;AACF,KAAA,CAAA,OAASY,KAAAA,EAAO;AACd,MAAA,MAAMe,aAAazB,MAAAA,GACf,CAAA,4DAAA,EAA+DA,MAAAA,CAAAA,4BAAAA,EAAqCA,MAAAA,CAAAA,WAAAA,CAAAA,GACpG,EAAA;AACJ,MAAA,MAAM0B,YAAAA,GAAe,sEAAsEzB,QAAAA,CAAAA;QACzFwB,UAAAA;kCAC0BxC,KAAAA,CAAAA,qBAAAA,EAA6ByB,MAAMZ,OAAO,CAAA,CAAA,CAAA;AACtE,MAAA,MAAM,IAAIa,UAAUe,YAAAA,CAAAA;AACtB;AACF;AAEQP,EAAAA,eAAAA,CAAgBlC,KAAAA,EAAwB;AAC9C,IAAA,MAAM0C,eAAAA,GAAkB1C,KAAAA,CAAM2C,IAAAA,EAAI,CAAGC,WAAAA,EAAW;AAChD,IAAA,OACEF,eAAAA,CAAgBG,UAAAA,CAAW,QAAA,CAAA,IAC3BH,eAAAA,CAAgBG,WAAW,MAAA,CAAA,IAC3BH,eAAAA,CAAgBG,UAAAA,CAAW,MAAA,CAAA;AAE/B;AAEA,EAAA,MAAaC,OAAAA,GAAyB;AAEpC,IAAA,MAAMC,KAAAA,GAAQC,KAAAA,CAAMC,WAAAA,CAAY,IAAA,EAAM,YAAA,CAAA;AACtC,IAAA,MAAMC,KAAAA,GAAQH,MAAMI,GAAAA,EAAG;AAEvB,IAAA,IAAID,KAAAA,EAAO;AACTH,MAAAA,KAAAA,CAAMK,KAAAA,EAAK;AAEX,MAAA,IAAI;AACF,QAAA,MAAMF,KAAAA,CAAMG,KAAKC,KAAAA,EAAK;AACxB,OAAA,CAAA,OAAS7B,KAAAA,EAAO;AACd,QAAA,MAAM,IAAIC,UAAU,CAAA,uCAAA,CAAA,EAA2C;AAACD,UAAAA;AAAM,SAAA,CAAA;AACxE;AACF;AACF;AACF","file":"sql.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport {\n  Tool,\n  ToolInput,\n  ToolError,\n  BaseToolOptions,\n  BaseToolRunOptions,\n  JSONToolOutput,\n  ToolInputValidationError,\n  ToolEmitter,\n} from \"@/tools/base.js\";\nimport { z } from \"zod\";\nimport { Sequelize, Options } from \"sequelize\";\nimport { Provider, getMetadata } from \"@/tools/database/metadata.js\";\nimport { Cache } from \"@/cache/decoratorCache.js\";\nimport { ValidationError } from \"ajv\";\nimport { AnyToolSchemaLike } from \"@/internals/helpers/schema.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\n\ninterface ToolOptions extends BaseToolOptions {\n  provider: Provider;\n  connection: Options;\n}\n\ntype ToolRunOptions = BaseToolRunOptions;\n\nexport const SQLToolAction = {\n  GetMetadata: \"GET_METADATA\",\n  Query: \"QUERY\",\n} as const;\n\nexport class SQLToolOutput extends JSONToolOutput<any> {}\n\nexport class SQLTool extends Tool<SQLToolOutput, ToolOptions, ToolRunOptions> {\n  name = \"SQLTool\";\n\n  description = `Converts natural language to SQL query and executes it. IMPORTANT: strictly follow this order of actions:\n   1. ${SQLToolAction.GetMetadata} - get database tables structure (metadata)\n   2. ${SQLToolAction.Query} - execute the generated SQL query`;\n\n  inputSchema() {\n    return z.object({\n      action: z\n        .nativeEnum(SQLToolAction)\n        .describe(\n          `The action to perform. ${SQLToolAction.GetMetadata} get database tables structure, ${SQLToolAction.Query} execute the SQL query`,\n        ),\n      query: z\n        .string()\n        .optional()\n        .describe(`The SQL query to be executed, required for ${SQLToolAction.Query} action`),\n    });\n  }\n\n  public readonly emitter: ToolEmitter<ToolInput<this>, SQLToolOutput> = Emitter.root.child({\n    namespace: [\"tool\", \"database\", \"sql\"],\n    creator: this,\n  });\n\n  public constructor(options: ToolOptions) {\n    super(options);\n    if (!options.connection.dialect) {\n      throw new ValidationError([\n        {\n          message: \"Property is required\",\n          propertyName: \"connection.dialect\",\n        },\n      ]);\n    }\n    if (\n      !options.connection.schema &&\n      (options.provider === \"oracle\" || options.provider === \"db2\")\n    ) {\n      throw new ValidationError([\n        {\n          message: `Property is required for ${options.provider}`,\n          propertyName: \"connection.schema\",\n        },\n      ]);\n    }\n    if (!options.connection.storage && options.provider === \"sqlite\") {\n      throw new ValidationError([\n        {\n          message: `Property is required for ${options.provider}`,\n          propertyName: \"connection.storage\",\n        },\n      ]);\n    }\n  }\n\n  protected validateInput(\n    schema: AnyToolSchemaLike,\n    input: unknown,\n  ): asserts input is ToolInput<this> {\n    super.validateInput(schema, input);\n    if (input.action === SQLToolAction.Query && !input.query) {\n      throw new ToolInputValidationError(\n        `SQL Query is required for ${SQLToolAction.Query} action.`,\n      );\n    }\n  }\n\n  static {\n    this.register();\n  }\n\n  @Cache()\n  protected async connection(): Promise<Sequelize> {\n    try {\n      const sequelize = new Sequelize(this.options.connection);\n\n      await sequelize.authenticate();\n      return sequelize;\n    } catch (error) {\n      throw new ToolError(`Unable to connect to database: ${error}`, [], {\n        isRetryable: false,\n        isFatal: true,\n      });\n    }\n  }\n\n  protected async _run(\n    input: ToolInput<this>,\n    _options: Partial<ToolRunOptions>,\n  ): Promise<SQLToolOutput> {\n    const { provider, connection } = this.options;\n    const { schema } = connection;\n\n    if (input.action === SQLToolAction.GetMetadata) {\n      const sequelize = await this.connection();\n      const metadata = await getMetadata(sequelize, provider, schema);\n      return new SQLToolOutput(metadata);\n    }\n\n    if (input.action === SQLToolAction.Query) {\n      return await this.executeQuery(input.query!, provider, schema);\n    }\n\n    throw new ToolError(`Invalid action specified: ${input.action}`);\n  }\n\n  protected async executeQuery(\n    query: string,\n    provider: Provider,\n    schema: string | undefined,\n  ): Promise<SQLToolOutput> {\n    if (!this.isReadOnlyQuery(query)) {\n      return new JSONToolOutput({\n        success: false,\n        error: \"Invalid query. Only SELECT queries are allowed.\",\n      });\n    }\n\n    try {\n      const sequelize = await this.connection();\n      const [results] = await sequelize.query(query);\n      if (Array.isArray(results) && results.length > 0) {\n        return new JSONToolOutput({ success: true, results });\n      }\n\n      return new JSONToolOutput({\n        success: false,\n        message: `No rows selected`,\n      });\n    } catch (error) {\n      const schemaHint = schema\n        ? `Fully qualify the table names by appending the schema name \"${schema}\" as a prefix, for example: ${schema}.table_name`\n        : \"\";\n      const errorMessage = `Generate a correct query that retrieves data using the appropriate ${provider} dialect.\n      ${schemaHint}\n      The original request was: ${query}, and the error was: ${error.message}.`;\n      throw new ToolError(errorMessage);\n    }\n  }\n\n  private isReadOnlyQuery(query: string): boolean {\n    const normalizedQuery = query.trim().toUpperCase();\n    return (\n      normalizedQuery.startsWith(\"SELECT\") ||\n      normalizedQuery.startsWith(\"SHOW\") ||\n      normalizedQuery.startsWith(\"DESC\")\n    );\n  }\n\n  public async destroy(): Promise<void> {\n    // @ts-expect-error protected property\n    const cache = Cache.getInstance(this, \"connection\");\n    const entry = cache.get();\n\n    if (entry) {\n      cache.clear();\n\n      try {\n        await entry.data.close();\n      } catch (error) {\n        throw new ToolError(`Failed to close the database connection`, [error]);\n      }\n    }\n  }\n}\n"]}
+{"version":3,"sources":["../../../src/tools/database/sql.ts"],"names":["SQLToolAction","GetMetadata","Query","SQLToolOutput","JSONToolOutput","SQLTool","Tool","name","description","inputSchema","z","object","action","nativeEnum","describe","query","string","optional","emitter","Emitter","root","child","namespace","creator","options","connection","dialect","ValidationError","message","propertyName","schema","provider","storage","validateInput","input","ToolInputValidationError","register","sequelize","Sequelize","authenticate","error","ToolError","isRetryable","isFatal","_run","_options","metadata","getMetadata","executeQuery","isReadOnlyQuery","success","results","Array","isArray","length","schemaHint","errorMessage","normalizedQuery","trim","toUpperCase","startsWith","destroy","cache","Cache","getInstance","entry","get","clear","data","close"],"mappings":";;;;;;;;;;;;;;;;;;;;;AA8BO,MAAMA,aAAAA,GAAgB;EAC3BC,WAAAA,EAAa,cAAA;EACbC,KAAAA,EAAO;AACT;AAEO,MAAMC,sBAAsBC,cAAAA,CAAAA;AAAAA,EAAAA;;;AAAqB;AAEjD,MAAMC,gBAAgBC,IAAAA,CAAAA;AAAAA,EAAAA;;;EAC3BC,IAAAA,GAAO,SAAA;EAEPC,WAAAA,GAAc,CAAA;AACRR,MAAAA,EAAAA,aAAAA,CAAcC,WAAW,CAAA;AACzBD,MAAAA,EAAAA,aAAAA,CAAcE,KAAK,CAAA,kCAAA,CAAA;EAEzBO,WAAAA,GAAc;AACZ,IAAA,OAAOC,EAAEC,MAAAA,CAAO;MACdC,MAAAA,EAAQF,CAAAA,CACLG,UAAAA,CAAWb,aAAAA,CAAAA,CACXc,QAAAA,CACC,CAAA,uBAAA,EAA0Bd,aAAAA,CAAcC,WAAW,CAAA,gCAAA,EAAmCD,aAAAA,CAAcE,KAAK,CAAA,sBAAA,CAAwB,CAAA;MAErIa,KAAAA,EAAOL,CAAAA,CACJM,QAAM,CACNC,QAAAA,GACAH,QAAAA,CAAS,CAAA,2CAAA,EAA8Cd,aAAAA,CAAcE,KAAK,CAAA,OAAA,CAAS;KACxF,CAAA;AACF,EAAA;EAEgBgB,OAAAA,GAAuDC,OAAAA,CAAQC,KAAKC,KAAAA,CAAM;IACxFC,SAAAA,EAAW;AAAC,MAAA,MAAA;AAAQ,MAAA,UAAA;AAAY,MAAA;;IAChCC,OAAAA,EAAS;GACX,CAAA;AAEA,EAAA,WAAA,CAAmBC,OAAAA,EAAsB;AACvC,IAAA,KAAA,CAAMA,OAAAA,CAAAA;AACN,IAAA,IAAI,CAACA,OAAAA,CAAQC,UAAAA,CAAWC,OAAAA,EAAS;AAC/B,MAAA,MAAM,IAAIC,eAAAA,CAAgB;AACxB,QAAA;UACEC,OAAAA,EAAS,sBAAA;UACTC,YAAAA,EAAc;AAChB;AACD,OAAA,CAAA;AACH,IAAA;AACA,IAAA,IACE,CAACL,QAAQC,UAAAA,CAAWK,MAAAA,KACnBN,QAAQO,QAAAA,KAAa,QAAA,IAAYP,OAAAA,CAAQO,QAAAA,KAAa,KAAA,CAAA,EACvD;AACA,MAAA,MAAM,IAAIJ,eAAAA,CAAgB;AACxB,QAAA;UACEC,OAAAA,EAAS,CAAA,yBAAA,EAA4BJ,QAAQO,QAAQ,CAAA,CAAA;UACrDF,YAAAA,EAAc;AAChB;AACD,OAAA,CAAA;AACH,IAAA;AACA,IAAA,IAAI,CAACL,OAAAA,CAAQC,UAAAA,CAAWO,OAAAA,IAAWR,OAAAA,CAAQO,aAAa,QAAA,EAAU;AAChE,MAAA,MAAM,IAAIJ,eAAAA,CAAgB;AACxB,QAAA;UACEC,OAAAA,EAAS,CAAA,yBAAA,EAA4BJ,QAAQO,QAAQ,CAAA,CAAA;UACrDF,YAAAA,EAAc;AAChB;AACD,OAAA,CAAA;AACH,IAAA;AACF,EAAA;AAEUI,EAAAA,aAAAA,CACRH,QACAI,KAAAA,EACkC;AAClC,IAAA,KAAA,CAAMD,aAAAA,CAAcH,QAAQI,KAAAA,CAAAA;AAC5B,IAAA,IAAIA,MAAMtB,MAAAA,KAAWZ,aAAAA,CAAcE,KAAAA,IAAS,CAACgC,MAAMnB,KAAAA,EAAO;AACxD,MAAA,MAAM,IAAIoB,wBAAAA,CACR,CAAA,0BAAA,EAA6BnC,aAAAA,CAAcE,KAAK,CAAA,QAAA,CAAU,CAAA;AAE9D,IAAA;AACF,EAAA;EAEA;AACE,IAAA,IAAA,CAAKkC,QAAAA,EAAQ;AACf;AAEA,EAAA,MACgBX,UAAAA,GAAiC;AAC/C,IAAA,IAAI;AACF,MAAA,MAAMY,SAAAA,GAAY,IAAIC,SAAAA,CAAU,IAAA,CAAKd,QAAQC,UAAU,CAAA;AAEvD,MAAA,MAAMY,UAAUE,YAAAA,EAAY;AAC5B,MAAA,OAAOF,SAAAA;AACT,IAAA,CAAA,CAAA,OAASG,KAAAA,EAAO;AACd,MAAA,MAAM,IAAIC,SAAAA,CAAU,CAAA,+BAAA,EAAkCD,KAAAA,CAAAA,CAAAA,EAAS,EAAA,EAAI;QACjEE,WAAAA,EAAa,KAAA;QACbC,OAAAA,EAAS;OACX,CAAA;AACF,IAAA;AACF,EAAA;EAEA,MAAgBC,IAAAA,CACdV,OACAW,QAAAA,EACwB;AACxB,IAAA,MAAM,EAAEd,QAAAA,EAAUN,UAAAA,EAAU,GAAK,IAAA,CAAKD,OAAAA;AACtC,IAAA,MAAM,EAAEM,QAAM,GAAKL,UAAAA;AAEnB,IAAA,IAAIS,KAAAA,CAAMtB,MAAAA,KAAWZ,aAAAA,CAAcC,WAAAA,EAAa;AAC9C,MAAA,MAAMoC,SAAAA,GAAY,MAAM,IAAA,CAAKZ,UAAAA,EAAU;AACvC,MAAA,MAAMqB,QAAAA,GAAW,MAAMC,WAAAA,CAAYV,SAAAA,EAAWN,UAAUD,MAAAA,CAAAA;AACxD,MAAA,OAAO,IAAI3B,cAAc2C,QAAAA,CAAAA;AAC3B,IAAA;AAEA,IAAA,IAAIZ,KAAAA,CAAMtB,MAAAA,KAAWZ,aAAAA,CAAcE,KAAAA,EAAO;AACxC,MAAA,OAAO,MAAM,IAAA,CAAK8C,YAAAA,CAAad,KAAAA,CAAMnB,KAAAA,EAAQgB,UAAUD,MAAAA,CAAAA;AACzD,IAAA;AAEA,IAAA,MAAM,IAAIW,SAAAA,CAAU,CAAA,0BAAA,EAA6BP,KAAAA,CAAMtB,MAAM,CAAA,CAAE,CAAA;AACjE,EAAA;EAEA,MAAgBoC,YAAAA,CACdjC,KAAAA,EACAgB,QAAAA,EACAD,MAAAA,EACwB;AACxB,IAAA,IAAI,CAAC,IAAA,CAAKmB,eAAAA,CAAgBlC,KAAAA,CAAAA,EAAQ;AAChC,MAAA,OAAO,IAAIX,cAAAA,CAAe;QACxB8C,OAAAA,EAAS,KAAA;QACTV,KAAAA,EAAO;OACT,CAAA;AACF,IAAA;AAEA,IAAA,IAAI;AACF,MAAA,MAAMH,SAAAA,GAAY,MAAM,IAAA,CAAKZ,UAAAA,EAAU;AACvC,MAAA,MAAM,CAAC0B,OAAAA,CAAAA,GAAW,MAAMd,SAAAA,CAAUtB,MAAMA,KAAAA,CAAAA;AACxC,MAAA,IAAIqC,MAAMC,OAAAA,CAAQF,OAAAA,CAAAA,IAAYA,OAAAA,CAAQG,SAAS,CAAA,EAAG;AAChD,QAAA,OAAO,IAAIlD,cAAAA,CAAe;UAAE8C,OAAAA,EAAS,IAAA;AAAMC,UAAAA;SAAQ,CAAA;AACrD,MAAA;AAEA,MAAA,OAAO,IAAI/C,cAAAA,CAAe;QACxB8C,OAAAA,EAAS,KAAA;QACTtB,OAAAA,EAAS,CAAA,gBAAA;OACX,CAAA;AACF,IAAA,CAAA,CAAA,OAASY,KAAAA,EAAO;AACd,MAAA,MAAMe,aAAazB,MAAAA,GACf,CAAA,4DAAA,EAA+DA,MAAAA,CAAAA,4BAAAA,EAAqCA,MAAAA,CAAAA,WAAAA,CAAAA,GACpG,EAAA;AACJ,MAAA,MAAM0B,YAAAA,GAAe,sEAAsEzB,QAAAA,CAAAA;QACzFwB,UAAAA;kCAC0BxC,KAAAA,CAAAA,qBAAAA,EAA6ByB,MAAMZ,OAAO,CAAA,CAAA,CAAA;AACtE,MAAA,MAAM,IAAIa,UAAUe,YAAAA,CAAAA;AACtB,IAAA;AACF,EAAA;AAEQP,EAAAA,eAAAA,CAAgBlC,KAAAA,EAAwB;AAC9C,IAAA,MAAM0C,eAAAA,GAAkB1C,KAAAA,CAAM2C,IAAAA,EAAI,CAAGC,WAAAA,EAAW;AAChD,IAAA,OACEF,eAAAA,CAAgBG,UAAAA,CAAW,QAAA,CAAA,IAC3BH,eAAAA,CAAgBG,WAAW,MAAA,CAAA,IAC3BH,eAAAA,CAAgBG,UAAAA,CAAW,MAAA,CAAA;AAE/B,EAAA;AAEA,EAAA,MAAaC,OAAAA,GAAyB;AAEpC,IAAA,MAAMC,KAAAA,GAAQC,KAAAA,CAAMC,WAAAA,CAAY,IAAA,EAAM,YAAA,CAAA;AACtC,IAAA,MAAMC,KAAAA,GAAQH,MAAMI,GAAAA,EAAG;AAEvB,IAAA,IAAID,KAAAA,EAAO;AACTH,MAAAA,KAAAA,CAAMK,KAAAA,EAAK;AAEX,MAAA,IAAI;AACF,QAAA,MAAMF,KAAAA,CAAMG,KAAKC,KAAAA,EAAK;AACxB,MAAA,CAAA,CAAA,OAAS7B,KAAAA,EAAO;AACd,QAAA,MAAM,IAAIC,UAAU,CAAA,uCAAA,CAAA,EAA2C;AAACD,UAAAA;AAAM,SAAA,CAAA;AACxE,MAAA;AACF,IAAA;AACF,EAAA;AACF","file":"sql.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport {\n  Tool,\n  ToolInput,\n  ToolError,\n  BaseToolOptions,\n  BaseToolRunOptions,\n  JSONToolOutput,\n  ToolInputValidationError,\n  ToolEmitter,\n} from \"@/tools/base.js\";\nimport { z } from \"zod\";\nimport { Sequelize, Options } from \"sequelize\";\nimport { Provider, getMetadata } from \"@/tools/database/metadata.js\";\nimport { Cache } from \"@/cache/decoratorCache.js\";\nimport { ValidationError } from \"ajv\";\nimport { AnyToolSchemaLike } from \"@/internals/helpers/schema.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\n\ninterface ToolOptions extends BaseToolOptions {\n  provider: Provider;\n  connection: Options;\n}\n\ntype ToolRunOptions = BaseToolRunOptions;\n\nexport const SQLToolAction = {\n  GetMetadata: \"GET_METADATA\",\n  Query: \"QUERY\",\n} as const;\n\nexport class SQLToolOutput extends JSONToolOutput<any> {}\n\nexport class SQLTool extends Tool<SQLToolOutput, ToolOptions, ToolRunOptions> {\n  name = \"SQLTool\";\n\n  description = `Converts natural language to SQL query and executes it. IMPORTANT: strictly follow this order of actions:\n   1. ${SQLToolAction.GetMetadata} - get database tables structure (metadata)\n   2. ${SQLToolAction.Query} - execute the generated SQL query`;\n\n  inputSchema() {\n    return z.object({\n      action: z\n        .nativeEnum(SQLToolAction)\n        .describe(\n          `The action to perform. ${SQLToolAction.GetMetadata} get database tables structure, ${SQLToolAction.Query} execute the SQL query`,\n        ),\n      query: z\n        .string()\n        .optional()\n        .describe(`The SQL query to be executed, required for ${SQLToolAction.Query} action`),\n    });\n  }\n\n  public readonly emitter: ToolEmitter<ToolInput<this>, SQLToolOutput> = Emitter.root.child({\n    namespace: [\"tool\", \"database\", \"sql\"],\n    creator: this,\n  });\n\n  public constructor(options: ToolOptions) {\n    super(options);\n    if (!options.connection.dialect) {\n      throw new ValidationError([\n        {\n          message: \"Property is required\",\n          propertyName: \"connection.dialect\",\n        },\n      ]);\n    }\n    if (\n      !options.connection.schema &&\n      (options.provider === \"oracle\" || options.provider === \"db2\")\n    ) {\n      throw new ValidationError([\n        {\n          message: `Property is required for ${options.provider}`,\n          propertyName: \"connection.schema\",\n        },\n      ]);\n    }\n    if (!options.connection.storage && options.provider === \"sqlite\") {\n      throw new ValidationError([\n        {\n          message: `Property is required for ${options.provider}`,\n          propertyName: \"connection.storage\",\n        },\n      ]);\n    }\n  }\n\n  protected validateInput(\n    schema: AnyToolSchemaLike,\n    input: unknown,\n  ): asserts input is ToolInput<this> {\n    super.validateInput(schema, input);\n    if (input.action === SQLToolAction.Query && !input.query) {\n      throw new ToolInputValidationError(\n        `SQL Query is required for ${SQLToolAction.Query} action.`,\n      );\n    }\n  }\n\n  static {\n    this.register();\n  }\n\n  @Cache()\n  protected async connection(): Promise<Sequelize> {\n    try {\n      const sequelize = new Sequelize(this.options.connection);\n\n      await sequelize.authenticate();\n      return sequelize;\n    } catch (error) {\n      throw new ToolError(`Unable to connect to database: ${error}`, [], {\n        isRetryable: false,\n        isFatal: true,\n      });\n    }\n  }\n\n  protected async _run(\n    input: ToolInput<this>,\n    _options: Partial<ToolRunOptions>,\n  ): Promise<SQLToolOutput> {\n    const { provider, connection } = this.options;\n    const { schema } = connection;\n\n    if (input.action === SQLToolAction.GetMetadata) {\n      const sequelize = await this.connection();\n      const metadata = await getMetadata(sequelize, provider, schema);\n      return new SQLToolOutput(metadata);\n    }\n\n    if (input.action === SQLToolAction.Query) {\n      return await this.executeQuery(input.query!, provider, schema);\n    }\n\n    throw new ToolError(`Invalid action specified: ${input.action}`);\n  }\n\n  protected async executeQuery(\n    query: string,\n    provider: Provider,\n    schema: string | undefined,\n  ): Promise<SQLToolOutput> {\n    if (!this.isReadOnlyQuery(query)) {\n      return new JSONToolOutput({\n        success: false,\n        error: \"Invalid query. Only SELECT queries are allowed.\",\n      });\n    }\n\n    try {\n      const sequelize = await this.connection();\n      const [results] = await sequelize.query(query);\n      if (Array.isArray(results) && results.length > 0) {\n        return new JSONToolOutput({ success: true, results });\n      }\n\n      return new JSONToolOutput({\n        success: false,\n        message: `No rows selected`,\n      });\n    } catch (error) {\n      const schemaHint = schema\n        ? `Fully qualify the table names by appending the schema name \"${schema}\" as a prefix, for example: ${schema}.table_name`\n        : \"\";\n      const errorMessage = `Generate a correct query that retrieves data using the appropriate ${provider} dialect.\n      ${schemaHint}\n      The original request was: ${query}, and the error was: ${error.message}.`;\n      throw new ToolError(errorMessage);\n    }\n  }\n\n  private isReadOnlyQuery(query: string): boolean {\n    const normalizedQuery = query.trim().toUpperCase();\n    return (\n      normalizedQuery.startsWith(\"SELECT\") ||\n      normalizedQuery.startsWith(\"SHOW\") ||\n      normalizedQuery.startsWith(\"DESC\")\n    );\n  }\n\n  public async destroy(): Promise<void> {\n    // @ts-expect-error protected property\n    const cache = Cache.getInstance(this, \"connection\");\n    const entry = cache.get();\n\n    if (entry) {\n      cache.clear();\n\n      try {\n        await entry.data.close();\n      } catch (error) {\n        throw new ToolError(`Failed to close the database connection`, [error]);\n      }\n    }\n  }\n}\n"]}

package/dist/tools/handoff.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/tools/handoff.ts"],"names":["HandoffTool","Tool","propagateInputs","name","description","emitter","Emitter","root","child","namespace","creator","inputSchema","z","object","task","string","describe","target","options","toCamelCase","meta","_run","input","run","memory","getProp","context","contextKeys","Memory","ToolError","isRetryable","isFatal","messages","filter","msg","SystemMessage","lastValidMsgIndex","findLastIndex","AssistantMessage","getToolCalls","length","slice","push","UserMessage","clone","reset","addMany","response","prompt","StringToolOutput","result","text"],"mappings":";;;;;;;;;;;AA4BO,MAAMA,oBAAoBC,aAAAA,CAAAA;EA5BjC;;;;AA6BmBC,EAAAA,eAAAA;AACVC,EAAAA,IAAAA;AACAC,EAAAA,WAAAA;EAESC,OAAAA,GAA0DC,mBAAAA,CAAQC,KAAKC,KAAAA,CAAM;IAC3FC,SAAAA,EAAW;AAAC,MAAA,MAAA;AAAQ,MAAA;;IACpBC,OAAAA,EAAS;GACX,CAAA;EAEAC,WAAAA,GAAc;AACZ,IAAA,OAAOC,MAAEC,MAAAA,CAAO;AACdC,MAAAA,IAAAA,EAAMF,KAAAA,CACHG,MAAAA,EAAM,CACNC,QAAAA,CAAS,uEAAA;KACd,CAAA;AACF;AAEA,EAAA,WAAA,CACUC,QACRC,OAAAA,EACA;AACA,IAAA,KAAA,EAAK,EAAA,KAHGD,MAAAA,GAAAA,MAAAA;AAIR,IAAA,IAAA,CAAKd,OAAOgB,kBAAAA,CAAYD,OAAAA,EAASf,QAAQ,IAAA,CAAKc,MAAAA,CAAOG,KAAKjB,IAAI,CAAA;AAC9D,IAAA,IAAA,CAAKC,WAAAA,GAAcc,OAAAA,EAASd,WAAAA,IAAe,IAAA,CAAKa,OAAOG,IAAAA,CAAKhB,WAAAA;AAC5D,IAAA,IAAA,CAAKF,eAAAA,GAAkBgB,SAAShB,eAAAA,IAAmB,IAAA;AACrD;EAEA,MAAgBmB,IAAAA,CACdC,KAAAA,EACAJ,OAAAA,EACAK,GAAAA,EAC2B;AAC3B,IAAA,MAAMC,MAAAA,GAASC,kBAAAA,CAAQF,GAAAA,CAAIG,OAAAA,EAAS;AAACzB,MAAAA,aAAAA,CAAK0B,WAAAA,CAAYC;AAAO,KAAA,CAAA;AAE7D,IAAA,IAAI,CAACJ,MAAAA,EAAQ;AACX,MAAA,MAAM,IAAIK,kBAAAA,CAAU,CAAA,+BAAA,CAAA,EAAmC,EAAA,EAAI;QACzDC,WAAAA,EAAa,KAAA;QACbC,OAAAA,EAAS;OACX,CAAA;AACF;AAEA,IAAA,IAAIC,QAAAA,GAAsBR,OAAOQ,QAAAA,CAASC,MAAAA,CAAO,CAACC,GAAAA,KAAQ,EAAEA,eAAeC,yBAAAA,CAAY,CAAA;AAEvF,IAAA,MAAMC,iBAAAA,GAAoBC,oBAAAA,CACxBL,QAAAA,EACA,CAACE,GAAAA,KAAQA,GAAAA,YAAeI,4BAAAA,IAAAA,CAAqBJ,GAAAA,CAAIK,YAAAA,EAAY,EAAIC,MAAAA,IAAU,CAAA,IAAK,CAAA,CAAA;AAGlF,IAAA,IAAIJ,sBAAsB,EAAA,EAAI;AAC5BJ,MAAAA,QAAAA,GAAWA,QAAAA,CAASS,KAAAA,CAAM,CAAA,EAAGL,iBAAAA,CAAAA;AAC/B;AAEA,IAAA,IAAI,KAAKlC,eAAAA,EAAiB;AACxB8B,MAAAA,QAAAA,CAASU,IAAAA,CAAK,IAAIC,uBAAAA,CAAYrB,KAAAA,CAAMR,IAAI,CAAA,CAAA;AAC1C;AAEA,IAAA,MAAMG,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAAA,CAAO2B,KAAAA,EAAK;AACtC3B,IAAAA,MAAAA,CAAOO,OAAOqB,KAAAA,EAAK;AACnB,IAAA,MAAM5B,MAAAA,CAAOO,MAAAA,CAAOsB,OAAAA,CAAQd,QAAAA,CAAAA;AAE5B,IAAA,MAAMe,QAAAA,GAAW,MAAM9B,MAAAA,CAAOM,GAAAA,CAAI;MAAEyB,MAAAA,EAAQ;KAAK,CAAA;AAEjD,IAAA,OAAO,IAAIC,yBAAAA,CAAiBF,QAAAA,CAASG,MAAAA,CAAOC,IAAI,CAAA;AAClD;AACF","file":"handoff.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { z } from \"zod\";\nimport {\n  BaseToolRunOptions,\n  StringToolOutput,\n  Tool,\n  ToolEmitter,\n  ToolError,\n  ToolInput,\n} from \"./base.js\";\nimport { BaseMemory } from \"@/memory/base.js\";\nimport { AssistantMessage, Message, SystemMessage, UserMessage } from \"@/backend/message.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { AnyAgent } from \"@/agents/types.js\";\nimport { GetRunContext } from \"@/context.js\";\nimport { getProp } from \"@/internals/helpers/object.js\";\nimport { findLastIndex, toCamelCase } from \"remeda\";\n\nexport interface HandoffToolInput {\n  name?: string;\n  description?: string;\n  propagateInputs?: boolean;\n}\n\nexport class HandoffTool extends Tool<StringToolOutput> {\n  private readonly propagateInputs: boolean;\n  public name: string;\n  public description: string;\n\n  public readonly emitter: ToolEmitter<ToolInput<this>, StringToolOutput> = Emitter.root.child({\n    namespace: [\"tool\", \"handoff\"],\n    creator: this,\n  });\n\n  inputSchema() {\n    return z.object({\n      task: z\n        .string()\n        .describe(\"Clearly defined task for the agent to work on based on his abilities.\"),\n    });\n  }\n\n  constructor(\n    private target: AnyAgent,\n    options?: HandoffToolInput,\n  ) {\n    super();\n    this.name = toCamelCase(options?.name || this.target.meta.name);\n    this.description = options?.description || this.target.meta.description;\n    this.propagateInputs = options?.propagateInputs ?? true;\n  }\n\n  protected async _run(\n    input: ToolInput<this>,\n    options: Partial<BaseToolRunOptions>,\n    run: GetRunContext<typeof this>,\n  ): Promise<StringToolOutput> {\n    const memory = getProp(run.context, [Tool.contextKeys.Memory]) as BaseMemory;\n\n    if (!memory) {\n      throw new ToolError(`No memory found in the context.`, [], {\n        isRetryable: false,\n        isFatal: true,\n      });\n    }\n\n    let messages: Message[] = memory.messages.filter((msg) => !(msg instanceof SystemMessage));\n\n    const lastValidMsgIndex = findLastIndex(\n      messages,\n      (msg) => msg instanceof AssistantMessage && (msg.getToolCalls()?.length ?? 0) > 0,\n    );\n\n    if (lastValidMsgIndex !== -1) {\n      messages = messages.slice(0, lastValidMsgIndex);\n    }\n\n    if (this.propagateInputs) {\n      messages.push(new UserMessage(input.task));\n    }\n\n    const target = await this.target.clone();\n    target.memory.reset();\n    await target.memory.addMany(messages);\n\n    const response = await target.run({ prompt: null });\n\n    return new StringToolOutput(response.result.text);\n  }\n}\n"]}
+{"version":3,"sources":["../../src/tools/handoff.ts"],"names":["HandoffTool","Tool","propagateInputs","name","description","emitter","Emitter","root","child","namespace","creator","inputSchema","z","object","task","string","describe","target","options","toCamelCase","meta","_run","input","run","memory","getProp","context","contextKeys","Memory","ToolError","isRetryable","isFatal","messages","filter","msg","SystemMessage","lastValidMsgIndex","findLastIndex","AssistantMessage","getToolCalls","length","slice","push","UserMessage","clone","reset","addMany","response","prompt","StringToolOutput","result","text"],"mappings":";;;;;;;;;;;AA4BO,MAAMA,oBAAoBC,aAAAA,CAAAA;EA5BjC;;;;AA6BmBC,EAAAA,eAAAA;AACVC,EAAAA,IAAAA;AACAC,EAAAA,WAAAA;EAESC,OAAAA,GAA0DC,mBAAAA,CAAQC,KAAKC,KAAAA,CAAM;IAC3FC,SAAAA,EAAW;AAAC,MAAA,MAAA;AAAQ,MAAA;;IACpBC,OAAAA,EAAS;GACX,CAAA;EAEAC,WAAAA,GAAc;AACZ,IAAA,OAAOC,MAAEC,MAAAA,CAAO;AACdC,MAAAA,IAAAA,EAAMF,KAAAA,CACHG,MAAAA,EAAM,CACNC,QAAAA,CAAS,uEAAA;KACd,CAAA;AACF,EAAA;AAEA,EAAA,WAAA,CACUC,QACRC,OAAAA,EACA;AACA,IAAA,KAAA,EAAK,EAAA,KAHGD,MAAAA,GAAAA,MAAAA;AAIR,IAAA,IAAA,CAAKd,OAAOgB,kBAAAA,CAAYD,OAAAA,EAASf,QAAQ,IAAA,CAAKc,MAAAA,CAAOG,KAAKjB,IAAI,CAAA;AAC9D,IAAA,IAAA,CAAKC,WAAAA,GAAcc,OAAAA,EAASd,WAAAA,IAAe,IAAA,CAAKa,OAAOG,IAAAA,CAAKhB,WAAAA;AAC5D,IAAA,IAAA,CAAKF,eAAAA,GAAkBgB,SAAShB,eAAAA,IAAmB,IAAA;AACrD,EAAA;EAEA,MAAgBmB,IAAAA,CACdC,KAAAA,EACAJ,OAAAA,EACAK,GAAAA,EAC2B;AAC3B,IAAA,MAAMC,MAAAA,GAASC,kBAAAA,CAAQF,GAAAA,CAAIG,OAAAA,EAAS;AAACzB,MAAAA,aAAAA,CAAK0B,WAAAA,CAAYC;AAAO,KAAA,CAAA;AAE7D,IAAA,IAAI,CAACJ,MAAAA,EAAQ;AACX,MAAA,MAAM,IAAIK,kBAAAA,CAAU,CAAA,+BAAA,CAAA,EAAmC,EAAA,EAAI;QACzDC,WAAAA,EAAa,KAAA;QACbC,OAAAA,EAAS;OACX,CAAA;AACF,IAAA;AAEA,IAAA,IAAIC,QAAAA,GAAsBR,OAAOQ,QAAAA,CAASC,MAAAA,CAAO,CAACC,GAAAA,KAAQ,EAAEA,eAAeC,yBAAAA,CAAY,CAAA;AAEvF,IAAA,MAAMC,iBAAAA,GAAoBC,oBAAAA,CACxBL,QAAAA,EACA,CAACE,GAAAA,KAAQA,GAAAA,YAAeI,4BAAAA,IAAAA,CAAqBJ,GAAAA,CAAIK,YAAAA,EAAY,EAAIC,MAAAA,IAAU,CAAA,IAAK,CAAA,CAAA;AAGlF,IAAA,IAAIJ,sBAAsB,EAAA,EAAI;AAC5BJ,MAAAA,QAAAA,GAAWA,QAAAA,CAASS,KAAAA,CAAM,CAAA,EAAGL,iBAAAA,CAAAA;AAC/B,IAAA;AAEA,IAAA,IAAI,KAAKlC,eAAAA,EAAiB;AACxB8B,MAAAA,QAAAA,CAASU,IAAAA,CAAK,IAAIC,uBAAAA,CAAYrB,KAAAA,CAAMR,IAAI,CAAA,CAAA;AAC1C,IAAA;AAEA,IAAA,MAAMG,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAAA,CAAO2B,KAAAA,EAAK;AACtC3B,IAAAA,MAAAA,CAAOO,OAAOqB,KAAAA,EAAK;AACnB,IAAA,MAAM5B,MAAAA,CAAOO,MAAAA,CAAOsB,OAAAA,CAAQd,QAAAA,CAAAA;AAE5B,IAAA,MAAMe,QAAAA,GAAW,MAAM9B,MAAAA,CAAOM,GAAAA,CAAI;MAAEyB,MAAAA,EAAQ;KAAK,CAAA;AAEjD,IAAA,OAAO,IAAIC,yBAAAA,CAAiBF,QAAAA,CAASG,MAAAA,CAAOC,IAAI,CAAA;AAClD,EAAA;AACF","file":"handoff.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { z } from \"zod\";\nimport {\n  BaseToolRunOptions,\n  StringToolOutput,\n  Tool,\n  ToolEmitter,\n  ToolError,\n  ToolInput,\n} from \"./base.js\";\nimport { BaseMemory } from \"@/memory/base.js\";\nimport { AssistantMessage, Message, SystemMessage, UserMessage } from \"@/backend/message.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { AnyAgent } from \"@/agents/types.js\";\nimport { GetRunContext } from \"@/context.js\";\nimport { getProp } from \"@/internals/helpers/object.js\";\nimport { findLastIndex, toCamelCase } from \"remeda\";\n\nexport interface HandoffToolInput {\n  name?: string;\n  description?: string;\n  propagateInputs?: boolean;\n}\n\nexport class HandoffTool extends Tool<StringToolOutput> {\n  private readonly propagateInputs: boolean;\n  public name: string;\n  public description: string;\n\n  public readonly emitter: ToolEmitter<ToolInput<this>, StringToolOutput> = Emitter.root.child({\n    namespace: [\"tool\", \"handoff\"],\n    creator: this,\n  });\n\n  inputSchema() {\n    return z.object({\n      task: z\n        .string()\n        .describe(\"Clearly defined task for the agent to work on based on his abilities.\"),\n    });\n  }\n\n  constructor(\n    private target: AnyAgent,\n    options?: HandoffToolInput,\n  ) {\n    super();\n    this.name = toCamelCase(options?.name || this.target.meta.name);\n    this.description = options?.description || this.target.meta.description;\n    this.propagateInputs = options?.propagateInputs ?? true;\n  }\n\n  protected async _run(\n    input: ToolInput<this>,\n    options: Partial<BaseToolRunOptions>,\n    run: GetRunContext<typeof this>,\n  ): Promise<StringToolOutput> {\n    const memory = getProp(run.context, [Tool.contextKeys.Memory]) as BaseMemory;\n\n    if (!memory) {\n      throw new ToolError(`No memory found in the context.`, [], {\n        isRetryable: false,\n        isFatal: true,\n      });\n    }\n\n    let messages: Message[] = memory.messages.filter((msg) => !(msg instanceof SystemMessage));\n\n    const lastValidMsgIndex = findLastIndex(\n      messages,\n      (msg) => msg instanceof AssistantMessage && (msg.getToolCalls()?.length ?? 0) > 0,\n    );\n\n    if (lastValidMsgIndex !== -1) {\n      messages = messages.slice(0, lastValidMsgIndex);\n    }\n\n    if (this.propagateInputs) {\n      messages.push(new UserMessage(input.task));\n    }\n\n    const target = await this.target.clone();\n    target.memory.reset();\n    await target.memory.addMany(messages);\n\n    const response = await target.run({ prompt: null });\n\n    return new StringToolOutput(response.result.text);\n  }\n}\n"]}

package/dist/tools/handoff.d.cts

@@ -1,6 +1,6 @@
 import { z } from 'zod';
 import { Tool, StringToolOutput, ToolEmitter, ToolInput, BaseToolRunOptions } from './base.cjs';
-import { b as AnyAgent } from '../base-BqzN2QwB.cjs';
+import { b as AnyAgent } from '../base-s3ioVIE9.cjs';
 import { GetRunContext } from '../context.cjs';
 import 'ajv';
 import '../errors.cjs';
@@ -11,7 +11,7 @@ import 'promise-based-task';
 import '../cache/base.cjs';
 import '../internals/helpers/schema.cjs';
 import 'zod-to-json-schema';
-import '../emitter-UkIkNrek.cjs';
+import '../emitter-D4OcelZ9.cjs';
 import '../internals/helpers/promise.cjs';
 import '../memory/base.cjs';
 import '../backend/message.cjs';

package/dist/tools/handoff.d.ts

@@ -1,6 +1,6 @@
 import { z } from 'zod';
 import { Tool, StringToolOutput, ToolEmitter, ToolInput, BaseToolRunOptions } from './base.js';
-import { b as AnyAgent } from '../base-BfhNzUYf.js';
+import { b as AnyAgent } from '../base-BjzyQ8k6.js';
 import { GetRunContext } from '../context.js';
 import 'ajv';
 import '../errors.js';
@@ -11,7 +11,7 @@ import 'promise-based-task';
 import '../cache/base.js';
 import '../internals/helpers/schema.js';
 import 'zod-to-json-schema';
-import '../emitter-CL56qab4.js';
+import '../emitter-kHxkUxco.js';
 import '../internals/helpers/promise.js';
 import '../memory/base.js';
 import '../backend/message.js';

package/dist/tools/handoff.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/tools/handoff.ts"],"names":["HandoffTool","Tool","propagateInputs","name","description","emitter","Emitter","root","child","namespace","creator","inputSchema","z","object","task","string","describe","target","options","toCamelCase","meta","_run","input","run","memory","getProp","context","contextKeys","Memory","ToolError","isRetryable","isFatal","messages","filter","msg","SystemMessage","lastValidMsgIndex","findLastIndex","AssistantMessage","getToolCalls","length","slice","push","UserMessage","clone","reset","addMany","response","prompt","StringToolOutput","result","text"],"mappings":";;;;;;;;;AA4BO,MAAMA,oBAAoBC,IAAAA,CAAAA;EA5BjC;;;;AA6BmBC,EAAAA,eAAAA;AACVC,EAAAA,IAAAA;AACAC,EAAAA,WAAAA;EAESC,OAAAA,GAA0DC,OAAAA,CAAQC,KAAKC,KAAAA,CAAM;IAC3FC,SAAAA,EAAW;AAAC,MAAA,MAAA;AAAQ,MAAA;;IACpBC,OAAAA,EAAS;GACX,CAAA;EAEAC,WAAAA,GAAc;AACZ,IAAA,OAAOC,EAAEC,MAAAA,CAAO;AACdC,MAAAA,IAAAA,EAAMF,CAAAA,CACHG,MAAAA,EAAM,CACNC,QAAAA,CAAS,uEAAA;KACd,CAAA;AACF;AAEA,EAAA,WAAA,CACUC,QACRC,OAAAA,EACA;AACA,IAAA,KAAA,EAAK,EAAA,KAHGD,MAAAA,GAAAA,MAAAA;AAIR,IAAA,IAAA,CAAKd,OAAOgB,WAAAA,CAAYD,OAAAA,EAASf,QAAQ,IAAA,CAAKc,MAAAA,CAAOG,KAAKjB,IAAI,CAAA;AAC9D,IAAA,IAAA,CAAKC,WAAAA,GAAcc,OAAAA,EAASd,WAAAA,IAAe,IAAA,CAAKa,OAAOG,IAAAA,CAAKhB,WAAAA;AAC5D,IAAA,IAAA,CAAKF,eAAAA,GAAkBgB,SAAShB,eAAAA,IAAmB,IAAA;AACrD;EAEA,MAAgBmB,IAAAA,CACdC,KAAAA,EACAJ,OAAAA,EACAK,GAAAA,EAC2B;AAC3B,IAAA,MAAMC,MAAAA,GAASC,OAAAA,CAAQF,GAAAA,CAAIG,OAAAA,EAAS;AAACzB,MAAAA,IAAAA,CAAK0B,WAAAA,CAAYC;AAAO,KAAA,CAAA;AAE7D,IAAA,IAAI,CAACJ,MAAAA,EAAQ;AACX,MAAA,MAAM,IAAIK,SAAAA,CAAU,CAAA,+BAAA,CAAA,EAAmC,EAAA,EAAI;QACzDC,WAAAA,EAAa,KAAA;QACbC,OAAAA,EAAS;OACX,CAAA;AACF;AAEA,IAAA,IAAIC,QAAAA,GAAsBR,OAAOQ,QAAAA,CAASC,MAAAA,CAAO,CAACC,GAAAA,KAAQ,EAAEA,eAAeC,aAAAA,CAAY,CAAA;AAEvF,IAAA,MAAMC,iBAAAA,GAAoBC,aAAAA,CACxBL,QAAAA,EACA,CAACE,GAAAA,KAAQA,GAAAA,YAAeI,gBAAAA,IAAAA,CAAqBJ,GAAAA,CAAIK,YAAAA,EAAY,EAAIC,MAAAA,IAAU,CAAA,IAAK,CAAA,CAAA;AAGlF,IAAA,IAAIJ,sBAAsB,EAAA,EAAI;AAC5BJ,MAAAA,QAAAA,GAAWA,QAAAA,CAASS,KAAAA,CAAM,CAAA,EAAGL,iBAAAA,CAAAA;AAC/B;AAEA,IAAA,IAAI,KAAKlC,eAAAA,EAAiB;AACxB8B,MAAAA,QAAAA,CAASU,IAAAA,CAAK,IAAIC,WAAAA,CAAYrB,KAAAA,CAAMR,IAAI,CAAA,CAAA;AAC1C;AAEA,IAAA,MAAMG,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAAA,CAAO2B,KAAAA,EAAK;AACtC3B,IAAAA,MAAAA,CAAOO,OAAOqB,KAAAA,EAAK;AACnB,IAAA,MAAM5B,MAAAA,CAAOO,MAAAA,CAAOsB,OAAAA,CAAQd,QAAAA,CAAAA;AAE5B,IAAA,MAAMe,QAAAA,GAAW,MAAM9B,MAAAA,CAAOM,GAAAA,CAAI;MAAEyB,MAAAA,EAAQ;KAAK,CAAA;AAEjD,IAAA,OAAO,IAAIC,gBAAAA,CAAiBF,QAAAA,CAASG,MAAAA,CAAOC,IAAI,CAAA;AAClD;AACF","file":"handoff.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { z } from \"zod\";\nimport {\n  BaseToolRunOptions,\n  StringToolOutput,\n  Tool,\n  ToolEmitter,\n  ToolError,\n  ToolInput,\n} from \"./base.js\";\nimport { BaseMemory } from \"@/memory/base.js\";\nimport { AssistantMessage, Message, SystemMessage, UserMessage } from \"@/backend/message.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { AnyAgent } from \"@/agents/types.js\";\nimport { GetRunContext } from \"@/context.js\";\nimport { getProp } from \"@/internals/helpers/object.js\";\nimport { findLastIndex, toCamelCase } from \"remeda\";\n\nexport interface HandoffToolInput {\n  name?: string;\n  description?: string;\n  propagateInputs?: boolean;\n}\n\nexport class HandoffTool extends Tool<StringToolOutput> {\n  private readonly propagateInputs: boolean;\n  public name: string;\n  public description: string;\n\n  public readonly emitter: ToolEmitter<ToolInput<this>, StringToolOutput> = Emitter.root.child({\n    namespace: [\"tool\", \"handoff\"],\n    creator: this,\n  });\n\n  inputSchema() {\n    return z.object({\n      task: z\n        .string()\n        .describe(\"Clearly defined task for the agent to work on based on his abilities.\"),\n    });\n  }\n\n  constructor(\n    private target: AnyAgent,\n    options?: HandoffToolInput,\n  ) {\n    super();\n    this.name = toCamelCase(options?.name || this.target.meta.name);\n    this.description = options?.description || this.target.meta.description;\n    this.propagateInputs = options?.propagateInputs ?? true;\n  }\n\n  protected async _run(\n    input: ToolInput<this>,\n    options: Partial<BaseToolRunOptions>,\n    run: GetRunContext<typeof this>,\n  ): Promise<StringToolOutput> {\n    const memory = getProp(run.context, [Tool.contextKeys.Memory]) as BaseMemory;\n\n    if (!memory) {\n      throw new ToolError(`No memory found in the context.`, [], {\n        isRetryable: false,\n        isFatal: true,\n      });\n    }\n\n    let messages: Message[] = memory.messages.filter((msg) => !(msg instanceof SystemMessage));\n\n    const lastValidMsgIndex = findLastIndex(\n      messages,\n      (msg) => msg instanceof AssistantMessage && (msg.getToolCalls()?.length ?? 0) > 0,\n    );\n\n    if (lastValidMsgIndex !== -1) {\n      messages = messages.slice(0, lastValidMsgIndex);\n    }\n\n    if (this.propagateInputs) {\n      messages.push(new UserMessage(input.task));\n    }\n\n    const target = await this.target.clone();\n    target.memory.reset();\n    await target.memory.addMany(messages);\n\n    const response = await target.run({ prompt: null });\n\n    return new StringToolOutput(response.result.text);\n  }\n}\n"]}
+{"version":3,"sources":["../../src/tools/handoff.ts"],"names":["HandoffTool","Tool","propagateInputs","name","description","emitter","Emitter","root","child","namespace","creator","inputSchema","z","object","task","string","describe","target","options","toCamelCase","meta","_run","input","run","memory","getProp","context","contextKeys","Memory","ToolError","isRetryable","isFatal","messages","filter","msg","SystemMessage","lastValidMsgIndex","findLastIndex","AssistantMessage","getToolCalls","length","slice","push","UserMessage","clone","reset","addMany","response","prompt","StringToolOutput","result","text"],"mappings":";;;;;;;;;AA4BO,MAAMA,oBAAoBC,IAAAA,CAAAA;EA5BjC;;;;AA6BmBC,EAAAA,eAAAA;AACVC,EAAAA,IAAAA;AACAC,EAAAA,WAAAA;EAESC,OAAAA,GAA0DC,OAAAA,CAAQC,KAAKC,KAAAA,CAAM;IAC3FC,SAAAA,EAAW;AAAC,MAAA,MAAA;AAAQ,MAAA;;IACpBC,OAAAA,EAAS;GACX,CAAA;EAEAC,WAAAA,GAAc;AACZ,IAAA,OAAOC,EAAEC,MAAAA,CAAO;AACdC,MAAAA,IAAAA,EAAMF,CAAAA,CACHG,MAAAA,EAAM,CACNC,QAAAA,CAAS,uEAAA;KACd,CAAA;AACF,EAAA;AAEA,EAAA,WAAA,CACUC,QACRC,OAAAA,EACA;AACA,IAAA,KAAA,EAAK,EAAA,KAHGD,MAAAA,GAAAA,MAAAA;AAIR,IAAA,IAAA,CAAKd,OAAOgB,WAAAA,CAAYD,OAAAA,EAASf,QAAQ,IAAA,CAAKc,MAAAA,CAAOG,KAAKjB,IAAI,CAAA;AAC9D,IAAA,IAAA,CAAKC,WAAAA,GAAcc,OAAAA,EAASd,WAAAA,IAAe,IAAA,CAAKa,OAAOG,IAAAA,CAAKhB,WAAAA;AAC5D,IAAA,IAAA,CAAKF,eAAAA,GAAkBgB,SAAShB,eAAAA,IAAmB,IAAA;AACrD,EAAA;EAEA,MAAgBmB,IAAAA,CACdC,KAAAA,EACAJ,OAAAA,EACAK,GAAAA,EAC2B;AAC3B,IAAA,MAAMC,MAAAA,GAASC,OAAAA,CAAQF,GAAAA,CAAIG,OAAAA,EAAS;AAACzB,MAAAA,IAAAA,CAAK0B,WAAAA,CAAYC;AAAO,KAAA,CAAA;AAE7D,IAAA,IAAI,CAACJ,MAAAA,EAAQ;AACX,MAAA,MAAM,IAAIK,SAAAA,CAAU,CAAA,+BAAA,CAAA,EAAmC,EAAA,EAAI;QACzDC,WAAAA,EAAa,KAAA;QACbC,OAAAA,EAAS;OACX,CAAA;AACF,IAAA;AAEA,IAAA,IAAIC,QAAAA,GAAsBR,OAAOQ,QAAAA,CAASC,MAAAA,CAAO,CAACC,GAAAA,KAAQ,EAAEA,eAAeC,aAAAA,CAAY,CAAA;AAEvF,IAAA,MAAMC,iBAAAA,GAAoBC,aAAAA,CACxBL,QAAAA,EACA,CAACE,GAAAA,KAAQA,GAAAA,YAAeI,gBAAAA,IAAAA,CAAqBJ,GAAAA,CAAIK,YAAAA,EAAY,EAAIC,MAAAA,IAAU,CAAA,IAAK,CAAA,CAAA;AAGlF,IAAA,IAAIJ,sBAAsB,EAAA,EAAI;AAC5BJ,MAAAA,QAAAA,GAAWA,QAAAA,CAASS,KAAAA,CAAM,CAAA,EAAGL,iBAAAA,CAAAA;AAC/B,IAAA;AAEA,IAAA,IAAI,KAAKlC,eAAAA,EAAiB;AACxB8B,MAAAA,QAAAA,CAASU,IAAAA,CAAK,IAAIC,WAAAA,CAAYrB,KAAAA,CAAMR,IAAI,CAAA,CAAA;AAC1C,IAAA;AAEA,IAAA,MAAMG,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAAA,CAAO2B,KAAAA,EAAK;AACtC3B,IAAAA,MAAAA,CAAOO,OAAOqB,KAAAA,EAAK;AACnB,IAAA,MAAM5B,MAAAA,CAAOO,MAAAA,CAAOsB,OAAAA,CAAQd,QAAAA,CAAAA;AAE5B,IAAA,MAAMe,QAAAA,GAAW,MAAM9B,MAAAA,CAAOM,GAAAA,CAAI;MAAEyB,MAAAA,EAAQ;KAAK,CAAA;AAEjD,IAAA,OAAO,IAAIC,gBAAAA,CAAiBF,QAAAA,CAASG,MAAAA,CAAOC,IAAI,CAAA;AAClD,EAAA;AACF","file":"handoff.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { z } from \"zod\";\nimport {\n  BaseToolRunOptions,\n  StringToolOutput,\n  Tool,\n  ToolEmitter,\n  ToolError,\n  ToolInput,\n} from \"./base.js\";\nimport { BaseMemory } from \"@/memory/base.js\";\nimport { AssistantMessage, Message, SystemMessage, UserMessage } from \"@/backend/message.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { AnyAgent } from \"@/agents/types.js\";\nimport { GetRunContext } from \"@/context.js\";\nimport { getProp } from \"@/internals/helpers/object.js\";\nimport { findLastIndex, toCamelCase } from \"remeda\";\n\nexport interface HandoffToolInput {\n  name?: string;\n  description?: string;\n  propagateInputs?: boolean;\n}\n\nexport class HandoffTool extends Tool<StringToolOutput> {\n  private readonly propagateInputs: boolean;\n  public name: string;\n  public description: string;\n\n  public readonly emitter: ToolEmitter<ToolInput<this>, StringToolOutput> = Emitter.root.child({\n    namespace: [\"tool\", \"handoff\"],\n    creator: this,\n  });\n\n  inputSchema() {\n    return z.object({\n      task: z\n        .string()\n        .describe(\"Clearly defined task for the agent to work on based on his abilities.\"),\n    });\n  }\n\n  constructor(\n    private target: AnyAgent,\n    options?: HandoffToolInput,\n  ) {\n    super();\n    this.name = toCamelCase(options?.name || this.target.meta.name);\n    this.description = options?.description || this.target.meta.description;\n    this.propagateInputs = options?.propagateInputs ?? true;\n  }\n\n  protected async _run(\n    input: ToolInput<this>,\n    options: Partial<BaseToolRunOptions>,\n    run: GetRunContext<typeof this>,\n  ): Promise<StringToolOutput> {\n    const memory = getProp(run.context, [Tool.contextKeys.Memory]) as BaseMemory;\n\n    if (!memory) {\n      throw new ToolError(`No memory found in the context.`, [], {\n        isRetryable: false,\n        isFatal: true,\n      });\n    }\n\n    let messages: Message[] = memory.messages.filter((msg) => !(msg instanceof SystemMessage));\n\n    const lastValidMsgIndex = findLastIndex(\n      messages,\n      (msg) => msg instanceof AssistantMessage && (msg.getToolCalls()?.length ?? 0) > 0,\n    );\n\n    if (lastValidMsgIndex !== -1) {\n      messages = messages.slice(0, lastValidMsgIndex);\n    }\n\n    if (this.propagateInputs) {\n      messages.push(new UserMessage(input.task));\n    }\n\n    const target = await this.target.clone();\n    target.memory.reset();\n    await target.memory.addMany(messages);\n\n    const response = await target.run({ prompt: null });\n\n    return new StringToolOutput(response.result.text);\n  }\n}\n"]}

package/dist/tools/llm.cjs

@@ -39,7 +39,7 @@ class LLMTool extends base_cjs.Tool {
     schema: zod.z.object({
       task: zod.z.string()
     }),
-    template: `You have to accomplish a task by using Using common sense and the information contained in the conversation up to this point, complete the following task. Do not follow any previously used formats or structures.
+    template: `Use common sense and the information contained in the conversation up to this point to complete the following task. Do not follow any previously used formats or structures.
 
 The Task: {{task}}`
   });

package/dist/tools/llm.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/tools/llm.ts"],"names":["LLMTool","Tool","name","description","input","emitter","Emitter","root","child","namespace","toCamelCase","filter","Boolean","creator","inputSchema","z","object","task","string","min","describe","template","PromptTemplate","schema","_run","_options","run","memory","getProp","context","contextKeys","Memory","ToolError","isFatal","isRetryable","options","output","llm","create","messages","SystemMessage","render","msg","role","Role","SYSTEM","UserMessage","StringToolOutput","getTextContent"],"mappings":";;;;;;;;;;;;AA+BO,MAAMA,gBAAgBC,aAAAA,CAAAA;EA/B7B;;;;EAgCEC,IAAAA,GAAO,KAAA;EACPC,WAAAA,GACE,wHAAA;AAGF,EAAA,WAAA,CAA+BC,KAAAA,EAAqB;AAClD,IAAA,KAAA,CAAMA,KAAAA,CAAAA,EAAAA,IAAAA,CADuBA,KAAAA,GAAAA,KAAAA;AAE7B,IAAA,IAAA,CAAKF,IAAAA,GAAOE,KAAAA,EAAOF,IAAAA,IAAQ,IAAA,CAAKA,IAAAA;AAChC,IAAA,IAAA,CAAKC,WAAAA,GAAcC,KAAAA,EAAOD,WAAAA,IAAe,IAAA,CAAKA,WAAAA;AAC9C,IAAA,IAAA,CAAKE,OAAAA,GAAUC,mBAAAA,CAAQC,IAAAA,CAAKC,KAAAA,CAAM;MAChCC,SAAAA,EAAW;AAAC,QAAA,MAAA;AAAQ,QAAA,KAAA;QAAOC,kBAAAA,CAAYN,KAAAA,EAAOF,QAAQ,EAAA;AAAKS,OAAAA,CAAAA,MAAAA,CAAOC,OAAAA,CAAAA;MAClEC,OAAAA,EAAS;KACX,CAAA;AACF;EAEAC,WAAAA,GAAc;AACZ,IAAA,OAAOC,MAAEC,MAAAA,CAAO;AACdC,MAAAA,IAAAA,EAAMF,MAAEG,MAAAA,EAAM,CAAGC,IAAI,CAAA,CAAA,CAAGC,SAAS,iDAAA;KACnC,CAAA;AACF;EAEA,OAAgBC,QAAAA,GAAW,IAAIC,2BAAAA,CAAe;AAC5CC,IAAAA,MAAAA,EAAQR,MAAEC,MAAAA,CAAO;AACfC,MAAAA,IAAAA,EAAMF,MAAEG,MAAAA;KACV,CAAA;IACAG,QAAAA,EAAU,CAAA;;;GAGZ,CAAA;EAEA,MAAgBG,IAAAA,CACdpB,KAAAA,EACAqB,QAAAA,EACAC,GAAAA,EACA;AACA,IAAA,MAAMC,MAAAA,GAASC,kBAAAA,CAAQF,GAAAA,CAAIG,OAAAA,EAAS;AAAC5B,MAAAA,aAAAA,CAAK6B,WAAAA,CAAYC;AAAO,KAAA,CAAA;AAC7D,IAAA,IAAI,CAACJ,MAAAA,EAAQ;AACX,MAAA,MAAM,IAAIK,kBAAAA,CAAU,CAAA,6BAAA,CAAA,EAAiC,EAAA,EAAI;QACvDC,OAAAA,EAAS,IAAA;QACTC,WAAAA,EAAa;OACf,CAAA;AACF;AAEA,IAAA,MAAMb,QAAAA,GAAW,IAAA,CAAKc,OAAAA,EAASd,QAAAA,IAAYrB,OAAAA,CAAQqB,QAAAA;AACnD,IAAA,MAAMe,MAAAA,GAAS,MAAM,IAAA,CAAKhC,KAAAA,CAAMiC,IAAIC,MAAAA,CAAO;MACzCC,QAAAA,EAAU;QACR,IAAIC,yBAAAA,CACFnB,SAASoB,MAAAA,CAAO;AACdxB,UAAAA,IAAAA,EAAMb,KAAAA,CAAMa;AACd,SAAA,CAAA,CAAA;AAECU,QAAAA,GAAAA,MAAAA,CAAOY,SAAS5B,MAAAA,CAAO,CAAC+B,QAAQA,GAAAA,CAAIC,IAAAA,KAASC,iBAAKC,MAAM,CAAA;QAC3D,IAAIC,uBAAAA,CACFzB,SAASoB,MAAAA,CAAO;AACdxB,UAAAA,IAAAA,EAAMb,KAAAA,CAAMa;AACd,SAAA,CAAA;;KAGN,CAAA;AAEA,IAAA,OAAO,IAAI8B,yBAAAA,CAAiBX,MAAAA,CAAOY,cAAAA,EAAc,CAAA;AACnD;AACF","file":"llm.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport {\n  BaseToolOptions,\n  BaseToolRunOptions,\n  StringToolOutput,\n  Tool,\n  ToolEmitter,\n  ToolError,\n  ToolInput,\n} from \"@/tools/base.js\";\nimport { z } from \"zod\";\nimport { GetRunContext } from \"@/context.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { PromptTemplate } from \"@/template.js\";\nimport { getProp } from \"@/internals/helpers/object.js\";\nimport type { BaseMemory } from \"@/memory/base.js\";\nimport { toCamelCase } from \"remeda\";\nimport { Role, SystemMessage, UserMessage } from \"@/backend/message.js\";\nimport { ChatModel } from \"@/backend/chat.js\";\n\nexport interface LLMToolInput extends BaseToolOptions {\n  llm: ChatModel;\n  name?: string;\n  description?: string;\n  template?: typeof LLMTool.template;\n}\n\nexport class LLMTool extends Tool<StringToolOutput, LLMToolInput> {\n  name = \"LLM\";\n  description =\n    \"Uses expert LLM to work with data in the existing conversation (classification, entity extraction, summarization, ...)\";\n  declare readonly emitter: ToolEmitter<ToolInput<this>, StringToolOutput>;\n\n  constructor(protected readonly input: LLMToolInput) {\n    super(input);\n    this.name = input?.name || this.name;\n    this.description = input?.description || this.description;\n    this.emitter = Emitter.root.child({\n      namespace: [\"tool\", \"llm\", toCamelCase(input?.name ?? \"\")].filter(Boolean),\n      creator: this,\n    });\n  }\n\n  inputSchema() {\n    return z.object({\n      task: z.string().min(1).describe(\"A clearly defined task for the LLM to complete.\"),\n    });\n  }\n\n  static readonly template = new PromptTemplate({\n    schema: z.object({\n      task: z.string(),\n    }),\n    template: `You have to accomplish a task by using Using common sense and the information contained in the conversation up to this point, complete the following task. Do not follow any previously used formats or structures.\n\nThe Task: {{task}}`,\n  });\n\n  protected async _run(\n    input: ToolInput<this>,\n    _options: Partial<BaseToolRunOptions>,\n    run: GetRunContext<this>,\n  ) {\n    const memory = getProp(run.context, [Tool.contextKeys.Memory]) as BaseMemory;\n    if (!memory) {\n      throw new ToolError(`No context has been provided!`, [], {\n        isFatal: true,\n        isRetryable: false,\n      });\n    }\n\n    const template = this.options?.template ?? LLMTool.template;\n    const output = await this.input.llm.create({\n      messages: [\n        new SystemMessage(\n          template.render({\n            task: input.task,\n          }),\n        ),\n        ...memory.messages.filter((msg) => msg.role !== Role.SYSTEM),\n        new UserMessage(\n          template.render({\n            task: input.task,\n          }),\n        ),\n      ],\n    });\n\n    return new StringToolOutput(output.getTextContent());\n  }\n}\n"]}
+{"version":3,"sources":["../../src/tools/llm.ts"],"names":["LLMTool","Tool","name","description","input","emitter","Emitter","root","child","namespace","toCamelCase","filter","Boolean","creator","inputSchema","z","object","task","string","min","describe","template","PromptTemplate","schema","_run","_options","run","memory","getProp","context","contextKeys","Memory","ToolError","isFatal","isRetryable","options","output","llm","create","messages","SystemMessage","render","msg","role","Role","SYSTEM","UserMessage","StringToolOutput","getTextContent"],"mappings":";;;;;;;;;;;;AA+BO,MAAMA,gBAAgBC,aAAAA,CAAAA;EA/B7B;;;;EAgCEC,IAAAA,GAAO,KAAA;EACPC,WAAAA,GACE,wHAAA;AAGF,EAAA,WAAA,CAA+BC,KAAAA,EAAqB;AAClD,IAAA,KAAA,CAAMA,KAAAA,CAAAA,EAAAA,IAAAA,CADuBA,KAAAA,GAAAA,KAAAA;AAE7B,IAAA,IAAA,CAAKF,IAAAA,GAAOE,KAAAA,EAAOF,IAAAA,IAAQ,IAAA,CAAKA,IAAAA;AAChC,IAAA,IAAA,CAAKC,WAAAA,GAAcC,KAAAA,EAAOD,WAAAA,IAAe,IAAA,CAAKA,WAAAA;AAC9C,IAAA,IAAA,CAAKE,OAAAA,GAAUC,mBAAAA,CAAQC,IAAAA,CAAKC,KAAAA,CAAM;MAChCC,SAAAA,EAAW;AAAC,QAAA,MAAA;AAAQ,QAAA,KAAA;QAAOC,kBAAAA,CAAYN,KAAAA,EAAOF,QAAQ,EAAA;AAAKS,OAAAA,CAAAA,MAAAA,CAAOC,OAAAA,CAAAA;MAClEC,OAAAA,EAAS;KACX,CAAA;AACF,EAAA;EAEAC,WAAAA,GAAc;AACZ,IAAA,OAAOC,MAAEC,MAAAA,CAAO;AACdC,MAAAA,IAAAA,EAAMF,MAAEG,MAAAA,EAAM,CAAGC,IAAI,CAAA,CAAA,CAAGC,SAAS,iDAAA;KACnC,CAAA;AACF,EAAA;EAEA,OAAgBC,QAAAA,GAAW,IAAIC,2BAAAA,CAAe;AAC5CC,IAAAA,MAAAA,EAAQR,MAAEC,MAAAA,CAAO;AACfC,MAAAA,IAAAA,EAAMF,MAAEG,MAAAA;KACV,CAAA;IACAG,QAAAA,EAAU,CAAA;;;GAGZ,CAAA;EAEA,MAAgBG,IAAAA,CACdpB,KAAAA,EACAqB,QAAAA,EACAC,GAAAA,EACA;AACA,IAAA,MAAMC,MAAAA,GAASC,kBAAAA,CAAQF,GAAAA,CAAIG,OAAAA,EAAS;AAAC5B,MAAAA,aAAAA,CAAK6B,WAAAA,CAAYC;AAAO,KAAA,CAAA;AAC7D,IAAA,IAAI,CAACJ,MAAAA,EAAQ;AACX,MAAA,MAAM,IAAIK,kBAAAA,CAAU,CAAA,6BAAA,CAAA,EAAiC,EAAA,EAAI;QACvDC,OAAAA,EAAS,IAAA;QACTC,WAAAA,EAAa;OACf,CAAA;AACF,IAAA;AAEA,IAAA,MAAMb,QAAAA,GAAW,IAAA,CAAKc,OAAAA,EAASd,QAAAA,IAAYrB,OAAAA,CAAQqB,QAAAA;AACnD,IAAA,MAAMe,MAAAA,GAAS,MAAM,IAAA,CAAKhC,KAAAA,CAAMiC,IAAIC,MAAAA,CAAO;MACzCC,QAAAA,EAAU;QACR,IAAIC,yBAAAA,CACFnB,SAASoB,MAAAA,CAAO;AACdxB,UAAAA,IAAAA,EAAMb,KAAAA,CAAMa;AACd,SAAA,CAAA,CAAA;AAECU,QAAAA,GAAAA,MAAAA,CAAOY,SAAS5B,MAAAA,CAAO,CAAC+B,QAAQA,GAAAA,CAAIC,IAAAA,KAASC,iBAAKC,MAAM,CAAA;QAC3D,IAAIC,uBAAAA,CACFzB,SAASoB,MAAAA,CAAO;AACdxB,UAAAA,IAAAA,EAAMb,KAAAA,CAAMa;AACd,SAAA,CAAA;;KAGN,CAAA;AAEA,IAAA,OAAO,IAAI8B,yBAAAA,CAAiBX,MAAAA,CAAOY,cAAAA,EAAc,CAAA;AACnD,EAAA;AACF","file":"llm.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport {\n  BaseToolOptions,\n  BaseToolRunOptions,\n  StringToolOutput,\n  Tool,\n  ToolEmitter,\n  ToolError,\n  ToolInput,\n} from \"@/tools/base.js\";\nimport { z } from \"zod\";\nimport { GetRunContext } from \"@/context.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { PromptTemplate } from \"@/template.js\";\nimport { getProp } from \"@/internals/helpers/object.js\";\nimport type { BaseMemory } from \"@/memory/base.js\";\nimport { toCamelCase } from \"remeda\";\nimport { Role, SystemMessage, UserMessage } from \"@/backend/message.js\";\nimport { ChatModel } from \"@/backend/chat.js\";\n\nexport interface LLMToolInput extends BaseToolOptions {\n  llm: ChatModel;\n  name?: string;\n  description?: string;\n  template?: typeof LLMTool.template;\n}\n\nexport class LLMTool extends Tool<StringToolOutput, LLMToolInput> {\n  name = \"LLM\";\n  description =\n    \"Uses expert LLM to work with data in the existing conversation (classification, entity extraction, summarization, ...)\";\n  declare readonly emitter: ToolEmitter<ToolInput<this>, StringToolOutput>;\n\n  constructor(protected readonly input: LLMToolInput) {\n    super(input);\n    this.name = input?.name || this.name;\n    this.description = input?.description || this.description;\n    this.emitter = Emitter.root.child({\n      namespace: [\"tool\", \"llm\", toCamelCase(input?.name ?? \"\")].filter(Boolean),\n      creator: this,\n    });\n  }\n\n  inputSchema() {\n    return z.object({\n      task: z.string().min(1).describe(\"A clearly defined task for the LLM to complete.\"),\n    });\n  }\n\n  static readonly template = new PromptTemplate({\n    schema: z.object({\n      task: z.string(),\n    }),\n    template: `Use common sense and the information contained in the conversation up to this point to complete the following task. Do not follow any previously used formats or structures.\n\nThe Task: {{task}}`,\n  });\n\n  protected async _run(\n    input: ToolInput<this>,\n    _options: Partial<BaseToolRunOptions>,\n    run: GetRunContext<this>,\n  ) {\n    const memory = getProp(run.context, [Tool.contextKeys.Memory]) as BaseMemory;\n    if (!memory) {\n      throw new ToolError(`No context has been provided!`, [], {\n        isFatal: true,\n        isRetryable: false,\n      });\n    }\n\n    const template = this.options?.template ?? LLMTool.template;\n    const output = await this.input.llm.create({\n      messages: [\n        new SystemMessage(\n          template.render({\n            task: input.task,\n          }),\n        ),\n        ...memory.messages.filter((msg) => msg.role !== Role.SYSTEM),\n        new UserMessage(\n          template.render({\n            task: input.task,\n          }),\n        ),\n      ],\n    });\n\n    return new StringToolOutput(output.getTextContent());\n  }\n}\n"]}

package/dist/tools/llm.d.cts

@@ -1,8 +1,8 @@
-import { BaseToolOptions, Tool, StringToolOutput, ToolEmitter, ToolInput, BaseToolRunOptions } from './base.cjs';
+import { Tool, StringToolOutput, BaseToolOptions, ToolEmitter, ToolInput, BaseToolRunOptions } from './base.cjs';
 import { z } from 'zod';
 import { GetRunContext } from '../context.cjs';
 import { PromptTemplate } from '../template.cjs';
-import { m as ChatModel } from '../chat-DaqV3woF.cjs';
+import { a as ChatModel } from '../chat-C20nYj46.cjs';
 import 'ajv';
 import '../errors.cjs';
 import '../internals/types.cjs';
@@ -12,7 +12,7 @@ import 'promise-based-task';
 import '../cache/base.cjs';
 import '../internals/helpers/schema.cjs';
 import 'zod-to-json-schema';
-import '../emitter-UkIkNrek.cjs';
+import '../emitter-D4OcelZ9.cjs';
 import '../internals/helpers/promise.cjs';
 import '../backend/message.cjs';
 import 'ai';

package/dist/tools/llm.d.ts

@@ -1,8 +1,8 @@
-import { BaseToolOptions, Tool, StringToolOutput, ToolEmitter, ToolInput, BaseToolRunOptions } from './base.js';
+import { Tool, StringToolOutput, BaseToolOptions, ToolEmitter, ToolInput, BaseToolRunOptions } from './base.js';
 import { z } from 'zod';
 import { GetRunContext } from '../context.js';
 import { PromptTemplate } from '../template.js';
-import { m as ChatModel } from '../chat-DTxmaKP0.js';
+import { a as ChatModel } from '../chat-CfTNfR60.js';
 import 'ajv';
 import '../errors.js';
 import '../internals/types.js';
@@ -12,7 +12,7 @@ import 'promise-based-task';
 import '../cache/base.js';
 import '../internals/helpers/schema.js';
 import 'zod-to-json-schema';
-import '../emitter-CL56qab4.js';
+import '../emitter-kHxkUxco.js';
 import '../internals/helpers/promise.js';
 import '../backend/message.js';
 import 'ai';

package/dist/tools/llm.js

@@ -37,7 +37,7 @@ class LLMTool extends Tool {
     schema: z.object({
       task: z.string()
     }),
-    template: `You have to accomplish a task by using Using common sense and the information contained in the conversation up to this point, complete the following task. Do not follow any previously used formats or structures.
+    template: `Use common sense and the information contained in the conversation up to this point to complete the following task. Do not follow any previously used formats or structures.
 
 The Task: {{task}}`
   });

package/dist/tools/llm.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/tools/llm.ts"],"names":["LLMTool","Tool","name","description","input","emitter","Emitter","root","child","namespace","toCamelCase","filter","Boolean","creator","inputSchema","z","object","task","string","min","describe","template","PromptTemplate","schema","_run","_options","run","memory","getProp","context","contextKeys","Memory","ToolError","isFatal","isRetryable","options","output","llm","create","messages","SystemMessage","render","msg","role","Role","SYSTEM","UserMessage","StringToolOutput","getTextContent"],"mappings":";;;;;;;;;;AA+BO,MAAMA,gBAAgBC,IAAAA,CAAAA;EA/B7B;;;;EAgCEC,IAAAA,GAAO,KAAA;EACPC,WAAAA,GACE,wHAAA;AAGF,EAAA,WAAA,CAA+BC,KAAAA,EAAqB;AAClD,IAAA,KAAA,CAAMA,KAAAA,CAAAA,EAAAA,IAAAA,CADuBA,KAAAA,GAAAA,KAAAA;AAE7B,IAAA,IAAA,CAAKF,IAAAA,GAAOE,KAAAA,EAAOF,IAAAA,IAAQ,IAAA,CAAKA,IAAAA;AAChC,IAAA,IAAA,CAAKC,WAAAA,GAAcC,KAAAA,EAAOD,WAAAA,IAAe,IAAA,CAAKA,WAAAA;AAC9C,IAAA,IAAA,CAAKE,OAAAA,GAAUC,OAAAA,CAAQC,IAAAA,CAAKC,KAAAA,CAAM;MAChCC,SAAAA,EAAW;AAAC,QAAA,MAAA;AAAQ,QAAA,KAAA;QAAOC,WAAAA,CAAYN,KAAAA,EAAOF,QAAQ,EAAA;AAAKS,OAAAA,CAAAA,MAAAA,CAAOC,OAAAA,CAAAA;MAClEC,OAAAA,EAAS;KACX,CAAA;AACF;EAEAC,WAAAA,GAAc;AACZ,IAAA,OAAOC,EAAEC,MAAAA,CAAO;AACdC,MAAAA,IAAAA,EAAMF,EAAEG,MAAAA,EAAM,CAAGC,IAAI,CAAA,CAAA,CAAGC,SAAS,iDAAA;KACnC,CAAA;AACF;EAEA,OAAgBC,QAAAA,GAAW,IAAIC,cAAAA,CAAe;AAC5CC,IAAAA,MAAAA,EAAQR,EAAEC,MAAAA,CAAO;AACfC,MAAAA,IAAAA,EAAMF,EAAEG,MAAAA;KACV,CAAA;IACAG,QAAAA,EAAU,CAAA;;;GAGZ,CAAA;EAEA,MAAgBG,IAAAA,CACdpB,KAAAA,EACAqB,QAAAA,EACAC,GAAAA,EACA;AACA,IAAA,MAAMC,MAAAA,GAASC,OAAAA,CAAQF,GAAAA,CAAIG,OAAAA,EAAS;AAAC5B,MAAAA,IAAAA,CAAK6B,WAAAA,CAAYC;AAAO,KAAA,CAAA;AAC7D,IAAA,IAAI,CAACJ,MAAAA,EAAQ;AACX,MAAA,MAAM,IAAIK,SAAAA,CAAU,CAAA,6BAAA,CAAA,EAAiC,EAAA,EAAI;QACvDC,OAAAA,EAAS,IAAA;QACTC,WAAAA,EAAa;OACf,CAAA;AACF;AAEA,IAAA,MAAMb,QAAAA,GAAW,IAAA,CAAKc,OAAAA,EAASd,QAAAA,IAAYrB,OAAAA,CAAQqB,QAAAA;AACnD,IAAA,MAAMe,MAAAA,GAAS,MAAM,IAAA,CAAKhC,KAAAA,CAAMiC,IAAIC,MAAAA,CAAO;MACzCC,QAAAA,EAAU;QACR,IAAIC,aAAAA,CACFnB,SAASoB,MAAAA,CAAO;AACdxB,UAAAA,IAAAA,EAAMb,KAAAA,CAAMa;AACd,SAAA,CAAA,CAAA;AAECU,QAAAA,GAAAA,MAAAA,CAAOY,SAAS5B,MAAAA,CAAO,CAAC+B,QAAQA,GAAAA,CAAIC,IAAAA,KAASC,KAAKC,MAAM,CAAA;QAC3D,IAAIC,WAAAA,CACFzB,SAASoB,MAAAA,CAAO;AACdxB,UAAAA,IAAAA,EAAMb,KAAAA,CAAMa;AACd,SAAA,CAAA;;KAGN,CAAA;AAEA,IAAA,OAAO,IAAI8B,gBAAAA,CAAiBX,MAAAA,CAAOY,cAAAA,EAAc,CAAA;AACnD;AACF","file":"llm.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport {\n  BaseToolOptions,\n  BaseToolRunOptions,\n  StringToolOutput,\n  Tool,\n  ToolEmitter,\n  ToolError,\n  ToolInput,\n} from \"@/tools/base.js\";\nimport { z } from \"zod\";\nimport { GetRunContext } from \"@/context.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { PromptTemplate } from \"@/template.js\";\nimport { getProp } from \"@/internals/helpers/object.js\";\nimport type { BaseMemory } from \"@/memory/base.js\";\nimport { toCamelCase } from \"remeda\";\nimport { Role, SystemMessage, UserMessage } from \"@/backend/message.js\";\nimport { ChatModel } from \"@/backend/chat.js\";\n\nexport interface LLMToolInput extends BaseToolOptions {\n  llm: ChatModel;\n  name?: string;\n  description?: string;\n  template?: typeof LLMTool.template;\n}\n\nexport class LLMTool extends Tool<StringToolOutput, LLMToolInput> {\n  name = \"LLM\";\n  description =\n    \"Uses expert LLM to work with data in the existing conversation (classification, entity extraction, summarization, ...)\";\n  declare readonly emitter: ToolEmitter<ToolInput<this>, StringToolOutput>;\n\n  constructor(protected readonly input: LLMToolInput) {\n    super(input);\n    this.name = input?.name || this.name;\n    this.description = input?.description || this.description;\n    this.emitter = Emitter.root.child({\n      namespace: [\"tool\", \"llm\", toCamelCase(input?.name ?? \"\")].filter(Boolean),\n      creator: this,\n    });\n  }\n\n  inputSchema() {\n    return z.object({\n      task: z.string().min(1).describe(\"A clearly defined task for the LLM to complete.\"),\n    });\n  }\n\n  static readonly template = new PromptTemplate({\n    schema: z.object({\n      task: z.string(),\n    }),\n    template: `You have to accomplish a task by using Using common sense and the information contained in the conversation up to this point, complete the following task. Do not follow any previously used formats or structures.\n\nThe Task: {{task}}`,\n  });\n\n  protected async _run(\n    input: ToolInput<this>,\n    _options: Partial<BaseToolRunOptions>,\n    run: GetRunContext<this>,\n  ) {\n    const memory = getProp(run.context, [Tool.contextKeys.Memory]) as BaseMemory;\n    if (!memory) {\n      throw new ToolError(`No context has been provided!`, [], {\n        isFatal: true,\n        isRetryable: false,\n      });\n    }\n\n    const template = this.options?.template ?? LLMTool.template;\n    const output = await this.input.llm.create({\n      messages: [\n        new SystemMessage(\n          template.render({\n            task: input.task,\n          }),\n        ),\n        ...memory.messages.filter((msg) => msg.role !== Role.SYSTEM),\n        new UserMessage(\n          template.render({\n            task: input.task,\n          }),\n        ),\n      ],\n    });\n\n    return new StringToolOutput(output.getTextContent());\n  }\n}\n"]}
+{"version":3,"sources":["../../src/tools/llm.ts"],"names":["LLMTool","Tool","name","description","input","emitter","Emitter","root","child","namespace","toCamelCase","filter","Boolean","creator","inputSchema","z","object","task","string","min","describe","template","PromptTemplate","schema","_run","_options","run","memory","getProp","context","contextKeys","Memory","ToolError","isFatal","isRetryable","options","output","llm","create","messages","SystemMessage","render","msg","role","Role","SYSTEM","UserMessage","StringToolOutput","getTextContent"],"mappings":";;;;;;;;;;AA+BO,MAAMA,gBAAgBC,IAAAA,CAAAA;EA/B7B;;;;EAgCEC,IAAAA,GAAO,KAAA;EACPC,WAAAA,GACE,wHAAA;AAGF,EAAA,WAAA,CAA+BC,KAAAA,EAAqB;AAClD,IAAA,KAAA,CAAMA,KAAAA,CAAAA,EAAAA,IAAAA,CADuBA,KAAAA,GAAAA,KAAAA;AAE7B,IAAA,IAAA,CAAKF,IAAAA,GAAOE,KAAAA,EAAOF,IAAAA,IAAQ,IAAA,CAAKA,IAAAA;AAChC,IAAA,IAAA,CAAKC,WAAAA,GAAcC,KAAAA,EAAOD,WAAAA,IAAe,IAAA,CAAKA,WAAAA;AAC9C,IAAA,IAAA,CAAKE,OAAAA,GAAUC,OAAAA,CAAQC,IAAAA,CAAKC,KAAAA,CAAM;MAChCC,SAAAA,EAAW;AAAC,QAAA,MAAA;AAAQ,QAAA,KAAA;QAAOC,WAAAA,CAAYN,KAAAA,EAAOF,QAAQ,EAAA;AAAKS,OAAAA,CAAAA,MAAAA,CAAOC,OAAAA,CAAAA;MAClEC,OAAAA,EAAS;KACX,CAAA;AACF,EAAA;EAEAC,WAAAA,GAAc;AACZ,IAAA,OAAOC,EAAEC,MAAAA,CAAO;AACdC,MAAAA,IAAAA,EAAMF,EAAEG,MAAAA,EAAM,CAAGC,IAAI,CAAA,CAAA,CAAGC,SAAS,iDAAA;KACnC,CAAA;AACF,EAAA;EAEA,OAAgBC,QAAAA,GAAW,IAAIC,cAAAA,CAAe;AAC5CC,IAAAA,MAAAA,EAAQR,EAAEC,MAAAA,CAAO;AACfC,MAAAA,IAAAA,EAAMF,EAAEG,MAAAA;KACV,CAAA;IACAG,QAAAA,EAAU,CAAA;;;GAGZ,CAAA;EAEA,MAAgBG,IAAAA,CACdpB,KAAAA,EACAqB,QAAAA,EACAC,GAAAA,EACA;AACA,IAAA,MAAMC,MAAAA,GAASC,OAAAA,CAAQF,GAAAA,CAAIG,OAAAA,EAAS;AAAC5B,MAAAA,IAAAA,CAAK6B,WAAAA,CAAYC;AAAO,KAAA,CAAA;AAC7D,IAAA,IAAI,CAACJ,MAAAA,EAAQ;AACX,MAAA,MAAM,IAAIK,SAAAA,CAAU,CAAA,6BAAA,CAAA,EAAiC,EAAA,EAAI;QACvDC,OAAAA,EAAS,IAAA;QACTC,WAAAA,EAAa;OACf,CAAA;AACF,IAAA;AAEA,IAAA,MAAMb,QAAAA,GAAW,IAAA,CAAKc,OAAAA,EAASd,QAAAA,IAAYrB,OAAAA,CAAQqB,QAAAA;AACnD,IAAA,MAAMe,MAAAA,GAAS,MAAM,IAAA,CAAKhC,KAAAA,CAAMiC,IAAIC,MAAAA,CAAO;MACzCC,QAAAA,EAAU;QACR,IAAIC,aAAAA,CACFnB,SAASoB,MAAAA,CAAO;AACdxB,UAAAA,IAAAA,EAAMb,KAAAA,CAAMa;AACd,SAAA,CAAA,CAAA;AAECU,QAAAA,GAAAA,MAAAA,CAAOY,SAAS5B,MAAAA,CAAO,CAAC+B,QAAQA,GAAAA,CAAIC,IAAAA,KAASC,KAAKC,MAAM,CAAA;QAC3D,IAAIC,WAAAA,CACFzB,SAASoB,MAAAA,CAAO;AACdxB,UAAAA,IAAAA,EAAMb,KAAAA,CAAMa;AACd,SAAA,CAAA;;KAGN,CAAA;AAEA,IAAA,OAAO,IAAI8B,gBAAAA,CAAiBX,MAAAA,CAAOY,cAAAA,EAAc,CAAA;AACnD,EAAA;AACF","file":"llm.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport {\n  BaseToolOptions,\n  BaseToolRunOptions,\n  StringToolOutput,\n  Tool,\n  ToolEmitter,\n  ToolError,\n  ToolInput,\n} from \"@/tools/base.js\";\nimport { z } from \"zod\";\nimport { GetRunContext } from \"@/context.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { PromptTemplate } from \"@/template.js\";\nimport { getProp } from \"@/internals/helpers/object.js\";\nimport type { BaseMemory } from \"@/memory/base.js\";\nimport { toCamelCase } from \"remeda\";\nimport { Role, SystemMessage, UserMessage } from \"@/backend/message.js\";\nimport { ChatModel } from \"@/backend/chat.js\";\n\nexport interface LLMToolInput extends BaseToolOptions {\n  llm: ChatModel;\n  name?: string;\n  description?: string;\n  template?: typeof LLMTool.template;\n}\n\nexport class LLMTool extends Tool<StringToolOutput, LLMToolInput> {\n  name = \"LLM\";\n  description =\n    \"Uses expert LLM to work with data in the existing conversation (classification, entity extraction, summarization, ...)\";\n  declare readonly emitter: ToolEmitter<ToolInput<this>, StringToolOutput>;\n\n  constructor(protected readonly input: LLMToolInput) {\n    super(input);\n    this.name = input?.name || this.name;\n    this.description = input?.description || this.description;\n    this.emitter = Emitter.root.child({\n      namespace: [\"tool\", \"llm\", toCamelCase(input?.name ?? \"\")].filter(Boolean),\n      creator: this,\n    });\n  }\n\n  inputSchema() {\n    return z.object({\n      task: z.string().min(1).describe(\"A clearly defined task for the LLM to complete.\"),\n    });\n  }\n\n  static readonly template = new PromptTemplate({\n    schema: z.object({\n      task: z.string(),\n    }),\n    template: `Use common sense and the information contained in the conversation up to this point to complete the following task. Do not follow any previously used formats or structures.\n\nThe Task: {{task}}`,\n  });\n\n  protected async _run(\n    input: ToolInput<this>,\n    _options: Partial<BaseToolRunOptions>,\n    run: GetRunContext<this>,\n  ) {\n    const memory = getProp(run.context, [Tool.contextKeys.Memory]) as BaseMemory;\n    if (!memory) {\n      throw new ToolError(`No context has been provided!`, [], {\n        isFatal: true,\n        isRetryable: false,\n      });\n    }\n\n    const template = this.options?.template ?? LLMTool.template;\n    const output = await this.input.llm.create({\n      messages: [\n        new SystemMessage(\n          template.render({\n            task: input.task,\n          }),\n        ),\n        ...memory.messages.filter((msg) => msg.role !== Role.SYSTEM),\n        new UserMessage(\n          template.render({\n            task: input.task,\n          }),\n        ),\n      ],\n    });\n\n    return new StringToolOutput(output.getTextContent());\n  }\n}\n"]}

package/dist/tools/mcp.cjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../../src/tools/mcp.ts"],"names":["MCPToolOutput","JSONToolOutput","MCPTool","Tool","name","description","emitter","client","tool","options","Emitter","root","child","namespace","toCamelCase","creator","inputSchema","_run","input","_options","run","result","callTool","arguments","undefined","signal","fromClient","tools","paginate","size","Infinity","handler","cursor","nextCursor","listTools","data","map"],"mappings":";;;;;;;;;AAmBO,MAAMA,sBAAsBC,uBAAAA,CAAAA;EAnBnC;;;AAmBwD;AAEjD,MAAMC,gBAAgBC,aAAAA,CAAAA;EArB7B;;;AAsBkBC,EAAAA,IAAAA;AACAC,EAAAA,WAAAA;AACAC,EAAAA,OAAAA;AAEAC,EAAAA,MAAAA;AACCC,EAAAA,IAAAA;AAEjB,EAAA,WAAA,CAAmB,EAAED,MAAAA,EAAQC,IAAAA,EAAM,GAAGC,SAAAA,EAAyB;AAC7D,IAAA,KAAA,CAAMA,OAAAA,CAAAA;AACN,IAAA,IAAA,CAAKF,MAAAA,GAASA,MAAAA;AACd,IAAA,IAAA,CAAKC,IAAAA,GAAOA,IAAAA;AACZ,IAAA,IAAA,CAAKJ,OAAOI,IAAAA,CAAKJ,IAAAA;AACjB,IAAA,IAAA,CAAKC,WAAAA,GACHG,KAAKH,WAAAA,IAAe,sEAAA;AACtB,IAAA,IAAA,CAAKC,OAAAA,GAAUI,mBAAAA,CAAQC,IAAAA,CAAKC,KAAAA,CAAM;MAChCC,SAAAA,EAAW;AAAC,QAAA,MAAA;AAAQ,QAAA,KAAA;AAAOC,QAAAA,kBAAAA,CAAY,KAAKV,IAAI;;MAChDW,OAAAA,EAAS;KACX,CAAA;AACF;EAEOC,WAAAA,GAAc;AACnB,IAAA,OAAO,KAAKR,IAAAA,CAAKQ,WAAAA;AACnB;EAEA,MAAgBC,IAAAA,CACdC,KAAAA,EACAC,QAAAA,EACAC,GAAAA,EACA;AACA,IAAA,MAAMC,MAAAA,GAAS,MAAM,IAAA,CAAKd,MAAAA,CAAOe,QAAAA,CAAS;AAAElB,MAAAA,IAAAA,EAAM,IAAA,CAAKA,IAAAA;MAAMmB,SAAAA,EAAWL;AAAM,KAAA,EAAGM,MAAAA,EAAW;AAC1FC,MAAAA,MAAAA,EAAQL,GAAAA,CAAIK;KACd,CAAA;AACA,IAAA,OAAO,IAAIzB,cAAcqB,MAAAA,CAAAA;AAC3B;AAEA,EAAA,aAAoBK,WAAWnB,MAAAA,EAAuC;AACpE,IAAA,MAAMoB,KAAAA,GAAQ,MAAMC,qBAAAA,CAAS;MAC3BC,IAAAA,EAAMC,QAAAA;MACNC,OAAAA,kBAAS,MAAA,CAAA,OAAO,EAAEC,MAAAA,EAAM,KAAuB;AAC7C,QAAA,MAAM,EAAEL,KAAAA,EAAAA,MAAAA,EAAOM,YAAU,GAAK,MAAM1B,OAAO2B,SAAAA,CAAU;AAAEF,UAAAA;SAAO,CAAA;AAC9D,QAAA,OAAO;UAAEG,IAAAA,EAAMR,MAAAA;AAAOM,UAAAA;AAAW,SAAA;OACnC,EAHS,SAAA;KAIX,CAAA;AACA,IAAA,OAAON,KAAAA,CAAMS,GAAAA,CAAI,CAAC5B,IAAAA,KAAS,IAAIN,OAAAA,CAAQ;AAAEK,MAAAA,MAAAA;AAAQC,MAAAA;AAAK,KAAA,CAAA,CAAA;AACxD;AACF","file":"mcp.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { BaseToolRunOptions, ToolEmitter, ToolInput, JSONToolOutput, Tool } from \"@/tools/base.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { GetRunContext } from \"@/context.js\";\nimport { Client as MCPClient } from \"@modelcontextprotocol/sdk/client/index.js\";\nimport { ListToolsResult } from \"@modelcontextprotocol/sdk/types.js\";\nimport { SchemaObject } from \"ajv\";\nimport { paginate } from \"@/internals/helpers/paginate.js\";\nimport { toCamelCase } from \"remeda\";\n\nexport interface MCPToolInput {\n  client: MCPClient;\n  tool: ListToolsResult[\"tools\"][number];\n}\n\nexport class MCPToolOutput extends JSONToolOutput<any> {}\n\nexport class MCPTool extends Tool<MCPToolOutput> {\n  public readonly name: string;\n  public readonly description: string;\n  public readonly emitter: ToolEmitter<ToolInput<this>, MCPToolOutput>;\n\n  public readonly client: MCPClient;\n  private readonly tool: ListToolsResult[\"tools\"][number];\n\n  public constructor({ client, tool, ...options }: MCPToolInput) {\n    super(options);\n    this.client = client;\n    this.tool = tool;\n    this.name = tool.name;\n    this.description =\n      tool.description ?? \"No available description, use the tool based on its name and schema.\";\n    this.emitter = Emitter.root.child({\n      namespace: [\"tool\", \"mcp\", toCamelCase(this.name)],\n      creator: this,\n    });\n  }\n\n  public inputSchema() {\n    return this.tool.inputSchema as SchemaObject;\n  }\n\n  protected async _run(\n    input: ToolInput<this>,\n    _options: BaseToolRunOptions,\n    run: GetRunContext<typeof this>,\n  ) {\n    const result = await this.client.callTool({ name: this.name, arguments: input }, undefined, {\n      signal: run.signal,\n    });\n    return new MCPToolOutput(result);\n  }\n\n  public static async fromClient(client: MCPClient): Promise<MCPTool[]> {\n    const tools = await paginate({\n      size: Infinity,\n      handler: async ({ cursor }: { cursor?: string }) => {\n        const { tools, nextCursor } = await client.listTools({ cursor });\n        return { data: tools, nextCursor } as const;\n      },\n    });\n    return tools.map((tool) => new MCPTool({ client, tool }));\n  }\n}\n"]}
+{"version":3,"sources":["../../src/tools/mcp.ts"],"names":["MCPToolOutput","JSONToolOutput","MCPTool","Tool","name","description","emitter","client","tool","options","Emitter","root","child","namespace","toCamelCase","creator","inputSchema","_run","input","_options","run","result","callTool","arguments","undefined","signal","fromClient","tools","paginate","size","Infinity","handler","cursor","nextCursor","listTools","data","map"],"mappings":";;;;;;;;;AAmBO,MAAMA,sBAAsBC,uBAAAA,CAAAA;EAnBnC;;;AAmBwD;AAEjD,MAAMC,gBAAgBC,aAAAA,CAAAA;EArB7B;;;AAsBkBC,EAAAA,IAAAA;AACAC,EAAAA,WAAAA;AACAC,EAAAA,OAAAA;AAEAC,EAAAA,MAAAA;AACCC,EAAAA,IAAAA;AAEjB,EAAA,WAAA,CAAmB,EAAED,MAAAA,EAAQC,IAAAA,EAAM,GAAGC,SAAAA,EAAyB;AAC7D,IAAA,KAAA,CAAMA,OAAAA,CAAAA;AACN,IAAA,IAAA,CAAKF,MAAAA,GAASA,MAAAA;AACd,IAAA,IAAA,CAAKC,IAAAA,GAAOA,IAAAA;AACZ,IAAA,IAAA,CAAKJ,OAAOI,IAAAA,CAAKJ,IAAAA;AACjB,IAAA,IAAA,CAAKC,WAAAA,GACHG,KAAKH,WAAAA,IAAe,sEAAA;AACtB,IAAA,IAAA,CAAKC,OAAAA,GAAUI,mBAAAA,CAAQC,IAAAA,CAAKC,KAAAA,CAAM;MAChCC,SAAAA,EAAW;AAAC,QAAA,MAAA;AAAQ,QAAA,KAAA;AAAOC,QAAAA,kBAAAA,CAAY,KAAKV,IAAI;;MAChDW,OAAAA,EAAS;KACX,CAAA;AACF,EAAA;EAEOC,WAAAA,GAAc;AACnB,IAAA,OAAO,KAAKR,IAAAA,CAAKQ,WAAAA;AACnB,EAAA;EAEA,MAAgBC,IAAAA,CACdC,KAAAA,EACAC,QAAAA,EACAC,GAAAA,EACA;AACA,IAAA,MAAMC,MAAAA,GAAS,MAAM,IAAA,CAAKd,MAAAA,CAAOe,QAAAA,CAAS;AAAElB,MAAAA,IAAAA,EAAM,IAAA,CAAKA,IAAAA;MAAMmB,SAAAA,EAAWL;AAAM,KAAA,EAAGM,MAAAA,EAAW;AAC1FC,MAAAA,MAAAA,EAAQL,GAAAA,CAAIK;KACd,CAAA;AACA,IAAA,OAAO,IAAIzB,cAAcqB,MAAAA,CAAAA;AAC3B,EAAA;AAEA,EAAA,aAAoBK,WAAWnB,MAAAA,EAAuC;AACpE,IAAA,MAAMoB,KAAAA,GAAQ,MAAMC,qBAAAA,CAAS;MAC3BC,IAAAA,EAAMC,QAAAA;MACNC,OAAAA,kBAAS,MAAA,CAAA,OAAO,EAAEC,MAAAA,EAAM,KAAuB;AAC7C,QAAA,MAAM,EAAEL,KAAAA,EAAAA,MAAAA,EAAOM,YAAU,GAAK,MAAM1B,OAAO2B,SAAAA,CAAU;AAAEF,UAAAA;SAAO,CAAA;AAC9D,QAAA,OAAO;UAAEG,IAAAA,EAAMR,MAAAA;AAAOM,UAAAA;AAAW,SAAA;MACnC,CAAA,EAHS,SAAA;KAIX,CAAA;AACA,IAAA,OAAON,KAAAA,CAAMS,GAAAA,CAAI,CAAC5B,IAAAA,KAAS,IAAIN,OAAAA,CAAQ;AAAEK,MAAAA,MAAAA;AAAQC,MAAAA;AAAK,KAAA,CAAA,CAAA;AACxD,EAAA;AACF","file":"mcp.cjs","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n * SPDX-License-Identifier: Apache-2.0\n */\n\nimport { BaseToolRunOptions, ToolEmitter, ToolInput, JSONToolOutput, Tool } from \"@/tools/base.js\";\nimport { Emitter } from \"@/emitter/emitter.js\";\nimport { GetRunContext } from \"@/context.js\";\nimport { Client as MCPClient } from \"@modelcontextprotocol/sdk/client/index.js\";\nimport { ListToolsResult } from \"@modelcontextprotocol/sdk/types.js\";\nimport { SchemaObject } from \"ajv\";\nimport { paginate } from \"@/internals/helpers/paginate.js\";\nimport { toCamelCase } from \"remeda\";\n\nexport interface MCPToolInput {\n  client: MCPClient;\n  tool: ListToolsResult[\"tools\"][number];\n}\n\nexport class MCPToolOutput extends JSONToolOutput<any> {}\n\nexport class MCPTool extends Tool<MCPToolOutput> {\n  public readonly name: string;\n  public readonly description: string;\n  public readonly emitter: ToolEmitter<ToolInput<this>, MCPToolOutput>;\n\n  public readonly client: MCPClient;\n  private readonly tool: ListToolsResult[\"tools\"][number];\n\n  public constructor({ client, tool, ...options }: MCPToolInput) {\n    super(options);\n    this.client = client;\n    this.tool = tool;\n    this.name = tool.name;\n    this.description =\n      tool.description ?? \"No available description, use the tool based on its name and schema.\";\n    this.emitter = Emitter.root.child({\n      namespace: [\"tool\", \"mcp\", toCamelCase(this.name)],\n      creator: this,\n    });\n  }\n\n  public inputSchema() {\n    return this.tool.inputSchema as SchemaObject;\n  }\n\n  protected async _run(\n    input: ToolInput<this>,\n    _options: BaseToolRunOptions,\n    run: GetRunContext<typeof this>,\n  ) {\n    const result = await this.client.callTool({ name: this.name, arguments: input }, undefined, {\n      signal: run.signal,\n    });\n    return new MCPToolOutput(result);\n  }\n\n  public static async fromClient(client: MCPClient): Promise<MCPTool[]> {\n    const tools = await paginate({\n      size: Infinity,\n      handler: async ({ cursor }: { cursor?: string }) => {\n        const { tools, nextCursor } = await client.listTools({ cursor });\n        return { data: tools, nextCursor } as const;\n      },\n    });\n    return tools.map((tool) => new MCPTool({ client, tool }));\n  }\n}\n"]}