Skip to content
Snippets Groups Projects
Commit 22127934 authored by Yi Ding's avatar Yi Ding
Browse files

more housekeeping

parent d6c6aefd
Branches
Tags
No related merge requests found
......@@ -174,24 +174,12 @@ export function getTopKMMREmbeddings(
}
export abstract class BaseEmbedding {
static similarity(
similarity(
embedding1: number[],
embedding2: number[],
mode: SimilarityType = SimilarityType.DOT_PRODUCT
mode: SimilarityType = SimilarityType.DEFAULT
): number {
if (embedding1.length !== embedding2.length) {
throw new Error("Embedding length mismatch");
}
if (mode === SimilarityType.DOT_PRODUCT) {
let result = 0;
for (let i = 0; i < embedding1.length; i++) {
result += embedding1[i] * embedding2[i];
}
return result;
} else {
throw new Error("Not implemented yet");
}
return similarity(embedding1, embedding2, mode);
}
abstract aGetTextEmbedding(text: string): Promise<number[]>;
......
......@@ -19,7 +19,7 @@ export function getNodesFromDocument(
const textSplits = getTextSplitsFromDocument(document, textSplitter);
textSplits.forEach((textSplit, index) => {
textSplits.forEach((textSplit) => {
const node = new TextNode({ text: textSplit });
node.relationships[NodeRelationship.SOURCE] = document.asRelatedNodeInfo();
nodes.push(node);
......
......@@ -182,6 +182,7 @@ export function getResponseBuilder(): BaseResponseBuilder {
return new SimpleResponseBuilder();
}
// TODO replace with Logan's new response_sythesizers/factory.py
export class ResponseSynthesizer {
responseBuilder: BaseResponseBuilder;
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment