langchain 0.0.7 → 0.0.9

package/agents.mjs

@@ -0,0 +1 @@
+export * from './dist/agents/index.js'

package/chains.mjs

@@ -0,0 +1 @@
+export * from './dist/chains/index.js'

package/dist/agents/tests/{agent.test.js → agent.int.test.js}

@@ -26,10 +26,9 @@ const initialize_1 = require("../initialize");
     const tools = [new serpapi_1.SerpAPI(), new calculator_1.Calculator()];
     const executor = await (0, initialize_1.initializeAgentExecutor)(tools, model, "zero-shot-react-description");
     console.log("Loaded agent.");
-    const input = "Who is Olivia Wilde's boyfriend?" +
-        " What is his current age raised to the 0.23 power?";
+    const input = `Who is Olivia Wilde's boyfriend? What is his current age raised to the 0.23 power?`;
     console.log(`Executing with input "${input}"...`);
     const result = await executor.call({ input });
     console.log(`Got output ${result.output}`);
 }, 30000);
-//# sourceMappingURL=agent.test.js.map
+//# sourceMappingURL=agent.int.test.js.map

package/dist/agents/tests/agent.int.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent.int.test.js","sourceRoot":"","sources":["../../../agents/tests/agent.int.test.ts"],"names":[],"mappings":";;AAAA,2CAAqC;AACrC,8CAA2C;AAC3C,kCAAoC;AACpC,oCAA+C;AAC/C,8CAA2C;AAC3C,oDAAiD;AACjD,8CAAwD;AAExD,IAAA,cAAI,EAAC,oBAAoB,EAAE,KAAK,IAAI,EAAE;IACpC,MAAM,KAAK,GAAG,IAAI,eAAM,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE,CAAC,CAAC;IAC7C,MAAM,KAAK,GAAW,CAAC,IAAI,iBAAO,EAAE,EAAE,IAAI,uBAAU,EAAE,CAAC,CAAC;IACxD,MAAM,KAAK,GAAG,MAAM,IAAA,gBAAS,EAC3B,oDAAoD,EACpD,EAAE,GAAG,EAAE,KAAK,EAAE,KAAK,EAAE,CACtB,CAAC;IACF,MAAM,QAAQ,GAAG,qBAAa,CAAC,iBAAiB,CAAC;QAC/C,KAAK;QACL,KAAK;QACL,uBAAuB,EAAE,IAAI;KAC9B,CAAC,CAAC;IACH,MAAM,GAAG,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC;QAC9B,KAAK,EACH,oFAAoF;KACvF,CAAC,CAAC;IACH,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;AACnB,CAAC,EAAE,KAAK,CAAC,CAAC;AAEV,IAAA,cAAI,EAAC,mBAAmB,EAAE,KAAK,IAAI,EAAE;IACnC,MAAM,KAAK,GAAG,IAAI,eAAM,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE,CAAC,CAAC;IAC7C,MAAM,KAAK,GAAG,CAAC,IAAI,iBAAO,EAAE,EAAE,IAAI,uBAAU,EAAE,CAAC,CAAC;IAEhD,MAAM,QAAQ,GAAG,MAAM,IAAA,oCAAuB,EAC5C,KAAK,EACL,KAAK,EACL,6BAA6B,CAC9B,CAAC;IACF,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;IAE7B,MAAM,KAAK,GAAG,oFAAoF,CAAC;IACnG,OAAO,CAAC,GAAG,CAAC,yBAAyB,KAAK,MAAM,CAAC,CAAC;IAElD,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;IAE9C,OAAO,CAAC,GAAG,CAAC,cAAc,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;AAC7C,CAAC,EAAE,KAAK,CAAC,CAAC"}

package/dist/agents/tools/serpapi.js

@@ -44,7 +44,7 @@ class SerpAPI extends base_1.Tool {
         });
         // Throw error at construction time.
         if (getJson === null) {
-            throw new Error("Please install serpapi as a dependency with, e.g. `npm i serpapi`");
+            throw new Error("Please install serpapi as a dependency with, e.g. `yarn add serpapi`");
         }
         if (!apiKey) {
             throw new Error("SerpAPI API key not set. You can set it as SERPAPI_API_KEY in your .env file, or pass it to SerpAPI.");

package/dist/agents/tools/serpapi.js.map

@@ -1 +1 @@
-{"version":3,"file":"serpapi.js","sourceRoot":"","sources":["../../../agents/tools/serpapi.ts"],"names":[],"mappings":";;;AAEA,iCAA8B;AAE9B,IAAI,OAAO,GAA2B,IAAI,CAAC;AAE3C,IAAI;IACF,4EAA4E;IAC5E,CAAC,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC;CACpC;AAAC,WAAM;IACN,eAAe;CAChB;AAED;;;;GAIG;AACH,MAAa,OAAQ,SAAQ,WAAI;IAK/B,YACE,SAA6B,OAAO,CAAC,GAAG,CAAC,eAAe,EACxD,SAAoC,EAAE;QAEtC,KAAK,EAAE,CAAC;QARV;;;;;WAAsB;QAEtB;;;;;WAA4C;QAyB5C;;;;mBAAO,QAAQ;WAAC;QAgDhB;;;;mBACE,qHAAqH;WAAC;QAlEtH,oCAAoC;QACpC,IAAI,OAAO,KAAK,IAAI,EAAE;YACpB,MAAM,IAAI,KAAK,CACb,mEAAmE,CACpE,CAAC;SACH;QAED,IAAI,CAAC,MAAM,EAAE;YACX,MAAM,IAAI,KAAK,CACb,sGAAsG,CACvG,CAAC;SACH;QAED,IAAI,CAAC,GAAG,GAAG,MAAM,CAAC;QAClB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAID;;OAEG;IACH,KAAK,CAAC,IAAI,CAAC,KAAa;;QACtB,IAAI,OAAO,KAAK,IAAI,EAAE;YACpB,MAAM,IAAI,KAAK,CACb,mEAAmE,CACpE,CAAC;SACH;QACD,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,QAAQ,EAAE;YAClC,GAAG,IAAI,CAAC,MAAM;YACd,OAAO,EAAE,IAAI,CAAC,GAAG;YACjB,CAAC,EAAE,KAAK;SACT,CAAC,CAAC;QAEH,IAAI,GAAG,CAAC,KAAK,EAAE;YACb,MAAM,IAAI,KAAK,CAAC,2BAA2B,GAAG,CAAC,KAAK,EAAE,CAAC,CAAC;SACzD;QAED,IAAI,MAAA,GAAG,CAAC,UAAU,0CAAE,MAAM,EAAE;YAC1B,OAAO,GAAG,CAAC,UAAU,CAAC,MAAM,CAAC;SAC9B;QAED,IAAI,MAAA,GAAG,CAAC,UAAU,0CAAE,OAAO,EAAE;YAC3B,OAAO,GAAG,CAAC,UAAU,CAAC,OAAO,CAAC;SAC/B;QAED,IAAI,MAAA,GAAG,CAAC,UAAU,0CAAE,yBAAyB,EAAE;YAC7C,OAAO,GAAG,CAAC,UAAU,CAAC,yBAAyB,CAAC,CAAC,CAAC,CAAC;SACpD;QAED,IAAI,MAAA,GAAG,CAAC,cAAc,0CAAE,cAAc,EAAE;YACtC,OAAO,GAAG,CAAC,cAAc,CAAC,cAAc,CAAC;SAC1C;QAED,IAAI,MAAA,GAAG,CAAC,eAAe,0CAAE,WAAW,EAAE;YACpC,OAAO,GAAG,CAAC,eAAe,CAAC,WAAW,CAAC;SACxC;QAED,IAAI,MAAA,MAAA,GAAG,CAAC,eAAe,0CAAG,CAAC,CAAC,0CAAE,OAAO,EAAE;YACrC,OAAO,GAAG,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;SACvC;QAED,OAAO,6BAA6B,CAAC;IACvC,CAAC;CAIF;AA9ED,0BA8EC"}
+{"version":3,"file":"serpapi.js","sourceRoot":"","sources":["../../../agents/tools/serpapi.ts"],"names":[],"mappings":";;;AAEA,iCAA8B;AAE9B,IAAI,OAAO,GAA2B,IAAI,CAAC;AAE3C,IAAI;IACF,4EAA4E;IAC5E,CAAC,EAAE,OAAO,EAAE,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC;CACpC;AAAC,WAAM;IACN,eAAe;CAChB;AAED;;;;GAIG;AACH,MAAa,OAAQ,SAAQ,WAAI;IAK/B,YACE,SAA6B,OAAO,CAAC,GAAG,CAAC,eAAe,EACxD,SAAoC,EAAE;QAEtC,KAAK,EAAE,CAAC;QARV;;;;;WAAsB;QAEtB;;;;;WAA4C;QAyB5C;;;;mBAAO,QAAQ;WAAC;QAgDhB;;;;mBACE,qHAAqH;WAAC;QAlEtH,oCAAoC;QACpC,IAAI,OAAO,KAAK,IAAI,EAAE;YACpB,MAAM,IAAI,KAAK,CACb,sEAAsE,CACvE,CAAC;SACH;QAED,IAAI,CAAC,MAAM,EAAE;YACX,MAAM,IAAI,KAAK,CACb,sGAAsG,CACvG,CAAC;SACH;QAED,IAAI,CAAC,GAAG,GAAG,MAAM,CAAC;QAClB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAID;;OAEG;IACH,KAAK,CAAC,IAAI,CAAC,KAAa;;QACtB,IAAI,OAAO,KAAK,IAAI,EAAE;YACpB,MAAM,IAAI,KAAK,CACb,mEAAmE,CACpE,CAAC;SACH;QACD,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,QAAQ,EAAE;YAClC,GAAG,IAAI,CAAC,MAAM;YACd,OAAO,EAAE,IAAI,CAAC,GAAG;YACjB,CAAC,EAAE,KAAK;SACT,CAAC,CAAC;QAEH,IAAI,GAAG,CAAC,KAAK,EAAE;YACb,MAAM,IAAI,KAAK,CAAC,2BAA2B,GAAG,CAAC,KAAK,EAAE,CAAC,CAAC;SACzD;QAED,IAAI,MAAA,GAAG,CAAC,UAAU,0CAAE,MAAM,EAAE;YAC1B,OAAO,GAAG,CAAC,UAAU,CAAC,MAAM,CAAC;SAC9B;QAED,IAAI,MAAA,GAAG,CAAC,UAAU,0CAAE,OAAO,EAAE;YAC3B,OAAO,GAAG,CAAC,UAAU,CAAC,OAAO,CAAC;SAC/B;QAED,IAAI,MAAA,GAAG,CAAC,UAAU,0CAAE,yBAAyB,EAAE;YAC7C,OAAO,GAAG,CAAC,UAAU,CAAC,yBAAyB,CAAC,CAAC,CAAC,CAAC;SACpD;QAED,IAAI,MAAA,GAAG,CAAC,cAAc,0CAAE,cAAc,EAAE;YACtC,OAAO,GAAG,CAAC,cAAc,CAAC,cAAc,CAAC;SAC1C;QAED,IAAI,MAAA,GAAG,CAAC,eAAe,0CAAE,WAAW,EAAE;YACpC,OAAO,GAAG,CAAC,eAAe,CAAC,WAAW,CAAC;SACxC;QAED,IAAI,MAAA,MAAA,GAAG,CAAC,eAAe,0CAAG,CAAC,CAAC,0CAAE,OAAO,EAAE;YACrC,OAAO,GAAG,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;SACvC;QAED,OAAO,6BAA6B,CAAC;IACvC,CAAC;CAIF;AA9ED,0BA8EC"}

package/dist/cache.d.ts

@@ -1,11 +1,12 @@
 import { Generation } from "./llms";
+export declare const getKey: (...strings: string[]) => string;
 export declare abstract class BaseCache<T = Generation[]> {
-    abstract lookup(prompt: string, llmKey: string): T | undefined;
-    abstract update(prompt: string, llmKey: string, value: T): void;
+    abstract lookup(key: string): T | undefined;
+    abstract update(key: string, value: T): void;
 }
 export declare class InMemoryCache<T = Generation[]> extends BaseCache<T> {
     private cache;
     constructor();
-    lookup(prompt: string, llmKey: string): T;
-    update(prompt: string, llmKey: string, value: T): void;
+    lookup(key: string): T;
+    update(key: string, value: T): void;
 }

package/dist/cache.js

@@ -1,7 +1,18 @@
 "use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.InMemoryCache = exports.BaseCache = void 0;
-const getKey = (prompt, llmKey) => `${prompt}_${llmKey}`;
+exports.InMemoryCache = exports.BaseCache = exports.getKey = void 0;
+const crypto_1 = __importDefault(require("crypto"));
+// Takes in an arbitrary number of strings and returns a hash of them
+// that can be used as a key in a cache.
+const getKey = (...strings) => {
+    const hash = crypto_1.default.createHash("sha256");
+    strings.forEach((s) => hash.update(s));
+    return hash.digest("hex");
+};
+exports.getKey = getKey;
 class BaseCache {
 }
 exports.BaseCache = BaseCache;
@@ -16,11 +27,11 @@ class InMemoryCache extends BaseCache {
         });
         this.cache = {};
     }
-    lookup(prompt, llmKey) {
-        return this.cache[getKey(prompt, llmKey)];
+    lookup(key) {
+        return this.cache[key];
     }
-    update(prompt, llmKey, value) {
-        this.cache[getKey(prompt, llmKey)] = value;
+    update(key, value) {
+        this.cache[key] = value;
     }
 }
 exports.InMemoryCache = InMemoryCache;

package/dist/cache.js.map

@@ -1 +1 @@
-{"version":3,"file":"cache.js","sourceRoot":"","sources":["../cache.ts"],"names":[],"mappings":";;;AAEA,MAAM,MAAM,GAAG,CAAC,MAAc,EAAE,MAAc,EAAU,EAAE,CACxD,GAAG,MAAM,IAAI,MAAM,EAAE,CAAC;AAExB,MAAsB,SAAS;CAI9B;AAJD,8BAIC;AAED,MAAa,aAAgC,SAAQ,SAAY;IAG/D;QACE,KAAK,EAAE,CAAC;QAHV;;;;;WAAiC;QAI/B,IAAI,CAAC,KAAK,GAAG,EAAE,CAAC;IAClB,CAAC;IAED,MAAM,CAAC,MAAc,EAAE,MAAc;QACnC,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC;IAC5C,CAAC;IAED,MAAM,CAAC,MAAc,EAAE,MAAc,EAAE,KAAQ;QAC7C,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,GAAG,KAAK,CAAC;IAC7C,CAAC;CACF;AAfD,sCAeC"}
+{"version":3,"file":"cache.js","sourceRoot":"","sources":["../cache.ts"],"names":[],"mappings":";;;;;;AAAA,oDAA4B;AAG5B,qEAAqE;AACrE,wCAAwC;AACjC,MAAM,MAAM,GAAG,CAAC,GAAG,OAAiB,EAAU,EAAE;IACrD,MAAM,IAAI,GAAG,gBAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;IACzC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;IACvC,OAAO,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AAC5B,CAAC,CAAC;AAJW,QAAA,MAAM,UAIjB;AAEF,MAAsB,SAAS;CAI9B;AAJD,8BAIC;AAED,MAAa,aAAgC,SAAQ,SAAY;IAG/D;QACE,KAAK,EAAE,CAAC;QAHV;;;;;WAAiC;QAI/B,IAAI,CAAC,KAAK,GAAG,EAAE,CAAC;IAClB,CAAC;IAED,MAAM,CAAC,GAAW;QAChB,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;IACzB,CAAC;IAED,MAAM,CAAC,GAAW,EAAE,KAAQ;QAC1B,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;IAC1B,CAAC;CACF;AAfD,sCAeC"}

package/dist/chains/base.d.ts

@@ -12,6 +12,7 @@ export interface ChainInputs {
  */
 export declare abstract class BaseChain implements ChainInputs {
     memory?: BaseMemory;
+    constructor(memory?: BaseMemory);
     /**
      * Run the core logic of this chain and return the output
      */

package/dist/chains/base.js

@@ -13,13 +13,14 @@ const chainClasses = [
  * Base interface that all chains must implement.
  */
 class BaseChain {
-    constructor() {
+    constructor(memory) {
         Object.defineProperty(this, "memory", {
             enumerable: true,
             configurable: true,
             writable: true,
             value: void 0
         });
+        this.memory = memory;
     }
     /**
      * Run the core logic of this chain and add to output if desired.
@@ -37,7 +38,7 @@ class BaseChain {
         // TODO(sean) add callback support
         const outputValues = this._call(fullValues);
         if (!(this.memory == null)) {
-            this.memory.saveContext(values, outputValues);
+            await this.memory.saveContext(values, outputValues);
         }
         return outputValues;
     }

package/dist/chains/base.js.map

@@ -1 +1 @@
-{"version":3,"file":"base.js","sourceRoot":"","sources":["../../chains/base.ts"],"names":[],"mappings":";;;AAAA,qCAAsC;AACtC,mCAKiB;AAQjB,MAAM,YAAY,GAAG;IACnB,gBAAQ;IACR,2BAAmB;IACnB,uBAAe;IACf,2BAAmB;CACpB,CAAC;AAUF;;GAEG;AACH,MAAsB,SAAS;IAA/B;QACE;;;;;WAAoB;IAmEtB,CAAC;IAlDC;;;;OAIG;IACH,KAAK,CAAC,IAAI,CAAC,MAAmB;QAC5B,MAAM,UAAU,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC;QACpC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,EAAE;YAC1B,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC;YAChE,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE;gBACpD,UAAU,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;aACzB;SACF;QACD,kCAAkC;QAClC,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;QAC5C,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,EAAE;YAC1B,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;SAC/C;QACD,OAAO,YAAY,CAAC;IACtB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,MAAqB;QACzB,OAAO,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC/B,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,WAAW,CAChB,IAAyB,EACzB,SAAqB,EAAE;QAEvB,QAAQ,IAAI,CAAC,KAAK,EAAE;YAClB,KAAK,WAAW;gBACd,OAAO,gBAAQ,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;YACpC,KAAK,uBAAuB;gBAC1B,OAAO,2BAAmB,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;YAC/C,KAAK,cAAc;gBACjB,OAAO,uBAAe,CAAC,WAAW,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;YACnD;gBACE,MAAM,IAAI,KAAK,CACb,kCACG,IAA4B,CAAC,KAChC,EAAE,CACH,CAAC;SACL;IACH,CAAC;CACF;AApED,8BAoEC"}
+{"version":3,"file":"base.js","sourceRoot":"","sources":["../../chains/base.ts"],"names":[],"mappings":";;;AAAA,qCAAsC;AACtC,mCAKiB;AAQjB,MAAM,YAAY,GAAG;IACnB,gBAAQ;IACR,2BAAmB;IACnB,uBAAe;IACf,2BAAmB;CACpB,CAAC;AAUF;;GAEG;AACH,MAAsB,SAAS;IAG7B,YAAY,MAAmB;QAF/B;;;;;WAAoB;QAGlB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAiBD;;;;OAIG;IACH,KAAK,CAAC,IAAI,CAAC,MAAmB;QAC5B,MAAM,UAAU,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC;QACpC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,EAAE;YAC1B,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC;YAChE,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE;gBACpD,UAAU,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;aACzB;SACF;QACD,kCAAkC;QAClC,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;QAC5C,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,EAAE;YAC1B,MAAM,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;SACrD;QACD,OAAO,YAAY,CAAC;IACtB,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,MAAqB;QACzB,OAAO,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC/B,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,WAAW,CAChB,IAAyB,EACzB,SAAqB,EAAE;QAEvB,QAAQ,IAAI,CAAC,KAAK,EAAE;YAClB,KAAK,WAAW;gBACd,OAAO,gBAAQ,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;YACpC,KAAK,uBAAuB;gBAC1B,OAAO,2BAAmB,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;YAC/C,KAAK,cAAc;gBACjB,OAAO,uBAAe,CAAC,WAAW,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;YACnD;gBACE,MAAM,IAAI,KAAK,CACb,kCACG,IAA4B,CAAC,KAChC,EAAE,CACH,CAAC;SACL;IACH,CAAC;CACF;AAxED,8BAwEC"}

package/dist/chains/index.d.ts

@@ -1,4 +1,4 @@
-export { BaseChain, ChainValues } from "./base";
+export { BaseChain, ChainValues, ChainInputs } from "./base";
 export { SerializedLLMChain, LLMChain, ConversationChain } from "./llm_chain";
 export { SerializedStuffDocumentsChain, StuffDocumentsChain, } from "./combine_docs_chain";
 export { ChatVectorDBQAChain, SerializedChatVectorDBQAChain, } from "./chat_vector_db_chain";

package/dist/chains/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../chains/index.ts"],"names":[],"mappings":";;;AAAA,+BAAgD;AAAvC,iGAAA,SAAS,OAAA;AAClB,yCAA8E;AAAjD,qGAAA,QAAQ,OAAA;AAAE,8GAAA,iBAAiB,OAAA;AACxD,2DAG8B;AAD5B,yHAAA,mBAAmB,OAAA;AAErB,+DAGgC;AAF9B,2HAAA,mBAAmB,OAAA;AAGrB,+CAA4E;AAAnE,+GAAA,eAAe,OAAA;AACxB,+BAAmC;AAA1B,iGAAA,SAAS,OAAA;AAClB,kDAAwD;AAA/C,mGAAA,WAAW,OAAA"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../chains/index.ts"],"names":[],"mappings":";;;AAAA,+BAA6D;AAApD,iGAAA,SAAS,OAAA;AAClB,yCAA8E;AAAjD,qGAAA,QAAQ,OAAA;AAAE,8GAAA,iBAAiB,OAAA;AACxD,2DAG8B;AAD5B,yHAAA,mBAAmB,OAAA;AAErB,+DAGgC;AAF9B,2HAAA,mBAAmB,OAAA;AAGrB,+CAA4E;AAAnE,+GAAA,eAAe,OAAA;AACxB,+BAAmC;AAA1B,iGAAA,SAAS,OAAA;AAClB,kDAAwD;AAA/C,mGAAA,WAAW,OAAA"}

package/dist/chains/llm_chain.d.ts

@@ -1,8 +1,8 @@
-import { BaseChain, ChainValues } from "./index";
+import { BaseChain, ChainValues, ChainInputs } from "./index";
 import { BaseLLM, SerializedLLM } from "../llms";
 import { BaseMemory } from "../memory";
 import { BasePromptTemplate, SerializedBasePromptTemplate } from "../prompts";
-export interface LLMChainInput {
+export interface LLMChainInput extends ChainInputs {
     /** Prompt object to use */
     prompt: BasePromptTemplate;
     /** LLM Wrapper to use */
@@ -37,6 +37,7 @@ export declare class LLMChain extends BaseChain implements LLMChainInput {
         prompt: BasePromptTemplate;
         llm: BaseLLM;
         outputKey?: string;
+        memory?: BaseMemory;
     });
     _call(values: ChainValues): Promise<ChainValues>;
     /**

package/dist/chains/llm_chain.js

@@ -21,7 +21,7 @@ const util_1 = require("../util");
 class LLMChain extends index_1.BaseChain {
     constructor(fields) {
         var _a;
-        super();
+        super(fields.memory);
         Object.defineProperty(this, "prompt", {
             enumerable: true,
             configurable: true,

package/dist/chains/llm_chain.js.map

@@ -1 +1 @@
-{"version":3,"file":"llm_chain.js","sourceRoot":"","sources":["../../chains/llm_chain.ts"],"names":[],"mappings":";;;AAAA,mCAAiD;AAEjD,kCAAiD;AAEjD,sCAAqD;AACrD,wCAIoB;AAEpB,kCAAgD;AAoBhD;;;;;;;;;;;GAWG;AACH,MAAa,QAAS,SAAQ,iBAAS;IAOrC,YAAY,MAIX;;QACC,KAAK,EAAE,CAAC;QAXV;;;;;WAA2B;QAE3B;;;;;WAAa;QAEb;;;;mBAAY,MAAM;WAAC;QAQjB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QAC5B,IAAI,CAAC,GAAG,GAAG,MAAM,CAAC,GAAG,CAAC;QACtB,IAAI,CAAC,SAAS,GAAG,MAAA,MAAM,CAAC,SAAS,mCAAI,IAAI,CAAC,SAAS,CAAC;IACtD,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,MAAmB;QAC7B,IAAI,IAAI,CAAC;QACT,IAAI,MAAM,IAAI,MAAM,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE;YAClD,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;SACpB;QACD,MAAM,eAAe,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QACnD,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,CAAC;QAC7D,MAAM,MAAM,GAAG,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,SAAS,EAAE,CAAC;QAC/C,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;;;;;;;;OAUG;IACH,KAAK,CAAC,OAAO,CAAC,MAAmB;QAC/B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACvC,OAAO,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAChC,CAAC;IAED,UAAU;QACR,OAAO,WAAoB,CAAC;IAC9B,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,IAAwB;QAC/C,MAAM,aAAa,GAAG,IAAA,4BAAqB,EACzC,KAAK,EACL,IAAI,CACL,CAAC;QACF,MAAM,gBAAgB,GAAG,IAAA,4BAAqB,EAG5C,QAAQ,EAAE,IAAI,CAAC,CAAC;QAElB,OAAO,IAAI,QAAQ,CAAC;YAClB,GAAG,EAAE,MAAM,cAAO,CAAC,WAAW,CAAC,aAAa,CAAC;YAC7C,MAAM,EAAE,MAAM,4BAAkB,CAAC,WAAW,CAAC,gBAAgB,CAAC;SAC/D,CAAC,CAAC;IACL,CAAC;IAED,SAAS;QACP,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,UAAU,EAAE;YACxB,GAAG,EAAE,IAAI,CAAC,GAAG,CAAC,SAAS,EAAE;YACzB,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE;SAChC,CAAC;IACJ,CAAC;CACF;AAxED,4BAwEC;AAED,mCAAmC;AACnC,MAAM,eAAe,GAAG;;;;;IAKpB,CAAC;AAEL,MAAM,aAAa,GAAG,IAAI,wBAAc,CAAC;IACvC,QAAQ,EAAE,eAAe;IACzB,cAAc,EAAE,CAAC,SAAS,EAAE,OAAO,CAAC;CACrC,CAAC,CAAC;AAEH,MAAa,iBAAkB,SAAQ,QAAQ;IAC7C,YAAY,MAKX;;QACC,KAAK,CAAC;YACJ,MAAM,EAAE,MAAA,MAAM,CAAC,MAAM,mCAAI,aAAa;YACtC,GAAG,EAAE,MAAM,CAAC,GAAG;YACf,SAAS,EAAE,MAAA,MAAM,CAAC,SAAS,mCAAI,UAAU;SAC1C,CAAC,CAAC;QACH,IAAI,CAAC,MAAM,GAAG,MAAA,MAAM,CAAC,MAAM,mCAAI,IAAI,qBAAY,EAAE,CAAC;IACpD,CAAC;CACF;AAdD,8CAcC"}
+{"version":3,"file":"llm_chain.js","sourceRoot":"","sources":["../../chains/llm_chain.ts"],"names":[],"mappings":";;;AAAA,mCAA8D;AAE9D,kCAAiD;AAEjD,sCAAqD;AACrD,wCAIoB;AAEpB,kCAAgD;AAoBhD;;;;;;;;;;;GAWG;AACH,MAAa,QAAS,SAAQ,iBAAS;IAOrC,YAAY,MAKX;;QACC,KAAK,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAZvB;;;;;WAA2B;QAE3B;;;;;WAAa;QAEb;;;;mBAAY,MAAM;WAAC;QASjB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QAC5B,IAAI,CAAC,GAAG,GAAG,MAAM,CAAC,GAAG,CAAC;QACtB,IAAI,CAAC,SAAS,GAAG,MAAA,MAAM,CAAC,SAAS,mCAAI,IAAI,CAAC,SAAS,CAAC;IACtD,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,MAAmB;QAC7B,IAAI,IAAI,CAAC;QACT,IAAI,MAAM,IAAI,MAAM,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE;YAClD,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;SACpB;QACD,MAAM,eAAe,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QACnD,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,eAAe,EAAE,IAAI,CAAC,CAAC;QAC7D,MAAM,MAAM,GAAG,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,SAAS,EAAE,CAAC;QAC/C,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;;;;;;;;OAUG;IACH,KAAK,CAAC,OAAO,CAAC,MAAmB;QAC/B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACvC,OAAO,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAChC,CAAC;IAED,UAAU;QACR,OAAO,WAAoB,CAAC;IAC9B,CAAC;IAED,MAAM,CAAC,KAAK,CAAC,WAAW,CAAC,IAAwB;QAC/C,MAAM,aAAa,GAAG,IAAA,4BAAqB,EACzC,KAAK,EACL,IAAI,CACL,CAAC;QACF,MAAM,gBAAgB,GAAG,IAAA,4BAAqB,EAG5C,QAAQ,EAAE,IAAI,CAAC,CAAC;QAElB,OAAO,IAAI,QAAQ,CAAC;YAClB,GAAG,EAAE,MAAM,cAAO,CAAC,WAAW,CAAC,aAAa,CAAC;YAC7C,MAAM,EAAE,MAAM,4BAAkB,CAAC,WAAW,CAAC,gBAAgB,CAAC;SAC/D,CAAC,CAAC;IACL,CAAC;IAED,SAAS;QACP,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,UAAU,EAAE;YACxB,GAAG,EAAE,IAAI,CAAC,GAAG,CAAC,SAAS,EAAE;YACzB,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE;SAChC,CAAC;IACJ,CAAC;CACF;AAzED,4BAyEC;AAED,mCAAmC;AACnC,MAAM,eAAe,GAAG;;;;;IAKpB,CAAC;AAEL,MAAM,aAAa,GAAG,IAAI,wBAAc,CAAC;IACvC,QAAQ,EAAE,eAAe;IACzB,cAAc,EAAE,CAAC,SAAS,EAAE,OAAO,CAAC;CACrC,CAAC,CAAC;AAEH,MAAa,iBAAkB,SAAQ,QAAQ;IAC7C,YAAY,MAKX;;QACC,KAAK,CAAC;YACJ,MAAM,EAAE,MAAA,MAAM,CAAC,MAAM,mCAAI,aAAa;YACtC,GAAG,EAAE,MAAM,CAAC,GAAG;YACf,SAAS,EAAE,MAAA,MAAM,CAAC,SAAS,mCAAI,UAAU;SAC1C,CAAC,CAAC;QACH,IAAI,CAAC,MAAM,GAAG,MAAA,MAAM,CAAC,MAAM,mCAAI,IAAI,qBAAY,EAAE,CAAC;IACpD,CAAC;CACF;AAdD,8CAcC"}

package/dist/chains/question_answering/tests/{load.test.js → load.int.test.js}

@@ -15,4 +15,4 @@ const document_1 = require("../../../document");
     const res = await chain.call({ input_documents: docs, question: "Whats up" });
     console.log({ res });
 });
-//# sourceMappingURL=load.test.js.map
+//# sourceMappingURL=load.int.test.js.map

package/dist/chains/question_answering/tests/load.int.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"load.int.test.js","sourceRoot":"","sources":["../../../../chains/question_answering/tests/load.int.test.ts"],"names":[],"mappings":";;AAAA,2CAAqC;AACrC,iDAA8C;AAC9C,kCAAsC;AACtC,gDAA6C;AAE7C,IAAA,cAAI,EAAC,kBAAkB,EAAE,KAAK,IAAI,EAAE;IAClC,MAAM,KAAK,GAAG,IAAI,eAAM,CAAC,EAAE,CAAC,CAAC;IAC7B,MAAM,KAAK,GAAG,IAAA,kBAAW,EAAC,KAAK,CAAC,CAAC;IACjC,MAAM,IAAI,GAAG;QACX,IAAI,mBAAQ,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC;QACpC,IAAI,mBAAQ,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC;QACpC,IAAI,mBAAQ,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC;KACrC,CAAC;IACF,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,EAAE,eAAe,EAAE,IAAI,EAAE,QAAQ,EAAE,UAAU,EAAE,CAAC,CAAC;IAC9E,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;AACvB,CAAC,CAAC,CAAC"}

package/dist/chains/tests/{chat_vector_db_qa_chain.test.js → chat_vector_db_qa_chain.int.test.js}

@@ -32,4 +32,4 @@ const embeddings_1 = require("../../embeddings");
     const res = await chain.call({ question: "foo", chat_history: "bar" });
     console.log({ res });
 });
-//# sourceMappingURL=chat_vector_db_qa_chain.test.js.map
+//# sourceMappingURL=chat_vector_db_qa_chain.int.test.js.map

package/dist/chains/tests/chat_vector_db_qa_chain.int.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chat_vector_db_qa_chain.int.test.js","sourceRoot":"","sources":["../../../chains/tests/chat_vector_db_qa_chain.int.test.ts"],"names":[],"mappings":";;AAAA,2CAAqC;AACrC,8CAA2C;AAC3C,2CAA+C;AAC/C,4CAAwC;AACxC,8DAA4D;AAC5D,kEAA8D;AAC9D,wDAAqD;AACrD,iDAAoD;AAEpD,IAAA,cAAI,EAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;IAC1C,MAAM,KAAK,GAAG,IAAI,eAAM,CAAC,EAAE,CAAC,CAAC;IAC7B,MAAM,MAAM,GAAG,wBAAc,CAAC,YAAY,CACxC,6CAA6C,CAC9C,CAAC;IACF,MAAM,WAAW,GAAG,MAAM,iBAAO,CAAC,SAAS,CACzC,CAAC,aAAa,EAAE,SAAS,EAAE,kBAAkB,EAAE,KAAK,EAAE,IAAI,CAAC,EAC3D,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,EACvD,IAAI,6BAAgB,EAAE,CACvB,CAAC;IACF,MAAM,QAAQ,GAAG,IAAI,oBAAQ,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC;IACtD,MAAM,gBAAgB,GAAG,IAAI,wCAAmB,CAAC;QAC/C,QAAQ;QACR,oBAAoB,EAAE,KAAK;KAC5B,CAAC,CAAC;IACH,MAAM,KAAK,GAAG,IAAI,0CAAmB,CAAC;QACpC,qBAAqB,EAAE,gBAAgB;QACvC,WAAW,EAAE,WAAW;QACxB,sBAAsB,EAAE,QAAQ;KACjC,CAAC,CAAC;IACH,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,YAAY,EAAE,KAAK,EAAE,CAAC,CAAC;IACvE,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;AACvB,CAAC,CAAC,CAAC;AAEH,IAAA,cAAI,EAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;IACnD,MAAM,KAAK,GAAG,IAAI,eAAM,CAAC,EAAE,CAAC,CAAC;IAC7B,MAAM,WAAW,GAAG,MAAM,iBAAO,CAAC,SAAS,CACzC,CAAC,aAAa,EAAE,SAAS,EAAE,kBAAkB,EAAE,KAAK,EAAE,IAAI,CAAC,EAC3D,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,EACvD,IAAI,6BAAgB,EAAE,CACvB,CAAC;IACF,MAAM,KAAK,GAAG,0CAAmB,CAAC,OAAO,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;IAC9D,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,KAAK,EAAE,YAAY,EAAE,KAAK,EAAE,CAAC,CAAC;IACvE,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;AACvB,CAAC,CAAC,CAAC"}

package/dist/chains/tests/{combine_docs_chain.test.js → combine_docs_chain.int.test.js}

@@ -36,4 +36,4 @@ const document_1 = require("../../document");
     const res = await chain.call({ input_documents: docs, question: "what up" });
     console.log({ res });
 });
-//# sourceMappingURL=combine_docs_chain.test.js.map
+//# sourceMappingURL=combine_docs_chain.int.test.js.map

package/dist/chains/tests/combine_docs_chain.int.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"combine_docs_chain.int.test.js","sourceRoot":"","sources":["../../../chains/tests/combine_docs_chain.int.test.ts"],"names":[],"mappings":";;AAAA,2CAAqC;AACrC,8CAA2C;AAC3C,2CAA+C;AAC/C,4CAAwC;AACxC,kCAAoC;AACpC,8DAA4D;AAC5D,6CAA0C;AAE1C,IAAA,cAAI,EAAC,0BAA0B,EAAE,KAAK,IAAI,EAAE;IAC1C,MAAM,KAAK,GAAG,IAAI,eAAM,CAAC,EAAE,CAAC,CAAC;IAC7B,MAAM,MAAM,GAAG,IAAI,wBAAc,CAAC;QAChC,QAAQ,EAAE,aAAa;QACvB,cAAc,EAAE,CAAC,KAAK,CAAC;KACxB,CAAC,CAAC;IACH,MAAM,QAAQ,GAAG,IAAI,oBAAQ,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC;IACtD,MAAM,KAAK,GAAG,IAAI,wCAAmB,CAAC;QACpC,QAAQ;QACR,oBAAoB,EAAE,KAAK;KAC5B,CAAC,CAAC;IACH,MAAM,IAAI,GAAG;QACX,IAAI,mBAAQ,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC;QACpC,IAAI,mBAAQ,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC;QACpC,IAAI,mBAAQ,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC;KACrC,CAAC;IACF,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,EAAE,eAAe,EAAE,IAAI,EAAE,CAAC,CAAC;IACxD,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;AACvB,CAAC,CAAC,CAAC;AAEH,IAAA,cAAI,EAAC,qBAAqB,EAAE,KAAK,IAAI,EAAE;IACrC,MAAM,KAAK,GAAG,MAAM,IAAA,gBAAS,EAC3B,iDAAiD,CAClD,CAAC;IACF,MAAM,IAAI,GAAG;QACX,IAAI,mBAAQ,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC;QACpC,IAAI,mBAAQ,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC;QACpC,IAAI,mBAAQ,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC;KACrC,CAAC;IACF,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,EAAE,eAAe,EAAE,IAAI,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC,CAAC;IAC7E,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;AACvB,CAAC,CAAC,CAAC"}

package/dist/chains/tests/{llm_chain.test.js → llm_chain.int.test.js}

@@ -26,4 +26,4 @@ const load_1 = require("../load");
     const res = await chain.call({ input: "my favorite color" });
     console.log({ res });
 });
-//# sourceMappingURL=llm_chain.test.js.map
+//# sourceMappingURL=llm_chain.int.test.js.map

package/dist/chains/tests/llm_chain.int.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"llm_chain.int.test.js","sourceRoot":"","sources":["../../../chains/tests/llm_chain.int.test.ts"],"names":[],"mappings":";;AAAA,2CAAqC;AACrC,8CAA2C;AAC3C,2CAA+C;AAC/C,4CAA2D;AAC3D,kCAAoC;AAEpC,IAAA,cAAI,EAAC,aAAa,EAAE,KAAK,IAAI,EAAE;IAC7B,MAAM,KAAK,GAAG,IAAI,eAAM,CAAC,EAAE,CAAC,CAAC;IAC7B,MAAM,MAAM,GAAG,IAAI,wBAAc,CAAC;QAChC,QAAQ,EAAE,aAAa;QACvB,cAAc,EAAE,CAAC,KAAK,CAAC;KACxB,CAAC,CAAC;IACH,MAAM,KAAK,GAAG,IAAI,oBAAQ,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC;IACnD,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,mBAAmB,EAAE,CAAC,CAAC;IAC3D,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;AACvB,CAAC,CAAC,CAAC;AAEH,IAAA,cAAI,EAAC,qBAAqB,EAAE,KAAK,IAAI,EAAE;IACrC,MAAM,KAAK,GAAG,MAAM,IAAA,gBAAS,EAAC,oCAAoC,CAAC,CAAC;IACpE,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,mBAAmB,EAAE,CAAC,CAAC;IAC7D,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;AACvB,CAAC,CAAC,CAAC;AAEH,IAAA,cAAI,EAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;IACxC,MAAM,KAAK,GAAG,IAAI,eAAM,CAAC,EAAE,CAAC,CAAC;IAC7B,MAAM,KAAK,GAAG,IAAI,6BAAiB,CAAC,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC;IACpD,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,mBAAmB,EAAE,CAAC,CAAC;IAC7D,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;AACvB,CAAC,CAAC,CAAC"}

package/dist/chains/tests/{vector_db_qa_chain.test.js → vector_db_qa_chain.int.test.js}

@@ -43,4 +43,4 @@ const embeddings_1 = require("../../embeddings");
     const res = await chain.call({ query: "what up" });
     console.log({ res });
 });
-//# sourceMappingURL=vector_db_qa_chain.test.js.map
+//# sourceMappingURL=vector_db_qa_chain.int.test.js.map

package/dist/chains/tests/vector_db_qa_chain.int.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"vector_db_qa_chain.int.test.js","sourceRoot":"","sources":["../../../chains/tests/vector_db_qa_chain.int.test.ts"],"names":[],"mappings":";;AAAA,2CAAqC;AACrC,8CAA2C;AAC3C,2CAA+C;AAC/C,4CAAwC;AACxC,kCAAoC;AACpC,8DAA4D;AAC5D,kDAAkD;AAClD,wDAAqD;AACrD,iDAAoD;AAEpD,IAAA,cAAI,EAAC,sBAAsB,EAAE,KAAK,IAAI,EAAE;IACtC,MAAM,KAAK,GAAG,IAAI,eAAM,CAAC,EAAE,CAAC,CAAC;IAC7B,MAAM,MAAM,GAAG,IAAI,wBAAc,CAAC;QAChC,QAAQ,EAAE,aAAa;QACvB,cAAc,EAAE,CAAC,KAAK,CAAC;KACxB,CAAC,CAAC;IACH,MAAM,WAAW,GAAG,MAAM,iBAAO,CAAC,SAAS,CACzC,CAAC,aAAa,EAAE,SAAS,EAAE,kBAAkB,EAAE,KAAK,EAAE,IAAI,CAAC,EAC3D,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,EACvD,IAAI,6BAAgB,EAAE,CACvB,CAAC;IACF,MAAM,QAAQ,GAAG,IAAI,oBAAQ,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC;IACtD,MAAM,gBAAgB,GAAG,IAAI,wCAAmB,CAAC;QAC/C,QAAQ;QACR,oBAAoB,EAAE,KAAK;KAC5B,CAAC,CAAC;IACH,MAAM,KAAK,GAAG,IAAI,8BAAe,CAAC;QAChC,qBAAqB,EAAE,gBAAgB;QACvC,WAAW,EAAE,WAAW;KACzB,CAAC,CAAC;IACH,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC;IACnD,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;AACvB,CAAC,CAAC,CAAC;AAEH,IAAA,cAAI,EAAC,+BAA+B,EAAE,KAAK,IAAI,EAAE;IAC/C,MAAM,KAAK,GAAG,IAAI,eAAM,CAAC,EAAE,CAAC,CAAC;IAC7B,MAAM,WAAW,GAAG,MAAM,iBAAO,CAAC,SAAS,CACzC,CAAC,aAAa,EAAE,SAAS,EAAE,kBAAkB,EAAE,KAAK,EAAE,IAAI,CAAC,EAC3D,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,EACvD,IAAI,6BAAgB,EAAE,CACvB,CAAC;IACF,MAAM,KAAK,GAAG,8BAAe,CAAC,OAAO,CAAC,KAAK,EAAE,WAAW,CAAC,CAAC;IAC1D,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC;IACnD,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;AACvB,CAAC,CAAC,CAAC;AAEH,IAAA,cAAI,EAAC,qBAAqB,EAAE,KAAK,IAAI,EAAE;IACrC,MAAM,WAAW,GAAG,MAAM,iBAAO,CAAC,SAAS,CACzC,CAAC,aAAa,EAAE,SAAS,EAAE,kBAAkB,EAAE,KAAK,EAAE,IAAI,CAAC,EAC3D,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,EACvD,IAAI,6BAAgB,EAAE,CACvB,CAAC;IACF,MAAM,KAAK,GAAG,MAAM,IAAA,gBAAS,EAAC,2CAA2C,EAAE;QACzE,WAAW,EAAE,WAAW;KACzB,CAAC,CAAC;IACH,MAAM,GAAG,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC,CAAC;IACnD,OAAO,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;AACvB,CAAC,CAAC,CAAC"}

package/dist/docstore/base.d.ts

@@ -0,0 +1,5 @@
+import { Document } from "../document";
+export declare abstract class Docstore {
+    abstract search(search: string): Document | string;
+    abstract add(texts: Record<string, Document>): void;
+}

package/dist/docstore/base.js

@@ -0,0 +1,7 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Docstore = void 0;
+class Docstore {
+}
+exports.Docstore = Docstore;
+//# sourceMappingURL=base.js.map

package/dist/docstore/base.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"base.js","sourceRoot":"","sources":["../../docstore/base.ts"],"names":[],"mappings":";;;AAEA,MAAsB,QAAQ;CAI7B;AAJD,4BAIC"}

package/dist/docstore/in_memory.d.ts

@@ -0,0 +1,10 @@
+import { Document } from "../document";
+import { Docstore } from "./base";
+export declare class InMemoryDocstore extends Docstore {
+    _docs: Map<string, Document>;
+    constructor(docs?: Map<string, Document>);
+    /** Method for getting count of documents in _docs */
+    get count(): number;
+    search(search: string): Document | string;
+    add(texts: Record<string, Document>): void;
+}

package/dist/docstore/in_memory.js

@@ -0,0 +1,36 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.InMemoryDocstore = void 0;
+const base_1 = require("./base");
+class InMemoryDocstore extends base_1.Docstore {
+    constructor(docs) {
+        super();
+        Object.defineProperty(this, "_docs", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this._docs = docs !== null && docs !== void 0 ? docs : new Map();
+    }
+    /** Method for getting count of documents in _docs */
+    get count() {
+        return this._docs.size;
+    }
+    search(search) {
+        var _a;
+        return (_a = this._docs.get(search)) !== null && _a !== void 0 ? _a : `ID ${search} not found.`;
+    }
+    add(texts) {
+        const keys = [...this._docs.keys()];
+        const overlapping = Object.keys(texts).filter((x) => keys.includes(x));
+        if (overlapping.length > 0) {
+            throw new Error(`Tried to add ids that already exist: ${overlapping}`);
+        }
+        for (const [key, value] of Object.entries(texts)) {
+            this._docs.set(key, value);
+        }
+    }
+}
+exports.InMemoryDocstore = InMemoryDocstore;
+//# sourceMappingURL=in_memory.js.map

package/dist/docstore/in_memory.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"in_memory.js","sourceRoot":"","sources":["../../docstore/in_memory.ts"],"names":[],"mappings":";;;AACA,iCAAkC;AAElC,MAAa,gBAAiB,SAAQ,eAAQ;IAG5C,YAAY,IAA4B;QACtC,KAAK,EAAE,CAAC;QAHV;;;;;WAA6B;QAI3B,IAAI,CAAC,KAAK,GAAG,IAAI,aAAJ,IAAI,cAAJ,IAAI,GAAI,IAAI,GAAG,EAAE,CAAC;IACjC,CAAC;IAED,qDAAqD;IACrD,IAAI,KAAK;QACP,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC;IACzB,CAAC;IAED,MAAM,CAAC,MAAc;;QACnB,OAAO,MAAA,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,mCAAI,MAAM,MAAM,aAAa,CAAC;IAC7D,CAAC;IAED,GAAG,CAAC,KAA+B;QACjC,MAAM,IAAI,GAAG,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;QACpC,MAAM,WAAW,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;QAEvE,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE;YAC1B,MAAM,IAAI,KAAK,CAAC,wCAAwC,WAAW,EAAE,CAAC,CAAC;SACxE;QAED,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE;YAChD,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;SAC5B;IACH,CAAC;CACF;AA7BD,4CA6BC"}

package/dist/docstore/index.d.ts

@@ -0,0 +1,3 @@
+export { Document } from "../document";
+export { Docstore } from "./base";
+export { InMemoryDocstore } from "./in_memory";

package/dist/docstore/index.js

@@ -0,0 +1,10 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.InMemoryDocstore = exports.Docstore = exports.Document = void 0;
+var document_1 = require("../document");
+Object.defineProperty(exports, "Document", { enumerable: true, get: function () { return document_1.Document; } });
+var base_1 = require("./base");
+Object.defineProperty(exports, "Docstore", { enumerable: true, get: function () { return base_1.Docstore; } });
+var in_memory_1 = require("./in_memory");
+Object.defineProperty(exports, "InMemoryDocstore", { enumerable: true, get: function () { return in_memory_1.InMemoryDocstore; } });
+//# sourceMappingURL=index.js.map

package/dist/docstore/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../docstore/index.ts"],"names":[],"mappings":";;;AAAA,wCAAuC;AAA9B,oGAAA,QAAQ,OAAA;AACjB,+BAAkC;AAAzB,gGAAA,QAAQ,OAAA;AACjB,yCAA+C;AAAtC,6GAAA,gBAAgB,OAAA"}

package/dist/document.d.ts

@@ -1,4 +1,4 @@
-interface DocumentParams {
+export interface DocumentParams {
     pageContent: string;
     metadata: Record<string, any>;
 }
@@ -10,4 +10,3 @@ export declare class Document implements DocumentParams {
     metadata: Record<string, any>;
     constructor(fields?: Partial<DocumentParams>);
 }
-export {};

package/dist/document_loaders/base.d.ts

@@ -0,0 +1,10 @@
+import { TextSplitter } from "../text_splitter";
+import { Document } from "../document";
+export interface DocumentLoader {
+    load(): Promise<Document[]>;
+    loadAndSplit(textSplitter?: TextSplitter): Promise<Document[]>;
+}
+export declare abstract class BaseDocumentLoader implements DocumentLoader {
+    abstract load(): Promise<Document[]>;
+    loadAndSplit(splitter?: TextSplitter): Promise<Document[]>;
+}

package/dist/document_loaders/base.js

@@ -0,0 +1,12 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.BaseDocumentLoader = void 0;
+const text_splitter_1 = require("../text_splitter");
+class BaseDocumentLoader {
+    async loadAndSplit(splitter = new text_splitter_1.RecursiveCharacterTextSplitter()) {
+        const docs = await this.load();
+        return splitter.splitDocuments(docs);
+    }
+}
+exports.BaseDocumentLoader = BaseDocumentLoader;
+//# sourceMappingURL=base.js.map

package/dist/document_loaders/base.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"base.js","sourceRoot":"","sources":["../../document_loaders/base.ts"],"names":[],"mappings":";;;AAAA,oDAAgF;AAQhF,MAAsB,kBAAkB;IAGtC,KAAK,CAAC,YAAY,CAChB,WAAyB,IAAI,8CAA8B,EAAE;QAE7D,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;QAC/B,OAAO,QAAQ,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;IACvC,CAAC;CACF;AATD,gDASC"}

package/dist/document_loaders/cheerio_web_base.d.ts

@@ -0,0 +1,10 @@
+import type { CheerioAPI } from "cheerio";
+import { Document } from "../document";
+import { BaseDocumentLoader } from "./base";
+import type { DocumentLoader } from "./base";
+export declare class CheerioWebBaseLoader extends BaseDocumentLoader implements DocumentLoader {
+    webPath: string;
+    constructor(webPath: string);
+    scrape(): Promise<CheerioAPI>;
+    load(): Promise<Document[]>;
+}

package/dist/document_loaders/cheerio_web_base.js

@@ -0,0 +1,47 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CheerioWebBaseLoader = void 0;
+const document_1 = require("../document");
+const base_1 = require("./base");
+let load = null;
+try {
+    // eslint-disable-next-line global-require,import/no-extraneous-dependencies
+    ({ load } = require("cheerio"));
+}
+catch (_a) {
+    // ignore error, will be throw in constructor
+}
+class CheerioWebBaseLoader extends base_1.BaseDocumentLoader {
+    constructor(webPath) {
+        super();
+        Object.defineProperty(this, "webPath", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: webPath
+        });
+        /**
+         * Throw error at construction time
+         * if cheerio package is not installed.
+         */
+        if (load === null) {
+            throw new Error("Please install cheerio as a dependency with, e.g. `yarn add cheerio`");
+        }
+    }
+    async scrape() {
+        if (load === null) {
+            throw new Error("Please install cheerio as a dependency with, e.g. `yarn add cheerio`");
+        }
+        const response = await fetch(this.webPath);
+        const html = await response.text();
+        return load(html);
+    }
+    async load() {
+        const $ = await this.scrape();
+        const text = $("body").text();
+        const metadata = { source: this.webPath };
+        return [new document_1.Document({ pageContent: text, metadata })];
+    }
+}
+exports.CheerioWebBaseLoader = CheerioWebBaseLoader;
+//# sourceMappingURL=cheerio_web_base.js.map

package/dist/document_loaders/cheerio_web_base.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cheerio_web_base.js","sourceRoot":"","sources":["../../document_loaders/cheerio_web_base.ts"],"names":[],"mappings":";;;AACA,0CAAuC;AACvC,iCAA4C;AAG5C,IAAI,IAAI,GAAwB,IAAI,CAAC;AAErC,IAAI;IACF,4EAA4E;IAC5E,CAAC,EAAE,IAAI,EAAE,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC;CACjC;AAAC,WAAM;IACN,6CAA6C;CAC9C;AAED,MAAa,oBACX,SAAQ,yBAAkB;IAG1B,YAAmB,OAAe;QAChC,KAAK,EAAE,CAAC;;;;;mBADS;;QAGjB;;;WAGG;QACH,IAAI,IAAI,KAAK,IAAI,EAAE;YACjB,MAAM,IAAI,KAAK,CACb,sEAAsE,CACvE,CAAC;SACH;IACH,CAAC;IAED,KAAK,CAAC,MAAM;QACV,IAAI,IAAI,KAAK,IAAI,EAAE;YACjB,MAAM,IAAI,KAAK,CACb,sEAAsE,CACvE,CAAC;SACH;QAED,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC3C,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QACnC,OAAO,IAAI,CAAC,IAAI,CAAC,CAAC;IACpB,CAAC;IAED,KAAK,CAAC,IAAI;QACR,MAAM,CAAC,GAAG,MAAM,IAAI,CAAC,MAAM,EAAE,CAAC;QAC9B,MAAM,IAAI,GAAG,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,CAAC;QAC9B,MAAM,QAAQ,GAAG,EAAE,MAAM,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC;QAC1C,OAAO,CAAC,IAAI,mBAAQ,CAAC,EAAE,WAAW,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC;IACzD,CAAC;CACF;AApCD,oDAoCC"}

package/dist/document_loaders/college_confidential.d.ts

@@ -0,0 +1,6 @@
+import { Document } from "../document";
+import { CheerioWebBaseLoader } from "./cheerio_web_base";
+export declare class CollegeConfidentialLoader extends CheerioWebBaseLoader {
+    constructor(webPath: string);
+    load(): Promise<Document[]>;
+}

package/dist/document_loaders/college_confidential.js

@@ -0,0 +1,18 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CollegeConfidentialLoader = void 0;
+const document_1 = require("../document");
+const cheerio_web_base_1 = require("./cheerio_web_base");
+class CollegeConfidentialLoader extends cheerio_web_base_1.CheerioWebBaseLoader {
+    constructor(webPath) {
+        super(webPath);
+    }
+    async load() {
+        const $ = await this.scrape();
+        const text = $("main[class='skin-handler']").text();
+        const metadata = { source: this.webPath };
+        return [new document_1.Document({ pageContent: text, metadata })];
+    }
+}
+exports.CollegeConfidentialLoader = CollegeConfidentialLoader;
+//# sourceMappingURL=college_confidential.js.map

package/dist/document_loaders/college_confidential.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"college_confidential.js","sourceRoot":"","sources":["../../document_loaders/college_confidential.ts"],"names":[],"mappings":";;;AAAA,0CAAuC;AACvC,yDAA0D;AAE1D,MAAa,yBAA0B,SAAQ,uCAAoB;IACjE,YAAY,OAAe;QACzB,KAAK,CAAC,OAAO,CAAC,CAAC;IACjB,CAAC;IAEM,KAAK,CAAC,IAAI;QACf,MAAM,CAAC,GAAG,MAAM,IAAI,CAAC,MAAM,EAAE,CAAC;QAC9B,MAAM,IAAI,GAAG,CAAC,CAAC,4BAA4B,CAAC,CAAC,IAAI,EAAE,CAAC;QACpD,MAAM,QAAQ,GAAG,EAAE,MAAM,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC;QAC1C,OAAO,CAAC,IAAI,mBAAQ,CAAC,EAAE,WAAW,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC;IACzD,CAAC;CACF;AAXD,8DAWC"}

package/dist/document_loaders/hn.d.ts

@@ -0,0 +1,9 @@
+import { Document } from "../document";
+import { CheerioWebBaseLoader } from "./cheerio_web_base";
+export declare class HNLoader extends CheerioWebBaseLoader {
+    webPath: string;
+    constructor(webPath: string);
+    load(): Promise<Document[]>;
+    private loadComments;
+    private loadResults;
+}

package/dist/document_loaders/hn.js

@@ -0,0 +1,53 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.HNLoader = void 0;
+const document_1 = require("../document");
+const cheerio_web_base_1 = require("./cheerio_web_base");
+class HNLoader extends cheerio_web_base_1.CheerioWebBaseLoader {
+    constructor(webPath) {
+        super(webPath);
+        Object.defineProperty(this, "webPath", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: webPath
+        });
+    }
+    async load() {
+        const $ = await this.scrape();
+        if (this.webPath.includes("item")) {
+            return this.loadComments($);
+        }
+        return this.loadResults($);
+    }
+    loadComments($) {
+        const comments = $("tr[class='athing comtr']");
+        const title = $("tr[id='pagespace']").attr("title");
+        const documents = [];
+        comments.each((_index, comment) => {
+            const text = $(comment).text().trim();
+            const metadata = { source: this.webPath, title };
+            documents.push(new document_1.Document({ pageContent: text, metadata }));
+        });
+        return documents;
+    }
+    loadResults($) {
+        const items = $("tr[class='athing']");
+        const documents = [];
+        items.each((_index, item) => {
+            const ranking = $(item).find("span[class='rank']").text();
+            const link = $(item).find("span[class='titleline'] a").attr("href");
+            const title = $(item).find("span[class='titleline']").text().trim();
+            const metadata = {
+                source: this.webPath,
+                title,
+                link,
+                ranking,
+            };
+            documents.push(new document_1.Document({ pageContent: title, metadata }));
+        });
+        return documents;
+    }
+}
+exports.HNLoader = HNLoader;
+//# sourceMappingURL=hn.js.map

package/dist/document_loaders/hn.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"hn.js","sourceRoot":"","sources":["../../document_loaders/hn.ts"],"names":[],"mappings":";;;AACA,0CAAuC;AACvC,yDAA0D;AAE1D,MAAa,QAAS,SAAQ,uCAAoB;IAChD,YAAmB,OAAe;QAChC,KAAK,CAAC,OAAO,CAAC,CAAC;;;;;mBADE;;IAEnB,CAAC;IAEM,KAAK,CAAC,IAAI;QACf,MAAM,CAAC,GAAG,MAAM,IAAI,CAAC,MAAM,EAAE,CAAC;QAC9B,IAAI,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE;YACjC,OAAO,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;SAC7B;QACD,OAAO,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC;IAC7B,CAAC;IAEO,YAAY,CAAC,CAAa;QAChC,MAAM,QAAQ,GAAG,CAAC,CAAC,0BAA0B,CAAC,CAAC;QAC/C,MAAM,KAAK,GAAG,CAAC,CAAC,oBAAoB,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACpD,MAAM,SAAS,GAAe,EAAE,CAAC;QACjC,QAAQ,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE;YAChC,MAAM,IAAI,GAAG,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC;YACtC,MAAM,QAAQ,GAAG,EAAE,MAAM,EAAE,IAAI,CAAC,OAAO,EAAE,KAAK,EAAE,CAAC;YACjD,SAAS,CAAC,IAAI,CAAC,IAAI,mBAAQ,CAAC,EAAE,WAAW,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QACH,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,WAAW,CAAC,CAAa;QAC/B,MAAM,KAAK,GAAG,CAAC,CAAC,oBAAoB,CAAC,CAAC;QACtC,MAAM,SAAS,GAAe,EAAE,CAAC;QACjC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,IAAI,EAAE,EAAE;YAC1B,MAAM,OAAO,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC,IAAI,EAAE,CAAC;YAC1D,MAAM,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACpE,MAAM,KAAK,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC;YACpE,MAAM,QAAQ,GAAG;gBACf,MAAM,EAAE,IAAI,CAAC,OAAO;gBACpB,KAAK;gBACL,IAAI;gBACJ,OAAO;aACR,CAAC;YACF,SAAS,CAAC,IAAI,CAAC,IAAI,mBAAQ,CAAC,EAAE,WAAW,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;QACH,OAAO,SAAS,CAAC;IACnB,CAAC;CACF;AA1CD,4BA0CC"}

package/dist/document_loaders/imsdb.d.ts

@@ -0,0 +1,7 @@
+import { Document } from "../document";
+import { CheerioWebBaseLoader } from "./cheerio_web_base";
+export declare class IMSDBLoader extends CheerioWebBaseLoader {
+    webPath: string;
+    constructor(webPath: string);
+    load(): Promise<Document[]>;
+}