chonkie
Version:
🦛 CHONK your texts in TS with Chonkie!✨The no-nonsense lightweight and efficient chunking library.
42 lines • 1.89 kB
JavaScript
;
/** Refinery clients for Chonkie API. */
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
return new (P || (P = Promise))(function (resolve, reject) {
function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
step((generator = generator.apply(thisArg, _arguments || [])).next());
});
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.EmbeddingsRefinery = void 0;
const base_1 = require("./base");
const base_2 = require("../types/base");
class EmbeddingsRefinery extends base_1.CloudClient {
constructor(apiKey, config) {
super({ apiKey });
if (!config.embeddingModel) {
throw new Error("Embedding model is required for embeddings refinement");
}
this.config = {
embeddingModel: config.embeddingModel,
};
}
refine(chunks) {
return __awaiter(this, void 0, void 0, function* () {
const response = yield this.request("/v1/refine/embeddings", {
body: {
chunks: chunks.map(chunk => chunk.toDict()),
embedding_model: this.config.embeddingModel,
},
headers: {
"Content-Type": "application/json",
},
});
return response.map(chunk => base_2.Chunk.fromDict(chunk));
});
}
}
exports.EmbeddingsRefinery = EmbeddingsRefinery;
//# sourceMappingURL=embeddings_refinery.js.map