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( ...@@ -174,24 +174,12 @@ export function getTopKMMREmbeddings(
} }
export abstract class BaseEmbedding { export abstract class BaseEmbedding {
static similarity( similarity(
embedding1: number[], embedding1: number[],
embedding2: number[], embedding2: number[],
mode: SimilarityType = SimilarityType.DOT_PRODUCT mode: SimilarityType = SimilarityType.DEFAULT
): number { ): number {
if (embedding1.length !== embedding2.length) { return similarity(embedding1, embedding2, mode);
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");
}
} }
abstract aGetTextEmbedding(text: string): Promise<number[]>; abstract aGetTextEmbedding(text: string): Promise<number[]>;
......
...@@ -19,7 +19,7 @@ export function getNodesFromDocument( ...@@ -19,7 +19,7 @@ export function getNodesFromDocument(
const textSplits = getTextSplitsFromDocument(document, textSplitter); const textSplits = getTextSplitsFromDocument(document, textSplitter);
textSplits.forEach((textSplit, index) => { textSplits.forEach((textSplit) => {
const node = new TextNode({ text: textSplit }); const node = new TextNode({ text: textSplit });
node.relationships[NodeRelationship.SOURCE] = document.asRelatedNodeInfo(); node.relationships[NodeRelationship.SOURCE] = document.asRelatedNodeInfo();
nodes.push(node); nodes.push(node);
......
...@@ -182,6 +182,7 @@ export function getResponseBuilder(): BaseResponseBuilder { ...@@ -182,6 +182,7 @@ export function getResponseBuilder(): BaseResponseBuilder {
return new SimpleResponseBuilder(); return new SimpleResponseBuilder();
} }
// TODO replace with Logan's new response_sythesizers/factory.py
export class ResponseSynthesizer { export class ResponseSynthesizer {
responseBuilder: BaseResponseBuilder; responseBuilder: BaseResponseBuilder;
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment