agent-swarm-kit 1.0.228 → 1.0.229

package/build/index.cjs

@@ -8757,16 +8757,34 @@ const CREATE_EMBEDDING_FN = async (item, self) => {
             id: item.id,
         });
     const index = await self.params.createIndex(item);
-    const hash = createSHA256Hash(index);
-    let embeddings = await self.params.readEmbeddingCache(self.params.embedding, hash);
-    if (!embeddings) {
-        embeddings = await self.params.createEmbedding(index, self.params.embedding);
-        await self.params.writeEmbeddingCache(embeddings, self.params.embedding, hash);
+    const result = [];
+    if (typeof index === "string") {
+        const hash = createSHA256Hash(index);
+        let embeddings = await self.params.readEmbeddingCache(self.params.embedding, hash);
+        if (!embeddings) {
+            embeddings = await self.params.createEmbedding(index, self.params.embedding);
+            await self.params.writeEmbeddingCache(embeddings, self.params.embedding, hash);
+        }
+        if (self.params.onCreate) {
+            self.params.onCreate(index, embeddings, self.params.clientId, self.params.embedding);
+        }
+        result.push([embeddings, index]);
     }
-    if (self.params.onCreate) {
-        self.params.onCreate(index, embeddings, self.params.clientId, self.params.embedding);
+    else {
+        for (const value of Object.values(index)) {
+            const hash = createSHA256Hash(value);
+            let embeddings = await self.params.readEmbeddingCache(self.params.embedding, hash);
+            if (!embeddings) {
+                embeddings = await self.params.createEmbedding(value, self.params.embedding);
+                await self.params.writeEmbeddingCache(embeddings, self.params.embedding, hash);
+            }
+            if (self.params.onCreate) {
+                self.params.onCreate(value, embeddings, self.params.clientId, self.params.embedding);
+            }
+            result.push([embeddings, value]);
+        }
     }
-    return [embeddings, index];
+    return result;
 };
 /**
  * Waits for the ClientStorage instance to initialize by loading initial data and creating embeddings.
@@ -8992,12 +9010,15 @@ class ClientStorage {
             this.params.onCreate(search, searchEmbeddings, this.params.clientId, this.params.embedding);
         }
         await Promise.all(Array.from(this._itemMap.values()).map(functoolsKit.execpool(async (item) => {
-            const [targetEmbeddings, index] = await this._createEmbedding(item);
-            const score = await this.params.calculateSimilarity(searchEmbeddings, targetEmbeddings);
-            if (this.params.onCompare) {
-                this.params.onCompare(search, index, score, this.params.clientId, this.params.embedding);
+            let maxScore = Number.NEGATIVE_INFINITY;
+            for (const [targetEmbeddings, index] of await this._createEmbedding(item)) {
+                const score = await this.params.calculateSimilarity(searchEmbeddings, targetEmbeddings);
+                if (this.params.onCompare) {
+                    this.params.onCompare(search, index, score, this.params.clientId, this.params.embedding);
+                }
+                maxScore = Math.max(score, maxScore);
             }
-            indexed.push(item, score);
+            indexed.push(item, maxScore);
         }, {
             delay: 10,
             maxExec: GLOBAL_CONFIG.CC_STORAGE_SEARCH_POOL,

package/build/index.mjs

@@ -8755,16 +8755,34 @@ const CREATE_EMBEDDING_FN = async (item, self) => {
             id: item.id,
         });
     const index = await self.params.createIndex(item);
-    const hash = createSHA256Hash(index);
-    let embeddings = await self.params.readEmbeddingCache(self.params.embedding, hash);
-    if (!embeddings) {
-        embeddings = await self.params.createEmbedding(index, self.params.embedding);
-        await self.params.writeEmbeddingCache(embeddings, self.params.embedding, hash);
+    const result = [];
+    if (typeof index === "string") {
+        const hash = createSHA256Hash(index);
+        let embeddings = await self.params.readEmbeddingCache(self.params.embedding, hash);
+        if (!embeddings) {
+            embeddings = await self.params.createEmbedding(index, self.params.embedding);
+            await self.params.writeEmbeddingCache(embeddings, self.params.embedding, hash);
+        }
+        if (self.params.onCreate) {
+            self.params.onCreate(index, embeddings, self.params.clientId, self.params.embedding);
+        }
+        result.push([embeddings, index]);
     }
-    if (self.params.onCreate) {
-        self.params.onCreate(index, embeddings, self.params.clientId, self.params.embedding);
+    else {
+        for (const value of Object.values(index)) {
+            const hash = createSHA256Hash(value);
+            let embeddings = await self.params.readEmbeddingCache(self.params.embedding, hash);
+            if (!embeddings) {
+                embeddings = await self.params.createEmbedding(value, self.params.embedding);
+                await self.params.writeEmbeddingCache(embeddings, self.params.embedding, hash);
+            }
+            if (self.params.onCreate) {
+                self.params.onCreate(value, embeddings, self.params.clientId, self.params.embedding);
+            }
+            result.push([embeddings, value]);
+        }
     }
-    return [embeddings, index];
+    return result;
 };
 /**
  * Waits for the ClientStorage instance to initialize by loading initial data and creating embeddings.
@@ -8990,12 +9008,15 @@ class ClientStorage {
             this.params.onCreate(search, searchEmbeddings, this.params.clientId, this.params.embedding);
         }
         await Promise.all(Array.from(this._itemMap.values()).map(execpool(async (item) => {
-            const [targetEmbeddings, index] = await this._createEmbedding(item);
-            const score = await this.params.calculateSimilarity(searchEmbeddings, targetEmbeddings);
-            if (this.params.onCompare) {
-                this.params.onCompare(search, index, score, this.params.clientId, this.params.embedding);
+            let maxScore = Number.NEGATIVE_INFINITY;
+            for (const [targetEmbeddings, index] of await this._createEmbedding(item)) {
+                const score = await this.params.calculateSimilarity(searchEmbeddings, targetEmbeddings);
+                if (this.params.onCompare) {
+                    this.params.onCompare(search, index, score, this.params.clientId, this.params.embedding);
+                }
+                maxScore = Math.max(score, maxScore);
             }
-            indexed.push(item, score);
+            indexed.push(item, maxScore);
         }, {
             delay: 10,
             maxExec: GLOBAL_CONFIG.CC_STORAGE_SEARCH_POOL,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "agent-swarm-kit",
-  "version": "1.0.228",
+  "version": "1.0.229",
   "description": "A TypeScript library for building orchestrated framework-agnostic multi-agent AI systems",
   "author": {
     "name": "Petr Tripolsky",

package/types.d.ts

@@ -220,7 +220,7 @@ interface IStorageSchema<T extends IStorageData = IStorageData> {
      * @param {T} item - The storage item to index.
      * @returns {Promise<string> | string} The index string for the item, synchronously or asynchronously.
      */
-    createIndex(item: T): Promise<string> | string;
+    createIndex(item: T): Promise<string> | string | Record<string, string> | Promise<Record<string, string>>;
     /** The name of the embedding mechanism used for indexing and searching storage data. */
     embedding: EmbeddingName;
     /** The unique name of the storage within the swarm. */
@@ -6006,7 +6006,7 @@ declare class ClientStorage<T extends IStorageData = IStorageData> implements IS
      * @returns {Promise<readonly [Embeddings, string]>} A tuple of embeddings (from Embedding.interface) and index string.
      * @private
      */
-    _createEmbedding: ((item: T) => Promise<readonly [Embeddings, string]>) & functools_kit.IClearableMemoize<string | number> & functools_kit.IControlMemoize<string | number, Promise<readonly [Embeddings, string]>>;
+    _createEmbedding: ((item: T) => Promise<readonly [Embeddings, string][]>) & functools_kit.IClearableMemoize<string | number> & functools_kit.IControlMemoize<string | number, Promise<readonly [Embeddings, string][]>>;
     /**
      * Waits for the initialization of the storage, loading initial data and creating embeddings via WAIT_FOR_INIT_FN.
      * Ensures initialization happens only once using singleshot, supporting StorageConnectionService’s lifecycle.