This commit is contained in:
Jin Yucong 2024-07-05 14:36:50 +08:00
parent 5b1aaee605
commit 3b737a078a
63 changed files with 1132 additions and 1853 deletions

View file

@ -1,17 +1,17 @@
import dot from 'compute-dot';
import cosineSimilarity from 'compute-cosine-similarity';
import { getSimilarityMeasure } from '../config';
import dot from "compute-dot";
import cosineSimilarity from "compute-cosine-similarity";
import { getSimilarityMeasure } from "../config";
const computeSimilarity = (x: number[], y: number[]): number => {
const similarityMeasure = getSimilarityMeasure();
if (similarityMeasure === 'cosine') {
if (similarityMeasure === "cosine") {
return cosineSimilarity(x, y);
} else if (similarityMeasure === 'dot') {
} else if (similarityMeasure === "dot") {
return dot(x, y);
}
throw new Error('Invalid similarity measure');
throw new Error("Invalid similarity measure");
};
export default computeSimilarity;

View file

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

View file

@ -1,20 +1,14 @@
import winston from 'winston';
import winston from "winston";
const logger = winston.createLogger({
level: 'info',
level: "info",
transports: [
new winston.transports.Console({
format: winston.format.combine(
winston.format.colorize(),
winston.format.simple(),
),
format: winston.format.combine(winston.format.colorize(), winston.format.simple()),
}),
new winston.transports.File({
filename: 'app.log',
format: winston.format.combine(
winston.format.timestamp(),
winston.format.json(),
),
filename: "app.log",
format: winston.format.combine(winston.format.timestamp(), winston.format.json()),
}),
],
});