prettier

Author: mishig25

src/lib/buildPrompt.ts

@@ -50,7 +50,7 @@ export async function buildPrompt({
 				`,
 			},
 		];
-	}else if (pdfSearch && pdfSearch.context) {
+	} else if (pdfSearch && pdfSearch.context) {
 		const lastMsg = messages.slice(-1)[0];
 		const messagesWithoutLastUsrMsg = messages.slice(0, -1);
 		const previousUserMessages = messages.filter((el) => el.from === "user").slice(0, -1);
@@ -61,7 +61,7 @@ export async function buildPrompt({
 						.map(({ content }) => `- ${content}`)
 						.join("\n")}`
 				: "";
-		
+
 		messages = [
 			...messagesWithoutLastUsrMsg,
 			{

src/lib/components/UploadBtn.svelte

@@ -1,21 +1,21 @@
 <script lang="ts">
 	import { PdfUploadStatus } from "$lib/types/PdfChat";
-	import {createEventDispatcher, onDestroy} from "svelte";
+	import { createEventDispatcher, onDestroy } from "svelte";
 	import CarbonUpload from "~icons/carbon/upload";
 
 	export let classNames = "";
 	export let multimodal = false;
 	export let files: File[];
 	export let uploadPdfStatus: PdfUploadStatus;
-	
+
 	const accept = multimodal ? "image/*,.pdf" : ".pdf";
 	const label = multimodal ? "Upload image or PDF" : "Upload PDF";
 	let fileInput: HTMLInputElement;
 	let interval: ReturnType<typeof setInterval>;
 
 	$: uploading = uploadPdfStatus === PdfUploadStatus.Uploading;
-	$:{
-		if(uploadPdfStatus === PdfUploadStatus.Uploaded){
+	$: {
+		if (uploadPdfStatus === PdfUploadStatus.Uploaded) {
 			interval = setInterval(() => {
 				uploadPdfStatus = PdfUploadStatus.Ready;
 			}, 1500);
@@ -27,25 +27,25 @@
 	}>();
 
 	function onChange() {
-		if(!fileInput.files){
+		if (!fileInput.files) {
 			return;
 		}
-		
+
 		const file = fileInput.files?.[0];
 		if (file?.type === "application/pdf") {
 			// pdf upload
 			dispatch("uploadpdf", file);
-		}else{
+		} else {
 			// image files for multimodal models
 			files = Array.from(fileInput.files);
 		}
 	}
 
 	onDestroy(() => {
-		if(interval){
+		if (interval) {
 			clearInterval(interval);
 		}
-	})
+	});
 </script>
 
 <button

src/lib/components/chat/ChatMessage.svelte

@@ -109,14 +109,14 @@
 
 	let searchUpdates: WebSearchUpdate[] = [];
 
-	$: searchUpdates = ((RAGMessages.filter(({type}) => type === "webSearch").length > 0
-		? RAGMessages.filter(({type}) => type === "webSearch")
+	$: searchUpdates = ((RAGMessages.filter(({ type }) => type === "webSearch").length > 0
+		? RAGMessages.filter(({ type }) => type === "webSearch")
 		: message.updates?.filter(({ type }) => type === "webSearch")) ?? []) as WebSearchUpdate[];
-	
+
 	let pdfUpdates: PdfSearchUpdate[] = [];
 
-	$: pdfUpdates = ((RAGMessages.filter(({type}) => type === "pdfSearch").length > 0
-		? RAGMessages.filter(({type}) => type === "pdfSearch")
+	$: pdfUpdates = ((RAGMessages.filter(({ type }) => type === "pdfSearch").length > 0
+		? RAGMessages.filter(({ type }) => type === "pdfSearch")
 		: message.updates?.filter(({ type }) => type === "pdfSearch")) ?? []) as PdfSearchUpdate[];
 
 	$: downloadLink =

src/lib/components/chat/ChatWindow.svelte

@@ -155,7 +155,13 @@
 							})}
 					/>
 				{/if}
-				<UploadBtn bind:files on:uploadpdf classNames="ml-auto" multimodal={currentModel.multimodal} {uploadPdfStatus} />
+				<UploadBtn
+					bind:files
+					on:uploadpdf
+					classNames="ml-auto"
+					multimodal={currentModel.multimodal}
+					{uploadPdfStatus}
+				/>
 			</div>
 			<form
 				on:dragover={onDragOver}

src/lib/server/embeddings.ts

@@ -48,4 +48,4 @@ export function findSimilarSentences(
 
 	// Return the indexes of the closest topK sentences
 	return distancesFromQuery.slice(0, topK).map((item) => item.index);
-}
+}

src/lib/server/files/downloadFile.ts

@@ -40,7 +40,7 @@ export async function downloadPdfEmbeddings(
 ) {
 	const fileId = collections.bucket.find({ filename: `${convId.toString()}-pdf` });
 	let textChunks: string[] = [];
-	let dims: number[] = []
+	let dims: number[] = [];
 
 	const content = await fileId.next().then(async (file) => {
 		if (!file) {

src/lib/server/files/uploadFile.ts

@@ -21,7 +21,11 @@ export async function uploadImgFile(file: Blob, conv: Conversation): Promise<str
 	});
 }
 
-export async function uploadPdfEmbeddings(embeddings: Tensor, textChunks: string[], conv: Conversation): Promise<void> {
+export async function uploadPdfEmbeddings(
+	embeddings: Tensor,
+	textChunks: string[],
+	conv: Conversation
+): Promise<void> {
 	const filename = `${conv._id}-pdf`;
 
 	// Step 1: Check if the file exists

src/lib/server/pdfSearch.ts

@@ -1,8 +1,5 @@
 import type { PdfSearch } from "$lib/types/PdfChat";
-import {
-	createEmbeddings,
-	findSimilarSentences,
-} from "$lib/server/embeddings";
+import { createEmbeddings, findSimilarSentences } from "$lib/server/embeddings";
 import type { Conversation } from "$lib/types/Conversation";
 import type { MessageUpdate } from "$lib/types/MessageUpdate";
 import { downloadPdfEmbeddings } from "./files/downloadFile";
@@ -29,11 +26,15 @@ export async function runPdfSearch(
 		const { content, textChunks, dims } = await downloadPdfEmbeddings(conv._id);
 		// reconstruct pdfEmbeddings
 		const buffer = Buffer.from(content);
-		const data = new Float32Array(buffer.buffer, buffer.byteOffset, buffer.length / Float32Array.BYTES_PER_ELEMENT);
-		const pdfEmbeddings = new Tensor('float32', data, dims);
+		const data = new Float32Array(
+			buffer.buffer,
+			buffer.byteOffset,
+			buffer.length / Float32Array.BYTES_PER_ELEMENT
+		);
+		const pdfEmbeddings = new Tensor("float32", data, dims);
 		const promptEmbeddings = await createEmbeddings([prompt]);
 
-		const indices = findSimilarSentences(pdfEmbeddings, promptEmbeddings, {topK: 5});
+		const indices = findSimilarSentences(pdfEmbeddings, promptEmbeddings, { topK: 5 });
 		pdfSearch.context = indices.map((idx) => textChunks[idx]).join(" ");
 
 		appendUpdate("Done", [], "done");

src/routes/conversation/[id]/+page.svelte

@@ -278,7 +278,7 @@
 		uploadPdfStatus = PdfUploadStatus.Uploading;
 
 		const formData = new FormData();
-        formData.append('pdf', file);
+		formData.append("pdf", file);
 
 		const res = await fetch(`${base}/conversation/${$page.params.id}/upload-pdf`, {
 			method: "POST",
@@ -297,10 +297,10 @@
 		// only used in case of creating new conversations (from the parent POST endpoint)
 		if ($pendingMessage) {
 			files = $pendingMessage.files;
-			if($pendingMessage.content){
+			if ($pendingMessage.content) {
 				await writeMessage($pendingMessage.content);
 			}
-			if($pendingMessage.pdfFile){
+			if ($pendingMessage.pdfFile) {
 				await uploadPdf($pendingMessage.pdfFile);
 			}
 			$pendingMessage = undefined;
@@ -353,7 +353,7 @@
 	{messages}
 	shared={data.shared}
 	preprompt={data.preprompt}
-	bind:RAGMessages={RAGMessages}
+	bind:RAGMessages
 	bind:files
 	on:message={onMessage}
 	on:retry={onRetry}

src/routes/conversation/[id]/+server.ts

@@ -244,7 +244,7 @@ export async function POST({ request, locals, params, getClientAddress }) {
 
 			let pdfSearchResults: PdfSearch | undefined;
 			const pdfSearch = await collections.files.findOne({ filename: `${convId.toString()}-pdf` });
-			if(pdfSearch){
+			if (pdfSearch) {
 				pdfSearchResults = await runPdfSearch(conv, newPrompt, update);
 			}