@cloudbase/agent-adapter-langchain 0.0.2

package/dist/index.mjs

@@ -0,0 +1,406 @@
+// src/store/tdai.ts
+import { BaseStore } from "@langchain/core/stores";
+var TDAIStore = class extends BaseStore {
+  constructor(fields) {
+    super();
+    this.lc_namespace = ["langchain", "storage", "tdai"];
+    this.yieldKeysScanBatchSize = 1e3;
+    this.client = fields.client;
+    this.namespace = fields.namespace;
+    this.yieldKeysScanBatchSize = fields.yieldKeysScanBatchSize ?? this.yieldKeysScanBatchSize;
+    this.defaultSessionId = fields.defaultSessionId ?? "langchain_store_session";
+    this.defaultStrategy = fields.defaultStrategy ?? "kv_cache";
+    this.ttlSeconds = fields.ttlSeconds;
+  }
+  /**
+   * Get prefixed key with namespace
+   */
+  _getPrefixedKey(key) {
+    if (this.namespace) {
+      const delimiter = "/";
+      return `${this.namespace}${delimiter}${key}`;
+    }
+    return key;
+  }
+  /**
+   * Remove prefix from key
+   */
+  _getDeprefixedKey(key) {
+    if (this.namespace) {
+      const delimiter = "/";
+      return key.slice(this.namespace.length + delimiter.length);
+    }
+    return key;
+  }
+  /**
+   * Get or create session for the store
+   */
+  async _getSession() {
+    if (this.sessionCache) {
+      return this.sessionCache;
+    }
+    try {
+      const sessions = await this.client.querySessions({
+        limit: 1
+      });
+      if (sessions.sessions && sessions.sessions.length > 0) {
+        this.sessionCache = sessions.sessions[0].sessionId;
+      } else {
+        const result = await this.client.createSession({
+          name: `langchain_store_${this.namespace || "default"}`
+        });
+        this.sessionCache = result.sessionId;
+      }
+      return this.sessionCache;
+    } catch (error) {
+      return this.defaultSessionId;
+    }
+  }
+  /**
+   * Create record content for storage
+   */
+  _createRecordContent(key, value) {
+    const decoder = new TextDecoder();
+    const content = {
+      key: this._getPrefixedKey(key),
+      value: decoder.decode(value),
+      timestamp: Date.now(),
+      ttl: this.ttlSeconds
+    };
+    return JSON.stringify(content);
+  }
+  /**
+   * Parse record content from storage
+   */
+  _parseRecordContent(content) {
+    try {
+      const data = JSON.parse(content);
+      const encoder = new TextEncoder();
+      if (data.ttl && data.timestamp) {
+        const now = Date.now();
+        if (now > data.timestamp + data.ttl * 1e3) {
+          return null;
+        }
+      }
+      return {
+        key: data.key,
+        value: encoder.encode(data.value),
+        timestamp: data.timestamp,
+        ttl: data.ttl
+      };
+    } catch (error) {
+      return null;
+    }
+  }
+  /**
+   * Gets multiple keys from the TDAI store.
+   * @param keys Array of keys to be retrieved.
+   * @returns An array of retrieved values.
+   */
+  async mget(keys) {
+    if (keys.length === 0) {
+      return [];
+    }
+    try {
+      const sessionId = await this._getSession();
+      const results = new Array(keys.length);
+      for (let i = 0; i < keys.length; i++) {
+        const prefixedKey = this._getPrefixedKey(keys[i]);
+        try {
+          const searchResult = await this.client.searchRecords({
+            content: prefixedKey,
+            sessionId,
+            strategies: [this.defaultStrategy],
+            limit: 1
+          });
+          if (searchResult.records && searchResult.records.length > 0) {
+            const record = searchResult.records[0];
+            const parsed = this._parseRecordContent(record.record_content);
+            if (parsed && parsed.key === prefixedKey) {
+              results[i] = parsed.value;
+            } else {
+              results[i] = void 0;
+            }
+          } else {
+            results[i] = void 0;
+          }
+        } catch (error) {
+          results[i] = void 0;
+        }
+      }
+      return results;
+    } catch (error) {
+      return new Array(keys.length).fill(void 0);
+    }
+  }
+  /**
+   * Sets multiple keys in the TDAI store.
+   * @param keyValuePairs Array of key-value pairs to be set.
+   * @returns Promise that resolves when all keys have been set.
+   */
+  async mset(keyValuePairs) {
+    if (keyValuePairs.length === 0) {
+      return;
+    }
+    try {
+      const sessionId = await this._getSession();
+      for (const [key, value] of keyValuePairs) {
+        const content = this._createRecordContent(key, value);
+        try {
+          await this.client.appendRecord({
+            sessionId,
+            content,
+            strategy: this.defaultStrategy
+          });
+        } catch (error) {
+          console.warn(`Failed to set key ${key}:`, error);
+        }
+      }
+    } catch (error) {
+      throw new Error(`Failed to set keys: ${error}`);
+    }
+  }
+  /**
+   * Deletes multiple keys from the TDAI store.
+   * @param keys Array of keys to be deleted.
+   * @returns Promise that resolves when all keys have been deleted.
+   */
+  async mdelete(keys) {
+    if (keys.length === 0) {
+      return;
+    }
+    try {
+      const sessionId = await this._getSession();
+      for (const key of keys) {
+        const prefixedKey = this._getPrefixedKey(key);
+        try {
+          const searchResult = await this.client.searchRecords({
+            content: prefixedKey,
+            sessionId,
+            strategies: [this.defaultStrategy],
+            limit: 1
+          });
+          if (searchResult.records && searchResult.records.length > 0) {
+            const record = searchResult.records[0];
+            await this.client.deleteRecord({
+              sessionId,
+              recordId: record.record_id
+            });
+          }
+        } catch (error) {
+          console.warn(`Failed to delete key ${key}:`, error);
+        }
+      }
+    } catch (error) {
+      throw new Error(`Failed to delete keys: ${error}`);
+    }
+  }
+  /**
+   * Yields keys from the TDAI store.
+   * @param prefix Optional prefix to filter the keys.
+   * @returns An AsyncGenerator that yields keys from the TDAI store.
+   */
+  async *yieldKeys(prefix) {
+    try {
+      const sessionId = await this._getSession();
+      let offset = 0;
+      let hasMore = true;
+      while (hasMore) {
+        try {
+          const result = await this.client.queryRecords({
+            sessionId,
+            strategies: [this.defaultStrategy],
+            limit: this.yieldKeysScanBatchSize,
+            offset
+          });
+          if (!result.records || result.records.length === 0) {
+            hasMore = false;
+            break;
+          }
+          for (const record of result.records) {
+            const parsed = this._parseRecordContent(record.record_content);
+            if (parsed) {
+              const deprefixedKey = this._getDeprefixedKey(parsed.key);
+              if (!prefix || deprefixedKey.startsWith(prefix)) {
+                yield deprefixedKey;
+              }
+            }
+          }
+          offset += result.records.length;
+          if (result.records.length < this.yieldKeysScanBatchSize) {
+            hasMore = false;
+          }
+        } catch (error) {
+          console.warn(`Error yielding keys at offset ${offset}:`, error);
+          hasMore = false;
+        }
+      }
+    } catch (error) {
+      console.warn("Error yielding keys:", error);
+    }
+  }
+  /**
+   * Close the TDAI client connection
+   */
+  close() {
+    this.client.close();
+  }
+};
+
+// src/tool.ts
+import { tool as langChainTool } from "@langchain/core/tools";
+import {
+  tool as agKitTool,
+  ToolResult
+} from "@cloudbase/agent-tools";
+import { z } from "zod/v4";
+function convert2LangChain(agkitTool, implFunc) {
+  const func = implFunc || (async (input) => {
+    const result = await agkitTool.invoke(input);
+    return result;
+  });
+  return langChainTool(func, {
+    name: agkitTool.name,
+    description: agkitTool.description || agkitTool.name,
+    schema: agkitTool.schema || z.object({})
+  });
+}
+function convertLangChain2AGKit(langchainTool) {
+  const func = async (input, context) => {
+    try {
+      const result = await langchainTool.invoke(input);
+      return new ToolResult({
+        success: true,
+        data: result
+      });
+    } catch (error) {
+      return new ToolResult({
+        success: false,
+        error: error instanceof Error ? error.message : String(error),
+        error_type: "execution"
+      });
+    }
+  };
+  return agKitTool(func, {
+    name: langchainTool.name,
+    description: langchainTool.description || langchainTool.name,
+    schema: langchainTool.schema || z.object({})
+  });
+}
+
+// src/agent.ts
+import { z as z2 } from "zod/v3";
+import { LanggraphAgent } from "@cloudbase/agent-adapter-langgraph";
+import { createMiddleware } from "langchain";
+var LangchainAgent = class extends LanggraphAgent {
+  constructor(config) {
+    super({ ...config, compiledWorkflow: config.agent });
+  }
+};
+function agKitClientTools() {
+  return createMiddleware({
+    name: "AGKitClientTools",
+    stateSchema: z2.object({
+      agKit: z2.object({
+        actions: z2.array(
+          z2.object({
+            name: z2.string(),
+            description: z2.string(),
+            schema: z2.any()
+          })
+        )
+      })
+    }),
+    beforeModel: (state) => state,
+    wrapModelCall: (request, hanlder) => {
+      const clientTools = (request.state.agKit.actions || []).map((tool) => {
+        const { name, description, schema } = tool;
+        return {
+          name,
+          description,
+          schema
+        };
+      });
+      return hanlder({
+        ...request,
+        tools: [...request.tools || [], ...clientTools]
+      });
+    }
+  });
+}
+
+// src/chat_history.ts
+import { BaseListChatMessageHistory } from "@langchain/core/chat_history";
+import {
+  mapStoredMessagesToChatMessages,
+  mapChatMessagesToStoredMessages
+} from "@langchain/core/messages";
+import { MemoryClient } from "@cloudbase/agent-agents";
+var TDAIChatHistory = class extends BaseListChatMessageHistory {
+  constructor(options) {
+    super();
+    this.lc_namespace = ["langchain", "stores", "message", "tdai"];
+    const { sessionId, ...clientOptions } = options;
+    this.client = new MemoryClient({
+      ...clientOptions
+    });
+    this.sessionId = options.sessionId;
+  }
+  async getMessages() {
+    const { events = [] } = await this.client.queryEvents({
+      sessionId: this.sessionId
+    });
+    return mapStoredMessagesToChatMessages(
+      events.map((event) => {
+        return {
+          type: event.type,
+          data: event.data
+        };
+      })
+    );
+  }
+  /**
+   * Method to add a new message to the Firestore collection. The message is
+   * passed as a BaseMessage object.
+   * @param message The message to be added as a BaseMessage object.
+   */
+  async addMessage(message) {
+    const messages = mapChatMessagesToStoredMessages([message]);
+    await this.appendMessage(messages[0]);
+  }
+  async appendMessage(message) {
+    await this.client.appendEvent({
+      sessionId: this.sessionId,
+      messages: message
+    });
+  }
+  /**
+   * Method to delete all messages from the Firestore collection associated
+   * with the current session.
+   */
+  async clear() {
+    const eventIds = await this.client.queryEvents({
+      sessionId: this.sessionId
+    }).then(({ events = [] }) => events.map((event) => event.event_id));
+    await Promise.all(
+      eventIds.map((id) => {
+        return this.client.deleteEvent({
+          sessionId: this.sessionId,
+          eventId: id
+        });
+      })
+    );
+    await this.client.deleteSession({
+      sessionId: this.sessionId
+    });
+  }
+};
+export {
+  LangchainAgent,
+  TDAIChatHistory,
+  TDAIStore,
+  agKitClientTools,
+  convert2LangChain,
+  convertLangChain2AGKit
+};
+//# sourceMappingURL=index.mjs.map

