85 lines
2.8 KiB
JavaScript
85 lines
2.8 KiB
JavaScript
|
"use strict";
|
||
|
|
Object.defineProperty(exports, "__esModule", { value: true });
|
||
|
|
exports.DataberryRetriever = void 0;
|
||
|
|
const retrievers_1 = require("@langchain/core/retrievers");
|
||
|
|
const documents_1 = require("@langchain/core/documents");
|
||
|
|
const async_caller_1 = require("@langchain/core/utils/async_caller");
|
||
|
|
/**
|
||
|
|
* A specific implementation of a document retriever for the Databerry
|
||
|
|
* API. It extends the BaseRetriever class, which is an abstract base
|
||
|
|
* class for a document retrieval system in LangChain.
|
||
|
|
*/
|
||
|
|
/** @deprecated Use "langchain/retrievers/chaindesk" instead */
|
||
|
|
class DataberryRetriever extends retrievers_1.BaseRetriever {
|
||
|
|
static lc_name() {
|
||
|
|
return "DataberryRetriever";
|
||
|
|
}
|
||
|
|
get lc_secrets() {
|
||
|
|
return { apiKey: "DATABERRY_API_KEY" };
|
||
|
|
}
|
||
|
|
get lc_aliases() {
|
||
|
|
return { apiKey: "api_key" };
|
||
|
|
}
|
||
|
|
constructor(fields) {
|
||
|
|
super(fields);
|
||
|
|
Object.defineProperty(this, "lc_namespace", {
|
||
|
|
enumerable: true,
|
||
|
|
configurable: true,
|
||
|
|
writable: true,
|
||
|
|
value: ["langchain", "retrievers", "databerry"]
|
||
|
|
});
|
||
|
|
Object.defineProperty(this, "caller", {
|
||
|
|
enumerable: true,
|
||
|
|
configurable: true,
|
||
|
|
writable: true,
|
||
|
|
value: void 0
|
||
|
|
});
|
||
|
|
Object.defineProperty(this, "datastoreUrl", {
|
||
|
|
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
|
||
|
|
});
|
||
|
|
const { datastoreUrl, apiKey, topK, ...rest } = fields;
|
||
|
|
this.caller = new async_caller_1.AsyncCaller(rest);
|
||
|
|
this.datastoreUrl = datastoreUrl;
|
||
|
|
this.apiKey = apiKey;
|
||
|
|
this.topK = topK;
|
||
|
|
}
|
||
|
|
async _getRelevantDocuments(query) {
|
||
|
|
const r = await this.caller.call(fetch, this.datastoreUrl, {
|
||
|
|
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 documents_1.Document({
|
||
|
|
pageContent: text,
|
||
|
|
metadata: {
|
||
|
|
score,
|
||
|
|
source,
|
||
|
|
...rest,
|
||
|
|
},
|
||
|
|
}));
|
||
|
|
}
|
||
|
|
}
|
||
|
|
exports.DataberryRetriever = DataberryRetriever;
|