beeai-framework 0.1.2

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/tools/database/elasticsearch.ts"],"names":["ElasticSearchAction","ListIndices","GetIndexDetails","Search","ElasticSearchTool","Tool","name","description","inputSchema","z","object","action","nativeEnum","describe","indexName","string","optional","query","start","coerce","number","int","min","default","size","max","emitter","Emitter","root","child","namespace","creator","validateInput","schema","input","ToolInputValidationError","register","options","connection","cloud","node","nodes","ValidationError","message","propertyName","client","Client","info","error","ToolError","isRetryable","isFatal","_run","_options","run","indices","listIndices","signal","JSONToolOutput","indexDetails","getIndexDetails","response","search","aggregations","hits","map","hit","_source","cat","expand_wildcards","h","format","filter","record","index","startsWith","getMapping","parsedQuery","parseBrokenJson","searchBody","from","body"],"mappings":";;;;;;;;;;AAcC,SAAA,YAAA,CAAA,UAAA,EAAA,MAAA,EAAA,GAAA,EAAA,IAAA,EAAA;;;;;;AAAA,MAAA,CAAA,YAAA,EAAA,cAAA,CAAA;;;;;AA8BM,MAAMA,mBAAsB,GAAA;EACjCC,WAAa,EAAA,cAAA;EACbC,eAAiB,EAAA,mBAAA;EACjBC,MAAQ,EAAA;AACV;AAEO,MAAMC,0BAA0BC,IAAAA,CAAAA;EAlDvC;;;EAsDEC,IAAO,GAAA,mBAAA;EAEPC,WAAc,GAAA,CAAA;AACRP,MAAAA,EAAAA,mBAAAA,CAAoBC,WAAW,CAAA;AAC/BD,MAAAA,EAAAA,mBAAAA,CAAoBE,eAAe,CAAA;AACnCF,MAAAA,EAAAA,mBAAAA,CAAoBG,MAAM,CAAA,qLAAA,CAAA;EAEhCK,WAAc,GAAA;AACZ,IAAA,OAAOC,EAAEC,MAAO,CAAA;AACdC,MAAAA,MAAAA,EAAQF,CACLG,CAAAA,UAAAA,CAAWZ,mBAAAA,CAAAA,CACXa,SACC,CAA0Bb,uBAAAA,EAAAA,mBAAAA,CAAoBC,WAAW,CAAA,oBAAA,EAAuBD,mBAAoBE,CAAAA,eAAe,CAA+CF,4CAAAA,EAAAA,mBAAAA,CAAoBG,MAAM,CAAyC,uCAAA,CAAA,CAAA;AAEzOW,MAAAA,SAAAA,EAAWL,CACRM,CAAAA,MAAAA,EACAC,CAAAA,QAAAA,EACAH,CAAAA,QAAAA,CACC,CAAgDb,6CAAAA,EAAAA,mBAAAA,CAAoBE,eAAe,CAAA,KAAA,EAAQF,mBAAoBG,CAAAA,MAAM,CAAE,CAAA,CAAA;MAE3Hc,KAAOR,EAAAA,CAAAA,CACJM,QACAC,CAAAA,QAAAA,GACAH,QACC,CAAA,CAAA,yDAAA,EAA4Db,mBAAoBG,CAAAA,MAAM,CAAS,OAAA,CAAA,CAAA;AAEnGe,MAAAA,KAAAA,EAAOT,CAAEU,CAAAA,MAAAA,CACNC,MAAM,EAAA,CACNC,KACAC,CAAAA,GAAAA,CAAI,CAAA,CAAA,CACJC,QAAQ,CAAA,CAAA,CACRP,QAAQ,EAAA,CACRH,SACC,qHAAA,CAAA;AAEJW,MAAAA,IAAAA,EAAMf,EAAEU,MACLC,CAAAA,MAAAA,GACAC,GAAG,EAAA,CACHC,IAAI,CAAA,CAAA,CACJG,GAAI,CAAA,EAAA,EACJF,OAAQ,CAAA,EAAA,EACRP,QAAQ,EAAA,CACRH,SAAS,gFAAA;KACd,CAAA;AACF;EAEgBa,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,MAAMvB,MAAWX,KAAAA,mBAAAA,CAAoBE,eAAmB,IAAA,CAACgC,MAAMpB,SAAW,EAAA;AAC5E,MAAA,MAAM,IAAIqB,wBAAAA,CACR,CAA8BnC,2BAAAA,EAAAA,mBAAAA,CAAoBE,eAAe,CAAU,QAAA,CAAA,CAAA;AAE/E;AACA,IAAIgC,IAAAA,KAAAA,CAAMvB,WAAWX,mBAAoBG,CAAAA,MAAAA,KAAW,CAAC+B,KAAMpB,CAAAA,SAAAA,IAAa,CAACoB,KAAAA,CAAMjB,KAAQ,CAAA,EAAA;AACrF,MAAA,MAAM,IAAIkB,wBAAAA,CACR,CAA8CnC,2CAAAA,EAAAA,mBAAAA,CAAoBG,MAAM,CAAU,QAAA,CAAA,CAAA;AAEtF;AACF;EAEA;AACE,IAAA,IAAA,CAAKiC,QAAQ,EAAA;AACf;AAEA,EAAA,WAAA,CAAmBC,OAAmC,EAAA;AACpD,IAAA,KAAA,CAAMA,OAAAA,CAAAA;AACN,IAAI,IAAA,CAACA,OAAQC,CAAAA,UAAAA,CAAWC,KAAS,IAAA,CAACF,OAAQC,CAAAA,UAAAA,CAAWE,IAAQ,IAAA,CAACH,OAAQC,CAAAA,UAAAA,CAAWG,KAAO,EAAA;AACtF,MAAA,MAAM,IAAIC,eAAgB,CAAA;AACxB,QAAA;UACEC,OAAS,EAAA,iDAAA;UACTC,YAAc,EAAA;AAChB;AACD,OAAA,CAAA;AACH;AACF;AAEA,EAAA,MACgBC,MAA0B,GAAA;AACxC,IAAI,IAAA;AACF,MAAA,MAAMA,MAAS,GAAA,IAAIC,MAAO,CAAA,IAAA,CAAKT,QAAQC,UAAU,CAAA;AACjD,MAAA,MAAMO,OAAOE,IAAI,EAAA;AACjB,MAAOF,OAAAA,MAAAA;AACT,KAAA,CAAA,OAASG,KAAO,EAAA;AACd,MAAM,MAAA,IAAIC,UAAU,CAAuC,mCAAA,CAAA,EAAA;AAACD,QAAAA;AAAQ,OAAA,EAAA;QAClEE,WAAa,EAAA,KAAA;QACbC,OAAS,EAAA;OACX,CAAA;AACF;AACF;EAEA,MAAgBC,IAAAA,CACdlB,KACAmB,EAAAA,QAAAA,EACAC,GAC8B,EAAA;AAC9B,IAAIpB,IAAAA,KAAAA,CAAMvB,MAAWX,KAAAA,mBAAAA,CAAoBC,WAAa,EAAA;AACpD,MAAA,MAAMsD,OAAU,GAAA,MAAM,IAAKC,CAAAA,WAAAA,CAAYF,IAAIG,MAAM,CAAA;AACjD,MAAO,OAAA,IAAIC,eAAeH,OAAAA,CAAAA;KACjBrB,MAAAA,IAAAA,KAAAA,CAAMvB,MAAWX,KAAAA,mBAAAA,CAAoBE,eAAiB,EAAA;AAC/D,MAAA,MAAMyD,eAAe,MAAM,IAAA,CAAKC,eAAgB1B,CAAAA,KAAAA,EAAOoB,IAAIG,MAAM,CAAA;AACjE,MAAO,OAAA,IAAIC,eAAeC,YAAAA,CAAAA;KACjBzB,MAAAA,IAAAA,KAAAA,CAAMvB,MAAWX,KAAAA,mBAAAA,CAAoBG,MAAQ,EAAA;AACtD,MAAA,MAAM0D,WAAW,MAAM,IAAA,CAAKC,MAAO5B,CAAAA,KAAAA,EAAOoB,IAAIG,MAAM,CAAA;AACpD,MAAA,IAAII,SAASE,YAAc,EAAA;AACzB,QAAO,OAAA,IAAIL,cAAeG,CAAAA,QAAAA,CAASE,YAAY,CAAA;OAC1C,MAAA;AACL,QAAO,OAAA,IAAIL,cAAeG,CAAAA,QAAAA,CAASG,IAAKA,CAAAA,IAAAA,CAAKC,IAAI,CAACC,GAAAA,KAA2BA,GAAIC,CAAAA,OAAO,CAAA,CAAA;AAC1F;KACK,MAAA;AACL,MAAA,MAAM,IAAIlB,SAAAA,CAAU,CAA6Bf,0BAAAA,EAAAA,KAAAA,CAAMvB,MAAM,CAAE,CAAA,CAAA;AACjE;AACF;AAEA,EAAA,MAAgB6C,YAAYC,MAA2D,EAAA;AACrF,IAAMZ,MAAAA,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAM,EAAA;AAChC,IAAA,MAAMgB,QAAW,GAAA,MAAMhB,MAAOuB,CAAAA,GAAAA,CAAIb,OAChC,CAAA;MACEc,gBAAkB,EAAA,MAAA;MAClBC,CAAG,EAAA,OAAA;MACHC,MAAQ,EAAA;KAEV,EAAA;AAAEd,MAAAA;KAAe,CAAA;AAEnB,IAAA,OAAOI,QACJW,CAAAA,MAAAA,CAAO,CAACC,MAAAA,KAAWA,OAAOC,KAAS,IAAA,CAACD,MAAOC,CAAAA,KAAAA,CAAMC,WAAW,GAAA,CAAA,CAC5DV,CAAAA,GAAAA,CAAI,CAACQ,MAAY,MAAA;AAAEC,MAAAA,KAAAA,EAAOD,MAAOC,CAAAA;KAAM,CAAA,CAAA;AAC5C;EAEA,MAAgBd,eAAAA,CACd1B,OACAuB,MAC4C,EAAA;AAC5C,IAAMZ,MAAAA,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAM,EAAA;AAChC,IAAO,OAAA,MAAMA,MAAOU,CAAAA,OAAAA,CAAQqB,UAC1B,CAAA;AACEF,MAAAA,KAAAA,EAAOxC,KAAMpB,CAAAA;KAEf,EAAA;AAAE2C,MAAAA;KAAe,CAAA;AAErB;EAEA,MAAgBK,MAAAA,CACd5B,OACAuB,MACiC,EAAA;AACjC,IAAMoB,MAAAA,WAAAA,GAAcC,eAAgB5C,CAAAA,KAAAA,CAAMjB,KAAK,CAAA;AAC/C,IAAA,MAAM8D,UAAoC,GAAA;MACxC,GAAGF,WAAAA;MACHG,IAAMH,EAAAA,WAAAA,CAAYG,QAAQ9C,KAAMhB,CAAAA,KAAAA;MAChCM,IAAMqD,EAAAA,WAAAA,CAAYrD,QAAQU,KAAMV,CAAAA;AAClC,KAAA;AACA,IAAMqB,MAAAA,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAM,EAAA;AAChC,IAAO,OAAA,MAAMA,OAAOiB,MAClB,CAAA;AACEY,MAAAA,KAAAA,EAAOxC,KAAMpB,CAAAA,SAAAA;MACbmE,IAAMF,EAAAA;KAER,EAAA;AAAEtB,MAAAA;KAAe,CAAA;AAErB;AACF","file":"elasticsearch.js","sourcesContent":["/**\n * Copyright 2025 IBM Corp.\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 { RunContext } from \"@/context.js\";\nimport { z } from \"zod\";\nimport { ValidationError } from \"ajv\";\nimport { AnyToolSchemaLike } from \"@/internals/helpers/schema.js\";\nimport { parseBrokenJson } from \"@/internals/helpers/schema.js\";\nimport { Client, ClientOptions, estypes as ESTypes } from \"@elastic/elasticsearch\";\nimport { Emitter } from \"@/emitter/emitter.js\";\n\nexport interface ElasticSearchToolOptions extends BaseToolOptions {\n  connection: ClientOptions;\n}\n\nexport type ElasticSearchToolResult =\n  | ESTypes.CatIndicesResponse\n  | ESTypes.IndicesGetMappingResponse\n  | ESTypes.SearchResponse;\n\nexport const ElasticSearchAction = {\n  ListIndices: \"LIST_INDICES\",\n  GetIndexDetails: \"GET_INDEX_DETAILS\",\n  Search: \"SEARCH\",\n} as const;\n\nexport class ElasticSearchTool extends Tool<\n  JSONToolOutput<ElasticSearchToolResult>,\n  ElasticSearchToolOptions\n> {\n  name = \"ElasticSearchTool\";\n\n  description = `Can query data from an ElasticSearch database. IMPORTANT: strictly follow this order of actions:\n   1. ${ElasticSearchAction.ListIndices} - retrieve a list of available indices\n   2. ${ElasticSearchAction.GetIndexDetails} - get details of index fields\n   3. ${ElasticSearchAction.Search} - perform search or aggregation query on a specific index or pass the original user query without modifications if it's a valid JSON ElasticSearch query after identifying the index`;\n\n  inputSchema() {\n    return z.object({\n      action: z\n        .nativeEnum(ElasticSearchAction)\n        .describe(\n          `The action to perform. ${ElasticSearchAction.ListIndices} lists all indices, ${ElasticSearchAction.GetIndexDetails} fetches details for a specified index, and ${ElasticSearchAction.Search} executes a search or aggregation query`,\n        ),\n      indexName: z\n        .string()\n        .optional()\n        .describe(\n          `The name of the index to query, required for ${ElasticSearchAction.GetIndexDetails} and ${ElasticSearchAction.Search}`,\n        ),\n      query: z\n        .string()\n        .optional()\n        .describe(\n          `Valid ElasticSearch JSON search or aggregation query for ${ElasticSearchAction.Search} action`,\n        ),\n      start: z.coerce\n        .number()\n        .int()\n        .min(0)\n        .default(0)\n        .optional()\n        .describe(\n          \"The record index from which the query will start. Increase by the size of the query to get the next page of results\",\n        ),\n      size: z.coerce\n        .number()\n        .int()\n        .min(0)\n        .max(10)\n        .default(10)\n        .optional()\n        .describe(\"How many records will be retrieved from the ElasticSearch query. Maximum is 10\"),\n    });\n  }\n\n  public readonly emitter: ToolEmitter<ToolInput<this>, JSONToolOutput<ElasticSearchToolResult>> =\n    Emitter.root.child({\n      namespace: [\"tool\", \"database\", \"elasticsearch\"],\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 === ElasticSearchAction.GetIndexDetails && !input.indexName) {\n      throw new ToolInputValidationError(\n        `Index name is required for ${ElasticSearchAction.GetIndexDetails} action.`,\n      );\n    }\n    if (input.action === ElasticSearchAction.Search && (!input.indexName || !input.query)) {\n      throw new ToolInputValidationError(\n        `Both index name and query are required for ${ElasticSearchAction.Search} action.`,\n      );\n    }\n  }\n\n  static {\n    this.register();\n  }\n\n  public constructor(options: ElasticSearchToolOptions) {\n    super(options);\n    if (!options.connection.cloud && !options.connection.node && !options.connection.nodes) {\n      throw new ValidationError([\n        {\n          message: \"At least one of the properties must be provided\",\n          propertyName: \"connection.cloud, connection.node, connection.nodes\",\n        },\n      ]);\n    }\n  }\n\n  @Cache()\n  protected async client(): Promise<Client> {\n    try {\n      const client = new Client(this.options.connection);\n      await client.info();\n      return client;\n    } catch (error) {\n      throw new ToolError(`Unable to connect to ElasticSearch.`, [error], {\n        isRetryable: false,\n        isFatal: true,\n      });\n    }\n  }\n\n  protected async _run(\n    input: ToolInput<this>,\n    _options: Partial<BaseToolRunOptions>,\n    run: RunContext<this>,\n  ): Promise<JSONToolOutput<any>> {\n    if (input.action === ElasticSearchAction.ListIndices) {\n      const indices = await this.listIndices(run.signal);\n      return new JSONToolOutput(indices);\n    } else if (input.action === ElasticSearchAction.GetIndexDetails) {\n      const indexDetails = await this.getIndexDetails(input, run.signal);\n      return new JSONToolOutput(indexDetails);\n    } else if (input.action === ElasticSearchAction.Search) {\n      const response = await this.search(input, run.signal);\n      if (response.aggregations) {\n        return new JSONToolOutput(response.aggregations);\n      } else {\n        return new JSONToolOutput(response.hits.hits.map((hit: ESTypes.SearchHit) => hit._source));\n      }\n    } else {\n      throw new ToolError(`Invalid action specified: ${input.action}`);\n    }\n  }\n\n  protected async listIndices(signal?: AbortSignal): Promise<ESTypes.CatIndicesResponse> {\n    const client = await this.client();\n    const response = await client.cat.indices(\n      {\n        expand_wildcards: \"open\",\n        h: \"index\",\n        format: \"json\",\n      },\n      { signal: signal },\n    );\n    return response\n      .filter((record) => record.index && !record.index.startsWith(\".\")) // Exclude system indices\n      .map((record) => ({ index: record.index }));\n  }\n\n  protected async getIndexDetails(\n    input: ToolInput<this>,\n    signal: AbortSignal,\n  ): Promise<ESTypes.IndicesGetMappingResponse> {\n    const client = await this.client();\n    return await client.indices.getMapping(\n      {\n        index: input.indexName,\n      },\n      { signal: signal },\n    );\n  }\n\n  protected async search(\n    input: ToolInput<this>,\n    signal: AbortSignal,\n  ): Promise<ESTypes.SearchResponse> {\n    const parsedQuery = parseBrokenJson(input.query);\n    const searchBody: ESTypes.SearchRequest = {\n      ...parsedQuery,\n      from: parsedQuery.from || input.start,\n      size: parsedQuery.size || input.size,\n    };\n    const client = await this.client();\n    return await client.search(\n      {\n        index: input.indexName,\n        body: searchBody,\n      },\n      { signal: signal },\n    );\n  }\n}\n"]}

package/dist/tools/database/metadata.cjs

@@ -0,0 +1,114 @@
+'use strict';
+
+var base_cjs = require('../base.cjs');
+
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+async function getMetadata(sequelize, provider, schema) {
+  try {
+    const query = getMetadataQuery(provider, schema);
+    const [metadata] = await sequelize.query(query);
+    const tableMap = /* @__PURE__ */ new Map();
+    metadata.forEach(({ tableName, columnName, dataType }) => {
+      if (!tableMap.has(tableName)) {
+        tableMap.set(tableName, [
+          `Table '${tableName}' with columns: ${columnName} (${dataType})`
+        ]);
+      } else {
+        tableMap.get(tableName).push(`${columnName} (${dataType})`);
+      }
+    });
+    return Array.from(tableMap.values()).map((columns) => columns.join(", ")).join("; ");
+  } catch (error) {
+    throw new base_cjs.ToolError(`Error initializing metadata: ${error}`, [], {
+      isRetryable: false
+    });
+  }
+}
+__name(getMetadata, "getMetadata");
+function getMetadataQuery(provider, schema) {
+  const schemaName = schema ?? getDefaultSchema(provider);
+  switch (provider) {
+    case "mysql":
+    case "mariadb":
+      return `
+          SELECT t.table_name AS tableName, c.column_name AS columnName, 
+                 c.data_type AS dataType
+          FROM information_schema.tables t
+          JOIN information_schema.columns c ON t.table_name = c.table_name
+          WHERE t.table_schema = DATABASE();
+        `;
+    case "postgres":
+      return `
+          SELECT t.table_name AS "tableName", c.column_name AS "columnName", 
+                 c.data_type AS "dataType"
+          FROM information_schema.tables t
+          JOIN information_schema.columns c ON t.table_name = c.table_name
+          WHERE t.table_schema = lower('${schemaName}');
+        `;
+    case "mssql":
+      return `
+          SELECT t.name AS tableName, c.name AS columnName,
+                 ty.name AS dataType
+           FROM sys.tables t
+           JOIN sys.columns c ON t.object_id = c.object_id
+           JOIN sys.types ty ON c.user_type_id = ty.user_type_id
+           JOIN sys.schemas s ON t.schema_id = s.schema_id
+           WHERE t.is_ms_shipped = 0 AND t.type = 'U'
+                AND s.name = lower('${schemaName}');
+        `;
+    case "db2":
+      return `
+          SELECT t.tabname AS "tableName", c.colname AS "columnName", 
+                 c.typename AS "dataType"
+          FROM SYSCAT.TABLES t
+          JOIN SYSCAT.COLUMNS c ON t.tabname = c.tabname
+          WHERE t.tabschema = upper('${schemaName}');
+        `;
+    case "sqlite":
+      return `
+          SELECT tbl_name AS "tableName", name AS "columnName", type AS "dataType"
+            FROM (
+                SELECT name AS tbl_name
+                FROM sqlite_master
+                WHERE type = 'table'
+            )
+            JOIN pragma_table_xinfo(tbl_name);
+        `;
+    case "oracle":
+      return `
+          SELECT t.table_name AS "tableName", c.column_name AS "columnName", 
+                 c.data_type AS "dataType"
+          FROM all_tables t
+          JOIN all_tab_columns c ON t.table_name = c.table_name
+          WHERE t.owner = upper('${schemaName}');
+        `;
+    default:
+      throw new base_cjs.ToolError(`Provider ${provider} is not supported`, [], {
+        isFatal: true,
+        isRetryable: false
+      });
+  }
+}
+__name(getMetadataQuery, "getMetadataQuery");
+function getDefaultSchema(provider) {
+  switch (provider) {
+    case "postgres":
+      return "public";
+    case "mssql":
+      return "dbo";
+    case "db2":
+    case "oracle":
+      throw new base_cjs.ToolError(`Schema name is required for ${provider}`, [], {
+        isRetryable: false,
+        isFatal: true
+      });
+    default:
+      return "";
+  }
+}
+__name(getDefaultSchema, "getDefaultSchema");
+
+exports.getMetadata = getMetadata;
+//# sourceMappingURL=metadata.cjs.map
+//# sourceMappingURL=metadata.cjs.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/tools/database/metadata.ts"],"names":["getMetadata","sequelize","provider","schema","query","getMetadataQuery","metadata","tableMap","Map","forEach","tableName","columnName","dataType","has","set","get","push","Array","from","values","map","columns","join","error","ToolError","isRetryable","schemaName","getDefaultSchema","isFatal"],"mappings":";;;;;;AA2BA,eAAsBA,WAAAA,CACpBC,SACAC,EAAAA,QAAAA,EACAC,MAAe,EAAA;AAEf,EAAI,IAAA;AACF,IAAMC,MAAAA,KAAAA,GAAQC,gBAAiBH,CAAAA,QAAAA,EAAUC,MAAAA,CAAAA;AAEzC,IAAA,MAAM,CAACG,QAAAA,CAAAA,GAAa,MAAML,SAAAA,CAAUG,MAAMA,KAAAA,CAAAA;AAE1C,IAAMG,MAAAA,QAAAA,uBAAeC,GAAAA,EAAAA;AAErBF,IAAAA,QAAAA,CAASG,QAAQ,CAAC,EAAEC,SAAWC,EAAAA,UAAAA,EAAYC,UAAU,KAAA;AACnD,MAAA,IAAI,CAACL,QAAAA,CAASM,GAAIH,CAAAA,SAAAA,CAAY,EAAA;AAC5BH,QAAAA,QAAAA,CAASO,IAAIJ,SAAW,EAAA;AAAC,UAAA,CAAA,OAAA,EAAUA,SAAAA,CAAAA,gBAAAA,EAA4BC,UAAAA,CAAAA,EAAAA,EAAeC,QAAAA,CAAAA,CAAAA;AAAY,SAAA,CAAA;OACrF,MAAA;AACLL,QAASQ,QAAAA,CAAAA,GAAAA,CAAIL,SAAAA,CAAYM,CAAAA,IAAAA,CAAK,GAAGL,UAAAA,CAAAA,EAAAA,EAAeC,QAAAA,CAAW,CAAA,CAAA,CAAA;AAC7D;KACF,CAAA;AAEA,IAAA,OAAOK,KAAMC,CAAAA,IAAAA,CAAKX,QAASY,CAAAA,MAAAA,EAAM,CAC9BC,CAAAA,GAAAA,CAAI,CAACC,OAAAA,KAAYA,QAAQC,IAAK,CAAA,IAAA,CAAA,CAAA,CAC9BA,KAAK,IAAA,CAAA;AACV,GAAA,CAAA,OAASC,KAAO,EAAA;AACd,IAAA,MAAM,IAAIC,kBAAU,CAAA,CAAA,6BAAA,EAAgCD,KAAAA,CAAAA,CAAAA,EAAS,EAAI,EAAA;MAC/DE,WAAa,EAAA;KACf,CAAA;AACF;AACF;AA5BsBzB,MAAAA,CAAAA,WAAAA,EAAAA,aAAAA,CAAAA;AA8BtB,SAASK,gBAAAA,CAAiBH,UAAoBC,MAAe,EAAA;AAC3D,EAAMuB,MAAAA,UAAAA,GAAavB,MAAUwB,IAAAA,gBAAAA,CAAiBzB,QAAAA,CAAAA;AAE9C,EAAA,QAAQA,QAAAA;IACN,KAAK,OAAA;IACL,KAAK,SAAA;AACH,MAAO,OAAA;;;;;;;IAQT,KAAK,UAAA;AACH,MAAO,OAAA;;;;;0CAK6BwB,UAAAA,CAAAA;;IAGtC,KAAK,OAAA;AACH,MAAO,OAAA;;;;;;;;sCAQyBA,UAAAA,CAAAA;;IAGlC,KAAK,KAAA;AACH,MAAO,OAAA;;;;;uCAK0BA,UAAAA,CAAAA;;IAGnC,KAAK,QAAA;AACH,MAAO,OAAA;;;;;;;;;IAUT,KAAK,QAAA;AACH,MAAO,OAAA;;;;;mCAKsBA,UAAAA,CAAAA;;AAG/B,IAAA;AACE,MAAA,MAAM,IAAIF,kBAAU,CAAA,CAAA,SAAA,EAAYtB,QAAAA,CAAAA,iBAAAA,CAAAA,EAA6B,EAAI,EAAA;QAC/D0B,OAAS,EAAA,IAAA;QACTH,WAAa,EAAA;OACf,CAAA;AACJ;AACF;AAtESpB,MAAAA,CAAAA,gBAAAA,EAAAA,kBAAAA,CAAAA;AAwET,SAASsB,iBAAiBzB,QAAkB,EAAA;AAC1C,EAAA,QAAQA,QAAAA;IACN,KAAK,UAAA;AACH,MAAO,OAAA,QAAA;IACT,KAAK,OAAA;AACH,MAAO,OAAA,KAAA;IACT,KAAK,KAAA;IACL,KAAK,QAAA;AACH,MAAA,MAAM,IAAIsB,kBAAU,CAAA,CAAA,4BAAA,EAA+BtB,QAAAA,CAAAA,CAAAA,EAAY,EAAI,EAAA;QACjEuB,WAAa,EAAA,KAAA;QACbG,OAAS,EAAA;OACX,CAAA;AACF,IAAA;AACE,MAAO,OAAA,EAAA;AACX;AACF;AAfSD,MAAAA,CAAAA,gBAAAA,EAAAA,kBAAAA,CAAAA","file":"metadata.cjs","sourcesContent":["/**\n * Copyright 2025 IBM Corp.\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 { ToolError } from \"@/tools/base.js\";\nimport { Sequelize } from \"sequelize\";\n\nexport type Provider = \"mysql\" | \"mariadb\" | \"postgres\" | \"mssql\" | \"db2\" | \"sqlite\" | \"oracle\";\n\nexport interface Metadata {\n  tableName: string;\n  columnName: string;\n  dataType: string;\n}\n\nexport async function getMetadata(\n  sequelize: Sequelize,\n  provider: Provider,\n  schema?: string,\n): Promise<string> {\n  try {\n    const query = getMetadataQuery(provider, schema);\n\n    const [metadata] = (await sequelize.query(query)) as [Metadata[], any];\n\n    const tableMap = new Map<string, string[]>();\n\n    metadata.forEach(({ tableName, columnName, dataType }) => {\n      if (!tableMap.has(tableName)) {\n        tableMap.set(tableName, [`Table '${tableName}' with columns: ${columnName} (${dataType})`]);\n      } else {\n        tableMap.get(tableName)!.push(`${columnName} (${dataType})`);\n      }\n    });\n\n    return Array.from(tableMap.values())\n      .map((columns) => columns.join(\", \"))\n      .join(\"; \");\n  } catch (error) {\n    throw new ToolError(`Error initializing metadata: ${error}`, [], {\n      isRetryable: false,\n    });\n  }\n}\n\nfunction getMetadataQuery(provider: Provider, schema?: string): string {\n  const schemaName = schema ?? getDefaultSchema(provider);\n\n  switch (provider) {\n    case \"mysql\":\n    case \"mariadb\":\n      return `\n          SELECT t.table_name AS tableName, c.column_name AS columnName, \n                 c.data_type AS dataType\n          FROM information_schema.tables t\n          JOIN information_schema.columns c ON t.table_name = c.table_name\n          WHERE t.table_schema = DATABASE();\n        `;\n\n    case \"postgres\":\n      return `\n          SELECT t.table_name AS \"tableName\", c.column_name AS \"columnName\", \n                 c.data_type AS \"dataType\"\n          FROM information_schema.tables t\n          JOIN information_schema.columns c ON t.table_name = c.table_name\n          WHERE t.table_schema = lower('${schemaName}');\n        `;\n\n    case \"mssql\":\n      return `\n          SELECT t.name AS tableName, c.name AS columnName,\n                 ty.name AS dataType\n           FROM sys.tables t\n           JOIN sys.columns c ON t.object_id = c.object_id\n           JOIN sys.types ty ON c.user_type_id = ty.user_type_id\n           JOIN sys.schemas s ON t.schema_id = s.schema_id\n           WHERE t.is_ms_shipped = 0 AND t.type = 'U'\n                AND s.name = lower('${schemaName}');\n        `;\n\n    case \"db2\":\n      return `\n          SELECT t.tabname AS \"tableName\", c.colname AS \"columnName\", \n                 c.typename AS \"dataType\"\n          FROM SYSCAT.TABLES t\n          JOIN SYSCAT.COLUMNS c ON t.tabname = c.tabname\n          WHERE t.tabschema = upper('${schemaName}');\n        `;\n\n    case \"sqlite\":\n      return `\n          SELECT tbl_name AS \"tableName\", name AS \"columnName\", type AS \"dataType\"\n            FROM (\n                SELECT name AS tbl_name\n                FROM sqlite_master\n                WHERE type = 'table'\n            )\n            JOIN pragma_table_xinfo(tbl_name);\n        `;\n\n    case \"oracle\":\n      return `\n          SELECT t.table_name AS \"tableName\", c.column_name AS \"columnName\", \n                 c.data_type AS \"dataType\"\n          FROM all_tables t\n          JOIN all_tab_columns c ON t.table_name = c.table_name\n          WHERE t.owner = upper('${schemaName}');\n        `;\n\n    default:\n      throw new ToolError(`Provider ${provider} is not supported`, [], {\n        isFatal: true,\n        isRetryable: false,\n      });\n  }\n}\n\nfunction getDefaultSchema(provider: Provider): string {\n  switch (provider) {\n    case \"postgres\":\n      return \"public\";\n    case \"mssql\":\n      return \"dbo\";\n    case \"db2\":\n    case \"oracle\":\n      throw new ToolError(`Schema name is required for ${provider}`, [], {\n        isRetryable: false,\n        isFatal: true,\n      });\n    default:\n      return \"\";\n  }\n}\n"]}

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

