Initial commit

src/utils/computeSimilarity.ts

@@ -0,0 +1,14 @@
+import dot from 'compute-dot';
+import cosineSimilarity from 'compute-cosine-similarity';
+
+const computeSimilarity = (x: number[], y: number[]): number => {
+  if (process.env.SIMILARITY_MEASURE === 'cosine') {
+    return cosineSimilarity(x, y);
+  } else if (process.env.SIMILARITY_MEASURE === 'dot') {
+    return dot(x, y);
+  }
+
+  throw new Error('Invalid similarity measure');
+};
+
+export default computeSimilarity;

src/utils/formatHistory.ts

@@ -0,0 +1,9 @@
+import { BaseMessage } from '@langchain/core/messages';
+
+const formatChatHistoryAsString = (history: BaseMessage[]) => {
+  return history
+    .map((message) => `${message._getType()}: ${message.content}`)
+    .join('\n');
+};
+
+export default formatChatHistoryAsString;