@dvina/agents 0.3.0 → 0.3.2

package/dist/index.cjs

@@ -12838,7 +12838,9 @@ var require_out4 = __commonJS({
 // src/index.ts
 var index_exports = {};
 __export(index_exports, {
+  InMemoryToolCache: () => InMemoryToolCache,
   LangchainAgentFactory: () => LangchainAgentFactory,
+  RedisToolCache: () => RedisToolCache,
   ToolRegistry: () => ToolRegistry
 });
 module.exports = __toCommonJS(index_exports);
@@ -21708,20 +21710,14 @@ var ToolRegistry = class {
   }
   invalidate(toolKit) {
     const entry = this.entries.get(toolKit);
-    if (entry && entry.type === "provider") {
-      void this.toolSpecCache?.del?.(this.toolSpecsCacheKey(toolKit));
-    }
-  }
-  toolSpecsCacheKey(toolKit) {
-    return `tool-specs:v1:${toolKit}`;
+    if (entry && entry.type === "provider") this.toolSpecCache?.del(toolKit);
   }
   async getToolKitSpecs(toolKit, provider, context) {
-    const cacheKey = this.toolSpecsCacheKey(toolKit);
     const cache = this.toolSpecCache;
     if (cache) {
       try {
-        const cached = await Promise.resolve(cache.get(cacheKey));
-        if (cached) return JSON.parse(cached);
+        const cached = await cache.get(toolKit);
+        if (cached) return cached;
       } catch (e) {
         console.error("Something went wrong when hitting the cache", e);
       }
@@ -21729,7 +21725,7 @@ var ToolRegistry = class {
     const specs = await provider.listToolKitSpecs(toolKit, context) ?? [];
     if (cache) {
       try {
-        await Promise.resolve(cache.set(cacheKey, JSON.stringify(specs)));
+        await cache.set(toolKit, specs);
       } catch (e) {
         console.error("Something went wrong when setting the cache", e);
       }
@@ -21737,9 +21733,58 @@ var ToolRegistry = class {
     return specs;
   }
 };
+
+// src/core/tools/tool-cache.ts
+var import_ioredis = __toESM(require("ioredis"), 1);
+var import_zod2 = require("zod");
+var RedisToolCache = class {
+  redis;
+  constructor(connectionString) {
+    this.redis = new import_ioredis.default(connectionString);
+  }
+  async get(toolKit) {
+    const data = await this.redis.get(`tool-cache:${toolKit}`);
+    if (!data) return null;
+    try {
+      const parsed = JSON.parse(data);
+      if (!Array.isArray(parsed)) return null;
+      return parsed.map((spec) => ({
+        ...spec,
+        inputSchema: import_zod2.z.fromJSONSchema(spec.inputSchema)
+      }));
+    } catch (e) {
+      console.error("Failed to parse cached tools", e);
+      return null;
+    }
+  }
+  async set(toolKit, specs) {
+    const serialized = specs.map((spec) => ({
+      ...spec,
+      inputSchema: spec.inputSchema.toJSONSchema()
+    }));
+    await this.redis.set(`tool-cache:${toolKit}`, JSON.stringify(serialized));
+  }
+  async del(toolKit) {
+    await this.redis.del(`tool-cache:${toolKit}`);
+  }
+};
+var InMemoryToolCache = class {
+  cache = /* @__PURE__ */ new Map();
+  get(toolKit) {
+    return this.cache.get(`tool-cache:${toolKit}`) ?? null;
+  }
+  async set(toolKit, specs) {
+    this.cache.set(`tool-cache:${toolKit}`, specs);
+  }
+  async del(toolKit) {
+    this.cache.delete(`tool-cache:${toolKit}`);
+  }
+};
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  InMemoryToolCache,
   LangchainAgentFactory,
+  RedisToolCache,
   ToolRegistry
 });
 /*! Bundled license information: