feat(routes): update routes to handle new llm types

This commit is contained in:
ItzCrazyKns
2025-11-23 19:24:17 +05:30
parent 6da6acbcd0
commit e22a39fd73
4 changed files with 39 additions and 53 deletions

View File

@@ -7,6 +7,7 @@ import { DocxLoader } from '@langchain/community/document_loaders/fs/docx';
import { RecursiveCharacterTextSplitter } from '@langchain/textsplitters';
import { Document } from '@langchain/core/documents';
import ModelRegistry from '@/lib/models/registry';
import { Chunk } from '@/lib/types';
interface FileRes {
fileName: string;
@@ -87,9 +88,17 @@ export async function POST(req: Request) {
}),
);
const embeddings = await model.embedDocuments(
splitted.map((doc) => doc.pageContent),
const chunks: Chunk[] = splitted.map((doc) => {
return {
content: doc.pageContent,
metadata: doc.metadata,
}
});
const embeddings = await model.embedChunks(
chunks
);
const embeddingsDataPath = filePath.replace(
/\.\w+$/,
'-embeddings.json',