beeai-framework 0.1.12 → 0.1.14

package/dist/tools/database/qdrant.js

@@ -0,0 +1,171 @@
+import { Tool, ToolInputValidationError, ToolError, JSONToolOutput } from '../base.js';
+import { Cache } from '../../cache/decoratorCache.js';
+import { QdrantClient } from '@qdrant/js-client-rest';
+import { v4 } from 'uuid';
+import { z } from 'zod';
+import { Emitter } from '../../emitter/emitter.js';
+
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+function _ts_decorate(decorators, target, key, desc) {
+  var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+  if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+  else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+  return c > 3 && r && Object.defineProperty(target, key, r), r;
+}
+__name(_ts_decorate, "_ts_decorate");
+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) {
+  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");
+  }
+  name = "QdrantDatabaseTool";
+  description = `Can query data from a Qdrant vector database. IMPORTANT: strictly follow this order of actions:
+     1. ${"ListCollections"} - List all the Qdrant collections
+     2. ${"GetCollectionInfo"} - Get information about a Qdrant collection
+     3. ${"Insert"} - Insert data into a Qdrant collection
+     3. ${"Search"} - Perform search on a Qdrant collection
+     4. ${"Delete"} - Delete from a Qdrant collection`;
+  inputSchema() {
+    return z.object({
+      action: z.nativeEnum(QdrantAction).describe(`The action to perform. ${"ListCollections"} lists all collections, ${"GetCollectionInfo"} fetches details for a specified collection, ${"Search"} executes a vector search, ${"Insert"} inserts new vectors, and ${"Delete"} removes vectors.`),
+      collectionName: z.string().optional().describe(`The name of the collection to query, required for ${"GetCollectionInfo"}, ${"Search"}, ${"Insert"}, and ${"Delete"}`),
+      vector: z.array(z.number()).optional().describe(`The vector to search for, required for ${"Search"}.`),
+      vectors: z.array(z.array(z.number())).optional().describe(`The vectors to insert, required for ${"Insert"}.`),
+      topK: z.coerce.number().int().default(10).optional().describe(`The number of nearest neighbors to return.`),
+      filter: z.record(z.string(), z.any()).optional().describe(`Optional filter for ${"Search"}.`),
+      payload: z.array(z.record(z.string(), z.any())).optional().describe(`Additional payload to insert with vectors.`),
+      ids: z.array(z.string().or(z.number())).optional().describe(`Array of IDs to delete or insert.`)
+    });
+  }
+  emitter = Emitter.root.child({
+    namespace: [
+      "tool",
+      "database",
+      "qdrant"
+    ],
+    creator: this
+  });
+  validateInput(schema, input) {
+    super.validateInput(schema, input);
+    if (input.action != "ListCollections" && !input.collectionName) {
+      throw new ToolInputValidationError(`Collection name is required for ${"GetCollectionInfo"}, ${"Search"}, ${"Insert"}, and ${"Delete"} actions.`);
+    }
+    if (input.action === "Search" && (!input.collectionName || !input.vector)) {
+      throw new ToolInputValidationError(`Vector is required for ${"Search"} action.`);
+    }
+    if (input.action === "Insert" && (!input.collectionName || !input.vectors)) {
+      throw new ToolInputValidationError(`Vectors are required for ${"Insert"} action.`);
+    }
+  }
+  static {
+    this.register();
+  }
+  async client() {
+    return new QdrantClient(this.options.connection);
+  }
+  async _run(input, _options) {
+    switch (input.action) {
+      case "ListCollections": {
+        const collections = await this.listCollections();
+        return new JSONToolOutput(collections);
+      }
+      case "GetCollectionInfo": {
+        if (!input.collectionName) {
+          throw new ToolError("A collection name is required for Qdrant GetCollectionInfo action");
+        }
+        const collectionInfo = await this.getCollectionInfo(input.collectionName);
+        return new JSONToolOutput(collectionInfo);
+      }
+      case "Search": {
+        if (!input.collectionName || !input.vector) {
+          throw new ToolError("A collection name and vector are required for Qdrant Search action");
+        }
+        const searchResults = await this.search(input);
+        return new JSONToolOutput(searchResults);
+      }
+      case "Insert": {
+        if (!input.collectionName || !input.vectors) {
+          throw new ToolError("A collection name and vectors are required for Qdrant Insert action");
+        }
+        const insertResults = await this.insert(input);
+        return new JSONToolOutput(insertResults);
+      }
+      case "Delete": {
+        if (!input.collectionName || !input.ids) {
+          throw new ToolError("Collection name and ids are required for Qdrant Delete action");
+        }
+        const deleteResults = await this.delete(input);
+        return new JSONToolOutput(deleteResults);
+      }
+      default: {
+        throw new ToolError(`Invalid action specified: ${input.action}`);
+      }
+    }
+  }
+  async listCollections() {
+    try {
+      const client = await this.client();
+      const response = await client.getCollections();
+      return response.collections.map((collection) => collection.name);
+    } catch (error) {
+      throw new ToolError(`Failed to list collections from Qdrant: ${error}`);
+    }
+  }
+  async getCollectionInfo(collectionName) {
+    const client = await this.client();
+    const response = await client.getCollection(collectionName);
+    return response;
+  }
+  async insert(input) {
+    const client = await this.client();
+    const points = input.vectors.map((vector, index) => ({
+      id: input?.ids?.[index] ?? v4(),
+      vector,
+      payload: input?.payload?.[index] || {}
+    }));
+    const response = await client.upsert(input.collectionName, {
+      points
+    });
+    return response;
+  }
+  async search(input) {
+    const client = await this.client();
+    const searchParams = {
+      query: input.vector,
+      limit: input.topK || 10,
+      with_payload: true,
+      filter: input.filter
+    };
+    const response = await client.query(input.collectionName, searchParams);
+    return response;
+  }
+  async delete(input) {
+    const client = await this.client();
+    const response = await client.delete(input.collectionName, {
+      points: input.ids
+    });
+    return response;
+  }
+}
+_ts_decorate([
+  Cache(),
+  _ts_metadata("design:type", Function),
+  _ts_metadata("design:paramtypes", []),
+  _ts_metadata("design:returntype", Promise)
+], QdrantDatabaseTool.prototype, "client", null);
+
+export { QdrantAction, QdrantDatabaseTool };
+//# sourceMappingURL=qdrant.js.map
+//# sourceMappingURL=qdrant.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["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":";;;;;;;;;AAcC,SAAA,YAAA,CAAA,UAAA,EAAA,MAAA,EAAA,GAAA,EAAA,IAAA,EAAA;;;;;;AAAA,MAAA,CAAA,YAAA,EAAA,cAAA,CAAA;;;;;AA0BWA,IAAAA,YAAAA,4BAAAA,aAAAA,EAAAA;;;;;;AAAAA,EAAAA,OAAAA,aAAAA;;AAQL,MAAMC,2BAA2BC,IAAAA,CAAAA;EAhDxC;;;EAoDEC,IAAO,GAAA,oBAAA;EAEPC,WAAc,GAAA,CAAA;UACR,iBAAA,CAAA;UACA,mBAAA,CAAA;UACA,QAAA,CAAA;UACA,QAAA,CAAA;UACA,QAAA,CAAA,kCAAA,CAAA;EAENC,WAAc,GAAA;AACZ,IAAA,OAAOC,EAAEC,MAAO,CAAA;AACdC,MAAAA,MAAAA,EAAQF,CACLG,CAAAA,UAAAA,CAAWT,YAAAA,CAAAA,CACXU,SACC,CAAwB,uBAAA,EAAA,iBAAA,CAAuD,wBAAA,EAAA,mBAAA,gDAA8E,QAAA,CAAA,2BAAA,EAAiD,QAAA,CAAA,0BAAA,EAAgD,QAAA,CAAwC,iBAAA,CAAA,CAAA;AAE1SC,MAAAA,cAAAA,EAAgBL,CACbM,CAAAA,MAAAA,EACAC,CAAAA,QAAAA,GACAH,QACC,CAAA,CAAA,kDAAA,EAAmD,mBAAA,CAAA,EAAA,EAAmC,QAAA,CAAA,EAAA,EAAwB,QAAA,CAAA,MAAA,EAA4B,QAAA,CAAuB,CAAA,CAAA;MAErKI,MAAQR,EAAAA,CAAAA,CACLS,KAAMT,CAAAA,CAAAA,CAAEU,MAAM,EAAA,CACdH,CAAAA,QAAAA,EACAH,CAAAA,QAAAA,CAAS,CAAwC,uCAAA,EAAA,QAAA,CAAwB,CAAA,CAAA,CAAA;AAC5EO,MAAAA,OAAAA,EAASX,CACNS,CAAAA,KAAAA,CAAMT,CAAES,CAAAA,KAAAA,CAAMT,EAAEU,MAAM,EAAA,CAAA,CAAA,CACtBH,QAAQ,EAAA,CACRH,QAAS,CAAA,CAAA,oCAAA,EAAqC,QAAA,CAAwB,CAAA,CAAA,CAAA;AACzEQ,MAAAA,IAAAA,EAAMZ,CAAEa,CAAAA,MAAAA,CACLH,MAAM,EAAA,CACNI,GAAG,EAAA,CACHC,OAAQ,CAAA,EAAA,CACRR,CAAAA,QAAAA,EACAH,CAAAA,QAAAA,CAAS,CAA4C,0CAAA,CAAA,CAAA;AACxDY,MAAAA,MAAAA,EAAQhB,CACLiB,CAAAA,MAAAA,CAAOjB,CAAEM,CAAAA,MAAAA,IAAUN,CAAEkB,CAAAA,GAAAA,EAAG,CAAA,CACxBX,QAAQ,EAAA,CACRH,QAAS,CAAA,CAAA,oBAAA,EAAqB,QAAA,CAAwB,CAAA,CAAA,CAAA;AACzDe,MAAAA,OAAAA,EAASnB,CACNS,CAAAA,KAAAA,CAAMT,CAAEiB,CAAAA,MAAAA,CAAOjB,EAAEM,MAAM,EAAA,EAAIN,CAAEkB,CAAAA,GAAAA,EAAG,CAAA,CAAA,CAChCX,QAAQ,EAAA,CACRH,SAAS,CAA4C,0CAAA,CAAA,CAAA;AACxDgB,MAAAA,GAAAA,EAAKpB,CACFS,CAAAA,KAAAA,CAAMT,CAAEM,CAAAA,MAAAA,GAASe,EAAGrB,CAAAA,CAAAA,CAAEU,MAAM,EAAA,CAAA,CAAA,CAC5BH,QAAQ,EAAA,CACRH,SAAS,CAAmC,iCAAA,CAAA;KACjD,CAAA;AACF;EAEgBkB,OACdC,GAAAA,OAAAA,CAAQC,KAAKC,KAAM,CAAA;IACjBC,SAAW,EAAA;AAAC,MAAA,MAAA;AAAQ,MAAA,UAAA;AAAY,MAAA;;IAChCC,OAAS,EAAA;GACX,CAAA;AAEQC,EAAAA,aAAAA,CACRC,QACAC,KACkC,EAAA;AAClC,IAAMF,KAAAA,CAAAA,aAAAA,CAAcC,QAAQC,KAAAA,CAAAA;AAC5B,IAAA,IAAIA,KAAM5B,CAAAA,MAAAA,IAAM,iBAAoC,IAAA,CAAC4B,MAAMzB,cAAgB,EAAA;AACzE,MAAM,MAAA,IAAI0B,wBACR,CAAA,CAAA,gCAAA,EAAiC,mBAAA,CAAA,EAAA,EAAmC,QAAA,CAAwB,EAAA,EAAA,QAAA,CAA4B,MAAA,EAAA,QAAA,CAAgC,SAAA,CAAA,CAAA;AAE5J;AACA,IAAID,IAAAA,KAAAA,CAAM5B,WAAM,QAA6B,KAAA,CAAC4B,MAAMzB,cAAkB,IAAA,CAACyB,MAAMtB,MAAS,CAAA,EAAA;AACpF,MAAA,MAAM,IAAIuB,wBAAAA,CAAyB,CAAwB,uBAAA,EAAA,QAAA,CAA+B,QAAA,CAAA,CAAA;AAC5F;AACA,IAAID,IAAAA,KAAAA,CAAM5B,WAAM,QAA6B,KAAA,CAAC4B,MAAMzB,cAAkB,IAAA,CAACyB,MAAMnB,OAAU,CAAA,EAAA;AACrF,MAAA,MAAM,IAAIoB,wBAAAA,CAAyB,CAA0B,yBAAA,EAAA,QAAA,CAA+B,QAAA,CAAA,CAAA;AAC9F;AACF;EAEA;AACE,IAAA,IAAA,CAAKC,QAAQ,EAAA;AACf;AAEA,EAAA,MACgBC,MAAgC,GAAA;AAC9C,IAAA,OAAO,IAAIC,YAAAA,CAAa,IAAKC,CAAAA,OAAAA,CAAQC,UAAU,CAAA;AACjD;EAEA,MAAgBC,IAAAA,CACdP,OACAQ,QAC8B,EAAA;AAC9B,IAAA,QAAQR,MAAM5B,MAAM;AAClB,MAAA,KAAA,iBAAmC,EAAA;AACjC,QAAMqC,MAAAA,WAAAA,GAAc,MAAM,IAAA,CAAKC,eAAe,EAAA;AAC9C,QAAO,OAAA,IAAIC,eAAeF,WAAAA,CAAAA;AAC5B;AAEA,MAAA,KAAA,mBAAqC,EAAA;AACnC,QAAI,IAAA,CAACT,MAAMzB,cAAgB,EAAA;AACzB,UAAM,MAAA,IAAIqC,UAAU,mEAAA,CAAA;AACtB;AACA,QAAA,MAAMC,cAAiB,GAAA,MAAM,IAAKC,CAAAA,iBAAAA,CAAkBd,MAAMzB,cAAc,CAAA;AACxE,QAAO,OAAA,IAAIoC,eAAeE,cAAAA,CAAAA;AAC5B;AAEA,MAAA,KAAA,QAA0B,EAAA;AACxB,QAAA,IAAI,CAACb,KAAAA,CAAMzB,cAAkB,IAAA,CAACyB,MAAMtB,MAAQ,EAAA;AAC1C,UAAM,MAAA,IAAIkC,UAAU,oEAAA,CAAA;AACtB;AACA,QAAA,MAAMG,aAAgB,GAAA,MAAM,IAAKC,CAAAA,MAAAA,CAAOhB,KAAAA,CAAAA;AACxC,QAAO,OAAA,IAAIW,eAAeI,aAAAA,CAAAA;AAC5B;AAEA,MAAA,KAAA,QAA0B,EAAA;AACxB,QAAA,IAAI,CAACf,KAAAA,CAAMzB,cAAkB,IAAA,CAACyB,MAAMnB,OAAS,EAAA;AAC3C,UAAM,MAAA,IAAI+B,UACR,qEAAA,CAAA;AAEJ;AACA,QAAA,MAAMK,aAAgB,GAAA,MAAM,IAAKC,CAAAA,MAAAA,CAAOlB,KAAAA,CAAAA;AACxC,QAAO,OAAA,IAAIW,eAAeM,aAAAA,CAAAA;AAC5B;AAEA,MAAA,KAAA,QAA0B,EAAA;AACxB,QAAA,IAAI,CAACjB,KAAAA,CAAMzB,cAAkB,IAAA,CAACyB,MAAMV,GAAK,EAAA;AACvC,UAAM,MAAA,IAAIsB,UAAU,+DAAA,CAAA;AACtB;AACA,QAAA,MAAMO,aAAgB,GAAA,MAAM,IAAKC,CAAAA,MAAAA,CAAOpB,KAAAA,CAAAA;AACxC,QAAO,OAAA,IAAIW,eAAeQ,aAAAA,CAAAA;AAC5B;MAEA,SAAS;AACP,QAAA,MAAM,IAAIP,SAAAA,CAAU,CAA6BZ,0BAAAA,EAAAA,KAAAA,CAAM5B,MAAM,CAAE,CAAA,CAAA;AACjE;AACF;AACF;AAEA,EAAA,MAAgBsC,eAAqC,GAAA;AACnD,IAAI,IAAA;AACF,MAAMP,MAAAA,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAM,EAAA;AAChC,MAAMkB,MAAAA,QAAAA,GAAW,MAAMlB,MAAAA,CAAOmB,cAAc,EAAA;AAE5C,MAAA,OAAOD,SAASZ,WAAYc,CAAAA,GAAAA,CAAI,CAACC,UAAAA,KAAeA,WAAWzD,IAAI,CAAA;AACjE,KAAA,CAAA,OAAS0D,KAAO,EAAA;AACd,MAAA,MAAM,IAAIb,SAAAA,CAAU,CAA2Ca,wCAAAA,EAAAA,KAAAA,CAAO,CAAA,CAAA;AACxE;AACF;AAEA,EAAA,MAAgBX,kBAAkBvC,cAAsC,EAAA;AACtE,IAAM4B,MAAAA,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAM,EAAA;AAChC,IAAA,MAAMkB,QAAW,GAAA,MAAMlB,MAAOuB,CAAAA,aAAAA,CAAcnD,cAAAA,CAAAA;AAC5C,IAAO8C,OAAAA,QAAAA;AACT;AAEA,EAAA,MAAgBH,OAAOlB,KAAsC,EAAA;AAC3D,IAAMG,MAAAA,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAM,EAAA;AAEhC,IAAA,MAAMwB,SAAS3B,KAAMnB,CAAAA,OAAAA,CAAS0C,GAAI,CAAA,CAAC7C,QAAQkD,KAAW,MAAA;AACpDC,MAAAA,EAAAA,EAAI7B,KAAOV,EAAAA,GAAAA,GAAMsC,KAAAA,CAAAA,IAAUE,EAAAA,EAAAA;AAC3BpD,MAAAA,MAAAA;AACAW,MAAAA,OAAAA,EAASW,KAAOX,EAAAA,OAAAA,GAAUuC,KAAAA,CAAAA,IAAU;KACtC,CAAA,CAAA;AAEA,IAAA,MAAMP,QAAW,GAAA,MAAMlB,MAAO4B,CAAAA,MAAAA,CAAO/B,MAAMzB,cAA0B,EAAA;AACnEoD,MAAAA;KACF,CAAA;AAEA,IAAON,OAAAA,QAAAA;AACT;AAEA,EAAA,MAAgBL,OAAOhB,KAAsC,EAAA;AAC3D,IAAMG,MAAAA,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAM,EAAA;AAEhC,IAAA,MAAM6B,YAAoB,GAAA;AACxBC,MAAAA,KAAAA,EAAOjC,KAAMtB,CAAAA,MAAAA;AACbwD,MAAAA,KAAAA,EAAOlC,MAAMlB,IAAQ,IAAA,EAAA;MACrBqD,YAAc,EAAA,IAAA;AACdjD,MAAAA,MAAAA,EAAQc,KAAMd,CAAAA;AAChB,KAAA;AAEA,IAAA,MAAMmC,WAAW,MAAMlB,MAAAA,CAAO8B,KAAMjC,CAAAA,KAAAA,CAAMzB,gBAA0ByD,YAAAA,CAAAA;AACpE,IAAOX,OAAAA,QAAAA;AACT;AAEA,EAAA,MAAgBD,OAAOpB,KAAsC,EAAA;AAC3D,IAAMG,MAAAA,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAM,EAAA;AAEhC,IAAA,MAAMkB,QAAW,GAAA,MAAMlB,MAAOiB,CAAAA,MAAAA,CAAOpB,MAAMzB,cAA0B,EAAA;AACnEoD,MAAAA,MAAAA,EAAQ3B,KAAMV,CAAAA;KAChB,CAAA;AAEA,IAAO+B,OAAAA,QAAAA;AACT;AACF","file":"qdrant.js","sourcesContent":["/**\n * Copyright 2025 © BeeAI a Series of LF Projects, LLC\n *\n * Licensed under the Apache License, Version 2.0 (the \"License\");\n * you may not use this file except in compliance with the License.\n * You may obtain a copy of the License at\n *\n *     http://www.apache.org/licenses/LICENSE-2.0\n *\n * Unless required by applicable law or agreed to in writing, software\n * distributed under the License is distributed on an \"AS IS\" BASIS,\n * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\n * See the License for the specific language governing permissions and\n * limitations under the License.\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/version.cjs

@@ -2,7 +2,7 @@
 
 exports.Version = "0.0.0";
 try {
-  exports.Version = "0.1.12";
+  exports.Version = "0.1.14";
 } catch {
 }
 //# sourceMappingURL=version.cjs.map

package/dist/version.js

@@ -1,6 +1,6 @@
 let Version = "0.0.0";
 try {
-  Version = "0.1.12";
+  Version = "0.1.14";
 } catch {
 }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "beeai-framework",
-  "version": "0.1.12",
+  "version": "0.1.14",
   "license": "Apache-2.0",
   "description": "BeeAI Framework - LLM Agent Framework",
   "author": "BeeAI a Series of LF Projects, LLC",
@@ -183,7 +183,7 @@
   "dependencies": {
     "@ai-zen/node-fetch-event-source": "^2.1.4",
     "@streamparser/json": "^0.0.22",
-    "ai": "^4.3.8",
+    "ai": "^4.3.13",
     "ajv": "^8.17.1",
     "ajv-formats": "^3.0.1",
     "duck-duck-scrape": "^2.2.7",
@@ -213,14 +213,14 @@
     "@ai-sdk/azure": "^1.3.14",
     "@ai-sdk/google-vertex": "^2.2.15",
     "@ai-sdk/groq": "^1.2.8",
-    "@ai-sdk/openai": "^1.3.13",
+    "@ai-sdk/openai": "^1.3.21",
     "@aws-sdk/client-bedrock-runtime": "^3.772.0",
     "@elastic/elasticsearch": "^8.0.0",
     "@googleapis/customsearch": "^3.2.0",
-    "@i-am-bee/acp-sdk": "^0.0.6",
     "@langchain/community": ">=0.2.28",
     "@langchain/core": ">=0.2.27",
     "@modelcontextprotocol/sdk": "^1.0.4",
+    "@qdrant/js-client-rest": "^1.14.0",
     "@zilliz/milvus2-sdk-node": "^2.4.9",
     "ollama-ai-provider": "^1.2.0",
     "sequelize": "^6.37.3",
@@ -251,9 +251,6 @@
     "@googleapis/customsearch": {
       "optional": true
     },
-    "@i-am-bee/acp-sdk": {
-      "optional": true
-    },
     "@ibm-cloud/watsonx-ai": {
       "optional": true
     },
@@ -266,6 +263,9 @@
     "@modelcontextprotocol/sdk": {
       "optional": true
     },
+    "@qdrant/js-client-rest": {
+      "optional": true
+    },
     "@zilliz/milvus2-sdk-node": {
       "optional": true
     },
@@ -292,13 +292,13 @@
     "@eslint/js": "^9.16.0",
     "@eslint/markdown": "^6.2.1",
     "@googleapis/customsearch": "^3.2.0",
-    "@i-am-bee/acp-sdk": "^0.0.4",
     "@ibm-cloud/watsonx-ai": "^1.6.0",
     "@langchain/community": "0.3.34",
     "@langchain/core": "0.3.42",
     "@langchain/langgraph": "0.2.55",
     "@langchain/ollama": "^0.2.0",
     "@modelcontextprotocol/sdk": "^1.0.4",
+    "@qdrant/js-client-rest": "^1.14.0",
     "@release-it/conventional-changelog": "^8.0.2",
     "@rollup/plugin-commonjs": "^28.0.1",
     "@stylistic/eslint-plugin-js": "^2.12.1",