@@ -0,0 +1,27 @@
+import { Sequelize } from 'sequelize';
+
+/**
+ * Copyright 2025 IBM Corp.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+type Provider = "mysql" | "mariadb" | "postgres" | "mssql" | "db2" | "sqlite" | "oracle";
+interface Metadata {
+    tableName: string;
+    columnName: string;
+    dataType: string;
+}
+declare function getMetadata(sequelize: Sequelize, provider: Provider, schema?: string): Promise<string>;
+
+export { type Metadata, type Provider, getMetadata };

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

@@ -0,0 +1,27 @@
+import { Sequelize } from 'sequelize';
+
+/**
+ * Copyright 2025 IBM Corp.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+type Provider = "mysql" | "mariadb" | "postgres" | "mssql" | "db2" | "sqlite" | "oracle";
+interface Metadata {
+    tableName: string;
+    columnName: string;
+    dataType: string;
+}
+declare function getMetadata(sequelize: Sequelize, provider: Provider, schema?: string): Promise<string>;
+
+export { type Metadata, type Provider, getMetadata };

package/dist/tools/database/metadata.js

@@ -0,0 +1,112 @@
+import { ToolError } from '../base.js';
+
+var __defProp = Object.defineProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+async function getMetadata(sequelize, provider, schema) {
+  try {
+    const query = getMetadataQuery(provider, schema);
+    const [metadata] = await sequelize.query(query);
+    const tableMap = /* @__PURE__ */ new Map();
+    metadata.forEach(({ tableName, columnName, dataType }) => {
+      if (!tableMap.has(tableName)) {
+        tableMap.set(tableName, [
+          `Table '${tableName}' with columns: ${columnName} (${dataType})`
+        ]);
+      } else {
+        tableMap.get(tableName).push(`${columnName} (${dataType})`);
+      }
+    });
+    return Array.from(tableMap.values()).map((columns) => columns.join(", ")).join("; ");
+  } catch (error) {
+    throw new ToolError(`Error initializing metadata: ${error}`, [], {
+      isRetryable: false
+    });
+  }
+}
+__name(getMetadata, "getMetadata");
+function getMetadataQuery(provider, schema) {
+  const schemaName = schema ?? getDefaultSchema(provider);
+  switch (provider) {
+    case "mysql":
+    case "mariadb":
+      return `
+          SELECT t.table_name AS tableName, c.column_name AS columnName, 
+                 c.data_type AS dataType
+          FROM information_schema.tables t
+          JOIN information_schema.columns c ON t.table_name = c.table_name
+          WHERE t.table_schema = DATABASE();
+        `;
+    case "postgres":
+      return `
+          SELECT t.table_name AS "tableName", c.column_name AS "columnName", 
+                 c.data_type AS "dataType"
+          FROM information_schema.tables t
+          JOIN information_schema.columns c ON t.table_name = c.table_name
+          WHERE t.table_schema = lower('${schemaName}');
+        `;
+    case "mssql":
+      return `
+          SELECT t.name AS tableName, c.name AS columnName,
+                 ty.name AS dataType
+           FROM sys.tables t
+           JOIN sys.columns c ON t.object_id = c.object_id
+           JOIN sys.types ty ON c.user_type_id = ty.user_type_id
+           JOIN sys.schemas s ON t.schema_id = s.schema_id
+           WHERE t.is_ms_shipped = 0 AND t.type = 'U'
+                AND s.name = lower('${schemaName}');
+        `;
+    case "db2":
+      return `
+          SELECT t.tabname AS "tableName", c.colname AS "columnName", 
+                 c.typename AS "dataType"
+          FROM SYSCAT.TABLES t
+          JOIN SYSCAT.COLUMNS c ON t.tabname = c.tabname
+          WHERE t.tabschema = upper('${schemaName}');
+        `;
+    case "sqlite":
+      return `
+          SELECT tbl_name AS "tableName", name AS "columnName", type AS "dataType"
+            FROM (
+                SELECT name AS tbl_name
+                FROM sqlite_master
+                WHERE type = 'table'
+            )
+            JOIN pragma_table_xinfo(tbl_name);
+        `;
+    case "oracle":
+      return `
+          SELECT t.table_name AS "tableName", c.column_name AS "columnName", 
+                 c.data_type AS "dataType"
+          FROM all_tables t
+          JOIN all_tab_columns c ON t.table_name = c.table_name
+          WHERE t.owner = upper('${schemaName}');
+        `;
+    default:
+      throw new ToolError(`Provider ${provider} is not supported`, [], {
+        isFatal: true,
+        isRetryable: false
+      });
+  }
+}
+__name(getMetadataQuery, "getMetadataQuery");
+function getDefaultSchema(provider) {
+  switch (provider) {
+    case "postgres":
+      return "public";
+    case "mssql":
+      return "dbo";
+    case "db2":
+    case "oracle":
+      throw new ToolError(`Schema name is required for ${provider}`, [], {
+        isRetryable: false,
+        isFatal: true
+      });
+    default:
+      return "";
+  }
+}
+__name(getDefaultSchema, "getDefaultSchema");
+
+export { getMetadata };
+//# sourceMappingURL=metadata.js.map
+//# sourceMappingURL=metadata.js.map

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

@@ -0,0 +1 @@
+{"version":3,"sources":["../../../src/tools/database/metadata.ts"],"names":["getMetadata","sequelize","provider","schema","query","getMetadataQuery","metadata","tableMap","Map","forEach","tableName","columnName","dataType","has","set","get","push","Array","from","values","map","columns","join","error","ToolError","isRetryable","schemaName","getDefaultSchema","isFatal"],"mappings":";;;;AA2BA,eAAsBA,WAAAA,CACpBC,SACAC,EAAAA,QAAAA,EACAC,MAAe,EAAA;AAEf,EAAI,IAAA;AACF,IAAMC,MAAAA,KAAAA,GAAQC,gBAAiBH,CAAAA,QAAAA,EAAUC,MAAAA,CAAAA;AAEzC,IAAA,MAAM,CAACG,QAAAA,CAAAA,GAAa,MAAML,SAAAA,CAAUG,MAAMA,KAAAA,CAAAA;AAE1C,IAAMG,MAAAA,QAAAA,uBAAeC,GAAAA,EAAAA;AAErBF,IAAAA,QAAAA,CAASG,QAAQ,CAAC,EAAEC,SAAWC,EAAAA,UAAAA,EAAYC,UAAU,KAAA;AACnD,MAAA,IAAI,CAACL,QAAAA,CAASM,GAAIH,CAAAA,SAAAA,CAAY,EAAA;AAC5BH,QAAAA,QAAAA,CAASO,IAAIJ,SAAW,EAAA;AAAC,UAAA,CAAA,OAAA,EAAUA,SAAAA,CAAAA,gBAAAA,EAA4BC,UAAAA,CAAAA,EAAAA,EAAeC,QAAAA,CAAAA,CAAAA;AAAY,SAAA,CAAA;OACrF,MAAA;AACLL,QAASQ,QAAAA,CAAAA,GAAAA,CAAIL,SAAAA,CAAYM,CAAAA,IAAAA,CAAK,GAAGL,UAAAA,CAAAA,EAAAA,EAAeC,QAAAA,CAAW,CAAA,CAAA,CAAA;AAC7D;KACF,CAAA;AAEA,IAAA,OAAOK,KAAMC,CAAAA,IAAAA,CAAKX,QAASY,CAAAA,MAAAA,EAAM,CAC9BC,CAAAA,GAAAA,CAAI,CAACC,OAAAA,KAAYA,QAAQC,IAAK,CAAA,IAAA,CAAA,CAAA,CAC9BA,KAAK,IAAA,CAAA;AACV,GAAA,CAAA,OAASC,KAAO,EAAA;AACd,IAAA,MAAM,IAAIC,SAAU,CAAA,CAAA,6BAAA,EAAgCD,KAAAA,CAAAA,CAAAA,EAAS,EAAI,EAAA;MAC/DE,WAAa,EAAA;KACf,CAAA;AACF;AACF;AA5BsBzB,MAAAA,CAAAA,WAAAA,EAAAA,aAAAA,CAAAA;AA8BtB,SAASK,gBAAAA,CAAiBH,UAAoBC,MAAe,EAAA;AAC3D,EAAMuB,MAAAA,UAAAA,GAAavB,MAAUwB,IAAAA,gBAAAA,CAAiBzB,QAAAA,CAAAA;AAE9C,EAAA,QAAQA,QAAAA;IACN,KAAK,OAAA;IACL,KAAK,SAAA;AACH,MAAO,OAAA;;;;;;;IAQT,KAAK,UAAA;AACH,MAAO,OAAA;;;;;0CAK6BwB,UAAAA,CAAAA;;IAGtC,KAAK,OAAA;AACH,MAAO,OAAA;;;;;;;;sCAQyBA,UAAAA,CAAAA;;IAGlC,KAAK,KAAA;AACH,MAAO,OAAA;;;;;uCAK0BA,UAAAA,CAAAA;;IAGnC,KAAK,QAAA;AACH,MAAO,OAAA;;;;;;;;;IAUT,KAAK,QAAA;AACH,MAAO,OAAA;;;;;mCAKsBA,UAAAA,CAAAA;;AAG/B,IAAA;AACE,MAAA,MAAM,IAAIF,SAAU,CAAA,CAAA,SAAA,EAAYtB,QAAAA,CAAAA,iBAAAA,CAAAA,EAA6B,EAAI,EAAA;QAC/D0B,OAAS,EAAA,IAAA;QACTH,WAAa,EAAA;OACf,CAAA;AACJ;AACF;AAtESpB,MAAAA,CAAAA,gBAAAA,EAAAA,kBAAAA,CAAAA;AAwET,SAASsB,iBAAiBzB,QAAkB,EAAA;AAC1C,EAAA,QAAQA,QAAAA;IACN,KAAK,UAAA;AACH,MAAO,OAAA,QAAA;IACT,KAAK,OAAA;AACH,MAAO,OAAA,KAAA;IACT,KAAK,KAAA;IACL,KAAK,QAAA;AACH,MAAA,MAAM,IAAIsB,SAAU,CAAA,CAAA,4BAAA,EAA+BtB,QAAAA,CAAAA,CAAAA,EAAY,EAAI,EAAA;QACjEuB,WAAa,EAAA,KAAA;QACbG,OAAS,EAAA;OACX,CAAA;AACF,IAAA;AACE,MAAO,OAAA,EAAA;AACX;AACF;AAfSD,MAAAA,CAAAA,gBAAAA,EAAAA,kBAAAA,CAAAA","file":"metadata.js","sourcesContent":["/**\n * Copyright 2025 IBM Corp.\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 { ToolError } from \"@/tools/base.js\";\nimport { Sequelize } from \"sequelize\";\n\nexport type Provider = \"mysql\" | \"mariadb\" | \"postgres\" | \"mssql\" | \"db2\" | \"sqlite\" | \"oracle\";\n\nexport interface Metadata {\n  tableName: string;\n  columnName: string;\n  dataType: string;\n}\n\nexport async function getMetadata(\n  sequelize: Sequelize,\n  provider: Provider,\n  schema?: string,\n): Promise<string> {\n  try {\n    const query = getMetadataQuery(provider, schema);\n\n    const [metadata] = (await sequelize.query(query)) as [Metadata[], any];\n\n    const tableMap = new Map<string, string[]>();\n\n    metadata.forEach(({ tableName, columnName, dataType }) => {\n      if (!tableMap.has(tableName)) {\n        tableMap.set(tableName, [`Table '${tableName}' with columns: ${columnName} (${dataType})`]);\n      } else {\n        tableMap.get(tableName)!.push(`${columnName} (${dataType})`);\n      }\n    });\n\n    return Array.from(tableMap.values())\n      .map((columns) => columns.join(\", \"))\n      .join(\"; \");\n  } catch (error) {\n    throw new ToolError(`Error initializing metadata: ${error}`, [], {\n      isRetryable: false,\n    });\n  }\n}\n\nfunction getMetadataQuery(provider: Provider, schema?: string): string {\n  const schemaName = schema ?? getDefaultSchema(provider);\n\n  switch (provider) {\n    case \"mysql\":\n    case \"mariadb\":\n      return `\n          SELECT t.table_name AS tableName, c.column_name AS columnName, \n                 c.data_type AS dataType\n          FROM information_schema.tables t\n          JOIN information_schema.columns c ON t.table_name = c.table_name\n          WHERE t.table_schema = DATABASE();\n        `;\n\n    case \"postgres\":\n      return `\n          SELECT t.table_name AS \"tableName\", c.column_name AS \"columnName\", \n                 c.data_type AS \"dataType\"\n          FROM information_schema.tables t\n          JOIN information_schema.columns c ON t.table_name = c.table_name\n          WHERE t.table_schema = lower('${schemaName}');\n        `;\n\n    case \"mssql\":\n      return `\n          SELECT t.name AS tableName, c.name AS columnName,\n                 ty.name AS dataType\n           FROM sys.tables t\n           JOIN sys.columns c ON t.object_id = c.object_id\n           JOIN sys.types ty ON c.user_type_id = ty.user_type_id\n           JOIN sys.schemas s ON t.schema_id = s.schema_id\n           WHERE t.is_ms_shipped = 0 AND t.type = 'U'\n                AND s.name = lower('${schemaName}');\n        `;\n\n    case \"db2\":\n      return `\n          SELECT t.tabname AS \"tableName\", c.colname AS \"columnName\", \n                 c.typename AS \"dataType\"\n          FROM SYSCAT.TABLES t\n          JOIN SYSCAT.COLUMNS c ON t.tabname = c.tabname\n          WHERE t.tabschema = upper('${schemaName}');\n        `;\n\n    case \"sqlite\":\n      return `\n          SELECT tbl_name AS \"tableName\", name AS \"columnName\", type AS \"dataType\"\n            FROM (\n                SELECT name AS tbl_name\n                FROM sqlite_master\n                WHERE type = 'table'\n            )\n            JOIN pragma_table_xinfo(tbl_name);\n        `;\n\n    case \"oracle\":\n      return `\n          SELECT t.table_name AS \"tableName\", c.column_name AS \"columnName\", \n                 c.data_type AS \"dataType\"\n          FROM all_tables t\n          JOIN all_tab_columns c ON t.table_name = c.table_name\n          WHERE t.owner = upper('${schemaName}');\n        `;\n\n    default:\n      throw new ToolError(`Provider ${provider} is not supported`, [], {\n        isFatal: true,\n        isRetryable: false,\n      });\n  }\n}\n\nfunction getDefaultSchema(provider: Provider): string {\n  switch (provider) {\n    case \"postgres\":\n      return \"public\";\n    case \"mssql\":\n      return \"dbo\";\n    case \"db2\":\n    case \"oracle\":\n      throw new ToolError(`Schema name is required for ${provider}`, [], {\n        isRetryable: false,\n        isFatal: true,\n      });\n    default:\n      return \"\";\n  }\n}\n"]}

package/dist/tools/database/milvus.cjs

@@ -0,0 +1,191 @@
+'use strict';
+
+var base_cjs = require('../base.cjs');
+var decoratorCache_cjs = require('../../cache/decoratorCache.cjs');
+var milvus2SdkNode = require('@zilliz/milvus2-sdk-node');
+var zod = require('zod');
+var emitter_cjs = require('../../emitter/emitter.cjs');
+
+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 MilvusAction = /* @__PURE__ */ function(MilvusAction2) {
+  MilvusAction2["ListCollections"] = "ListCollections";
+  MilvusAction2["GetCollectionInfo"] = "GetCollectionInfo";
+  MilvusAction2["Search"] = "Search";
+  MilvusAction2["Insert"] = "Insert";
+  MilvusAction2["Delete"] = "Delete";
+  return MilvusAction2;
+}({});
+class MilvusDatabaseTool extends base_cjs.Tool {
+  static {
+    __name(this, "MilvusDatabaseTool");
+  }
+  name = "MilvusDatabaseTool";
+  description = `Can query data from a Milvus database. IMPORTANT: strictly follow this order of actions:
+   1. ${"ListCollections"} - List all the Milvus collections
+   2. ${"GetCollectionInfo"} - Get information about into a Milvus collection
+   3. ${"Insert"} - Insert data into a Milvus collection
+   3. ${"Search"} - Perform search on a Milvus collection
+   4. ${"Delete"} - Delete from a Milvus collection`;
+  inputSchema() {
+    return zod.z.object({
+      action: zod.z.nativeEnum(MilvusAction).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: zod.z.string().optional().describe(`The name of the collection to query, required for ${"GetCollectionInfo"}, ${"Search"}, ${"Insert"}, and ${"Delete"}`),
+      vector: zod.z.array(zod.z.number()).optional().describe(`The vector to search for or insert, required for ${"Search"}`),
+      vectors: zod.z.array(zod.z.array(zod.z.number())).optional().describe(`The vectors to insert, required for ${"Insert"}`),
+      topK: zod.z.coerce.number().int().min(1).max(1e3).default(10).optional().describe(`The number of nearest neighbors to return for ${"Search"}. Maximum is 1000`),
+      filter: zod.z.string().optional().describe(`Optional filter expression for ${"Search"}`),
+      metadata: zod.z.record(zod.z.string(), zod.z.any()).optional().describe(`Additional metadata to insert with vectors for ${"Insert"}`),
+      ids: zod.z.array(zod.z.string().or(zod.z.number())).optional().describe(`Array of IDs to delete for ${"Delete"}`),
+      searchOutput: zod.z.array(zod.z.string()).optional().describe(`Fields to return in search results for ${"Search"}`)
+    });
+  }
+  emitter = emitter_cjs.Emitter.root.child({
+    namespace: [
+      "tool",
+      "database",
+      "milvus"
+    ],
+    creator: this
+  });
+  validateInput(schema, input) {
+    super.validateInput(schema, input);
+    if (input.action === "GetCollectionInfo" && !input.collectionName) {
+      throw new base_cjs.ToolInputValidationError(`Collection name is required for ${"GetCollectionInfo"}, ${"Search"}, ${"Insert"}, and ${"Delete"} actions.`);
+    }
+    if (input.action === "Search" && (!input.collectionName || !input.vector)) {
+      throw new base_cjs.ToolInputValidationError(`Both collection name and vector are required for ${"Search"} action.`);
+    }
+    if (input.action === "Insert" && (!input.collectionName || !input.vectors)) {
+      throw new base_cjs.ToolInputValidationError(`Both collection name and vectors are required for ${"Insert"} action.`);
+    }
+  }
+  static {
+    this.register();
+  }
+  async client() {
+    try {
+      const client = new milvus2SdkNode.MilvusClient(this.options.connection);
+      await client.listCollections();
+      return client;
+    } catch (error) {
+      throw new base_cjs.ToolError(`Unable to connect to Milvus.`, [
+        error
+      ], {
+        isRetryable: false,
+        isFatal: true
+      });
+    }
+  }
+  async _run(input, _options) {
+    switch (input.action) {
+      case "ListCollections": {
+        const collections = await this.listCollections();
+        return new base_cjs.JSONToolOutput(collections);
+      }
+      case "GetCollectionInfo": {
+        if (!input.collectionName) {
+          throw new base_cjs.ToolError("A collection name is required for Milvus GetCollectionInfo action");
+        }
+        const collectionInfo = await this.getCollectionInfo(input.collectionName);
+        return new base_cjs.JSONToolOutput(collectionInfo);
+      }
+      case "Search": {
+        if (!input.collectionName || !input.vector) {
+          throw new base_cjs.ToolError("A collection name and vector are required for Milvus Search action");
+        }
+        const searchResults = await this.search(input);
+        return new base_cjs.JSONToolOutput(searchResults);
+      }
+      case "Insert": {
+        if (!input.collectionName || !input.vectors) {
+          throw new base_cjs.ToolError("A collection name and vectors are required for Milvus Insert action");
+        }
+        const insertResults = await this.insert(input);
+        return new base_cjs.JSONToolOutput(insertResults);
+      }
+      case "Delete": {
+        if (!input.collectionName || !input.ids) {
+          throw new base_cjs.ToolError("Collection name and ids are required for Milvus Delete action");
+        }
+        const deleteResults = await this.delete(input);
+        return new base_cjs.JSONToolOutput(deleteResults);
+      }
+      default: {
+        throw new base_cjs.ToolError(`Invalid action specified: ${input.action}`);
+      }
+    }
+  }
+  async listCollections() {
+    try {
+      const client = await this.client();
+      const response = await client.listCollections({});
+      if (response && response.data && Array.isArray(response.data)) {
+        return response.data.map((collection) => collection.name);
+      } else {
+        return [];
+      }
+    } catch (error) {
+      throw new base_cjs.ToolError(`Failed to list collections from Milvus: ${error}`);
+    }
+  }
+  async getCollectionInfo(collectionName) {
+    const client = await this.client();
+    const response = client.describeCollection({
+      collection_name: collectionName
+    });
+    return response;
+  }
+  async insert(input) {
+    const client = await this.client();
+    const response = await client.insert({
+      collection_name: input.collectionName,
+      fields_data: input.vectors.map((vector, index) => ({
+        vector,
+        ...input.metadata?.[index]
+      }))
+    });
+    return response;
+  }
+  async search(input) {
+    const client = await this.client();
+    const response = await client.search({
+      collection_name: input.collectionName,
+      vector: input.vector,
+      limit: input.topK || 10,
+      filter: input.filter,
+      output_fields: input.searchOutput
+    });
+    return response.results;
+  }
+  async delete(input) {
+    const client = await this.client();
+    const response = client.delete({
+      collection_name: input.collectionName,
+      filter: `id in [${input.ids?.join(",")}]`
+    });
+    return response;
+  }
+}
+_ts_decorate([
+  decoratorCache_cjs.Cache(),
+  _ts_metadata("design:type", Function),
+  _ts_metadata("design:paramtypes", []),
+  _ts_metadata("design:returntype", Promise)
+], MilvusDatabaseTool.prototype, "client", null);
+
+exports.MilvusAction = MilvusAction;
+exports.MilvusDatabaseTool = MilvusDatabaseTool;
+//# sourceMappingURL=milvus.cjs.map
+//# sourceMappingURL=milvus.cjs.map

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

@@ -0,0 +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":";;;;;;;;;;AAcC,SAAA,YAAA,CAAA,UAAA,EAAA,MAAA,EAAA,GAAA,EAAA,IAAA,EAAA;;;;;;AAAA,MAAA,CAAA,YAAA,EAAA,cAAA,CAAA;;;;;AAiCWA,IAAAA,YAAAA,4BAAAA,aAAAA,EAAAA;;;;;;AAAAA,EAAAA,OAAAA,aAAAA;;AAQL,MAAMC,2BAA2BC,aAAAA,CAAAA;EAvDxC;;;EA2DEC,IAAO,GAAA,oBAAA;EAEPC,WAAc,GAAA,CAAA;QACV,iBAAA,CAAA;QACA,mBAAA,CAAA;QACA,QAAA,CAAA;QACA,QAAA,CAAA;QACA,QAAA,CAAA,kCAAA,CAAA;EAEJC,WAAc,GAAA;AACZ,IAAA,OAAOC,MAAEC,MAAO,CAAA;AACdC,MAAAA,MAAAA,EAAQF,KACLG,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,KACbM,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,KAAAA,CACLS,KAAMT,CAAAA,KAAAA,CAAEU,MAAM,EAAA,CACdH,CAAAA,QAAAA,EACAH,CAAAA,QAAAA,CAAS,CAAkD,iDAAA,EAAA,QAAA,CAAuB,CAAA,CAAA;AACrFO,MAAAA,OAAAA,EAASX,KACNS,CAAAA,KAAAA,CAAMT,KAAES,CAAAA,KAAAA,CAAMT,MAAEU,MAAM,EAAA,CAAA,CAAA,CACtBH,QAAQ,EAAA,CACRH,QAAS,CAAA,CAAA,oCAAA,EAAqC,QAAA,CAAuB,CAAA,CAAA;MACxEQ,IAAMZ,EAAAA,KAAAA,CAAEa,OACLH,MAAM,EAAA,CACNI,KACAC,CAAAA,GAAAA,CAAI,CAAA,CACJC,CAAAA,GAAAA,CAAI,GAAA,CACJC,CAAAA,OAAAA,CAAQ,EAAA,CACRV,CAAAA,QAAAA,GACAH,QACC,CAAA,CAAA,8CAAA,EAA+C,QAAA,CAAwC,iBAAA,CAAA,CAAA;MAE3Fc,MAAQlB,EAAAA,KAAAA,CACLM,QACAC,CAAAA,QAAAA,GACAH,QAAS,CAAA,CAAA,+BAAA,EAAgC,QAAA,CAAuB,CAAA,CAAA;AACnEe,MAAAA,QAAAA,EAAUnB,KACPoB,CAAAA,MAAAA,CAAOpB,KAAEM,CAAAA,MAAAA,IAAUN,KAAEqB,CAAAA,GAAAA,EAAG,CAAA,CACxBd,QAAQ,EAAA,CACRH,QAAS,CAAA,CAAA,+CAAA,EAAgD,QAAA,CAAuB,CAAA,CAAA;AACnFkB,MAAAA,GAAAA,EAAKtB,MACFS,KAAMT,CAAAA,KAAAA,CAAEM,MAAM,EAAA,CAAGiB,GAAGvB,KAAEU,CAAAA,MAAAA,EAAM,CAAA,EAC5BH,QAAQ,EAAA,CACRH,QAAS,CAAA,CAAA,2BAAA,EAA4B,QAAA,CAAuB,CAAA,CAAA;MAC/DoB,YAAcxB,EAAAA,KAAAA,CACXS,KAAMT,CAAAA,KAAAA,CAAEM,MAAM,EAAA,CACdC,CAAAA,QAAAA,EACAH,CAAAA,QAAAA,CAAS,CAAwC,uCAAA,EAAA,QAAA,CAAuB,CAAA;KAC7E,CAAA;AACF;EAEgBqB,OACdC,GAAAA,mBAAAA,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,KAAM/B,CAAAA,MAAAA,KAAM,mBAAuC,IAAA,CAAC+B,MAAM5B,cAAgB,EAAA;AAC5E,MAAM,MAAA,IAAI6B,iCACR,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,CAAM/B,WAAM,QAA6B,KAAA,CAAC+B,MAAM5B,cAAkB,IAAA,CAAC4B,MAAMzB,MAAS,CAAA,EAAA;AACpF,MAAA,MAAM,IAAI0B,iCAAAA,CACR,CAAkD,iDAAA,EAAA,QAAA,CAA+B,QAAA,CAAA,CAAA;AAErF;AACA,IAAID,IAAAA,KAAAA,CAAM/B,WAAM,QAA6B,KAAA,CAAC+B,MAAM5B,cAAkB,IAAA,CAAC4B,MAAMtB,OAAU,CAAA,EAAA;AACrF,MAAA,MAAM,IAAIuB,iCAAAA,CACR,CAAmD,kDAAA,EAAA,QAAA,CAA+B,QAAA,CAAA,CAAA;AAEtF;AACF;EAEA;AACE,IAAA,IAAA,CAAKC,QAAQ,EAAA;AACf;AAEA,EAAA,MACgBC,MAAgC,GAAA;AAC9C,IAAI,IAAA;AACF,MAAA,MAAMA,MAAS,GAAA,IAAIC,2BAAa,CAAA,IAAA,CAAKC,QAAQC,UAAU,CAAA;AAEvD,MAAA,MAAMH,OAAOI,eAAe,EAAA;AAC5B,MAAOJ,OAAAA,MAAAA;AACT,KAAA,CAAA,OAASK,KAAO,EAAA;AACd,MAAM,MAAA,IAAIC,mBAAU,CAAgC,4BAAA,CAAA,EAAA;AAACD,QAAAA;AAAQ,OAAA,EAAA;QAC3DE,WAAa,EAAA,KAAA;QACbC,OAAS,EAAA;OACX,CAAA;AACF;AACF;EAEA,MAAgBC,IAAAA,CACdZ,OACAa,QAC8B,EAAA;AAC9B,IAAA,QAAQb,MAAM/B,MAAM;AAClB,MAAA,KAAA,iBAAmC,EAAA;AACjC,QAAM6C,MAAAA,WAAAA,GAAc,MAAM,IAAA,CAAKP,eAAe,EAAA;AAC9C,QAAO,OAAA,IAAIQ,wBAAeD,WAAAA,CAAAA;AAC5B;AAEA,MAAA,KAAA,mBAAqC,EAAA;AACnC,QAAI,IAAA,CAACd,MAAM5B,cAAgB,EAAA;AACzB,UAAM,MAAA,IAAIqC,mBAAU,mEAAA,CAAA;AACtB;AACA,QAAA,MAAMO,cAAiB,GAAA,MAAM,IAAKC,CAAAA,iBAAAA,CAAkBjB,MAAM5B,cAAc,CAAA;AACxE,QAAO,OAAA,IAAI2C,wBAAeC,cAAAA,CAAAA;AAC5B;AAEA,MAAA,KAAA,QAA0B,EAAA;AACxB,QAAA,IAAI,CAAChB,KAAAA,CAAM5B,cAAkB,IAAA,CAAC4B,MAAMzB,MAAQ,EAAA;AAC1C,UAAM,MAAA,IAAIkC,mBAAU,oEAAA,CAAA;AACtB;AACA,QAAA,MAAMS,aAAgB,GAAA,MAAM,IAAKC,CAAAA,MAAAA,CAAOnB,KAAAA,CAAAA;AACxC,QAAO,OAAA,IAAIe,wBAAeG,aAAAA,CAAAA;AAC5B;AAEA,MAAA,KAAA,QAA0B,EAAA;AACxB,QAAA,IAAI,CAAClB,KAAAA,CAAM5B,cAAkB,IAAA,CAAC4B,MAAMtB,OAAS,EAAA;AAC3C,UAAM,MAAA,IAAI+B,mBACR,qEAAA,CAAA;AAEJ;AACA,QAAA,MAAMW,aAAgB,GAAA,MAAM,IAAKC,CAAAA,MAAAA,CAAOrB,KAAAA,CAAAA;AACxC,QAAO,OAAA,IAAIe,wBAAeK,aAAAA,CAAAA;AAC5B;AAEA,MAAA,KAAA,QAA0B,EAAA;AACxB,QAAA,IAAI,CAACpB,KAAAA,CAAM5B,cAAkB,IAAA,CAAC4B,MAAMX,GAAK,EAAA;AACvC,UAAM,MAAA,IAAIoB,mBAAU,+DAAA,CAAA;AACtB;AACA,QAAA,MAAMa,aAAgB,GAAA,MAAM,IAAKC,CAAAA,MAAAA,CAAOvB,KAAAA,CAAAA;AACxC,QAAO,OAAA,IAAIe,wBAAeO,aAAAA,CAAAA;AAC5B;MAEA,SAAS;AACP,QAAA,MAAM,IAAIb,kBAAAA,CAAU,CAA6BT,0BAAAA,EAAAA,KAAAA,CAAM/B,MAAM,CAAE,CAAA,CAAA;AACjE;AACF;AACF;AAEA,EAAA,MAAgBsC,eAAqC,GAAA;AACnD,IAAI,IAAA;AACF,MAAMJ,MAAAA,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAM,EAAA;AAChC,MAAA,MAAMqB,QAAW,GAAA,MAAMrB,MAAOI,CAAAA,eAAAA,CAAgB,EAAC,CAAA;AAE/C,MAAA,IAAIiB,YAAYA,QAASC,CAAAA,IAAAA,IAAQC,MAAMC,OAAQH,CAAAA,QAAAA,CAASC,IAAI,CAAG,EAAA;AAC7D,QAAA,OAAOD,SAASC,IAAKG,CAAAA,GAAAA,CAAI,CAACC,UAAAA,KAAeA,WAAWjE,IAAI,CAAA;OACnD,MAAA;AACL,QAAA,OAAO,EAAA;AACT;AACF,KAAA,CAAA,OAAS4C,KAAO,EAAA;AACd,MAAA,MAAM,IAAIC,kBAAAA,CAAU,CAA2CD,wCAAAA,EAAAA,KAAAA,CAAO,CAAA,CAAA;AACxE;AACF;AAEA,EAAA,MAAgBS,kBAAkB7C,cAAsC,EAAA;AACtE,IAAM+B,MAAAA,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAM,EAAA;AAChC,IAAMqB,MAAAA,QAAAA,GAAWrB,OAAO2B,kBAAmB,CAAA;MAAEC,eAAiB3D,EAAAA;KAAe,CAAA;AAC7E,IAAOoD,OAAAA,QAAAA;AACT;AAEA,EAAA,MAAgBH,OAAOrB,KAAsC,EAAA;AAC3D,IAAMG,MAAAA,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAM,EAAA;AAChC,IAAMqB,MAAAA,QAAAA,GAAW,MAAMrB,MAAAA,CAAOkB,MAAO,CAAA;AACnCU,MAAAA,eAAAA,EAAiB/B,KAAM5B,CAAAA,cAAAA;AACvB4D,MAAAA,WAAAA,EAAahC,KAAMtB,CAAAA,OAAAA,CAASkD,GAAI,CAAA,CAACrD,QAAQ0D,KAAW,MAAA;AAClD1D,QAAAA,MAAAA;QACA,GAAGyB,KAAAA,CAAMd,WAAW+C,KAAAA;OACtB,CAAA;KACF,CAAA;AACA,IAAOT,OAAAA,QAAAA;AACT;AAEA,EAAA,MAAgBL,OAAOnB,KAAsC,EAAA;AAC3D,IAAMG,MAAAA,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAM,EAAA;AAChC,IAAMqB,MAAAA,QAAAA,GAAW,MAAMrB,MAAAA,CAAOgB,MAAO,CAAA;AACnCY,MAAAA,eAAAA,EAAiB/B,KAAM5B,CAAAA,cAAAA;AACvBG,MAAAA,MAAAA,EAAQyB,KAAMzB,CAAAA,MAAAA;AACd2D,MAAAA,KAAAA,EAAOlC,MAAMrB,IAAQ,IAAA,EAAA;AACrBM,MAAAA,MAAAA,EAAQe,KAAMf,CAAAA,MAAAA;AACdkD,MAAAA,aAAAA,EAAenC,KAAMT,CAAAA;KACvB,CAAA;AACA,IAAA,OAAOiC,QAASY,CAAAA,OAAAA;AAClB;AAEA,EAAA,MAAgBb,OAAOvB,KAAsC,EAAA;AAC3D,IAAMG,MAAAA,MAAAA,GAAS,MAAM,IAAA,CAAKA,MAAM,EAAA;AAChC,IAAMqB,MAAAA,QAAAA,GAAWrB,OAAOoB,MAAO,CAAA;AAC7BQ,MAAAA,eAAAA,EAAiB/B,KAAM5B,CAAAA,cAAAA;AACvBa,MAAAA,MAAAA,EAAQ,CAAUe,OAAAA,EAAAA,KAAAA,CAAMX,GAAKgD,EAAAA,IAAAA,CAAK,GAAA,CAAA,CAAA,CAAA;KACpC,CAAA;AACA,IAAOb,OAAAA,QAAAA;AACT;AACF","file":"milvus.cjs","sourcesContent":["/**\n * Copyright 2025 IBM Corp.\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 {\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"]}