langchain 0.0.160 → 0.0.162

package/dist/experimental/hubs/makersuite/googlemakersuitehub.js

@@ -0,0 +1,291 @@
+import { GoogleAuth } from "google-auth-library";
+import { GooglePaLM } from "../../../llms/googlepalm.js";
+import { ChatGooglePaLM } from "../../../chat_models/googlepalm.js";
+import { PromptTemplate } from "../../../prompts/index.js";
+import { AsyncCaller, } from "../../../util/async_caller.js";
+import { GoogleConnection } from "../../../util/googlevertexai-connection.js";
+/**
+ * A class that represents the Prompt that has been created by MakerSuite
+ * and loaded from Google Drive. It exposes methods to turn this prompt
+ * into a Template, a Model, and into a chain consisting of these two elements.
+ * In general, this class should be created by the MakerSuiteHub class and
+ * not instantiated manually.
+ */
+export class MakerSuitePrompt {
+    constructor(promptData) {
+        Object.defineProperty(this, "promptType", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "promptData", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.promptData = promptData;
+        this._determinePromptType();
+    }
+    _determinePromptType() {
+        if (Object.hasOwn(this.promptData, "textPrompt")) {
+            this.promptType = "text";
+        }
+        else if (Object.hasOwn(this.promptData, "dataPrompt")) {
+            this.promptType = "data";
+        }
+        else if (Object.hasOwn(this.promptData, "multiturnPrompt")) {
+            this.promptType = "chat";
+        }
+        else {
+            const error = new Error("Unable to identify prompt type.");
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            error.promptData = this.promptData;
+            throw error;
+        }
+    }
+    _promptValueText() {
+        return (this.promptData?.textPrompt?.value ?? "");
+    }
+    _promptValueData() {
+        const promptData = this
+            .promptData;
+        const dataPrompt = promptData?.dataPrompt;
+        let prompt = `${dataPrompt?.preamble}\n` || "";
+        dataPrompt?.rows.forEach((row) => {
+            // Add the data for each row, as long as it is listed as used
+            if (dataPrompt?.rowsUsed.includes(row.rowId)) {
+                // Add each input column
+                dataPrompt?.columns.forEach((column) => {
+                    if (column.isInput) {
+                        prompt += `${column.displayName} ${row.columnBindings[column.columnId]}\n`;
+                    }
+                });
+                // Add each output column
+                dataPrompt?.columns.forEach((column) => {
+                    if (!column.isInput) {
+                        prompt += `${column.displayName} ${row.columnBindings[column.columnId]}\n`;
+                    }
+                });
+            }
+        });
+        // Add the input column prompts
+        dataPrompt?.columns.forEach((column) => {
+            if (column.isInput) {
+                prompt += `${column.displayName} {${column.displayName.replace(":", "")}}\n`;
+            }
+        });
+        // Add just the first output column
+        const firstOutput = dataPrompt?.columns.find((column) => !column.isInput);
+        prompt += `${firstOutput?.displayName} `;
+        return prompt;
+    }
+    _promptValueChat() {
+        return (this.promptData?.multiturnPrompt
+            ?.preamble ?? "");
+    }
+    _promptValue() {
+        switch (this.promptType) {
+            case "text":
+                return this._promptValueText();
+            case "data":
+                return this._promptValueData();
+            case "chat":
+                return this._promptValueChat();
+            default:
+                throw new Error(`Invalid promptType: ${this.promptType}`);
+        }
+    }
+    /**
+     * Create a template from the prompt, including any "test input" specified
+     * in MakerSuite as a template parameter.
+     */
+    toTemplate() {
+        const value = this._promptValue();
+        const promptString = value.replaceAll(/{{.*:(.+):.*}}/g, "{$1}");
+        return PromptTemplate.fromTemplate(promptString);
+    }
+    _modelName() {
+        let ret = this.promptData?.runSettings?.model;
+        if (!ret) {
+            ret =
+                this.promptType === "chat"
+                    ? "models/chat-bison-001"
+                    : "models/text-bison-001";
+        }
+        return ret;
+    }
+    _examples() {
+        const promptData = this
+            .promptData;
+        const ret = promptData?.multiturnPrompt?.primingExchanges
+            .map((exchange) => {
+            const example = {};
+            if (exchange?.request) {
+                example.input = {
+                    content: exchange.request,
+                };
+            }
+            if (exchange?.response) {
+                example.output = {
+                    content: exchange.response,
+                };
+            }
+            return example;
+        })
+            .filter((value) => Object.keys(value).length);
+        return ret;
+    }
+    /**
+     * Create a model from the prompt with all the parameters (model name,
+     * temperature, etc) set as they were in MakerSuite.
+     */
+    toModel() {
+        const modelName = this._modelName();
+        const modelSettings = {
+            modelName,
+            ...this.promptData?.runSettings,
+        };
+        if (this.promptType === "chat") {
+            const examples = this._examples();
+            return new ChatGooglePaLM({
+                examples,
+                ...modelSettings,
+            });
+        }
+        else {
+            return new GooglePaLM(modelSettings);
+        }
+    }
+    /**
+     * Create a RunnableSequence based on the template and model that can
+     * be created from this prompt. The template will have parameters available
+     * based on the "test input" that was set in MakerSuite, and the model
+     * will have the parameters (model name, temperature, etc) from those in
+     * MakerSuite.
+     */
+    toChain() {
+        return this.toTemplate().pipe(this.toModel());
+    }
+}
+export class DriveFileReadConnection extends GoogleConnection {
+    constructor(fields, caller) {
+        super(caller, new GoogleAuth({
+            scopes: "https://www.googleapis.com/auth/drive.readonly",
+            ...fields.authOptions,
+        }));
+        Object.defineProperty(this, "endpoint", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "apiVersion", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "fileId", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.endpoint = fields.endpoint ?? "www.googleapis.com";
+        this.apiVersion = fields.apiVersion ?? "v3";
+        this.fileId = fields.fileId;
+    }
+    async buildUrl() {
+        return `https://${this.endpoint}/drive/${this.apiVersion}/files/${this.fileId}?alt=media`;
+    }
+    buildMethod() {
+        return "GET";
+    }
+    async request(options) {
+        return this._request(undefined, options ?? {});
+    }
+}
+/**
+ * A class allowing access to MakerSuite prompts that have been saved in
+ * Google Drive.
+ * MakerSuite prompts are pulled based on their Google Drive ID (which is available
+ * from Google Drive or as part of the URL when the prompt is open in MakerSuite).
+ * There is a basic cache that will store the prompt in memory for a time specified
+ * in milliseconds. This defaults to 0, indicating the prompt should always be
+ * pulled from Google Drive.
+ */
+export class MakerSuiteHub {
+    constructor(config) {
+        Object.defineProperty(this, "cache", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: {}
+        });
+        Object.defineProperty(this, "cacheTimeout", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "caller", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.cacheTimeout = config?.cacheTimeout ?? 0;
+        this.caller = config?.caller ?? new AsyncCaller({});
+    }
+    /**
+     * Is the current cache entry valid, or does it need to be refreshed.
+     * It will need to be refreshed under any of the following conditions:
+     * - It does not currently exist in the cache
+     * - The cacheTimeout is 0
+     * - The cache was last updated longer ago than the cacheTimeout allows
+     * @param entry - The cache entry, including when this prompt was last refreshed
+     */
+    isValid(entry) {
+        // If we don't have a record, it can't be valid
+        // And if the cache timeout is 0, we will always refresh, so the cache is invalid
+        if (!entry || this.cacheTimeout === 0) {
+            return false;
+        }
+        const now = Date.now();
+        const expiration = entry.updated + this.cacheTimeout;
+        return expiration > now;
+    }
+    /**
+     * Get a MakerSuitePrompt that is specified by the Google Drive ID.
+     * This will always be loaded from Google Drive.
+     * @param id
+     * @return A MakerSuitePrompt which can be used to create a template, model, or chain.
+     */
+    async forcePull(id) {
+        const fields = {
+            fileId: id,
+        };
+        const connection = new DriveFileReadConnection(fields, this.caller);
+        const result = await connection.request();
+        const ret = new MakerSuitePrompt(result.data);
+        this.cache[id] = {
+            prompt: ret,
+            updated: Date.now(),
+        };
+        return ret;
+    }
+    /**
+     * Get a MakerSuitePrompt that is specified by the Google Drive ID. This may be
+     * loaded from Google Drive or, if there is a valid copy in the cache, the cached
+     * copy will be returned.
+     * @param id
+     * @return A MakerSuitePrompt which can be used to create a template, model, or chain.
+     */
+    async pull(id) {
+        const entry = this.cache[id];
+        const ret = this.isValid(entry) ? entry.prompt : await this.forcePull(id);
+        return ret;
+    }
+}

package/dist/llms/base.cjs

@@ -55,7 +55,7 @@ class BaseLLM extends index_js_2.BaseLanguageModel {
                 options: callOptions,
                 invocation_params: this?.invocationParams(callOptions),
             };
-            const runManagers = await callbackManager_?.handleLLMStart(this.toJSON(), [prompt.toString()], undefined, undefined, extra);
+            const runManagers = await callbackManager_?.handleLLMStart(this.toJSON(), [prompt.toString()], undefined, undefined, extra, undefined, undefined, runnableConfig.runName);
             let generation = new index_js_1.GenerationChunk({
                 text: "",
             });
@@ -129,7 +129,7 @@ class BaseLLM extends index_js_2.BaseLanguageModel {
             options: parsedOptions,
             invocation_params: this?.invocationParams(parsedOptions),
         };
-        const runManagers = await callbackManager_?.handleLLMStart(this.toJSON(), prompts, undefined, undefined, extra);
+        const runManagers = await callbackManager_?.handleLLMStart(this.toJSON(), prompts, undefined, undefined, extra, undefined, undefined, handledOptions?.runName);
         let output;
         try {
             output = await this._generate(prompts, parsedOptions, runManagers?.[0]);

package/dist/llms/base.js

@@ -52,7 +52,7 @@ export class BaseLLM extends BaseLanguageModel {
                 options: callOptions,
                 invocation_params: this?.invocationParams(callOptions),
             };
-            const runManagers = await callbackManager_?.handleLLMStart(this.toJSON(), [prompt.toString()], undefined, undefined, extra);
+            const runManagers = await callbackManager_?.handleLLMStart(this.toJSON(), [prompt.toString()], undefined, undefined, extra, undefined, undefined, runnableConfig.runName);
             let generation = new GenerationChunk({
                 text: "",
             });
@@ -126,7 +126,7 @@ export class BaseLLM extends BaseLanguageModel {
             options: parsedOptions,
             invocation_params: this?.invocationParams(parsedOptions),
         };
-        const runManagers = await callbackManager_?.handleLLMStart(this.toJSON(), prompts, undefined, undefined, extra);
+        const runManagers = await callbackManager_?.handleLLMStart(this.toJSON(), prompts, undefined, undefined, extra, undefined, undefined, handledOptions?.runName);
         let output;
         try {
             output = await this._generate(prompts, parsedOptions, runManagers?.[0]);

package/dist/load/import_constants.cjs

@@ -17,6 +17,7 @@ exports.optionalImportEntrypoints = [
     "langchain/chains/query_constructor/ir",
     "langchain/chains/sql_db",
     "langchain/chains/graph_qa/cypher",
+    "langchain/embeddings/bedrock",
     "langchain/embeddings/cloudflare_workersai",
     "langchain/embeddings/cohere",
     "langchain/embeddings/tensorflow",
@@ -140,4 +141,5 @@ exports.optionalImportEntrypoints = [
     "langchain/experimental/multimodal_embeddings/googlevertexai",
     "langchain/experimental/chat_models/anthropic_functions",
     "langchain/experimental/llms/bittensor",
+    "langchain/experimental/hubs/makersuite/googlemakersuitehub",
 ];

package/dist/load/import_constants.js

@@ -14,6 +14,7 @@ export const optionalImportEntrypoints = [
     "langchain/chains/query_constructor/ir",
     "langchain/chains/sql_db",
     "langchain/chains/graph_qa/cypher",
+    "langchain/embeddings/bedrock",
     "langchain/embeddings/cloudflare_workersai",
     "langchain/embeddings/cohere",
     "langchain/embeddings/tensorflow",
@@ -137,4 +138,5 @@ export const optionalImportEntrypoints = [
     "langchain/experimental/multimodal_embeddings/googlevertexai",
     "langchain/experimental/chat_models/anthropic_functions",
     "langchain/experimental/llms/bittensor",
+    "langchain/experimental/hubs/makersuite/googlemakersuitehub",
 ];

package/dist/load/import_map.cjs

@@ -25,7 +25,7 @@ var __importStar = (this && this.__importStar) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.retrievers__remote = exports.output_parsers = exports.callbacks = exports.schema__storage = exports.schema__runnable = exports.schema__retriever = exports.schema__query_constructor = exports.schema__output_parser = exports.schema__document = exports.schema = exports.chat_models__minimax = exports.chat_models__ollama = exports.chat_models__baiduwenxin = exports.chat_models__fireworks = exports.chat_models__anthropic = exports.chat_models__openai = exports.chat_models__base = exports.document_transformers__openai_functions = exports.document_loaders__web__sort_xyz_blockchain = exports.document_loaders__web__serpapi = exports.document_loaders__web__searchapi = exports.document_loaders__base = exports.document = exports.memory = exports.text_splitter = exports.vectorstores__xata = exports.vectorstores__vectara = exports.vectorstores__prisma = exports.vectorstores__memory = exports.vectorstores__base = exports.prompts = exports.llms__fireworks = exports.llms__ollama = exports.llms__aleph_alpha = exports.llms__ai21 = exports.llms__openai = exports.llms__base = exports.embeddings__minimax = exports.embeddings__openai = exports.embeddings__ollama = exports.embeddings__fake = exports.embeddings__cache_backed = exports.embeddings__base = exports.chains__openai_functions = exports.chains = exports.tools = exports.base_language = exports.agents__toolkits = exports.agents = exports.load__serializable = void 0;
-exports.evaluation = exports.experimental__chat_models__bittensor = exports.experimental__plan_and_execute = exports.experimental__generative_agents = exports.experimental__babyagi = exports.experimental__autogpt = exports.util__math = exports.storage__in_memory = exports.stores__message__in_memory = exports.stores__file__in_memory = exports.stores__doc__in_memory = exports.cache = exports.retrievers__vespa = exports.retrievers__score_threshold = exports.retrievers__hyde = exports.retrievers__document_compressors__chain_extract = exports.retrievers__time_weighted = exports.retrievers__parent_document = exports.retrievers__multi_vector = exports.retrievers__multi_query = exports.retrievers__document_compressors = exports.retrievers__contextual_compression = exports.retrievers__databerry = void 0;
+exports.evaluation = exports.experimental__chat_models__bittensor = exports.experimental__plan_and_execute = exports.experimental__generative_agents = exports.experimental__babyagi = exports.experimental__autogpt = exports.util__math = exports.storage__in_memory = exports.stores__message__in_memory = exports.stores__file__in_memory = exports.stores__doc__in_memory = exports.cache = exports.retrievers__vespa = exports.retrievers__score_threshold = exports.retrievers__hyde = exports.retrievers__document_compressors__embeddings_filter = exports.retrievers__document_compressors__chain_extract = exports.retrievers__time_weighted = exports.retrievers__tavily_search_api = exports.retrievers__parent_document = exports.retrievers__multi_vector = exports.retrievers__multi_query = exports.retrievers__document_compressors = exports.retrievers__contextual_compression = exports.retrievers__databerry = exports.retrievers__chaindesk = void 0;
 exports.load__serializable = __importStar(require("../load/serializable.cjs"));
 exports.agents = __importStar(require("../agents/index.cjs"));
 exports.agents__toolkits = __importStar(require("../agents/toolkits/index.cjs"));
@@ -76,14 +76,17 @@ exports.schema__storage = __importStar(require("../schema/storage.cjs"));
 exports.callbacks = __importStar(require("../callbacks/index.cjs"));
 exports.output_parsers = __importStar(require("../output_parsers/index.cjs"));
 exports.retrievers__remote = __importStar(require("../retrievers/remote/index.cjs"));
+exports.retrievers__chaindesk = __importStar(require("../retrievers/chaindesk.cjs"));
 exports.retrievers__databerry = __importStar(require("../retrievers/databerry.cjs"));
 exports.retrievers__contextual_compression = __importStar(require("../retrievers/contextual_compression.cjs"));
 exports.retrievers__document_compressors = __importStar(require("../retrievers/document_compressors/index.cjs"));
 exports.retrievers__multi_query = __importStar(require("../retrievers/multi_query.cjs"));
 exports.retrievers__multi_vector = __importStar(require("../retrievers/multi_vector.cjs"));
 exports.retrievers__parent_document = __importStar(require("../retrievers/parent_document.cjs"));
+exports.retrievers__tavily_search_api = __importStar(require("../retrievers/tavily_search_api.cjs"));
 exports.retrievers__time_weighted = __importStar(require("../retrievers/time_weighted.cjs"));
 exports.retrievers__document_compressors__chain_extract = __importStar(require("../retrievers/document_compressors/chain_extract.cjs"));
+exports.retrievers__document_compressors__embeddings_filter = __importStar(require("../retrievers/document_compressors/embeddings_filter.cjs"));
 exports.retrievers__hyde = __importStar(require("../retrievers/hyde.cjs"));
 exports.retrievers__score_threshold = __importStar(require("../retrievers/score_threshold.cjs"));
 exports.retrievers__vespa = __importStar(require("../retrievers/vespa.cjs"));

package/dist/load/import_map.d.ts

@@ -48,14 +48,17 @@ export * as schema__storage from "../schema/storage.js";
 export * as callbacks from "../callbacks/index.js";
 export * as output_parsers from "../output_parsers/index.js";
 export * as retrievers__remote from "../retrievers/remote/index.js";
+export * as retrievers__chaindesk from "../retrievers/chaindesk.js";
 export * as retrievers__databerry from "../retrievers/databerry.js";
 export * as retrievers__contextual_compression from "../retrievers/contextual_compression.js";
 export * as retrievers__document_compressors from "../retrievers/document_compressors/index.js";
 export * as retrievers__multi_query from "../retrievers/multi_query.js";
 export * as retrievers__multi_vector from "../retrievers/multi_vector.js";
 export * as retrievers__parent_document from "../retrievers/parent_document.js";
+export * as retrievers__tavily_search_api from "../retrievers/tavily_search_api.js";
 export * as retrievers__time_weighted from "../retrievers/time_weighted.js";
 export * as retrievers__document_compressors__chain_extract from "../retrievers/document_compressors/chain_extract.js";
+export * as retrievers__document_compressors__embeddings_filter from "../retrievers/document_compressors/embeddings_filter.js";
 export * as retrievers__hyde from "../retrievers/hyde.js";
 export * as retrievers__score_threshold from "../retrievers/score_threshold.js";
 export * as retrievers__vespa from "../retrievers/vespa.js";

package/dist/load/import_map.js

@@ -49,14 +49,17 @@ export * as schema__storage from "../schema/storage.js";
 export * as callbacks from "../callbacks/index.js";
 export * as output_parsers from "../output_parsers/index.js";
 export * as retrievers__remote from "../retrievers/remote/index.js";
+export * as retrievers__chaindesk from "../retrievers/chaindesk.js";
 export * as retrievers__databerry from "../retrievers/databerry.js";
 export * as retrievers__contextual_compression from "../retrievers/contextual_compression.js";
 export * as retrievers__document_compressors from "../retrievers/document_compressors/index.js";
 export * as retrievers__multi_query from "../retrievers/multi_query.js";
 export * as retrievers__multi_vector from "../retrievers/multi_vector.js";
 export * as retrievers__parent_document from "../retrievers/parent_document.js";
+export * as retrievers__tavily_search_api from "../retrievers/tavily_search_api.js";
 export * as retrievers__time_weighted from "../retrievers/time_weighted.js";
 export * as retrievers__document_compressors__chain_extract from "../retrievers/document_compressors/chain_extract.js";
+export * as retrievers__document_compressors__embeddings_filter from "../retrievers/document_compressors/embeddings_filter.js";
 export * as retrievers__hyde from "../retrievers/hyde.js";
 export * as retrievers__score_threshold from "../retrievers/score_threshold.js";
 export * as retrievers__vespa from "../retrievers/vespa.js";

package/dist/prompts/chat.cjs

@@ -3,9 +3,10 @@
 // Replace with "string" when we are comfortable with a breaking change.
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.ChatPromptTemplate = exports.SystemMessagePromptTemplate = exports.AIMessagePromptTemplate = exports.HumanMessagePromptTemplate = exports.ChatMessagePromptTemplate = exports.BaseChatPromptTemplate = exports.BaseMessageStringPromptTemplate = exports.MessagesPlaceholder = exports.ChatPromptValue = exports.BaseMessagePromptTemplate = void 0;
+const base_js_1 = require("../memory/base.cjs");
 const index_js_1 = require("../schema/index.cjs");
 const index_js_2 = require("../schema/runnable/index.cjs");
-const base_js_1 = require("./base.cjs");
+const base_js_2 = require("./base.cjs");
 const prompt_js_1 = require("./prompt.cjs");
 /**
  * Abstract class that serves as a base for creating message prompt
@@ -74,7 +75,7 @@ class ChatPromptValue extends index_js_1.BasePromptValue {
         this.messages = fields.messages;
     }
     toString() {
-        return JSON.stringify(this.messages);
+        return (0, base_js_1.getBufferString)(this.messages);
     }
     toChatMessages() {
         return this.messages;
@@ -142,7 +143,7 @@ exports.BaseMessageStringPromptTemplate = BaseMessageStringPromptTemplate;
  * Abstract class that serves as a base for creating chat prompt
  * templates. It extends the BasePromptTemplate.
  */
-class BaseChatPromptTemplate extends base_js_1.BasePromptTemplate {
+class BaseChatPromptTemplate extends base_js_2.BasePromptTemplate {
     constructor(input) {
         super(input);
     }

package/dist/prompts/chat.js

@@ -1,5 +1,6 @@
 // Default generic "any" values are for backwards compatibility.
 // Replace with "string" when we are comfortable with a breaking change.
+import { getBufferString } from "../memory/base.js";
 import { AIMessage, BaseMessage, BasePromptValue, ChatMessage, HumanMessage, SystemMessage, coerceMessageLikeToMessage, isBaseMessage, } from "../schema/index.js";
 import { Runnable } from "../schema/runnable/index.js";
 import { BasePromptTemplate, } from "./base.js";
@@ -70,7 +71,7 @@ export class ChatPromptValue extends BasePromptValue {
         this.messages = fields.messages;
     }
     toString() {
-        return JSON.stringify(this.messages);
+        return getBufferString(this.messages);
     }
     toChatMessages() {
         return this.messages;

package/dist/retrievers/chaindesk.cjs

@@ -0,0 +1,71 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ChaindeskRetriever = void 0;
+const retriever_js_1 = require("../schema/retriever.cjs");
+const document_js_1 = require("../document.cjs");
+const async_caller_js_1 = require("../util/async_caller.cjs");
+class ChaindeskRetriever extends retriever_js_1.BaseRetriever {
+    static lc_name() {
+        return "ChaindeskRetriever";
+    }
+    constructor({ datastoreId, apiKey, topK, ...rest }) {
+        super();
+        Object.defineProperty(this, "lc_namespace", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: ["langchain", "retrievers", "chaindesk"]
+        });
+        Object.defineProperty(this, "caller", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "datastoreId", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "topK", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "apiKey", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.caller = new async_caller_js_1.AsyncCaller(rest);
+        this.datastoreId = datastoreId;
+        this.apiKey = apiKey;
+        this.topK = topK;
+    }
+    async getRelevantDocuments(query) {
+        const r = await this.caller.call(fetch, `https://app.chaindesk.ai/api/datastores/${this.datastoreId}/query`, {
+            method: "POST",
+            body: JSON.stringify({
+                query,
+                ...(this.topK ? { topK: this.topK } : {}),
+            }),
+            headers: {
+                "Content-Type": "application/json",
+                ...(this.apiKey ? { Authorization: `Bearer ${this.apiKey}` } : {}),
+            },
+        });
+        const { results } = (await r.json());
+        return results.map(({ text, score, source, ...rest }) => new document_js_1.Document({
+            pageContent: text,
+            metadata: {
+                score,
+                source,
+                ...rest,
+            },
+        }));
+    }
+}
+exports.ChaindeskRetriever = ChaindeskRetriever;

package/dist/retrievers/chaindesk.d.ts

@@ -0,0 +1,18 @@
+import { BaseRetriever, type BaseRetrieverInput } from "../schema/retriever.js";
+import { Document } from "../document.js";
+import { AsyncCaller, type AsyncCallerParams } from "../util/async_caller.js";
+export interface ChaindeskRetrieverArgs extends AsyncCallerParams, BaseRetrieverInput {
+    datastoreId: string;
+    topK?: number;
+    apiKey?: string;
+}
+export declare class ChaindeskRetriever extends BaseRetriever {
+    static lc_name(): string;
+    lc_namespace: string[];
+    caller: AsyncCaller;
+    datastoreId: string;
+    topK?: number;
+    apiKey?: string;
+    constructor({ datastoreId, apiKey, topK, ...rest }: ChaindeskRetrieverArgs);
+    getRelevantDocuments(query: string): Promise<Document[]>;
+}

package/dist/retrievers/chaindesk.js

@@ -0,0 +1,67 @@
+import { BaseRetriever } from "../schema/retriever.js";
+import { Document } from "../document.js";
+import { AsyncCaller } from "../util/async_caller.js";
+export class ChaindeskRetriever extends BaseRetriever {
+    static lc_name() {
+        return "ChaindeskRetriever";
+    }
+    constructor({ datastoreId, apiKey, topK, ...rest }) {
+        super();
+        Object.defineProperty(this, "lc_namespace", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: ["langchain", "retrievers", "chaindesk"]
+        });
+        Object.defineProperty(this, "caller", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "datastoreId", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "topK", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "apiKey", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        this.caller = new AsyncCaller(rest);
+        this.datastoreId = datastoreId;
+        this.apiKey = apiKey;
+        this.topK = topK;
+    }
+    async getRelevantDocuments(query) {
+        const r = await this.caller.call(fetch, `https://app.chaindesk.ai/api/datastores/${this.datastoreId}/query`, {
+            method: "POST",
+            body: JSON.stringify({
+                query,
+                ...(this.topK ? { topK: this.topK } : {}),
+            }),
+            headers: {
+                "Content-Type": "application/json",
+                ...(this.apiKey ? { Authorization: `Bearer ${this.apiKey}` } : {}),
+            },
+        });
+        const { results } = (await r.json());
+        return results.map(({ text, score, source, ...rest }) => new Document({
+            pageContent: text,
+            metadata: {
+                score,
+                source,
+                ...rest,
+            },
+        }));
+    }
+}

package/dist/retrievers/contextual_compression.cjs

@@ -35,8 +35,8 @@ class ContextualCompressionRetriever extends retriever_js_1.BaseRetriever {
         this.baseRetriever = fields.baseRetriever;
     }
     async _getRelevantDocuments(query, runManager) {
-        const docs = await this.baseRetriever._getRelevantDocuments(query, runManager);
-        const compressedDocs = await this.baseCompressor.compressDocuments(docs, query);
+        const docs = await this.baseRetriever.getRelevantDocuments(query, runManager?.getChild("base_retriever"));
+        const compressedDocs = await this.baseCompressor.compressDocuments(docs, query, runManager?.getChild("base_compressor"));
         return compressedDocs;
     }
 }

package/dist/retrievers/contextual_compression.js

@@ -32,8 +32,8 @@ export class ContextualCompressionRetriever extends BaseRetriever {
         this.baseRetriever = fields.baseRetriever;
     }
     async _getRelevantDocuments(query, runManager) {
-        const docs = await this.baseRetriever._getRelevantDocuments(query, runManager);
-        const compressedDocs = await this.baseCompressor.compressDocuments(docs, query);
+        const docs = await this.baseRetriever.getRelevantDocuments(query, runManager?.getChild("base_retriever"));
+        const compressedDocs = await this.baseCompressor.compressDocuments(docs, query, runManager?.getChild("base_compressor"));
         return compressedDocs;
     }
 }