@langchain/community
Version:
Third-party integrations for LangChain.js
73 lines (72 loc) • 2.15 kB
JavaScript
Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
const require_runtime = require("../_virtual/_rolldown/runtime.cjs");
let _langchain_core_utils_async_caller = require("@langchain/core/utils/async_caller");
let _langchain_core_documents = require("@langchain/core/documents");
let _langchain_core_retrievers = require("@langchain/core/retrievers");
//#region src/retrievers/chaindesk.ts
var chaindesk_exports = /* @__PURE__ */ require_runtime.__exportAll({ ChaindeskRetriever: () => ChaindeskRetriever });
/**
* @example
* ```typescript
* const retriever = new ChaindeskRetriever({
* datastoreId: "DATASTORE_ID",
* apiKey: "CHAINDESK_API_KEY",
* topK: 8,
* });
* const docs = await retriever.getRelevantDocuments("hello");
* ```
*/
var ChaindeskRetriever = class extends _langchain_core_retrievers.BaseRetriever {
static lc_name() {
return "ChaindeskRetriever";
}
lc_namespace = [
"langchain",
"retrievers",
"chaindesk"
];
caller;
datastoreId;
topK;
filter;
apiKey;
constructor({ datastoreId, apiKey, topK, filter, ...rest }) {
super();
this.caller = new _langchain_core_utils_async_caller.AsyncCaller(rest);
this.datastoreId = datastoreId;
this.apiKey = apiKey;
this.topK = topK;
this.filter = filter;
}
async getRelevantDocuments(query) {
const { results } = await (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 } : {},
...this.filter ? { filters: this.filter } : {}
}),
headers: {
"Content-Type": "application/json",
...this.apiKey ? { Authorization: `Bearer ${this.apiKey}` } : {}
}
})).json();
return results.map(({ text, score, source, ...rest }) => new _langchain_core_documents.Document({
pageContent: text,
metadata: {
score,
source,
...rest
}
}));
}
};
//#endregion
exports.ChaindeskRetriever = ChaindeskRetriever;
Object.defineProperty(exports, "chaindesk_exports", {
enumerable: true,
get: function() {
return chaindesk_exports;
}
});
//# sourceMappingURL=chaindesk.cjs.map