package/dist/index.mjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../src/store/tdai.ts","../src/tool.ts","../src/agent.ts","../src/chat_history.ts"],"sourcesContent":["import { MemoryClient } from \"@cloudbase/agent-agents\";\nimport { BaseStore } from \"@langchain/core/stores\";\n\n/**\n * Configuration options for TDAIStore\n */\nexport interface TDAIStoreInput {\n  /**\n   * The TDAI Memory Client instance\n   */\n  client: MemoryClient;\n\n  /**\n   * The amount of keys to retrieve per batch when yielding keys.\n   * @default 1000\n   */\n  yieldKeysScanBatchSize?: number;\n\n  /**\n   * The namespace to use for the keys in the database.\n   */\n  namespace?: string;\n\n  /**\n   * Default session ID for storing records\n   */\n  defaultSessionId?: string;\n\n  /**\n   * Default strategy for storing records\n   */\n  defaultStrategy?: string;\n\n  /**\n   * TTL for records in seconds\n   */\n  ttlSeconds?: number;\n}\n\n/**\n * TDAI implementation of the BaseStore for key-value caching.\n * Uses TDAI Memory Client for persistent storage.\n *\n * @example\n * ```typescript\n * const client = new MemoryClient({\n *   endpoint: \"https://memory.tdai.tencentyun.com\",\n *   apiKey: \"your-api-key\",\n *   memoryId: \"your-memory-id\",\n * });\n *\n * const store = new TDAIStore({\n *   client,\n *   namespace: \"cache\",\n * });\n *\n * const encoder = new TextEncoder();\n * await store.mset([\n *   [\"key1\", encoder.encode(\"value1\")],\n *   [\"key2\", encoder.encode(\"value2\")],\n * ]);\n *\n * const values = await store.mget([\"key1\", \"key2\"]);\n * ```\n */\nexport class TDAIStore extends BaseStore<string, Uint8Array> {\n  lc_namespace = [\"langchain\", \"storage\", \"tdai\"];\n\n  protected client: MemoryClient;\n  protected namespace?: string;\n  protected yieldKeysScanBatchSize = 1000;\n  protected defaultSessionId: string;\n  protected defaultStrategy: string;\n  protected ttlSeconds?: number;\n  private sessionCache?: string;\n\n  constructor(fields: TDAIStoreInput) {\n    super();\n    this.client = fields.client;\n    this.namespace = fields.namespace;\n    this.yieldKeysScanBatchSize =\n      fields.yieldKeysScanBatchSize ?? this.yieldKeysScanBatchSize;\n    this.defaultSessionId =\n      fields.defaultSessionId ?? \"langchain_store_session\";\n    this.defaultStrategy = fields.defaultStrategy ?? \"kv_cache\";\n    this.ttlSeconds = fields.ttlSeconds;\n  }\n\n  /**\n   * Get prefixed key with namespace\n   */\n  private _getPrefixedKey(key: string): string {\n    if (this.namespace) {\n      const delimiter = \"/\";\n      return `${this.namespace}${delimiter}${key}`;\n    }\n    return key;\n  }\n\n  /**\n   * Remove prefix from key\n   */\n  private _getDeprefixedKey(key: string): string {\n    if (this.namespace) {\n      const delimiter = \"/\";\n      return key.slice(this.namespace.length + delimiter.length);\n    }\n    return key;\n  }\n\n  /**\n   * Get or create session for the store\n   */\n  private async _getSession(): Promise<string> {\n    if (this.sessionCache) {\n      return this.sessionCache;\n    }\n\n    try {\n      // Try to find existing session\n      const sessions = await this.client.querySessions({\n        limit: 1,\n      });\n\n      if (sessions.sessions && sessions.sessions.length > 0) {\n        this.sessionCache = sessions.sessions[0].sessionId;\n      } else {\n        // Create new session\n        const result = await this.client.createSession({\n          name: `langchain_store_${this.namespace || \"default\"}`,\n        });\n        this.sessionCache = result.sessionId;\n      }\n\n      return this.sessionCache!;\n    } catch (error) {\n      // Fallback to default session\n      return this.defaultSessionId;\n    }\n  }\n\n  /**\n   * Create record content for storage\n   */\n  private _createRecordContent(key: string, value: Uint8Array): string {\n    const decoder = new TextDecoder();\n    const content = {\n      key: this._getPrefixedKey(key),\n      value: decoder.decode(value),\n      timestamp: Date.now(),\n      ttl: this.ttlSeconds,\n    };\n    return JSON.stringify(content);\n  }\n\n  /**\n   * Parse record content from storage\n   */\n  private _parseRecordContent(\n    content: string\n  ): {\n    key: string;\n    value: Uint8Array;\n    timestamp: number;\n    ttl?: number;\n  } | null {\n    try {\n      const data = JSON.parse(content);\n      const encoder = new TextEncoder();\n\n      // Check TTL if present\n      if (data.ttl && data.timestamp) {\n        const now = Date.now();\n        if (now > data.timestamp + data.ttl * 1000) {\n          return null; // Expired\n        }\n      }\n\n      return {\n        key: data.key,\n        value: encoder.encode(data.value),\n        timestamp: data.timestamp,\n        ttl: data.ttl,\n      };\n    } catch (error) {\n      return null;\n    }\n  }\n\n  /**\n   * Gets multiple keys from the TDAI store.\n   * @param keys Array of keys to be retrieved.\n   * @returns An array of retrieved values.\n   */\n  async mget(keys: string[]): Promise<(Uint8Array | undefined)[]> {\n    if (keys.length === 0) {\n      return [];\n    }\n\n    try {\n      const sessionId = await this._getSession();\n      const results: (Uint8Array | undefined)[] = new Array(keys.length);\n\n      // Search for each key individually due to TDAI API limitations\n      for (let i = 0; i < keys.length; i++) {\n        const prefixedKey = this._getPrefixedKey(keys[i]);\n\n        try {\n          const searchResult = await this.client.searchRecords({\n            content: prefixedKey,\n            sessionId,\n            strategies: [this.defaultStrategy],\n            limit: 1,\n          });\n\n          if (searchResult.records && searchResult.records.length > 0) {\n            const record = searchResult.records[0];\n            const parsed = this._parseRecordContent(record.record_content);\n\n            if (parsed && parsed.key === prefixedKey) {\n              results[i] = parsed.value;\n            } else {\n              results[i] = undefined;\n            }\n          } else {\n            results[i] = undefined;\n          }\n        } catch (error) {\n          results[i] = undefined;\n        }\n      }\n\n      return results;\n    } catch (error) {\n      // Return array of undefined values on error\n      return new Array(keys.length).fill(undefined);\n    }\n  }\n\n  /**\n   * Sets multiple keys in the TDAI store.\n   * @param keyValuePairs Array of key-value pairs to be set.\n   * @returns Promise that resolves when all keys have been set.\n   */\n  async mset(keyValuePairs: [string, Uint8Array][]): Promise<void> {\n    if (keyValuePairs.length === 0) {\n      return;\n    }\n\n    try {\n      const sessionId = await this._getSession();\n\n      // Set each key-value pair\n      for (const [key, value] of keyValuePairs) {\n        const content = this._createRecordContent(key, value);\n\n        try {\n          await this.client.appendRecord({\n            sessionId,\n            content,\n            strategy: this.defaultStrategy,\n          });\n        } catch (error) {\n          console.warn(`Failed to set key ${key}:`, error);\n        }\n      }\n    } catch (error) {\n      throw new Error(`Failed to set keys: ${error}`);\n    }\n  }\n\n  /**\n   * Deletes multiple keys from the TDAI store.\n   * @param keys Array of keys to be deleted.\n   * @returns Promise that resolves when all keys have been deleted.\n   */\n  async mdelete(keys: string[]): Promise<void> {\n    if (keys.length === 0) {\n      return;\n    }\n\n    try {\n      const sessionId = await this._getSession();\n\n      // Delete each key individually\n      for (const key of keys) {\n        const prefixedKey = this._getPrefixedKey(key);\n\n        try {\n          // First find the record\n          const searchResult = await this.client.searchRecords({\n            content: prefixedKey,\n            sessionId,\n            strategies: [this.defaultStrategy],\n            limit: 1,\n          });\n\n          if (searchResult.records && searchResult.records.length > 0) {\n            const record = searchResult.records[0];\n            await this.client.deleteRecord({\n              sessionId,\n              recordId: record.record_id,\n            });\n          }\n        } catch (error) {\n          console.warn(`Failed to delete key ${key}:`, error);\n        }\n      }\n    } catch (error) {\n      throw new Error(`Failed to delete keys: ${error}`);\n    }\n  }\n\n  /**\n   * Yields keys from the TDAI store.\n   * @param prefix Optional prefix to filter the keys.\n   * @returns An AsyncGenerator that yields keys from the TDAI store.\n   */\n  async *yieldKeys(prefix?: string): AsyncGenerator<string> {\n    try {\n      const sessionId = await this._getSession();\n      let offset = 0;\n      let hasMore = true;\n\n      while (hasMore) {\n        try {\n          const result = await this.client.queryRecords({\n            sessionId,\n            strategies: [this.defaultStrategy],\n            limit: this.yieldKeysScanBatchSize,\n            offset,\n          });\n\n          if (!result.records || result.records.length === 0) {\n            hasMore = false;\n            break;\n          }\n\n          for (const record of result.records) {\n            const parsed = this._parseRecordContent(record.record_content);\n            if (parsed) {\n              const deprefixedKey = this._getDeprefixedKey(parsed.key);\n\n              if (!prefix || deprefixedKey.startsWith(prefix)) {\n                yield deprefixedKey;\n              }\n            }\n          }\n\n          offset += result.records.length;\n\n          // If we got fewer records than requested, we've reached the end\n          if (result.records.length < this.yieldKeysScanBatchSize) {\n            hasMore = false;\n          }\n        } catch (error) {\n          console.warn(`Error yielding keys at offset ${offset}:`, error);\n          hasMore = false;\n        }\n      }\n    } catch (error) {\n      console.warn(\"Error yielding keys:\", error);\n    }\n  }\n\n  /**\n   * Close the TDAI client connection\n   */\n  close(): void {\n    this.client.close();\n  }\n}\n\n","/**\n * AG-Kit BaseTool to LangChain Tool Converter and Vice Versa\n */\nimport { tool as langChainTool, StructuredTool } from \"@langchain/core/tools\";\nimport {\n  BaseTool,\n  tool as agKitTool,\n  ToolResult,\n  ToolExecutionContext,\n} from \"@cloudbase/agent-tools\";\nimport { z } from \"zod/v4\";\n\n/**\n * Convert AG-Kit BaseTool to LangChain DynamicStructuredTool\n *\n * @param agkitTool - AG-Kit BaseTool instance\n * @param implFunc - Optional custom implementation function\n * @returns LangChain DynamicStructuredTool instance\n */\nexport function convert2LangChain(\n  agkitTool: BaseTool,\n  implFunc?: (tool: BaseTool) => Function\n) {\n  const func =\n    implFunc ||\n    (async (input: any) => {\n      const result = await agkitTool.invoke(input);\n      return result;\n    });\n\n  return langChainTool(func as any, {\n    name: agkitTool.name,\n    description: agkitTool.description || agkitTool.name,\n    schema: (agkitTool as any).schema || z.object({}),\n  });\n}\n\n/**\n * Convert LangChain Tool to AG-Kit DynamicTool\n *\n * @param langchainTool - LangChain tool instance (DynamicStructuredTool or StructuredTool)\n * @returns AG-Kit DynamicTool instance\n */\nexport function convertLangChain2AGKit(langchainTool: StructuredTool) {\n  const func: any = async (\n    input: Record<string, any>,\n    context?: ToolExecutionContext\n  ) => {\n    try {\n      // Call the LangChain tool\n      const result = await langchainTool.invoke(input);\n\n      // Return successful result\n      return new ToolResult({\n        success: true,\n        data: result,\n      });\n    } catch (error) {\n      // Handle errors and return ToolResult with error\n      return new ToolResult({\n        success: false,\n        error: error instanceof Error ? error.message : String(error),\n        error_type: \"execution\",\n      });\n    }\n  };\n\n  return agKitTool(func, {\n    name: langchainTool.name,\n    description: langchainTool.description || langchainTool.name,\n    schema: (langchainTool as any).schema || (z.object({}) as z.ZodObject<any>),\n  });\n}\n\n","import { z } from \"zod/v3\";\nimport { AgentConfig } from \"@cloudbase/agent-agents/abstract\";\nimport { LanggraphAgent } from \"@cloudbase/agent-adapter-langgraph\";\nimport { createAgent, createMiddleware, DynamicTool } from \"langchain\";\n\nexport class LangchainAgent extends LanggraphAgent {\n  constructor(\n    config: AgentConfig & {\n      agent: ReturnType<typeof createAgent>;\n    }\n  ) {\n    super({ ...config, compiledWorkflow: config.agent });\n  }\n}\n\nexport function agKitClientTools() {\n  return createMiddleware({\n    name: \"AGKitClientTools\",\n    stateSchema: z.object({\n      agKit: z.object({\n        actions: z.array(\n          z.object({\n            name: z.string(),\n            description: z.string(),\n            schema: z.any(),\n          })\n        ),\n      }),\n    }),\n    beforeModel: (state) => state,\n    wrapModelCall: (request, hanlder) => {\n      const clientTools = (\n        (request.state.agKit.actions || []) as DynamicTool[]\n      ).map((tool) => {\n        const { name, description, schema } = tool;\n        return {\n          name,\n          description,\n          schema,\n        };\n      });\n\n      return hanlder({\n        ...request,\n        tools: [...(request.tools || []), ...clientTools],\n      });\n    },\n  });\n}\n\n","import { BaseListChatMessageHistory } from \"@langchain/core/chat_history\";\nimport {\n  BaseMessage,\n  mapStoredMessagesToChatMessages,\n  mapChatMessagesToStoredMessages,\n  StoredMessage,\n} from \"@langchain/core/messages\";\nimport { IMemoryClientOptions, MemoryClient } from \"@cloudbase/agent-agents\";\n\nexport class TDAIChatHistory extends BaseListChatMessageHistory {\n  lc_namespace = [\"langchain\", \"stores\", \"message\", \"tdai\"];\n\n  public client: MemoryClient;\n\n  private sessionId: string;\n\n  constructor(options: IMemoryClientOptions & { sessionId: string }) {\n    super();\n    const { sessionId, ...clientOptions } = options;\n    this.client = new MemoryClient({\n      ...clientOptions,\n    });\n    this.sessionId = options.sessionId;\n  }\n\n  async getMessages(): Promise<BaseMessage[]> {\n    const { events = [] } = await this.client.queryEvents({\n      sessionId: this.sessionId,\n    });\n\n    return mapStoredMessagesToChatMessages(\n      events.map((event) => {\n        return {\n          type: event.type,\n          data: event.data,\n        };\n      })\n    );\n  }\n\n  /**\n   * Method to add a new message to the Firestore collection. The message is\n   * passed as a BaseMessage object.\n   * @param message The message to be added as a BaseMessage object.\n   */\n  public async addMessage(message: BaseMessage) {\n    const messages = mapChatMessagesToStoredMessages([message]);\n    await this.appendMessage(messages[0]);\n  }\n\n  private async appendMessage(message: StoredMessage): Promise<void> {\n    await this.client.appendEvent({\n      sessionId: this.sessionId,\n      messages: message,\n    });\n  }\n\n  /**\n   * Method to delete all messages from the Firestore collection associated\n   * with the current session.\n   */\n  public async clear(): Promise<void> {\n    const eventIds = await this.client\n      .queryEvents({\n        sessionId: this.sessionId,\n      })\n      .then(({ events = [] }) => events.map((event: any) => event.event_id));\n\n    await Promise.all(\n      eventIds.map((id: string) => {\n        return this.client.deleteEvent({\n          sessionId: this.sessionId,\n          eventId: id,\n        });\n      })\n    );\n\n    await this.client.deleteSession({\n      sessionId: this.sessionId,\n    });\n  }\n}\n\n"],"mappings":";AACA,SAAS,iBAAiB;AAgEnB,IAAM,YAAN,cAAwB,UAA8B;AAAA,EAW3D,YAAY,QAAwB;AAClC,UAAM;AAXR,wBAAe,CAAC,aAAa,WAAW,MAAM;AAI9C,SAAU,yBAAyB;AAQjC,SAAK,SAAS,OAAO;AACrB,SAAK,YAAY,OAAO;AACxB,SAAK,yBACH,OAAO,0BAA0B,KAAK;AACxC,SAAK,mBACH,OAAO,oBAAoB;AAC7B,SAAK,kBAAkB,OAAO,mBAAmB;AACjD,SAAK,aAAa,OAAO;AAAA,EAC3B;AAAA;AAAA;AAAA;AAAA,EAKQ,gBAAgB,KAAqB;AAC3C,QAAI,KAAK,WAAW;AAClB,YAAM,YAAY;AAClB,aAAO,GAAG,KAAK,SAAS,GAAG,SAAS,GAAG,GAAG;AAAA,IAC5C;AACA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKQ,kBAAkB,KAAqB;AAC7C,QAAI,KAAK,WAAW;AAClB,YAAM,YAAY;AAClB,aAAO,IAAI,MAAM,KAAK,UAAU,SAAS,UAAU,MAAM;AAAA,IAC3D;AACA,WAAO;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKA,MAAc,cAA+B;AAC3C,QAAI,KAAK,cAAc;AACrB,aAAO,KAAK;AAAA,IACd;AAEA,QAAI;AAEF,YAAM,WAAW,MAAM,KAAK,OAAO,cAAc;AAAA,QAC/C,OAAO;AAAA,MACT,CAAC;AAED,UAAI,SAAS,YAAY,SAAS,SAAS,SAAS,GAAG;AACrD,aAAK,eAAe,SAAS,SAAS,CAAC,EAAE;AAAA,MAC3C,OAAO;AAEL,cAAM,SAAS,MAAM,KAAK,OAAO,cAAc;AAAA,UAC7C,MAAM,mBAAmB,KAAK,aAAa,SAAS;AAAA,QACtD,CAAC;AACD,aAAK,eAAe,OAAO;AAAA,MAC7B;AAEA,aAAO,KAAK;AAAA,IACd,SAAS,OAAO;AAEd,aAAO,KAAK;AAAA,IACd;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKQ,qBAAqB,KAAa,OAA2B;AACnE,UAAM,UAAU,IAAI,YAAY;AAChC,UAAM,UAAU;AAAA,MACd,KAAK,KAAK,gBAAgB,GAAG;AAAA,MAC7B,OAAO,QAAQ,OAAO,KAAK;AAAA,MAC3B,WAAW,KAAK,IAAI;AAAA,MACpB,KAAK,KAAK;AAAA,IACZ;AACA,WAAO,KAAK,UAAU,OAAO;AAAA,EAC/B;AAAA;AAAA;AAAA;AAAA,EAKQ,oBACN,SAMO;AACP,QAAI;AACF,YAAM,OAAO,KAAK,MAAM,OAAO;AAC/B,YAAM,UAAU,IAAI,YAAY;AAGhC,UAAI,KAAK,OAAO,KAAK,WAAW;AAC9B,cAAM,MAAM,KAAK,IAAI;AACrB,YAAI,MAAM,KAAK,YAAY,KAAK,MAAM,KAAM;AAC1C,iBAAO;AAAA,QACT;AAAA,MACF;AAEA,aAAO;AAAA,QACL,KAAK,KAAK;AAAA,QACV,OAAO,QAAQ,OAAO,KAAK,KAAK;AAAA,QAChC,WAAW,KAAK;AAAA,QAChB,KAAK,KAAK;AAAA,MACZ;AAAA,IACF,SAAS,OAAO;AACd,aAAO;AAAA,IACT;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,MAAM,KAAK,MAAqD;AAC9D,QAAI,KAAK,WAAW,GAAG;AACrB,aAAO,CAAC;AAAA,IACV;AAEA,QAAI;AACF,YAAM,YAAY,MAAM,KAAK,YAAY;AACzC,YAAM,UAAsC,IAAI,MAAM,KAAK,MAAM;AAGjE,eAAS,IAAI,GAAG,IAAI,KAAK,QAAQ,KAAK;AACpC,cAAM,cAAc,KAAK,gBAAgB,KAAK,CAAC,CAAC;AAEhD,YAAI;AACF,gBAAM,eAAe,MAAM,KAAK,OAAO,cAAc;AAAA,YACnD,SAAS;AAAA,YACT;AAAA,YACA,YAAY,CAAC,KAAK,eAAe;AAAA,YACjC,OAAO;AAAA,UACT,CAAC;AAED,cAAI,aAAa,WAAW,aAAa,QAAQ,SAAS,GAAG;AAC3D,kBAAM,SAAS,aAAa,QAAQ,CAAC;AACrC,kBAAM,SAAS,KAAK,oBAAoB,OAAO,cAAc;AAE7D,gBAAI,UAAU,OAAO,QAAQ,aAAa;AACxC,sBAAQ,CAAC,IAAI,OAAO;AAAA,YACtB,OAAO;AACL,sBAAQ,CAAC,IAAI;AAAA,YACf;AAAA,UACF,OAAO;AACL,oBAAQ,CAAC,IAAI;AAAA,UACf;AAAA,QACF,SAAS,OAAO;AACd,kBAAQ,CAAC,IAAI;AAAA,QACf;AAAA,MACF;AAEA,aAAO;AAAA,IACT,SAAS,OAAO;AAEd,aAAO,IAAI,MAAM,KAAK,MAAM,EAAE,KAAK,MAAS;AAAA,IAC9C;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,MAAM,KAAK,eAAsD;AAC/D,QAAI,cAAc,WAAW,GAAG;AAC9B;AAAA,IACF;AAEA,QAAI;AACF,YAAM,YAAY,MAAM,KAAK,YAAY;AAGzC,iBAAW,CAAC,KAAK,KAAK,KAAK,eAAe;AACxC,cAAM,UAAU,KAAK,qBAAqB,KAAK,KAAK;AAEpD,YAAI;AACF,gBAAM,KAAK,OAAO,aAAa;AAAA,YAC7B;AAAA,YACA;AAAA,YACA,UAAU,KAAK;AAAA,UACjB,CAAC;AAAA,QACH,SAAS,OAAO;AACd,kBAAQ,KAAK,qBAAqB,GAAG,KAAK,KAAK;AAAA,QACjD;AAAA,MACF;AAAA,IACF,SAAS,OAAO;AACd,YAAM,IAAI,MAAM,uBAAuB,KAAK,EAAE;AAAA,IAChD;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,MAAM,QAAQ,MAA+B;AAC3C,QAAI,KAAK,WAAW,GAAG;AACrB;AAAA,IACF;AAEA,QAAI;AACF,YAAM,YAAY,MAAM,KAAK,YAAY;AAGzC,iBAAW,OAAO,MAAM;AACtB,cAAM,cAAc,KAAK,gBAAgB,GAAG;AAE5C,YAAI;AAEF,gBAAM,eAAe,MAAM,KAAK,OAAO,cAAc;AAAA,YACnD,SAAS;AAAA,YACT;AAAA,YACA,YAAY,CAAC,KAAK,eAAe;AAAA,YACjC,OAAO;AAAA,UACT,CAAC;AAED,cAAI,aAAa,WAAW,aAAa,QAAQ,SAAS,GAAG;AAC3D,kBAAM,SAAS,aAAa,QAAQ,CAAC;AACrC,kBAAM,KAAK,OAAO,aAAa;AAAA,cAC7B;AAAA,cACA,UAAU,OAAO;AAAA,YACnB,CAAC;AAAA,UACH;AAAA,QACF,SAAS,OAAO;AACd,kBAAQ,KAAK,wBAAwB,GAAG,KAAK,KAAK;AAAA,QACpD;AAAA,MACF;AAAA,IACF,SAAS,OAAO;AACd,YAAM,IAAI,MAAM,0BAA0B,KAAK,EAAE;AAAA,IACnD;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,OAAO,UAAU,QAAyC;AACxD,QAAI;AACF,YAAM,YAAY,MAAM,KAAK,YAAY;AACzC,UAAI,SAAS;AACb,UAAI,UAAU;AAEd,aAAO,SAAS;AACd,YAAI;AACF,gBAAM,SAAS,MAAM,KAAK,OAAO,aAAa;AAAA,YAC5C;AAAA,YACA,YAAY,CAAC,KAAK,eAAe;AAAA,YACjC,OAAO,KAAK;AAAA,YACZ;AAAA,UACF,CAAC;AAED,cAAI,CAAC,OAAO,WAAW,OAAO,QAAQ,WAAW,GAAG;AAClD,sBAAU;AACV;AAAA,UACF;AAEA,qBAAW,UAAU,OAAO,SAAS;AACnC,kBAAM,SAAS,KAAK,oBAAoB,OAAO,cAAc;AAC7D,gBAAI,QAAQ;AACV,oBAAM,gBAAgB,KAAK,kBAAkB,OAAO,GAAG;AAEvD,kBAAI,CAAC,UAAU,cAAc,WAAW,MAAM,GAAG;AAC/C,sBAAM;AAAA,cACR;AAAA,YACF;AAAA,UACF;AAEA,oBAAU,OAAO,QAAQ;AAGzB,cAAI,OAAO,QAAQ,SAAS,KAAK,wBAAwB;AACvD,sBAAU;AAAA,UACZ;AAAA,QACF,SAAS,OAAO;AACd,kBAAQ,KAAK,iCAAiC,MAAM,KAAK,KAAK;AAC9D,oBAAU;AAAA,QACZ;AAAA,MACF;AAAA,IACF,SAAS,OAAO;AACd,cAAQ,KAAK,wBAAwB,KAAK;AAAA,IAC5C;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKA,QAAc;AACZ,SAAK,OAAO,MAAM;AAAA,EACpB;AACF;;;AChXA,SAAS,QAAQ,qBAAqC;AACtD;AAAA,EAEE,QAAQ;AAAA,EACR;AAAA,OAEK;AACP,SAAS,SAAS;AASX,SAAS,kBACd,WACA,UACA;AACA,QAAM,OACJ,aACC,OAAO,UAAe;AACrB,UAAM,SAAS,MAAM,UAAU,OAAO,KAAK;AAC3C,WAAO;AAAA,EACT;AAEF,SAAO,cAAc,MAAa;AAAA,IAChC,MAAM,UAAU;AAAA,IAChB,aAAa,UAAU,eAAe,UAAU;AAAA,IAChD,QAAS,UAAkB,UAAU,EAAE,OAAO,CAAC,CAAC;AAAA,EAClD,CAAC;AACH;AAQO,SAAS,uBAAuB,eAA+B;AACpE,QAAM,OAAY,OAChB,OACA,YACG;AACH,QAAI;AAEF,YAAM,SAAS,MAAM,cAAc,OAAO,KAAK;AAG/C,aAAO,IAAI,WAAW;AAAA,QACpB,SAAS;AAAA,QACT,MAAM;AAAA,MACR,CAAC;AAAA,IACH,SAAS,OAAO;AAEd,aAAO,IAAI,WAAW;AAAA,QACpB,SAAS;AAAA,QACT,OAAO,iBAAiB,QAAQ,MAAM,UAAU,OAAO,KAAK;AAAA,QAC5D,YAAY;AAAA,MACd,CAAC;AAAA,IACH;AAAA,EACF;AAEA,SAAO,UAAU,MAAM;AAAA,IACrB,MAAM,cAAc;AAAA,IACpB,aAAa,cAAc,eAAe,cAAc;AAAA,IACxD,QAAS,cAAsB,UAAW,EAAE,OAAO,CAAC,CAAC;AAAA,EACvD,CAAC;AACH;;;ACxEA,SAAS,KAAAA,UAAS;AAElB,SAAS,sBAAsB;AAC/B,SAAsB,wBAAqC;AAEpD,IAAM,iBAAN,cAA6B,eAAe;AAAA,EACjD,YACE,QAGA;AACA,UAAM,EAAE,GAAG,QAAQ,kBAAkB,OAAO,MAAM,CAAC;AAAA,EACrD;AACF;AAEO,SAAS,mBAAmB;AACjC,SAAO,iBAAiB;AAAA,IACtB,MAAM;AAAA,IACN,aAAaA,GAAE,OAAO;AAAA,MACpB,OAAOA,GAAE,OAAO;AAAA,QACd,SAASA,GAAE;AAAA,UACTA,GAAE,OAAO;AAAA,YACP,MAAMA,GAAE,OAAO;AAAA,YACf,aAAaA,GAAE,OAAO;AAAA,YACtB,QAAQA,GAAE,IAAI;AAAA,UAChB,CAAC;AAAA,QACH;AAAA,MACF,CAAC;AAAA,IACH,CAAC;AAAA,IACD,aAAa,CAAC,UAAU;AAAA,IACxB,eAAe,CAAC,SAAS,YAAY;AACnC,YAAM,eACH,QAAQ,MAAM,MAAM,WAAW,CAAC,GACjC,IAAI,CAAC,SAAS;AACd,cAAM,EAAE,MAAM,aAAa,OAAO,IAAI;AACtC,eAAO;AAAA,UACL;AAAA,UACA;AAAA,UACA;AAAA,QACF;AAAA,MACF,CAAC;AAED,aAAO,QAAQ;AAAA,QACb,GAAG;AAAA,QACH,OAAO,CAAC,GAAI,QAAQ,SAAS,CAAC,GAAI,GAAG,WAAW;AAAA,MAClD,CAAC;AAAA,IACH;AAAA,EACF,CAAC;AACH;;;AChDA,SAAS,kCAAkC;AAC3C;AAAA,EAEE;AAAA,EACA;AAAA,OAEK;AACP,SAA+B,oBAAoB;AAE5C,IAAM,kBAAN,cAA8B,2BAA2B;AAAA,EAO9D,YAAY,SAAuD;AACjE,UAAM;AAPR,wBAAe,CAAC,aAAa,UAAU,WAAW,MAAM;AAQtD,UAAM,EAAE,WAAW,GAAG,cAAc,IAAI;AACxC,SAAK,SAAS,IAAI,aAAa;AAAA,MAC7B,GAAG;AAAA,IACL,CAAC;AACD,SAAK,YAAY,QAAQ;AAAA,EAC3B;AAAA,EAEA,MAAM,cAAsC;AAC1C,UAAM,EAAE,SAAS,CAAC,EAAE,IAAI,MAAM,KAAK,OAAO,YAAY;AAAA,MACpD,WAAW,KAAK;AAAA,IAClB,CAAC;AAED,WAAO;AAAA,MACL,OAAO,IAAI,CAAC,UAAU;AACpB,eAAO;AAAA,UACL,MAAM,MAAM;AAAA,UACZ,MAAM,MAAM;AAAA,QACd;AAAA,MACF,CAAC;AAAA,IACH;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,MAAa,WAAW,SAAsB;AAC5C,UAAM,WAAW,gCAAgC,CAAC,OAAO,CAAC;AAC1D,UAAM,KAAK,cAAc,SAAS,CAAC,CAAC;AAAA,EACtC;AAAA,EAEA,MAAc,cAAc,SAAuC;AACjE,UAAM,KAAK,OAAO,YAAY;AAAA,MAC5B,WAAW,KAAK;AAAA,MAChB,UAAU;AAAA,IACZ,CAAC;AAAA,EACH;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAa,QAAuB;AAClC,UAAM,WAAW,MAAM,KAAK,OACzB,YAAY;AAAA,MACX,WAAW,KAAK;AAAA,IAClB,CAAC,EACA,KAAK,CAAC,EAAE,SAAS,CAAC,EAAE,MAAM,OAAO,IAAI,CAAC,UAAe,MAAM,QAAQ,CAAC;AAEvE,UAAM,QAAQ;AAAA,MACZ,SAAS,IAAI,CAAC,OAAe;AAC3B,eAAO,KAAK,OAAO,YAAY;AAAA,UAC7B,WAAW,KAAK;AAAA,UAChB,SAAS;AAAA,QACX,CAAC;AAAA,MACH,CAAC;AAAA,IACH;AAEA,UAAM,KAAK,OAAO,cAAc;AAAA,MAC9B,WAAW,KAAK;AAAA,IAClB,CAAC;AAAA,EACH;AACF;","names":["z"]}

package/package.json

@@ -0,0 +1,55 @@
+{
+  "name": "@cloudbase/agent-adapter-langchain",
+  "version": "0.0.2",
+  "description": "LangChain adapter for AG-Kit agents",
+  "files": [
+    "dist/",
+    "README.md",
+    "CHANGELOG.md"
+  ],
+  "main": "dist/index.js",
+  "module": "dist/index.mjs",
+  "types": "dist/index.d.ts",
+  "exports": {
+    ".": {
+      "import": {
+        "types": "./dist/index.d.mts",
+        "default": "./dist/index.mjs"
+      },
+      "require": {
+        "types": "./dist/index.d.ts",
+        "default": "./dist/index.js"
+      }
+    }
+  },
+  "keywords": [
+    "ag-kit",
+    "langchain",
+    "adapter",
+    "agent"
+  ],
+  "author": "",
+  "license": "ISC",
+  "devDependencies": {
+    "@types/node": "^20.0.0",
+    "tsup": "^8.5.0",
+    "typescript": "^5.0.0"
+  },
+  "dependencies": {
+    "@langchain/core": "^1.0.2",
+    "langchain": "^1.0.2",
+    "zod": "^3.25.0 || ^4.0.0",
+    "@cloudbase/agent-adapter-langgraph": "0.0.2",
+    "@cloudbase/agent-tools": "0.0.2",
+    "@cloudbase/agent-agents": "0.0.2"
+  },
+  "peerDependencies": {
+    "zod": "^3.25.0 || ^4.0.0",
+    "@cloudbase/agent-agents": "0.0.2",
+    "@cloudbase/agent-adapter-langgraph": "0.0.2"
+  },
+  "scripts": {
+    "test": "echo \"Error: no test specified\" && exit 1",
+    "build": "tsup --config tsup.config.ts"
+  }
+}