beeai-framework 0.1.20 → 0.1.21

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

@@ -1 +1 @@
-{"version":3,"sources":["../../../src/tools/database/milvus.ts"],"names":["MilvusAction","MilvusDatabaseTool","Tool","name","description","inputSchema","z","object","action","nativeEnum","describe","collectionName","string","optional","vector","array","number","vectors","topK","coerce","int","min","max","default","filter","metadata","record","any","ids","or","searchOutput","emitter","Emitter","root","child","namespace","creator","validateInput","schema","input","ToolInputValidationError","register","client","MilvusClient","options","connection","listCollections","error","ToolError","isRetryable","isFatal","_run","_options","collections","JSONToolOutput","collectionInfo","getCollectionInfo","searchResults","search","insertResults","insert","deleteResults","delete","response","data","Array","isArray","map","collection","describeCollection","collection_name","fields_data","index","limit","output_fields","results","join"],"mappings":";;;;;;;;;;AAGC,SAAA,YAAA,CAAA,UAAA,EAAA,MAAA,EAAA,GAAA,EAAA,IAAA,EAAA;;;;;;AAAA,MAAA,CAAA,YAAA,EAAA,cAAA,CAAA;;;;;AAiCM,IAAKA,YAAAA,4BAAAA,aAAAA,EAAAA;;;;;;AAAAA,EAAAA,OAAAA,aAAAA;;AAQL,MAAMC,2BAA2BC,aAAAA,CAAAA;EA5CxC;;;EAgDEC,IAAAA,GAAO,oBAAA;EAEPC,WAAAA,GAAc,CAAA;QACV,iBAAA,CAAA;QACA,mBAAA,CAAA;QACA,QAAA,CAAA;QACA,QAAA,CAAA;QACA,QAAA,CAAA,kCAAA,CAAA;EAEJC,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,iDAAA,EAAkD,QAAA,CAAA,CAAuB,CAAA;AACrFO,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,CAAuB,CAAA;MACxEQ,IAAAA,EAAMZ,KAAAA,CAAEa,OACLH,MAAAA,EAAM,CACNI,KAAG,CACHC,GAAAA,CAAI,CAAA,CAAA,CACJC,GAAAA,CAAI,GAAA,CAAA,CACJC,OAAAA,CAAQ,EAAA,CAAA,CACRV,QAAAA,GACAH,QAAAA,CACC,CAAA,8CAAA,EAA+C,QAAA,CAAA,iBAAA,CAAwC,CAAA;MAE3Fc,MAAAA,EAAQlB,KAAAA,CACLM,QAAM,CACNC,QAAAA,GACAH,QAAAA,CAAS,CAAA,+BAAA,EAAgC,QAAA,CAAA,CAAuB,CAAA;AACnEe,MAAAA,QAAAA,EAAUnB,KAAAA,CACPoB,MAAAA,CAAOpB,KAAAA,CAAEM,MAAAA,IAAUN,KAAAA,CAAEqB,GAAAA,EAAG,CAAA,CACxBd,QAAAA,EAAQ,CACRH,QAAAA,CAAS,CAAA,+CAAA,EAAgD,QAAA,CAAA,CAAuB,CAAA;AACnFkB,MAAAA,GAAAA,EAAKtB,MACFS,KAAAA,CAAMT,KAAAA,CAAEM,MAAAA,EAAM,CAAGiB,GAAGvB,KAAAA,CAAEU,MAAAA,EAAM,CAAA,EAC5BH,QAAAA,EAAQ,CACRH,QAAAA,CAAS,CAAA,2BAAA,EAA4B,QAAA,CAAA,CAAuB,CAAA;MAC/DoB,YAAAA,EAAcxB,KAAAA,CACXS,KAAAA,CAAMT,KAAAA,CAAEM,MAAAA,EAAM,CAAA,CACdC,QAAAA,EAAQ,CACRH,QAAAA,CAAS,CAAA,uCAAA,EAAwC,QAAA,CAAA,CAAuB;KAC7E,CAAA;AACF;EAEgBqB,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,CAAM/B,MAAAA,KAAM,mBAAA,IAAuC,CAAC+B,MAAM5B,cAAAA,EAAgB;AAC5E,MAAA,MAAM,IAAI6B,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,CAAM/B,WAAM,QAAA,KAA6B,CAAC+B,MAAM5B,cAAAA,IAAkB,CAAC4B,MAAMzB,MAAAA,CAAAA,EAAS;AACpF,MAAA,MAAM,IAAI0B,iCAAAA,CACR,CAAA,iDAAA,EAAkD,QAAA,CAAA,QAAA,CAA+B,CAAA;AAErF;AACA,IAAA,IAAID,KAAAA,CAAM/B,WAAM,QAAA,KAA6B,CAAC+B,MAAM5B,cAAAA,IAAkB,CAAC4B,MAAMtB,OAAAA,CAAAA,EAAU;AACrF,MAAA,MAAM,IAAIuB,iCAAAA,CACR,CAAA,kDAAA,EAAmD,QAAA,CAAA,QAAA,CAA+B,CAAA;AAEtF;AACF;EAEA;AACE,IAAA,IAAA,CAAKC,QAAAA,EAAQ;AACf;AAEA,EAAA,MACgBC,MAAAA,GAAgC;AAC9C,IAAA,IAAI;AACF,MAAA,MAAMA,MAAAA,GAAS,IAAIC,2BAAAA,CAAa,IAAA,CAAKC,QAAQC,UAAU,CAAA;AAEvD,MAAA,MAAMH,OAAOI,eAAAA,EAAe;AAC5B,MAAA,OAAOJ,MAAAA;AACT,KAAA,CAAA,OAASK,KAAAA,EAAO;AACd,MAAA,MAAM,IAAIC,mBAAU,CAAA,4BAAA,CAAA,EAAgC;AAACD,QAAAA;AAAQ,OAAA,EAAA;QAC3DE,WAAAA,EAAa,KAAA;QACbC,OAAAA,EAAS;OACX,CAAA;AACF;AACF;EAEA,MAAgBC,IAAAA,CACdZ,OACAa,QAAAA,EAC8B;AAC9B,IAAA,QAAQb,MAAM/B,MAAAA;AACZ,MAAA,KAAA,iBAAA,EAAmC;AACjC,QAAA,MAAM6C,WAAAA,GAAc,MAAM,IAAA,CAAKP,eAAAA,EAAe;AAC9C,QAAA,OAAO,IAAIQ,wBAAeD,WAAAA,CAAAA;AAC5B;AAEA,MAAA,KAAA,mBAAA,EAAqC;AACnC,QAAA,IAAI,CAACd,MAAM5B,cAAAA,EAAgB;AACzB,UAAA,MAAM,IAAIqC,mBAAU,mEAAA,CAAA;AACtB;AACA,QAAA,MAAMO,cAAAA,GAAiB,MAAM,IAAA,CAAKC,iBAAAA,CAAkBjB,MAAM5B,cAAc,CAAA;AACxE,QAAA,OAAO,IAAI2C,wBAAeC,cAAAA,CAAAA;AAC5B;AAEA,MAAA,KAAA,QAAA,EAA0B;AACxB,QAAA,IAAI,CAAChB,KAAAA,CAAM5B,cAAAA,IAAkB,CAAC4B,MAAMzB,MAAAA,EAAQ;AAC1C,UAAA,MAAM,IAAIkC,mBAAU,oEAAA,CAAA;AACtB;AACA,QAAA,MAAMS,aAAAA,GAAgB,MAAM,IAAA,CAAKC,MAAAA,CAAOnB,KAAAA,CAAAA;AACxC,QAAA,OAAO,IAAIe,wBAAeG,aAAAA,CAAAA;AAC5B;AAEA,MAAA,KAAA,QAAA,EAA0B;AACxB,QAAA,IAAI,CAAClB,KAAAA,CAAM5B,cAAAA,IAAkB,CAAC4B,MAAMtB,OAAAA,EAAS;AAC3C,UAAA,MAAM,IAAI+B,mBACR,qEAAA,CAAA;AAEJ;AACA,QAAA,MAAMW,aAAAA,GAAgB,MAAM,IAAA,CAAKC,MAAAA,CAAOrB,KAAAA,CAAAA;AACxC,QAAA,OAAO,IAAIe,wBAAeK,aAAAA,CAAAA;AAC5B;AAEA,MAAA,KAAA,QAAA,EAA0B;AACxB,QAAA,IAAI,CAACpB,KAAAA,CAAM5B,cAAAA,IAAkB,CAAC4B,MAAMX,GAAAA,EAAK;AACvC,UAAA,MAAM,IAAIoB,mBAAU,+DAAA,CAAA;AACtB;AACA,QAAA,MAAMa,aAAAA,GAAgB,MAAM,IAAA,CAAKC,MAAAA,CAAOvB,KAAAA,CAAAA;AACxC,QAAA,OAAO,IAAIe,wBAAeO,aAAAA,CAAAA;AAC5B;MAEA,SAAS;AACP,QAAA,MAAM,IAAIb,kBAAAA,CAAU,CAAA,0BAAA,EAA6BT,KAAAA,CAAM/B,MAAM,CAAA,CAAE,CAAA;AACjE;AACF;AACF;AAEA,EAAA,MAAgBsC,eAAAA,GAAqC;AACnD,IAAA,IAAI;AACF,MAAA,MAAMJ,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAAA,EAAM;AAChC,MAAA,MAAMqB,QAAAA,GAAW,MAAMrB,MAAAA,CAAOI,eAAAA,CAAgB,EAAC,CAAA;AAE/C,MAAA,IAAIiB,YAAYA,QAAAA,CAASC,IAAAA,IAAQC,MAAMC,OAAAA,CAAQH,QAAAA,CAASC,IAAI,CAAA,EAAG;AAC7D,QAAA,OAAOD,SAASC,IAAAA,CAAKG,GAAAA,CAAI,CAACC,UAAAA,KAAeA,WAAWjE,IAAI,CAAA;OAC1D,MAAO;AACL,QAAA,OAAO,EAAA;AACT;AACF,KAAA,CAAA,OAAS4C,KAAAA,EAAO;AACd,MAAA,MAAM,IAAIC,kBAAAA,CAAU,CAAA,wCAAA,EAA2CD,KAAAA,CAAAA,CAAO,CAAA;AACxE;AACF;AAEA,EAAA,MAAgBS,kBAAkB7C,cAAAA,EAAsC;AACtE,IAAA,MAAM+B,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAAA,EAAM;AAChC,IAAA,MAAMqB,QAAAA,GAAWrB,OAAO2B,kBAAAA,CAAmB;MAAEC,eAAAA,EAAiB3D;KAAe,CAAA;AAC7E,IAAA,OAAOoD,QAAAA;AACT;AAEA,EAAA,MAAgBH,OAAOrB,KAAAA,EAAsC;AAC3D,IAAA,MAAMG,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAAA,EAAM;AAChC,IAAA,MAAMqB,QAAAA,GAAW,MAAMrB,MAAAA,CAAOkB,MAAAA,CAAO;AACnCU,MAAAA,eAAAA,EAAiB/B,KAAAA,CAAM5B,cAAAA;AACvB4D,MAAAA,WAAAA,EAAahC,KAAAA,CAAMtB,OAAAA,CAASkD,GAAAA,CAAI,CAACrD,QAAQ0D,KAAAA,MAAW;AAClD1D,QAAAA,MAAAA;QACA,GAAGyB,KAAAA,CAAMd,WAAW+C,KAAAA;OACtB,CAAA;KACF,CAAA;AACA,IAAA,OAAOT,QAAAA;AACT;AAEA,EAAA,MAAgBL,OAAOnB,KAAAA,EAAsC;AAC3D,IAAA,MAAMG,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAAA,EAAM;AAChC,IAAA,MAAMqB,QAAAA,GAAW,MAAMrB,MAAAA,CAAOgB,MAAAA,CAAO;AACnCY,MAAAA,eAAAA,EAAiB/B,KAAAA,CAAM5B,cAAAA;AACvBG,MAAAA,MAAAA,EAAQyB,KAAAA,CAAMzB,MAAAA;AACd2D,MAAAA,KAAAA,EAAOlC,MAAMrB,IAAAA,IAAQ,EAAA;AACrBM,MAAAA,MAAAA,EAAQe,KAAAA,CAAMf,MAAAA;AACdkD,MAAAA,aAAAA,EAAenC,KAAAA,CAAMT;KACvB,CAAA;AACA,IAAA,OAAOiC,QAAAA,CAASY,OAAAA;AAClB;AAEA,EAAA,MAAgBb,OAAOvB,KAAAA,EAAsC;AAC3D,IAAA,MAAMG,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAAA,EAAM;AAChC,IAAA,MAAMqB,QAAAA,GAAWrB,OAAOoB,MAAAA,CAAO;AAC7BQ,MAAAA,eAAAA,EAAiB/B,KAAAA,CAAM5B,cAAAA;AACvBa,MAAAA,MAAAA,EAAQ,CAAA,OAAA,EAAUe,KAAAA,CAAMX,GAAAA,EAAKgD,IAAAA,CAAK,GAAA,CAAA,CAAA,CAAA;KACpC,CAAA;AACA,IAAA,OAAOb,QAAAA;AACT;AACF","file":"milvus.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 {\n  ClientConfig,\n  MilvusClient,\n  SearchRes,\n  ShowCollectionsResponse,\n  DescribeCollectionResponse,\n} from \"@zilliz/milvus2-sdk-node\";\nimport { z } from \"zod\";\nimport { Emitter } from \"@/emitter/emitter.js\";\n\nexport interface MilvusToolOptions extends BaseToolOptions {\n  connection: ClientConfig;\n}\n\nexport type MilvusSearchToolResult =\n  | ShowCollectionsResponse\n  | DescribeCollectionResponse\n  | SearchRes;\n\nexport enum MilvusAction {\n  ListCollections = \"ListCollections\",\n  GetCollectionInfo = \"GetCollectionInfo\",\n  Search = \"Search\",\n  Insert = \"Insert\",\n  Delete = \"Delete\",\n}\n\nexport class MilvusDatabaseTool extends Tool<\n  JSONToolOutput<MilvusSearchToolResult>,\n  MilvusToolOptions\n> {\n  name = \"MilvusDatabaseTool\";\n\n  description = `Can query data from a Milvus database. IMPORTANT: strictly follow this order of actions:\n   1. ${MilvusAction.ListCollections} - List all the Milvus collections\n   2. ${MilvusAction.GetCollectionInfo} - Get information about into a Milvus collection\n   3. ${MilvusAction.Insert} - Insert data into a Milvus collection\n   3. ${MilvusAction.Search} - Perform search on a Milvus collection\n   4. ${MilvusAction.Delete} - Delete from a Milvus collection`;\n\n  inputSchema() {\n    return z.object({\n      action: z\n        .nativeEnum(MilvusAction)\n        .describe(\n          `The action to perform. ${MilvusAction.ListCollections} lists all collections, ${MilvusAction.GetCollectionInfo} fetches details for a specified collection, ${MilvusAction.Search} executes a vector search, ${MilvusAction.Insert} inserts new vectors, and ${MilvusAction.Delete} removes vectors.`,\n        ),\n      collectionName: z\n        .string()\n        .optional()\n        .describe(\n          `The name of the collection to query, required for ${MilvusAction.GetCollectionInfo}, ${MilvusAction.Search}, ${MilvusAction.Insert}, and ${MilvusAction.Delete}`,\n        ),\n      vector: z\n        .array(z.number())\n        .optional()\n        .describe(`The vector to search for or insert, required for ${MilvusAction.Search}`),\n      vectors: z\n        .array(z.array(z.number()))\n        .optional()\n        .describe(`The vectors to insert, required for ${MilvusAction.Insert}`),\n      topK: z.coerce\n        .number()\n        .int()\n        .min(1)\n        .max(1000)\n        .default(10)\n        .optional()\n        .describe(\n          `The number of nearest neighbors to return for ${MilvusAction.Search}. Maximum is 1000`,\n        ),\n      filter: z\n        .string()\n        .optional()\n        .describe(`Optional filter expression for ${MilvusAction.Search}`),\n      metadata: z\n        .record(z.string(), z.any())\n        .optional()\n        .describe(`Additional metadata to insert with vectors for ${MilvusAction.Insert}`),\n      ids: z\n        .array(z.string().or(z.number()))\n        .optional()\n        .describe(`Array of IDs to delete for ${MilvusAction.Delete}`),\n      searchOutput: z\n        .array(z.string())\n        .optional()\n        .describe(`Fields to return in search results for ${MilvusAction.Search}`),\n    });\n  }\n\n  public readonly emitter: ToolEmitter<ToolInput<this>, JSONToolOutput<MilvusSearchToolResult>> =\n    Emitter.root.child({\n      namespace: [\"tool\", \"database\", \"milvus\"],\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 === MilvusAction.GetCollectionInfo && !input.collectionName) {\n      throw new ToolInputValidationError(\n        `Collection name is required for ${MilvusAction.GetCollectionInfo}, ${MilvusAction.Search}, ${MilvusAction.Insert}, and ${MilvusAction.Delete} actions.`,\n      );\n    }\n    if (input.action === MilvusAction.Search && (!input.collectionName || !input.vector)) {\n      throw new ToolInputValidationError(\n        `Both collection name and vector are required for ${MilvusAction.Search} action.`,\n      );\n    }\n    if (input.action === MilvusAction.Insert && (!input.collectionName || !input.vectors)) {\n      throw new ToolInputValidationError(\n        `Both collection name and vectors are required for ${MilvusAction.Insert} action.`,\n      );\n    }\n  }\n\n  static {\n    this.register();\n  }\n\n  @Cache()\n  protected async client(): Promise<MilvusClient> {\n    try {\n      const client = new MilvusClient(this.options.connection);\n      // operation expected to fail if connection is not established correctly.\n      await client.listCollections();\n      return client;\n    } catch (error) {\n      throw new ToolError(`Unable to connect to Milvus.`, [error], {\n        isRetryable: false,\n        isFatal: true,\n      });\n    }\n  }\n\n  protected async _run(\n    input: ToolInput<this>,\n    _options: Partial<BaseToolRunOptions>,\n  ): Promise<JSONToolOutput<any>> {\n    switch (input.action) {\n      case MilvusAction.ListCollections: {\n        const collections = await this.listCollections();\n        return new JSONToolOutput(collections);\n      }\n\n      case MilvusAction.GetCollectionInfo: {\n        if (!input.collectionName) {\n          throw new ToolError(\"A collection name is required for Milvus GetCollectionInfo action\");\n        }\n        const collectionInfo = await this.getCollectionInfo(input.collectionName);\n        return new JSONToolOutput(collectionInfo);\n      }\n\n      case MilvusAction.Search: {\n        if (!input.collectionName || !input.vector) {\n          throw new ToolError(\"A collection name and vector are required for Milvus Search action\");\n        }\n        const searchResults = await this.search(input);\n        return new JSONToolOutput(searchResults);\n      }\n\n      case MilvusAction.Insert: {\n        if (!input.collectionName || !input.vectors) {\n          throw new ToolError(\n            \"A collection name and vectors are required for Milvus Insert action\",\n          );\n        }\n        const insertResults = await this.insert(input);\n        return new JSONToolOutput(insertResults);\n      }\n\n      case MilvusAction.Delete: {\n        if (!input.collectionName || !input.ids) {\n          throw new ToolError(\"Collection name and ids are required for Milvus 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.listCollections({});\n\n      if (response && response.data && Array.isArray(response.data)) {\n        return response.data.map((collection) => collection.name);\n      } else {\n        return [];\n      }\n    } catch (error) {\n      throw new ToolError(`Failed to list collections from Milvus: ${error}`);\n    }\n  }\n\n  protected async getCollectionInfo(collectionName: string): Promise<any> {\n    const client = await this.client();\n    const response = client.describeCollection({ collection_name: collectionName });\n    return response;\n  }\n\n  protected async insert(input: ToolInput<this>): Promise<any> {\n    const client = await this.client();\n    const response = await client.insert({\n      collection_name: input.collectionName as string,\n      fields_data: input.vectors!.map((vector, index) => ({\n        vector: vector,\n        ...input.metadata?.[index],\n      })),\n    });\n    return response;\n  }\n\n  protected async search(input: ToolInput<this>): Promise<any> {\n    const client = await this.client();\n    const response = await client.search({\n      collection_name: input.collectionName as string,\n      vector: input.vector,\n      limit: input.topK || 10,\n      filter: input.filter,\n      output_fields: input.searchOutput,\n    });\n    return response.results;\n  }\n\n  protected async delete(input: ToolInput<this>): Promise<any> {\n    const client = await this.client();\n    const response = client.delete({\n      collection_name: input.collectionName as string,\n      filter: `id in [${input.ids?.join(\",\")}]`,\n    });\n    return response;\n  }\n}\n"]}
+{"version":3,"sources":["../../../src/tools/database/milvus.ts"],"names":["MilvusAction","MilvusDatabaseTool","Tool","name","description","inputSchema","z","object","action","nativeEnum","describe","collectionName","string","optional","vector","array","number","vectors","topK","coerce","int","min","max","default","filter","metadata","record","any","ids","or","searchOutput","emitter","Emitter","root","child","namespace","creator","validateInput","schema","input","ToolInputValidationError","register","client","MilvusClient","options","connection","listCollections","error","ToolError","isRetryable","isFatal","_run","_options","collections","JSONToolOutput","collectionInfo","getCollectionInfo","searchResults","search","insertResults","insert","deleteResults","delete","response","data","Array","isArray","map","collection","describeCollection","collection_name","fields_data","index","limit","output_fields","results","join"],"mappings":";;;;;;;;;;;;;;;;;;;;;AAoCO,IAAKA,YAAAA,6BAAAA,aAAAA,EAAAA;;;;;;AAAAA,EAAAA,OAAAA,aAAAA;;AAQL,MAAMC,2BAA2BC,aAAAA,CAAAA;AAAAA,EAAAA;;;EAItCC,IAAAA,GAAO,oBAAA;EAEPC,WAAAA,GAAc,CAAA;QACV,iBAAA,CAAA;QACA,mBAAA,CAAA;QACA,QAAA,CAAA;QACA,QAAA,CAAA;QACA,QAAA,CAAA,kCAAA,CAAA;EAEJC,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,iDAAA,EAAkD,QAAA,CAAA,CAAuB,CAAA;AACrFO,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,CAAuB,CAAA;MACxEQ,IAAAA,EAAMZ,KAAAA,CAAEa,OACLH,MAAAA,EAAM,CACNI,KAAG,CACHC,GAAAA,CAAI,CAAA,CAAA,CACJC,GAAAA,CAAI,GAAA,CAAA,CACJC,OAAAA,CAAQ,EAAA,CAAA,CACRV,QAAAA,GACAH,QAAAA,CACC,CAAA,8CAAA,EAA+C,QAAA,CAAA,iBAAA,CAAwC,CAAA;MAE3Fc,MAAAA,EAAQlB,KAAAA,CACLM,QAAM,CACNC,QAAAA,GACAH,QAAAA,CAAS,CAAA,+BAAA,EAAgC,QAAA,CAAA,CAAuB,CAAA;AACnEe,MAAAA,QAAAA,EAAUnB,KAAAA,CACPoB,MAAAA,CAAOpB,KAAAA,CAAEM,MAAAA,IAAUN,KAAAA,CAAEqB,GAAAA,EAAG,CAAA,CACxBd,QAAAA,EAAQ,CACRH,QAAAA,CAAS,CAAA,+CAAA,EAAgD,QAAA,CAAA,CAAuB,CAAA;AACnFkB,MAAAA,GAAAA,EAAKtB,MACFS,KAAAA,CAAMT,KAAAA,CAAEM,MAAAA,EAAM,CAAGiB,GAAGvB,KAAAA,CAAEU,MAAAA,EAAM,CAAA,EAC5BH,QAAAA,EAAQ,CACRH,QAAAA,CAAS,CAAA,2BAAA,EAA4B,QAAA,CAAA,CAAuB,CAAA;MAC/DoB,YAAAA,EAAcxB,KAAAA,CACXS,KAAAA,CAAMT,KAAAA,CAAEM,MAAAA,EAAM,CAAA,CACdC,QAAAA,EAAQ,CACRH,QAAAA,CAAS,CAAA,uCAAA,EAAwC,QAAA,CAAA,CAAuB;KAC7E,CAAA;AACF;EAEgBqB,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,CAAM/B,MAAAA,KAAM,mBAAA,IAAuC,CAAC+B,MAAM5B,cAAAA,EAAgB;AAC5E,MAAA,MAAM,IAAI6B,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,CAAM/B,WAAM,QAAA,KAA6B,CAAC+B,MAAM5B,cAAAA,IAAkB,CAAC4B,MAAMzB,MAAAA,CAAAA,EAAS;AACpF,MAAA,MAAM,IAAI0B,iCAAAA,CACR,CAAA,iDAAA,EAAkD,QAAA,CAAA,QAAA,CAA+B,CAAA;AAErF;AACA,IAAA,IAAID,KAAAA,CAAM/B,WAAM,QAAA,KAA6B,CAAC+B,MAAM5B,cAAAA,IAAkB,CAAC4B,MAAMtB,OAAAA,CAAAA,EAAU;AACrF,MAAA,MAAM,IAAIuB,iCAAAA,CACR,CAAA,kDAAA,EAAmD,QAAA,CAAA,QAAA,CAA+B,CAAA;AAEtF;AACF;EAEA;AACE,IAAA,IAAA,CAAKC,QAAAA,EAAQ;AACf;AAEA,EAAA,MACgBC,MAAAA,GAAgC;AAC9C,IAAA,IAAI;AACF,MAAA,MAAMA,MAAAA,GAAS,IAAIC,2BAAAA,CAAa,IAAA,CAAKC,QAAQC,UAAU,CAAA;AAEvD,MAAA,MAAMH,OAAOI,eAAAA,EAAe;AAC5B,MAAA,OAAOJ,MAAAA;AACT,KAAA,CAAA,OAASK,KAAAA,EAAO;AACd,MAAA,MAAM,IAAIC,mBAAU,CAAA,4BAAA,CAAA,EAAgC;AAACD,QAAAA;AAAQ,OAAA,EAAA;QAC3DE,WAAAA,EAAa,KAAA;QACbC,OAAAA,EAAS;OACX,CAAA;AACF;AACF;EAEA,MAAgBC,IAAAA,CACdZ,OACAa,QAAAA,EAC8B;AAC9B,IAAA,QAAQb,MAAM/B,MAAAA;AACZ,MAAA,KAAA,iBAAA,EAAmC;AACjC,QAAA,MAAM6C,WAAAA,GAAc,MAAM,IAAA,CAAKP,eAAAA,EAAe;AAC9C,QAAA,OAAO,IAAIQ,wBAAeD,WAAAA,CAAAA;AAC5B;AAEA,MAAA,KAAA,mBAAA,EAAqC;AACnC,QAAA,IAAI,CAACd,MAAM5B,cAAAA,EAAgB;AACzB,UAAA,MAAM,IAAIqC,mBAAU,mEAAA,CAAA;AACtB;AACA,QAAA,MAAMO,cAAAA,GAAiB,MAAM,IAAA,CAAKC,iBAAAA,CAAkBjB,MAAM5B,cAAc,CAAA;AACxE,QAAA,OAAO,IAAI2C,wBAAeC,cAAAA,CAAAA;AAC5B;AAEA,MAAA,KAAA,QAAA,EAA0B;AACxB,QAAA,IAAI,CAAChB,KAAAA,CAAM5B,cAAAA,IAAkB,CAAC4B,MAAMzB,MAAAA,EAAQ;AAC1C,UAAA,MAAM,IAAIkC,mBAAU,oEAAA,CAAA;AACtB;AACA,QAAA,MAAMS,aAAAA,GAAgB,MAAM,IAAA,CAAKC,MAAAA,CAAOnB,KAAAA,CAAAA;AACxC,QAAA,OAAO,IAAIe,wBAAeG,aAAAA,CAAAA;AAC5B;AAEA,MAAA,KAAA,QAAA,EAA0B;AACxB,QAAA,IAAI,CAAClB,KAAAA,CAAM5B,cAAAA,IAAkB,CAAC4B,MAAMtB,OAAAA,EAAS;AAC3C,UAAA,MAAM,IAAI+B,mBACR,qEAAA,CAAA;AAEJ;AACA,QAAA,MAAMW,aAAAA,GAAgB,MAAM,IAAA,CAAKC,MAAAA,CAAOrB,KAAAA,CAAAA;AACxC,QAAA,OAAO,IAAIe,wBAAeK,aAAAA,CAAAA;AAC5B;AAEA,MAAA,KAAA,QAAA,EAA0B;AACxB,QAAA,IAAI,CAACpB,KAAAA,CAAM5B,cAAAA,IAAkB,CAAC4B,MAAMX,GAAAA,EAAK;AACvC,UAAA,MAAM,IAAIoB,mBAAU,+DAAA,CAAA;AACtB;AACA,QAAA,MAAMa,aAAAA,GAAgB,MAAM,IAAA,CAAKC,MAAAA,CAAOvB,KAAAA,CAAAA;AACxC,QAAA,OAAO,IAAIe,wBAAeO,aAAAA,CAAAA;AAC5B;MAEA,SAAS;AACP,QAAA,MAAM,IAAIb,kBAAAA,CAAU,CAAA,0BAAA,EAA6BT,KAAAA,CAAM/B,MAAM,CAAA,CAAE,CAAA;AACjE;AACF;AACF;AAEA,EAAA,MAAgBsC,eAAAA,GAAqC;AACnD,IAAA,IAAI;AACF,MAAA,MAAMJ,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAAA,EAAM;AAChC,MAAA,MAAMqB,QAAAA,GAAW,MAAMrB,MAAAA,CAAOI,eAAAA,CAAgB,EAAC,CAAA;AAE/C,MAAA,IAAIiB,YAAYA,QAAAA,CAASC,IAAAA,IAAQC,MAAMC,OAAAA,CAAQH,QAAAA,CAASC,IAAI,CAAA,EAAG;AAC7D,QAAA,OAAOD,SAASC,IAAAA,CAAKG,GAAAA,CAAI,CAACC,UAAAA,KAAeA,WAAWjE,IAAI,CAAA;OAC1D,MAAO;AACL,QAAA,OAAO,EAAA;AACT;AACF,KAAA,CAAA,OAAS4C,KAAAA,EAAO;AACd,MAAA,MAAM,IAAIC,kBAAAA,CAAU,CAAA,wCAAA,EAA2CD,KAAAA,CAAAA,CAAO,CAAA;AACxE;AACF;AAEA,EAAA,MAAgBS,kBAAkB7C,cAAAA,EAAsC;AACtE,IAAA,MAAM+B,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAAA,EAAM;AAChC,IAAA,MAAMqB,QAAAA,GAAWrB,OAAO2B,kBAAAA,CAAmB;MAAEC,eAAAA,EAAiB3D;KAAe,CAAA;AAC7E,IAAA,OAAOoD,QAAAA;AACT;AAEA,EAAA,MAAgBH,OAAOrB,KAAAA,EAAsC;AAC3D,IAAA,MAAMG,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAAA,EAAM;AAChC,IAAA,MAAMqB,QAAAA,GAAW,MAAMrB,MAAAA,CAAOkB,MAAAA,CAAO;AACnCU,MAAAA,eAAAA,EAAiB/B,KAAAA,CAAM5B,cAAAA;AACvB4D,MAAAA,WAAAA,EAAahC,KAAAA,CAAMtB,OAAAA,CAASkD,GAAAA,CAAI,CAACrD,QAAQ0D,KAAAA,MAAW;AAClD1D,QAAAA,MAAAA;QACA,GAAGyB,KAAAA,CAAMd,WAAW+C,KAAAA;OACtB,CAAA;KACF,CAAA;AACA,IAAA,OAAOT,QAAAA;AACT;AAEA,EAAA,MAAgBL,OAAOnB,KAAAA,EAAsC;AAC3D,IAAA,MAAMG,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAAA,EAAM;AAChC,IAAA,MAAMqB,QAAAA,GAAW,MAAMrB,MAAAA,CAAOgB,MAAAA,CAAO;AACnCY,MAAAA,eAAAA,EAAiB/B,KAAAA,CAAM5B,cAAAA;MACvBqD,IAAAA,EAAMzB,KAAAA,CAAMzB,UAAU,EAAA;AACtB2D,MAAAA,KAAAA,EAAOlC,MAAMrB,IAAAA,IAAQ,EAAA;AACrBM,MAAAA,MAAAA,EAAQe,KAAAA,CAAMf,MAAAA;AACdkD,MAAAA,aAAAA,EAAenC,KAAAA,CAAMT;KACvB,CAAA;AACA,IAAA,OAAOiC,QAAAA,CAASY,OAAAA;AAClB;AAEA,EAAA,MAAgBb,OAAOvB,KAAAA,EAAsC;AAC3D,IAAA,MAAMG,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAAA,EAAM;AAChC,IAAA,MAAMqB,QAAAA,GAAWrB,OAAOoB,MAAAA,CAAO;AAC7BQ,MAAAA,eAAAA,EAAiB/B,KAAAA,CAAM5B,cAAAA;AACvBa,MAAAA,MAAAA,EAAQ,CAAA,OAAA,EAAUe,KAAAA,CAAMX,GAAAA,EAAKgD,IAAAA,CAAK,GAAA,CAAA,CAAA,CAAA;KACpC,CAAA;AACA,IAAA,OAAOb,QAAAA;AACT;AACF","file":"milvus.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 {\n  ClientConfig,\n  MilvusClient,\n  SearchRes,\n  ShowCollectionsResponse,\n  DescribeCollectionResponse,\n} from \"@zilliz/milvus2-sdk-node\";\nimport { z } from \"zod\";\nimport { Emitter } from \"@/emitter/emitter.js\";\n\nexport interface MilvusToolOptions extends BaseToolOptions {\n  connection: ClientConfig;\n}\n\nexport type MilvusSearchToolResult =\n  | ShowCollectionsResponse\n  | DescribeCollectionResponse\n  | SearchRes;\n\nexport enum MilvusAction {\n  ListCollections = \"ListCollections\",\n  GetCollectionInfo = \"GetCollectionInfo\",\n  Search = \"Search\",\n  Insert = \"Insert\",\n  Delete = \"Delete\",\n}\n\nexport class MilvusDatabaseTool extends Tool<\n  JSONToolOutput<MilvusSearchToolResult>,\n  MilvusToolOptions\n> {\n  name = \"MilvusDatabaseTool\";\n\n  description = `Can query data from a Milvus database. IMPORTANT: strictly follow this order of actions:\n   1. ${MilvusAction.ListCollections} - List all the Milvus collections\n   2. ${MilvusAction.GetCollectionInfo} - Get information about into a Milvus collection\n   3. ${MilvusAction.Insert} - Insert data into a Milvus collection\n   3. ${MilvusAction.Search} - Perform search on a Milvus collection\n   4. ${MilvusAction.Delete} - Delete from a Milvus collection`;\n\n  inputSchema() {\n    return z.object({\n      action: z\n        .nativeEnum(MilvusAction)\n        .describe(\n          `The action to perform. ${MilvusAction.ListCollections} lists all collections, ${MilvusAction.GetCollectionInfo} fetches details for a specified collection, ${MilvusAction.Search} executes a vector search, ${MilvusAction.Insert} inserts new vectors, and ${MilvusAction.Delete} removes vectors.`,\n        ),\n      collectionName: z\n        .string()\n        .optional()\n        .describe(\n          `The name of the collection to query, required for ${MilvusAction.GetCollectionInfo}, ${MilvusAction.Search}, ${MilvusAction.Insert}, and ${MilvusAction.Delete}`,\n        ),\n      vector: z\n        .array(z.number())\n        .optional()\n        .describe(`The vector to search for or insert, required for ${MilvusAction.Search}`),\n      vectors: z\n        .array(z.array(z.number()))\n        .optional()\n        .describe(`The vectors to insert, required for ${MilvusAction.Insert}`),\n      topK: z.coerce\n        .number()\n        .int()\n        .min(1)\n        .max(1000)\n        .default(10)\n        .optional()\n        .describe(\n          `The number of nearest neighbors to return for ${MilvusAction.Search}. Maximum is 1000`,\n        ),\n      filter: z\n        .string()\n        .optional()\n        .describe(`Optional filter expression for ${MilvusAction.Search}`),\n      metadata: z\n        .record(z.string(), z.any())\n        .optional()\n        .describe(`Additional metadata to insert with vectors for ${MilvusAction.Insert}`),\n      ids: z\n        .array(z.string().or(z.number()))\n        .optional()\n        .describe(`Array of IDs to delete for ${MilvusAction.Delete}`),\n      searchOutput: z\n        .array(z.string())\n        .optional()\n        .describe(`Fields to return in search results for ${MilvusAction.Search}`),\n    });\n  }\n\n  public readonly emitter: ToolEmitter<ToolInput<this>, JSONToolOutput<MilvusSearchToolResult>> =\n    Emitter.root.child({\n      namespace: [\"tool\", \"database\", \"milvus\"],\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 === MilvusAction.GetCollectionInfo && !input.collectionName) {\n      throw new ToolInputValidationError(\n        `Collection name is required for ${MilvusAction.GetCollectionInfo}, ${MilvusAction.Search}, ${MilvusAction.Insert}, and ${MilvusAction.Delete} actions.`,\n      );\n    }\n    if (input.action === MilvusAction.Search && (!input.collectionName || !input.vector)) {\n      throw new ToolInputValidationError(\n        `Both collection name and vector are required for ${MilvusAction.Search} action.`,\n      );\n    }\n    if (input.action === MilvusAction.Insert && (!input.collectionName || !input.vectors)) {\n      throw new ToolInputValidationError(\n        `Both collection name and vectors are required for ${MilvusAction.Insert} action.`,\n      );\n    }\n  }\n\n  static {\n    this.register();\n  }\n\n  @Cache()\n  protected async client(): Promise<MilvusClient> {\n    try {\n      const client = new MilvusClient(this.options.connection);\n      // operation expected to fail if connection is not established correctly.\n      await client.listCollections();\n      return client;\n    } catch (error) {\n      throw new ToolError(`Unable to connect to Milvus.`, [error], {\n        isRetryable: false,\n        isFatal: true,\n      });\n    }\n  }\n\n  protected async _run(\n    input: ToolInput<this>,\n    _options: Partial<BaseToolRunOptions>,\n  ): Promise<JSONToolOutput<any>> {\n    switch (input.action) {\n      case MilvusAction.ListCollections: {\n        const collections = await this.listCollections();\n        return new JSONToolOutput(collections);\n      }\n\n      case MilvusAction.GetCollectionInfo: {\n        if (!input.collectionName) {\n          throw new ToolError(\"A collection name is required for Milvus GetCollectionInfo action\");\n        }\n        const collectionInfo = await this.getCollectionInfo(input.collectionName);\n        return new JSONToolOutput(collectionInfo);\n      }\n\n      case MilvusAction.Search: {\n        if (!input.collectionName || !input.vector) {\n          throw new ToolError(\"A collection name and vector are required for Milvus Search action\");\n        }\n        const searchResults = await this.search(input);\n        return new JSONToolOutput(searchResults);\n      }\n\n      case MilvusAction.Insert: {\n        if (!input.collectionName || !input.vectors) {\n          throw new ToolError(\n            \"A collection name and vectors are required for Milvus Insert action\",\n          );\n        }\n        const insertResults = await this.insert(input);\n        return new JSONToolOutput(insertResults);\n      }\n\n      case MilvusAction.Delete: {\n        if (!input.collectionName || !input.ids) {\n          throw new ToolError(\"Collection name and ids are required for Milvus 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.listCollections({});\n\n      if (response && response.data && Array.isArray(response.data)) {\n        return response.data.map((collection) => collection.name);\n      } else {\n        return [];\n      }\n    } catch (error) {\n      throw new ToolError(`Failed to list collections from Milvus: ${error}`);\n    }\n  }\n\n  protected async getCollectionInfo(collectionName: string): Promise<any> {\n    const client = await this.client();\n    const response = client.describeCollection({ collection_name: collectionName });\n    return response;\n  }\n\n  protected async insert(input: ToolInput<this>): Promise<any> {\n    const client = await this.client();\n    const response = await client.insert({\n      collection_name: input.collectionName as string,\n      fields_data: input.vectors!.map((vector, index) => ({\n        vector: vector,\n        ...input.metadata?.[index],\n      })),\n    });\n    return response;\n  }\n\n  protected async search(input: ToolInput<this>): Promise<any> {\n    const client = await this.client();\n    const response = await client.search({\n      collection_name: input.collectionName as string,\n      data: input.vector ?? [],\n      limit: input.topK || 10,\n      filter: input.filter,\n      output_fields: input.searchOutput,\n    });\n    return response.results;\n  }\n\n  protected async delete(input: ToolInput<this>): Promise<any> {\n    const client = await this.client();\n    const response = client.delete({\n      collection_name: input.collectionName as string,\n      filter: `id in [${input.ids?.join(\",\")}]`,\n    });\n    return response;\n  }\n}\n"]}

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

@@ -4,7 +4,7 @@ import { ClientConfig, ShowCollectionsResponse, DescribeCollectionResponse, Sear
 import { z } from 'zod';
 import 'ajv';
 import '../../context.cjs';
-import '../../emitter-DWEsd-ud.cjs';
+import '../../emitter-DpqUYjXH.cjs';
 import '../../internals/types.cjs';
 import '../../internals/helpers/guards.cjs';
 import '../../internals/serializable.cjs';
@@ -45,8 +45,8 @@ declare class MilvusDatabaseTool extends Tool<JSONToolOutput<MilvusSearchToolRes
         searchOutput: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
     }, "strip", z.ZodTypeAny, {
         action: MilvusAction;
-        metadata?: Record<string, any> | undefined;
         filter?: string | undefined;
+        metadata?: Record<string, any> | undefined;
         collectionName?: string | undefined;
         vector?: number[] | undefined;
         vectors?: number[][] | undefined;
@@ -55,8 +55,8 @@ declare class MilvusDatabaseTool extends Tool<JSONToolOutput<MilvusSearchToolRes
         searchOutput?: string[] | undefined;
     }, {
         action: MilvusAction;
-        metadata?: Record<string, any> | undefined;
         filter?: string | undefined;
+        metadata?: Record<string, any> | undefined;
         collectionName?: string | undefined;
         vector?: number[] | undefined;
         vectors?: number[][] | undefined;

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

@@ -4,7 +4,7 @@ import { ClientConfig, ShowCollectionsResponse, DescribeCollectionResponse, Sear
 import { z } from 'zod';
 import 'ajv';
 import '../../context.js';
-import '../../emitter-C9nwGLt6.js';
+import '../../emitter-CZFbzlUi.js';
 import '../../internals/types.js';
 import '../../internals/helpers/guards.js';
 import '../../internals/serializable.js';
@@ -45,8 +45,8 @@ declare class MilvusDatabaseTool extends Tool<JSONToolOutput<MilvusSearchToolRes
         searchOutput: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
     }, "strip", z.ZodTypeAny, {
         action: MilvusAction;
-        metadata?: Record<string, any> | undefined;
         filter?: string | undefined;
+        metadata?: Record<string, any> | undefined;
         collectionName?: string | undefined;
         vector?: number[] | undefined;
         vectors?: number[][] | undefined;
@@ -55,8 +55,8 @@ declare class MilvusDatabaseTool extends Tool<JSONToolOutput<MilvusSearchToolRes
         searchOutput?: string[] | undefined;
     }, {
         action: MilvusAction;
-        metadata?: Record<string, any> | undefined;
         filter?: string | undefined;
+        metadata?: Record<string, any> | undefined;
         collectionName?: string | undefined;
         vector?: number[] | undefined;
         vectors?: number[][] | undefined;

package/dist/tools/database/milvus.js

@@ -17,14 +17,14 @@ function _ts_metadata(k, v) {
   if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
 }
 __name(_ts_metadata, "_ts_metadata");
-var MilvusAction = /* @__PURE__ */ function(MilvusAction2) {
+var MilvusAction = /* @__PURE__ */ (function(MilvusAction2) {
   MilvusAction2["ListCollections"] = "ListCollections";
   MilvusAction2["GetCollectionInfo"] = "GetCollectionInfo";
   MilvusAction2["Search"] = "Search";
   MilvusAction2["Insert"] = "Insert";
   MilvusAction2["Delete"] = "Delete";
   return MilvusAction2;
-}({});
+})({});
 class MilvusDatabaseTool extends Tool {
   static {
     __name(this, "MilvusDatabaseTool");
@@ -160,7 +160,7 @@ class MilvusDatabaseTool extends Tool {
     const client = await this.client();
     const response = await client.search({
       collection_name: input.collectionName,
-      vector: input.vector,
+      data: input.vector ?? [],
       limit: input.topK || 10,
       filter: input.filter,
       output_fields: input.searchOutput

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

@@ -1 +1 @@
-{"version":3,"sources":["../../../src/tools/database/milvus.ts"],"names":["MilvusAction","MilvusDatabaseTool","Tool","name","description","inputSchema","z","object","action","nativeEnum","describe","collectionName","string","optional","vector","array","number","vectors","topK","coerce","int","min","max","default","filter","metadata","record","any","ids","or","searchOutput","emitter","Emitter","root","child","namespace","creator","validateInput","schema","input","ToolInputValidationError","register","client","MilvusClient","options","connection","listCollections","error","ToolError","isRetryable","isFatal","_run","_options","collections","JSONToolOutput","collectionInfo","getCollectionInfo","searchResults","search","insertResults","insert","deleteResults","delete","response","data","Array","isArray","map","collection","describeCollection","collection_name","fields_data","index","limit","output_fields","results","join"],"mappings":";;;;;;;;AAGC,SAAA,YAAA,CAAA,UAAA,EAAA,MAAA,EAAA,GAAA,EAAA,IAAA,EAAA;;;;;;AAAA,MAAA,CAAA,YAAA,EAAA,cAAA,CAAA;;;;;AAiCM,IAAKA,YAAAA,4BAAAA,aAAAA,EAAAA;;;;;;AAAAA,EAAAA,OAAAA,aAAAA;;AAQL,MAAMC,2BAA2BC,IAAAA,CAAAA;EA5CxC;;;EAgDEC,IAAAA,GAAO,oBAAA;EAEPC,WAAAA,GAAc,CAAA;QACV,iBAAA,CAAA;QACA,mBAAA,CAAA;QACA,QAAA,CAAA;QACA,QAAA,CAAA;QACA,QAAA,CAAA,kCAAA,CAAA;EAEJC,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,iDAAA,EAAkD,QAAA,CAAA,CAAuB,CAAA;AACrFO,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,CAAuB,CAAA;MACxEQ,IAAAA,EAAMZ,CAAAA,CAAEa,OACLH,MAAAA,EAAM,CACNI,KAAG,CACHC,GAAAA,CAAI,CAAA,CAAA,CACJC,GAAAA,CAAI,GAAA,CAAA,CACJC,OAAAA,CAAQ,EAAA,CAAA,CACRV,QAAAA,GACAH,QAAAA,CACC,CAAA,8CAAA,EAA+C,QAAA,CAAA,iBAAA,CAAwC,CAAA;MAE3Fc,MAAAA,EAAQlB,CAAAA,CACLM,QAAM,CACNC,QAAAA,GACAH,QAAAA,CAAS,CAAA,+BAAA,EAAgC,QAAA,CAAA,CAAuB,CAAA;AACnEe,MAAAA,QAAAA,EAAUnB,CAAAA,CACPoB,MAAAA,CAAOpB,CAAAA,CAAEM,MAAAA,IAAUN,CAAAA,CAAEqB,GAAAA,EAAG,CAAA,CACxBd,QAAAA,EAAQ,CACRH,QAAAA,CAAS,CAAA,+CAAA,EAAgD,QAAA,CAAA,CAAuB,CAAA;AACnFkB,MAAAA,GAAAA,EAAKtB,EACFS,KAAAA,CAAMT,CAAAA,CAAEM,MAAAA,EAAM,CAAGiB,GAAGvB,CAAAA,CAAEU,MAAAA,EAAM,CAAA,EAC5BH,QAAAA,EAAQ,CACRH,QAAAA,CAAS,CAAA,2BAAA,EAA4B,QAAA,CAAA,CAAuB,CAAA;MAC/DoB,YAAAA,EAAcxB,CAAAA,CACXS,KAAAA,CAAMT,CAAAA,CAAEM,MAAAA,EAAM,CAAA,CACdC,QAAAA,EAAQ,CACRH,QAAAA,CAAS,CAAA,uCAAA,EAAwC,QAAA,CAAA,CAAuB;KAC7E,CAAA;AACF;EAEgBqB,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,CAAM/B,MAAAA,KAAM,mBAAA,IAAuC,CAAC+B,MAAM5B,cAAAA,EAAgB;AAC5E,MAAA,MAAM,IAAI6B,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,CAAM/B,WAAM,QAAA,KAA6B,CAAC+B,MAAM5B,cAAAA,IAAkB,CAAC4B,MAAMzB,MAAAA,CAAAA,EAAS;AACpF,MAAA,MAAM,IAAI0B,wBAAAA,CACR,CAAA,iDAAA,EAAkD,QAAA,CAAA,QAAA,CAA+B,CAAA;AAErF;AACA,IAAA,IAAID,KAAAA,CAAM/B,WAAM,QAAA,KAA6B,CAAC+B,MAAM5B,cAAAA,IAAkB,CAAC4B,MAAMtB,OAAAA,CAAAA,EAAU;AACrF,MAAA,MAAM,IAAIuB,wBAAAA,CACR,CAAA,kDAAA,EAAmD,QAAA,CAAA,QAAA,CAA+B,CAAA;AAEtF;AACF;EAEA;AACE,IAAA,IAAA,CAAKC,QAAAA,EAAQ;AACf;AAEA,EAAA,MACgBC,MAAAA,GAAgC;AAC9C,IAAA,IAAI;AACF,MAAA,MAAMA,MAAAA,GAAS,IAAIC,YAAAA,CAAa,IAAA,CAAKC,QAAQC,UAAU,CAAA;AAEvD,MAAA,MAAMH,OAAOI,eAAAA,EAAe;AAC5B,MAAA,OAAOJ,MAAAA;AACT,KAAA,CAAA,OAASK,KAAAA,EAAO;AACd,MAAA,MAAM,IAAIC,UAAU,CAAA,4BAAA,CAAA,EAAgC;AAACD,QAAAA;AAAQ,OAAA,EAAA;QAC3DE,WAAAA,EAAa,KAAA;QACbC,OAAAA,EAAS;OACX,CAAA;AACF;AACF;EAEA,MAAgBC,IAAAA,CACdZ,OACAa,QAAAA,EAC8B;AAC9B,IAAA,QAAQb,MAAM/B,MAAAA;AACZ,MAAA,KAAA,iBAAA,EAAmC;AACjC,QAAA,MAAM6C,WAAAA,GAAc,MAAM,IAAA,CAAKP,eAAAA,EAAe;AAC9C,QAAA,OAAO,IAAIQ,eAAeD,WAAAA,CAAAA;AAC5B;AAEA,MAAA,KAAA,mBAAA,EAAqC;AACnC,QAAA,IAAI,CAACd,MAAM5B,cAAAA,EAAgB;AACzB,UAAA,MAAM,IAAIqC,UAAU,mEAAA,CAAA;AACtB;AACA,QAAA,MAAMO,cAAAA,GAAiB,MAAM,IAAA,CAAKC,iBAAAA,CAAkBjB,MAAM5B,cAAc,CAAA;AACxE,QAAA,OAAO,IAAI2C,eAAeC,cAAAA,CAAAA;AAC5B;AAEA,MAAA,KAAA,QAAA,EAA0B;AACxB,QAAA,IAAI,CAAChB,KAAAA,CAAM5B,cAAAA,IAAkB,CAAC4B,MAAMzB,MAAAA,EAAQ;AAC1C,UAAA,MAAM,IAAIkC,UAAU,oEAAA,CAAA;AACtB;AACA,QAAA,MAAMS,aAAAA,GAAgB,MAAM,IAAA,CAAKC,MAAAA,CAAOnB,KAAAA,CAAAA;AACxC,QAAA,OAAO,IAAIe,eAAeG,aAAAA,CAAAA;AAC5B;AAEA,MAAA,KAAA,QAAA,EAA0B;AACxB,QAAA,IAAI,CAAClB,KAAAA,CAAM5B,cAAAA,IAAkB,CAAC4B,MAAMtB,OAAAA,EAAS;AAC3C,UAAA,MAAM,IAAI+B,UACR,qEAAA,CAAA;AAEJ;AACA,QAAA,MAAMW,aAAAA,GAAgB,MAAM,IAAA,CAAKC,MAAAA,CAAOrB,KAAAA,CAAAA;AACxC,QAAA,OAAO,IAAIe,eAAeK,aAAAA,CAAAA;AAC5B;AAEA,MAAA,KAAA,QAAA,EAA0B;AACxB,QAAA,IAAI,CAACpB,KAAAA,CAAM5B,cAAAA,IAAkB,CAAC4B,MAAMX,GAAAA,EAAK;AACvC,UAAA,MAAM,IAAIoB,UAAU,+DAAA,CAAA;AACtB;AACA,QAAA,MAAMa,aAAAA,GAAgB,MAAM,IAAA,CAAKC,MAAAA,CAAOvB,KAAAA,CAAAA;AACxC,QAAA,OAAO,IAAIe,eAAeO,aAAAA,CAAAA;AAC5B;MAEA,SAAS;AACP,QAAA,MAAM,IAAIb,SAAAA,CAAU,CAAA,0BAAA,EAA6BT,KAAAA,CAAM/B,MAAM,CAAA,CAAE,CAAA;AACjE;AACF;AACF;AAEA,EAAA,MAAgBsC,eAAAA,GAAqC;AACnD,IAAA,IAAI;AACF,MAAA,MAAMJ,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAAA,EAAM;AAChC,MAAA,MAAMqB,QAAAA,GAAW,MAAMrB,MAAAA,CAAOI,eAAAA,CAAgB,EAAC,CAAA;AAE/C,MAAA,IAAIiB,YAAYA,QAAAA,CAASC,IAAAA,IAAQC,MAAMC,OAAAA,CAAQH,QAAAA,CAASC,IAAI,CAAA,EAAG;AAC7D,QAAA,OAAOD,SAASC,IAAAA,CAAKG,GAAAA,CAAI,CAACC,UAAAA,KAAeA,WAAWjE,IAAI,CAAA;OAC1D,MAAO;AACL,QAAA,OAAO,EAAA;AACT;AACF,KAAA,CAAA,OAAS4C,KAAAA,EAAO;AACd,MAAA,MAAM,IAAIC,SAAAA,CAAU,CAAA,wCAAA,EAA2CD,KAAAA,CAAAA,CAAO,CAAA;AACxE;AACF;AAEA,EAAA,MAAgBS,kBAAkB7C,cAAAA,EAAsC;AACtE,IAAA,MAAM+B,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAAA,EAAM;AAChC,IAAA,MAAMqB,QAAAA,GAAWrB,OAAO2B,kBAAAA,CAAmB;MAAEC,eAAAA,EAAiB3D;KAAe,CAAA;AAC7E,IAAA,OAAOoD,QAAAA;AACT;AAEA,EAAA,MAAgBH,OAAOrB,KAAAA,EAAsC;AAC3D,IAAA,MAAMG,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAAA,EAAM;AAChC,IAAA,MAAMqB,QAAAA,GAAW,MAAMrB,MAAAA,CAAOkB,MAAAA,CAAO;AACnCU,MAAAA,eAAAA,EAAiB/B,KAAAA,CAAM5B,cAAAA;AACvB4D,MAAAA,WAAAA,EAAahC,KAAAA,CAAMtB,OAAAA,CAASkD,GAAAA,CAAI,CAACrD,QAAQ0D,KAAAA,MAAW;AAClD1D,QAAAA,MAAAA;QACA,GAAGyB,KAAAA,CAAMd,WAAW+C,KAAAA;OACtB,CAAA;KACF,CAAA;AACA,IAAA,OAAOT,QAAAA;AACT;AAEA,EAAA,MAAgBL,OAAOnB,KAAAA,EAAsC;AAC3D,IAAA,MAAMG,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAAA,EAAM;AAChC,IAAA,MAAMqB,QAAAA,GAAW,MAAMrB,MAAAA,CAAOgB,MAAAA,CAAO;AACnCY,MAAAA,eAAAA,EAAiB/B,KAAAA,CAAM5B,cAAAA;AACvBG,MAAAA,MAAAA,EAAQyB,KAAAA,CAAMzB,MAAAA;AACd2D,MAAAA,KAAAA,EAAOlC,MAAMrB,IAAAA,IAAQ,EAAA;AACrBM,MAAAA,MAAAA,EAAQe,KAAAA,CAAMf,MAAAA;AACdkD,MAAAA,aAAAA,EAAenC,KAAAA,CAAMT;KACvB,CAAA;AACA,IAAA,OAAOiC,QAAAA,CAASY,OAAAA;AAClB;AAEA,EAAA,MAAgBb,OAAOvB,KAAAA,EAAsC;AAC3D,IAAA,MAAMG,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAAA,EAAM;AAChC,IAAA,MAAMqB,QAAAA,GAAWrB,OAAOoB,MAAAA,CAAO;AAC7BQ,MAAAA,eAAAA,EAAiB/B,KAAAA,CAAM5B,cAAAA;AACvBa,MAAAA,MAAAA,EAAQ,CAAA,OAAA,EAAUe,KAAAA,CAAMX,GAAAA,EAAKgD,IAAAA,CAAK,GAAA,CAAA,CAAA,CAAA;KACpC,CAAA;AACA,IAAA,OAAOb,QAAAA;AACT;AACF","file":"milvus.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 {\n  ClientConfig,\n  MilvusClient,\n  SearchRes,\n  ShowCollectionsResponse,\n  DescribeCollectionResponse,\n} from \"@zilliz/milvus2-sdk-node\";\nimport { z } from \"zod\";\nimport { Emitter } from \"@/emitter/emitter.js\";\n\nexport interface MilvusToolOptions extends BaseToolOptions {\n  connection: ClientConfig;\n}\n\nexport type MilvusSearchToolResult =\n  | ShowCollectionsResponse\n  | DescribeCollectionResponse\n  | SearchRes;\n\nexport enum MilvusAction {\n  ListCollections = \"ListCollections\",\n  GetCollectionInfo = \"GetCollectionInfo\",\n  Search = \"Search\",\n  Insert = \"Insert\",\n  Delete = \"Delete\",\n}\n\nexport class MilvusDatabaseTool extends Tool<\n  JSONToolOutput<MilvusSearchToolResult>,\n  MilvusToolOptions\n> {\n  name = \"MilvusDatabaseTool\";\n\n  description = `Can query data from a Milvus database. IMPORTANT: strictly follow this order of actions:\n   1. ${MilvusAction.ListCollections} - List all the Milvus collections\n   2. ${MilvusAction.GetCollectionInfo} - Get information about into a Milvus collection\n   3. ${MilvusAction.Insert} - Insert data into a Milvus collection\n   3. ${MilvusAction.Search} - Perform search on a Milvus collection\n   4. ${MilvusAction.Delete} - Delete from a Milvus collection`;\n\n  inputSchema() {\n    return z.object({\n      action: z\n        .nativeEnum(MilvusAction)\n        .describe(\n          `The action to perform. ${MilvusAction.ListCollections} lists all collections, ${MilvusAction.GetCollectionInfo} fetches details for a specified collection, ${MilvusAction.Search} executes a vector search, ${MilvusAction.Insert} inserts new vectors, and ${MilvusAction.Delete} removes vectors.`,\n        ),\n      collectionName: z\n        .string()\n        .optional()\n        .describe(\n          `The name of the collection to query, required for ${MilvusAction.GetCollectionInfo}, ${MilvusAction.Search}, ${MilvusAction.Insert}, and ${MilvusAction.Delete}`,\n        ),\n      vector: z\n        .array(z.number())\n        .optional()\n        .describe(`The vector to search for or insert, required for ${MilvusAction.Search}`),\n      vectors: z\n        .array(z.array(z.number()))\n        .optional()\n        .describe(`The vectors to insert, required for ${MilvusAction.Insert}`),\n      topK: z.coerce\n        .number()\n        .int()\n        .min(1)\n        .max(1000)\n        .default(10)\n        .optional()\n        .describe(\n          `The number of nearest neighbors to return for ${MilvusAction.Search}. Maximum is 1000`,\n        ),\n      filter: z\n        .string()\n        .optional()\n        .describe(`Optional filter expression for ${MilvusAction.Search}`),\n      metadata: z\n        .record(z.string(), z.any())\n        .optional()\n        .describe(`Additional metadata to insert with vectors for ${MilvusAction.Insert}`),\n      ids: z\n        .array(z.string().or(z.number()))\n        .optional()\n        .describe(`Array of IDs to delete for ${MilvusAction.Delete}`),\n      searchOutput: z\n        .array(z.string())\n        .optional()\n        .describe(`Fields to return in search results for ${MilvusAction.Search}`),\n    });\n  }\n\n  public readonly emitter: ToolEmitter<ToolInput<this>, JSONToolOutput<MilvusSearchToolResult>> =\n    Emitter.root.child({\n      namespace: [\"tool\", \"database\", \"milvus\"],\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 === MilvusAction.GetCollectionInfo && !input.collectionName) {\n      throw new ToolInputValidationError(\n        `Collection name is required for ${MilvusAction.GetCollectionInfo}, ${MilvusAction.Search}, ${MilvusAction.Insert}, and ${MilvusAction.Delete} actions.`,\n      );\n    }\n    if (input.action === MilvusAction.Search && (!input.collectionName || !input.vector)) {\n      throw new ToolInputValidationError(\n        `Both collection name and vector are required for ${MilvusAction.Search} action.`,\n      );\n    }\n    if (input.action === MilvusAction.Insert && (!input.collectionName || !input.vectors)) {\n      throw new ToolInputValidationError(\n        `Both collection name and vectors are required for ${MilvusAction.Insert} action.`,\n      );\n    }\n  }\n\n  static {\n    this.register();\n  }\n\n  @Cache()\n  protected async client(): Promise<MilvusClient> {\n    try {\n      const client = new MilvusClient(this.options.connection);\n      // operation expected to fail if connection is not established correctly.\n      await client.listCollections();\n      return client;\n    } catch (error) {\n      throw new ToolError(`Unable to connect to Milvus.`, [error], {\n        isRetryable: false,\n        isFatal: true,\n      });\n    }\n  }\n\n  protected async _run(\n    input: ToolInput<this>,\n    _options: Partial<BaseToolRunOptions>,\n  ): Promise<JSONToolOutput<any>> {\n    switch (input.action) {\n      case MilvusAction.ListCollections: {\n        const collections = await this.listCollections();\n        return new JSONToolOutput(collections);\n      }\n\n      case MilvusAction.GetCollectionInfo: {\n        if (!input.collectionName) {\n          throw new ToolError(\"A collection name is required for Milvus GetCollectionInfo action\");\n        }\n        const collectionInfo = await this.getCollectionInfo(input.collectionName);\n        return new JSONToolOutput(collectionInfo);\n      }\n\n      case MilvusAction.Search: {\n        if (!input.collectionName || !input.vector) {\n          throw new ToolError(\"A collection name and vector are required for Milvus Search action\");\n        }\n        const searchResults = await this.search(input);\n        return new JSONToolOutput(searchResults);\n      }\n\n      case MilvusAction.Insert: {\n        if (!input.collectionName || !input.vectors) {\n          throw new ToolError(\n            \"A collection name and vectors are required for Milvus Insert action\",\n          );\n        }\n        const insertResults = await this.insert(input);\n        return new JSONToolOutput(insertResults);\n      }\n\n      case MilvusAction.Delete: {\n        if (!input.collectionName || !input.ids) {\n          throw new ToolError(\"Collection name and ids are required for Milvus 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.listCollections({});\n\n      if (response && response.data && Array.isArray(response.data)) {\n        return response.data.map((collection) => collection.name);\n      } else {\n        return [];\n      }\n    } catch (error) {\n      throw new ToolError(`Failed to list collections from Milvus: ${error}`);\n    }\n  }\n\n  protected async getCollectionInfo(collectionName: string): Promise<any> {\n    const client = await this.client();\n    const response = client.describeCollection({ collection_name: collectionName });\n    return response;\n  }\n\n  protected async insert(input: ToolInput<this>): Promise<any> {\n    const client = await this.client();\n    const response = await client.insert({\n      collection_name: input.collectionName as string,\n      fields_data: input.vectors!.map((vector, index) => ({\n        vector: vector,\n        ...input.metadata?.[index],\n      })),\n    });\n    return response;\n  }\n\n  protected async search(input: ToolInput<this>): Promise<any> {\n    const client = await this.client();\n    const response = await client.search({\n      collection_name: input.collectionName as string,\n      vector: input.vector,\n      limit: input.topK || 10,\n      filter: input.filter,\n      output_fields: input.searchOutput,\n    });\n    return response.results;\n  }\n\n  protected async delete(input: ToolInput<this>): Promise<any> {\n    const client = await this.client();\n    const response = client.delete({\n      collection_name: input.collectionName as string,\n      filter: `id in [${input.ids?.join(\",\")}]`,\n    });\n    return response;\n  }\n}\n"]}
+{"version":3,"sources":["../../../src/tools/database/milvus.ts"],"names":["MilvusAction","MilvusDatabaseTool","Tool","name","description","inputSchema","z","object","action","nativeEnum","describe","collectionName","string","optional","vector","array","number","vectors","topK","coerce","int","min","max","default","filter","metadata","record","any","ids","or","searchOutput","emitter","Emitter","root","child","namespace","creator","validateInput","schema","input","ToolInputValidationError","register","client","MilvusClient","options","connection","listCollections","error","ToolError","isRetryable","isFatal","_run","_options","collections","JSONToolOutput","collectionInfo","getCollectionInfo","searchResults","search","insertResults","insert","deleteResults","delete","response","data","Array","isArray","map","collection","describeCollection","collection_name","fields_data","index","limit","output_fields","results","join"],"mappings":";;;;;;;;;;;;;;;;;;;AAoCO,IAAKA,YAAAA,6BAAAA,aAAAA,EAAAA;;;;;;AAAAA,EAAAA,OAAAA,aAAAA;;AAQL,MAAMC,2BAA2BC,IAAAA,CAAAA;AAAAA,EAAAA;;;EAItCC,IAAAA,GAAO,oBAAA;EAEPC,WAAAA,GAAc,CAAA;QACV,iBAAA,CAAA;QACA,mBAAA,CAAA;QACA,QAAA,CAAA;QACA,QAAA,CAAA;QACA,QAAA,CAAA,kCAAA,CAAA;EAEJC,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,iDAAA,EAAkD,QAAA,CAAA,CAAuB,CAAA;AACrFO,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,CAAuB,CAAA;MACxEQ,IAAAA,EAAMZ,CAAAA,CAAEa,OACLH,MAAAA,EAAM,CACNI,KAAG,CACHC,GAAAA,CAAI,CAAA,CAAA,CACJC,GAAAA,CAAI,GAAA,CAAA,CACJC,OAAAA,CAAQ,EAAA,CAAA,CACRV,QAAAA,GACAH,QAAAA,CACC,CAAA,8CAAA,EAA+C,QAAA,CAAA,iBAAA,CAAwC,CAAA;MAE3Fc,MAAAA,EAAQlB,CAAAA,CACLM,QAAM,CACNC,QAAAA,GACAH,QAAAA,CAAS,CAAA,+BAAA,EAAgC,QAAA,CAAA,CAAuB,CAAA;AACnEe,MAAAA,QAAAA,EAAUnB,CAAAA,CACPoB,MAAAA,CAAOpB,CAAAA,CAAEM,MAAAA,IAAUN,CAAAA,CAAEqB,GAAAA,EAAG,CAAA,CACxBd,QAAAA,EAAQ,CACRH,QAAAA,CAAS,CAAA,+CAAA,EAAgD,QAAA,CAAA,CAAuB,CAAA;AACnFkB,MAAAA,GAAAA,EAAKtB,EACFS,KAAAA,CAAMT,CAAAA,CAAEM,MAAAA,EAAM,CAAGiB,GAAGvB,CAAAA,CAAEU,MAAAA,EAAM,CAAA,EAC5BH,QAAAA,EAAQ,CACRH,QAAAA,CAAS,CAAA,2BAAA,EAA4B,QAAA,CAAA,CAAuB,CAAA;MAC/DoB,YAAAA,EAAcxB,CAAAA,CACXS,KAAAA,CAAMT,CAAAA,CAAEM,MAAAA,EAAM,CAAA,CACdC,QAAAA,EAAQ,CACRH,QAAAA,CAAS,CAAA,uCAAA,EAAwC,QAAA,CAAA,CAAuB;KAC7E,CAAA;AACF;EAEgBqB,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,CAAM/B,MAAAA,KAAM,mBAAA,IAAuC,CAAC+B,MAAM5B,cAAAA,EAAgB;AAC5E,MAAA,MAAM,IAAI6B,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,CAAM/B,WAAM,QAAA,KAA6B,CAAC+B,MAAM5B,cAAAA,IAAkB,CAAC4B,MAAMzB,MAAAA,CAAAA,EAAS;AACpF,MAAA,MAAM,IAAI0B,wBAAAA,CACR,CAAA,iDAAA,EAAkD,QAAA,CAAA,QAAA,CAA+B,CAAA;AAErF;AACA,IAAA,IAAID,KAAAA,CAAM/B,WAAM,QAAA,KAA6B,CAAC+B,MAAM5B,cAAAA,IAAkB,CAAC4B,MAAMtB,OAAAA,CAAAA,EAAU;AACrF,MAAA,MAAM,IAAIuB,wBAAAA,CACR,CAAA,kDAAA,EAAmD,QAAA,CAAA,QAAA,CAA+B,CAAA;AAEtF;AACF;EAEA;AACE,IAAA,IAAA,CAAKC,QAAAA,EAAQ;AACf;AAEA,EAAA,MACgBC,MAAAA,GAAgC;AAC9C,IAAA,IAAI;AACF,MAAA,MAAMA,MAAAA,GAAS,IAAIC,YAAAA,CAAa,IAAA,CAAKC,QAAQC,UAAU,CAAA;AAEvD,MAAA,MAAMH,OAAOI,eAAAA,EAAe;AAC5B,MAAA,OAAOJ,MAAAA;AACT,KAAA,CAAA,OAASK,KAAAA,EAAO;AACd,MAAA,MAAM,IAAIC,UAAU,CAAA,4BAAA,CAAA,EAAgC;AAACD,QAAAA;AAAQ,OAAA,EAAA;QAC3DE,WAAAA,EAAa,KAAA;QACbC,OAAAA,EAAS;OACX,CAAA;AACF;AACF;EAEA,MAAgBC,IAAAA,CACdZ,OACAa,QAAAA,EAC8B;AAC9B,IAAA,QAAQb,MAAM/B,MAAAA;AACZ,MAAA,KAAA,iBAAA,EAAmC;AACjC,QAAA,MAAM6C,WAAAA,GAAc,MAAM,IAAA,CAAKP,eAAAA,EAAe;AAC9C,QAAA,OAAO,IAAIQ,eAAeD,WAAAA,CAAAA;AAC5B;AAEA,MAAA,KAAA,mBAAA,EAAqC;AACnC,QAAA,IAAI,CAACd,MAAM5B,cAAAA,EAAgB;AACzB,UAAA,MAAM,IAAIqC,UAAU,mEAAA,CAAA;AACtB;AACA,QAAA,MAAMO,cAAAA,GAAiB,MAAM,IAAA,CAAKC,iBAAAA,CAAkBjB,MAAM5B,cAAc,CAAA;AACxE,QAAA,OAAO,IAAI2C,eAAeC,cAAAA,CAAAA;AAC5B;AAEA,MAAA,KAAA,QAAA,EAA0B;AACxB,QAAA,IAAI,CAAChB,KAAAA,CAAM5B,cAAAA,IAAkB,CAAC4B,MAAMzB,MAAAA,EAAQ;AAC1C,UAAA,MAAM,IAAIkC,UAAU,oEAAA,CAAA;AACtB;AACA,QAAA,MAAMS,aAAAA,GAAgB,MAAM,IAAA,CAAKC,MAAAA,CAAOnB,KAAAA,CAAAA;AACxC,QAAA,OAAO,IAAIe,eAAeG,aAAAA,CAAAA;AAC5B;AAEA,MAAA,KAAA,QAAA,EAA0B;AACxB,QAAA,IAAI,CAAClB,KAAAA,CAAM5B,cAAAA,IAAkB,CAAC4B,MAAMtB,OAAAA,EAAS;AAC3C,UAAA,MAAM,IAAI+B,UACR,qEAAA,CAAA;AAEJ;AACA,QAAA,MAAMW,aAAAA,GAAgB,MAAM,IAAA,CAAKC,MAAAA,CAAOrB,KAAAA,CAAAA;AACxC,QAAA,OAAO,IAAIe,eAAeK,aAAAA,CAAAA;AAC5B;AAEA,MAAA,KAAA,QAAA,EAA0B;AACxB,QAAA,IAAI,CAACpB,KAAAA,CAAM5B,cAAAA,IAAkB,CAAC4B,MAAMX,GAAAA,EAAK;AACvC,UAAA,MAAM,IAAIoB,UAAU,+DAAA,CAAA;AACtB;AACA,QAAA,MAAMa,aAAAA,GAAgB,MAAM,IAAA,CAAKC,MAAAA,CAAOvB,KAAAA,CAAAA;AACxC,QAAA,OAAO,IAAIe,eAAeO,aAAAA,CAAAA;AAC5B;MAEA,SAAS;AACP,QAAA,MAAM,IAAIb,SAAAA,CAAU,CAAA,0BAAA,EAA6BT,KAAAA,CAAM/B,MAAM,CAAA,CAAE,CAAA;AACjE;AACF;AACF;AAEA,EAAA,MAAgBsC,eAAAA,GAAqC;AACnD,IAAA,IAAI;AACF,MAAA,MAAMJ,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAAA,EAAM;AAChC,MAAA,MAAMqB,QAAAA,GAAW,MAAMrB,MAAAA,CAAOI,eAAAA,CAAgB,EAAC,CAAA;AAE/C,MAAA,IAAIiB,YAAYA,QAAAA,CAASC,IAAAA,IAAQC,MAAMC,OAAAA,CAAQH,QAAAA,CAASC,IAAI,CAAA,EAAG;AAC7D,QAAA,OAAOD,SAASC,IAAAA,CAAKG,GAAAA,CAAI,CAACC,UAAAA,KAAeA,WAAWjE,IAAI,CAAA;OAC1D,MAAO;AACL,QAAA,OAAO,EAAA;AACT;AACF,KAAA,CAAA,OAAS4C,KAAAA,EAAO;AACd,MAAA,MAAM,IAAIC,SAAAA,CAAU,CAAA,wCAAA,EAA2CD,KAAAA,CAAAA,CAAO,CAAA;AACxE;AACF;AAEA,EAAA,MAAgBS,kBAAkB7C,cAAAA,EAAsC;AACtE,IAAA,MAAM+B,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAAA,EAAM;AAChC,IAAA,MAAMqB,QAAAA,GAAWrB,OAAO2B,kBAAAA,CAAmB;MAAEC,eAAAA,EAAiB3D;KAAe,CAAA;AAC7E,IAAA,OAAOoD,QAAAA;AACT;AAEA,EAAA,MAAgBH,OAAOrB,KAAAA,EAAsC;AAC3D,IAAA,MAAMG,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAAA,EAAM;AAChC,IAAA,MAAMqB,QAAAA,GAAW,MAAMrB,MAAAA,CAAOkB,MAAAA,CAAO;AACnCU,MAAAA,eAAAA,EAAiB/B,KAAAA,CAAM5B,cAAAA;AACvB4D,MAAAA,WAAAA,EAAahC,KAAAA,CAAMtB,OAAAA,CAASkD,GAAAA,CAAI,CAACrD,QAAQ0D,KAAAA,MAAW;AAClD1D,QAAAA,MAAAA;QACA,GAAGyB,KAAAA,CAAMd,WAAW+C,KAAAA;OACtB,CAAA;KACF,CAAA;AACA,IAAA,OAAOT,QAAAA;AACT;AAEA,EAAA,MAAgBL,OAAOnB,KAAAA,EAAsC;AAC3D,IAAA,MAAMG,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAAA,EAAM;AAChC,IAAA,MAAMqB,QAAAA,GAAW,MAAMrB,MAAAA,CAAOgB,MAAAA,CAAO;AACnCY,MAAAA,eAAAA,EAAiB/B,KAAAA,CAAM5B,cAAAA;MACvBqD,IAAAA,EAAMzB,KAAAA,CAAMzB,UAAU,EAAA;AACtB2D,MAAAA,KAAAA,EAAOlC,MAAMrB,IAAAA,IAAQ,EAAA;AACrBM,MAAAA,MAAAA,EAAQe,KAAAA,CAAMf,MAAAA;AACdkD,MAAAA,aAAAA,EAAenC,KAAAA,CAAMT;KACvB,CAAA;AACA,IAAA,OAAOiC,QAAAA,CAASY,OAAAA;AAClB;AAEA,EAAA,MAAgBb,OAAOvB,KAAAA,EAAsC;AAC3D,IAAA,MAAMG,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAAA,EAAM;AAChC,IAAA,MAAMqB,QAAAA,GAAWrB,OAAOoB,MAAAA,CAAO;AAC7BQ,MAAAA,eAAAA,EAAiB/B,KAAAA,CAAM5B,cAAAA;AACvBa,MAAAA,MAAAA,EAAQ,CAAA,OAAA,EAAUe,KAAAA,CAAMX,GAAAA,EAAKgD,IAAAA,CAAK,GAAA,CAAA,CAAA,CAAA;KACpC,CAAA;AACA,IAAA,OAAOb,QAAAA;AACT;AACF","file":"milvus.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 {\n  ClientConfig,\n  MilvusClient,\n  SearchRes,\n  ShowCollectionsResponse,\n  DescribeCollectionResponse,\n} from \"@zilliz/milvus2-sdk-node\";\nimport { z } from \"zod\";\nimport { Emitter } from \"@/emitter/emitter.js\";\n\nexport interface MilvusToolOptions extends BaseToolOptions {\n  connection: ClientConfig;\n}\n\nexport type MilvusSearchToolResult =\n  | ShowCollectionsResponse\n  | DescribeCollectionResponse\n  | SearchRes;\n\nexport enum MilvusAction {\n  ListCollections = \"ListCollections\",\n  GetCollectionInfo = \"GetCollectionInfo\",\n  Search = \"Search\",\n  Insert = \"Insert\",\n  Delete = \"Delete\",\n}\n\nexport class MilvusDatabaseTool extends Tool<\n  JSONToolOutput<MilvusSearchToolResult>,\n  MilvusToolOptions\n> {\n  name = \"MilvusDatabaseTool\";\n\n  description = `Can query data from a Milvus database. IMPORTANT: strictly follow this order of actions:\n   1. ${MilvusAction.ListCollections} - List all the Milvus collections\n   2. ${MilvusAction.GetCollectionInfo} - Get information about into a Milvus collection\n   3. ${MilvusAction.Insert} - Insert data into a Milvus collection\n   3. ${MilvusAction.Search} - Perform search on a Milvus collection\n   4. ${MilvusAction.Delete} - Delete from a Milvus collection`;\n\n  inputSchema() {\n    return z.object({\n      action: z\n        .nativeEnum(MilvusAction)\n        .describe(\n          `The action to perform. ${MilvusAction.ListCollections} lists all collections, ${MilvusAction.GetCollectionInfo} fetches details for a specified collection, ${MilvusAction.Search} executes a vector search, ${MilvusAction.Insert} inserts new vectors, and ${MilvusAction.Delete} removes vectors.`,\n        ),\n      collectionName: z\n        .string()\n        .optional()\n        .describe(\n          `The name of the collection to query, required for ${MilvusAction.GetCollectionInfo}, ${MilvusAction.Search}, ${MilvusAction.Insert}, and ${MilvusAction.Delete}`,\n        ),\n      vector: z\n        .array(z.number())\n        .optional()\n        .describe(`The vector to search for or insert, required for ${MilvusAction.Search}`),\n      vectors: z\n        .array(z.array(z.number()))\n        .optional()\n        .describe(`The vectors to insert, required for ${MilvusAction.Insert}`),\n      topK: z.coerce\n        .number()\n        .int()\n        .min(1)\n        .max(1000)\n        .default(10)\n        .optional()\n        .describe(\n          `The number of nearest neighbors to return for ${MilvusAction.Search}. Maximum is 1000`,\n        ),\n      filter: z\n        .string()\n        .optional()\n        .describe(`Optional filter expression for ${MilvusAction.Search}`),\n      metadata: z\n        .record(z.string(), z.any())\n        .optional()\n        .describe(`Additional metadata to insert with vectors for ${MilvusAction.Insert}`),\n      ids: z\n        .array(z.string().or(z.number()))\n        .optional()\n        .describe(`Array of IDs to delete for ${MilvusAction.Delete}`),\n      searchOutput: z\n        .array(z.string())\n        .optional()\n        .describe(`Fields to return in search results for ${MilvusAction.Search}`),\n    });\n  }\n\n  public readonly emitter: ToolEmitter<ToolInput<this>, JSONToolOutput<MilvusSearchToolResult>> =\n    Emitter.root.child({\n      namespace: [\"tool\", \"database\", \"milvus\"],\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 === MilvusAction.GetCollectionInfo && !input.collectionName) {\n      throw new ToolInputValidationError(\n        `Collection name is required for ${MilvusAction.GetCollectionInfo}, ${MilvusAction.Search}, ${MilvusAction.Insert}, and ${MilvusAction.Delete} actions.`,\n      );\n    }\n    if (input.action === MilvusAction.Search && (!input.collectionName || !input.vector)) {\n      throw new ToolInputValidationError(\n        `Both collection name and vector are required for ${MilvusAction.Search} action.`,\n      );\n    }\n    if (input.action === MilvusAction.Insert && (!input.collectionName || !input.vectors)) {\n      throw new ToolInputValidationError(\n        `Both collection name and vectors are required for ${MilvusAction.Insert} action.`,\n      );\n    }\n  }\n\n  static {\n    this.register();\n  }\n\n  @Cache()\n  protected async client(): Promise<MilvusClient> {\n    try {\n      const client = new MilvusClient(this.options.connection);\n      // operation expected to fail if connection is not established correctly.\n      await client.listCollections();\n      return client;\n    } catch (error) {\n      throw new ToolError(`Unable to connect to Milvus.`, [error], {\n        isRetryable: false,\n        isFatal: true,\n      });\n    }\n  }\n\n  protected async _run(\n    input: ToolInput<this>,\n    _options: Partial<BaseToolRunOptions>,\n  ): Promise<JSONToolOutput<any>> {\n    switch (input.action) {\n      case MilvusAction.ListCollections: {\n        const collections = await this.listCollections();\n        return new JSONToolOutput(collections);\n      }\n\n      case MilvusAction.GetCollectionInfo: {\n        if (!input.collectionName) {\n          throw new ToolError(\"A collection name is required for Milvus GetCollectionInfo action\");\n        }\n        const collectionInfo = await this.getCollectionInfo(input.collectionName);\n        return new JSONToolOutput(collectionInfo);\n      }\n\n      case MilvusAction.Search: {\n        if (!input.collectionName || !input.vector) {\n          throw new ToolError(\"A collection name and vector are required for Milvus Search action\");\n        }\n        const searchResults = await this.search(input);\n        return new JSONToolOutput(searchResults);\n      }\n\n      case MilvusAction.Insert: {\n        if (!input.collectionName || !input.vectors) {\n          throw new ToolError(\n            \"A collection name and vectors are required for Milvus Insert action\",\n          );\n        }\n        const insertResults = await this.insert(input);\n        return new JSONToolOutput(insertResults);\n      }\n\n      case MilvusAction.Delete: {\n        if (!input.collectionName || !input.ids) {\n          throw new ToolError(\"Collection name and ids are required for Milvus 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.listCollections({});\n\n      if (response && response.data && Array.isArray(response.data)) {\n        return response.data.map((collection) => collection.name);\n      } else {\n        return [];\n      }\n    } catch (error) {\n      throw new ToolError(`Failed to list collections from Milvus: ${error}`);\n    }\n  }\n\n  protected async getCollectionInfo(collectionName: string): Promise<any> {\n    const client = await this.client();\n    const response = client.describeCollection({ collection_name: collectionName });\n    return response;\n  }\n\n  protected async insert(input: ToolInput<this>): Promise<any> {\n    const client = await this.client();\n    const response = await client.insert({\n      collection_name: input.collectionName as string,\n      fields_data: input.vectors!.map((vector, index) => ({\n        vector: vector,\n        ...input.metadata?.[index],\n      })),\n    });\n    return response;\n  }\n\n  protected async search(input: ToolInput<this>): Promise<any> {\n    const client = await this.client();\n    const response = await client.search({\n      collection_name: input.collectionName as string,\n      data: input.vector ?? [],\n      limit: input.topK || 10,\n      filter: input.filter,\n      output_fields: input.searchOutput,\n    });\n    return response.results;\n  }\n\n  protected async delete(input: ToolInput<this>): Promise<any> {\n    const client = await this.client();\n    const response = client.delete({\n      collection_name: input.collectionName as string,\n      filter: `id in [${input.ids?.join(\",\")}]`,\n    });\n    return response;\n  }\n}\n"]}

package/dist/tools/database/qdrant.cjs

@@ -20,14 +20,14 @@ function _ts_metadata(k, v) {
   if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
 }
 __name(_ts_metadata, "_ts_metadata");
-var QdrantAction = /* @__PURE__ */ function(QdrantAction2) {
+var QdrantAction = /* @__PURE__ */ (function(QdrantAction2) {
   QdrantAction2["ListCollections"] = "ListCollections";
   QdrantAction2["GetCollectionInfo"] = "GetCollectionInfo";
   QdrantAction2["Search"] = "Search";
   QdrantAction2["Insert"] = "Insert";
   QdrantAction2["Delete"] = "Delete";
   return QdrantAction2;
-}({});
+})({});
 class QdrantDatabaseTool extends base_cjs.Tool {
   static {
     __name(this, "QdrantDatabaseTool");

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":";;;;;;;;;;;AAGC,SAAA,YAAA,CAAA,UAAA,EAAA,MAAA,EAAA,GAAA,EAAA,IAAA,EAAA;;;;;;AAAA,MAAA,CAAA,YAAA,EAAA,cAAA,CAAA;;;;;AA0BM,IAAKA,YAAAA,4BAAAA,aAAAA,EAAAA;;;;;;AAAAA,EAAAA,OAAAA,aAAAA;;AAQL,MAAMC,2BAA2BC,aAAAA,CAAAA;EArCxC;;;EAyCEC,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;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"]}

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

@@ -4,7 +4,7 @@ import { QdrantClientParams, QdrantClient } from '@qdrant/js-client-rest';
 import { z } from 'zod';
 import 'ajv';
 import '../../context.cjs';
-import '../../emitter-DWEsd-ud.cjs';
+import '../../emitter-DpqUYjXH.cjs';
 import '../../internals/types.cjs';
 import '../../internals/helpers/guards.cjs';
 import '../../internals/serializable.cjs';

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

@@ -4,7 +4,7 @@ import { QdrantClientParams, QdrantClient } from '@qdrant/js-client-rest';
 import { z } from 'zod';
 import 'ajv';
 import '../../context.js';
-import '../../emitter-C9nwGLt6.js';
+import '../../emitter-CZFbzlUi.js';
 import '../../internals/types.js';
 import '../../internals/helpers/guards.js';
 import '../../internals/serializable.js';

package/dist/tools/database/qdrant.js

@@ -18,14 +18,14 @@ function _ts_metadata(k, v) {
   if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
 }
 __name(_ts_metadata, "_ts_metadata");
-var QdrantAction = /* @__PURE__ */ function(QdrantAction2) {
+var QdrantAction = /* @__PURE__ */ (function(QdrantAction2) {
   QdrantAction2["ListCollections"] = "ListCollections";
   QdrantAction2["GetCollectionInfo"] = "GetCollectionInfo";
   QdrantAction2["Search"] = "Search";
   QdrantAction2["Insert"] = "Insert";
   QdrantAction2["Delete"] = "Delete";
   return QdrantAction2;
-}({});
+})({});
 class QdrantDatabaseTool extends Tool {
   static {
     __name(this, "QdrantDatabaseTool");

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":";;;;;;;;;AAGC,SAAA,YAAA,CAAA,UAAA,EAAA,MAAA,EAAA,GAAA,EAAA,IAAA,EAAA;;;;;;AAAA,MAAA,CAAA,YAAA,EAAA,cAAA,CAAA;;;;;AA0BM,IAAKA,YAAAA,4BAAAA,aAAAA,EAAAA;;;;;;AAAAA,EAAAA,OAAAA,aAAAA;;AAQL,MAAMC,2BAA2BC,IAAAA,CAAAA;EArCxC;;;EAyCEC,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;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"]}

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":";;;;;;;;;;;;AAGC,SAAA,YAAA,CAAA,UAAA,EAAA,MAAA,EAAA,GAAA,EAAA,IAAA,EAAA;;;;;;AAAA,MAAA,CAAA,YAAA,EAAA,cAAA,CAAA;;;;;AA2BM,MAAMA,aAAAA,GAAgB;EAC3BC,WAAAA,EAAa,cAAA;EACbC,KAAAA,EAAO;AACT;AAEO,MAAMC,sBAAsBC,uBAAAA,CAAAA;EAnCnC;;;AAmCwD;AAEjD,MAAMC,gBAAgBC,aAAAA,CAAAA;EArC7B;;;EAsCEC,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;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"]}

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

@@ -5,7 +5,7 @@ import { Provider } from './metadata.cjs';
 import { AnyToolSchemaLike } from '../../internals/helpers/schema.cjs';
 import 'ajv';
 import '../../context.cjs';
-import '../../emitter-DWEsd-ud.cjs';
+import '../../emitter-DpqUYjXH.cjs';
 import '../../internals/types.cjs';
 import '../../internals/helpers/guards.cjs';
 import '../../internals/serializable.cjs';

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

@@ -5,7 +5,7 @@ import { Provider } from './metadata.js';
 import { AnyToolSchemaLike } from '../../internals/helpers/schema.js';
 import 'ajv';
 import '../../context.js';
-import '../../emitter-C9nwGLt6.js';
+import '../../emitter-CZFbzlUi.js';
 import '../../internals/types.js';
 import '../../internals/helpers/guards.js';
 import '../../internals/serializable.js';