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

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

@@ -70,6 +70,7 @@ export async function POST({ request, fetch, locals, params, getClientAddress })
 
 	// fetch the model
 	const model = models.find((m) => m.id === conv.model);
+	const settings = await collections.settings.findOne(authCondition(locals));
 
 	if (!model) {
 		throw error(410, "Model not available anymore");
@@ -84,6 +85,7 @@ export async function POST({ request, fetch, locals, params, getClientAddress })
 		id: messageId,
 		is_retry,
 		web_search: webSearch,
+		domainFilters: domainFilters,
 	} = z
 		.object({
 			inputs: z.string().trim().min(1),
@@ -91,6 +93,7 @@ export async function POST({ request, fetch, locals, params, getClientAddress })
 			response_id: z.optional(z.string().uuid()),
 			is_retry: z.optional(z.boolean()),
 			web_search: z.optional(z.boolean()),
+			domainFilters: z.optional(z.string()),
 		})
 		.parse(json);
 
@@ -121,27 +124,6 @@ export async function POST({ request, fetch, locals, params, getClientAddress })
 		];
 	})() satisfies Message[];
 
-	if (conv.title.startsWith("Untitled")) {
-		try {
-			conv.title = (await summarize(newPrompt)) ?? conv.title;
-		} catch (e) {
-			console.error(e);
-		}
-	}
-
-	await collections.conversations.updateOne(
-		{
-			_id: convId,
-		},
-		{
-			$set: {
-				messages,
-				title: conv.title,
-				updatedAt: new Date(),
-			},
-		}
-	);
-
 	// we now build the stream
 	const stream = new ReadableStream({
 		async start(controller) {
@@ -159,7 +141,7 @@ export async function POST({ request, fetch, locals, params, getClientAddress })
 			let webSearchResults: WebSearch | undefined;
 
 			if (webSearch) {
-				webSearchResults = await runWebSearch(conv, newPrompt, update);
+				webSearchResults = await runWebSearch(conv, newPrompt, update, domainFilters);
 			}
 
 			// we can now build the prompt using the messages
@@ -167,7 +149,7 @@ export async function POST({ request, fetch, locals, params, getClientAddress })
 				messages,
 				model,
 				webSearch: webSearchResults,
-				preprompt: conv.preprompt ?? model.preprompt,
+				preprompt: settings?.customPrompts?.[model.id] ?? model.preprompt,
 				locals: locals,
 			});
 
@@ -189,7 +171,7 @@ export async function POST({ request, fetch, locals, params, getClientAddress })
 
 			async function saveLast(generated_text: string) {
 				if (!conv) {
-					throw error(404, "Conversation not found");
+					throw new Error("Conversation not found");
 				}
 
 				const lastMessage = messages[messages.length - 1];
@@ -220,7 +202,7 @@ export async function POST({ request, fetch, locals, params, getClientAddress })
 						{
 							$set: {
 								messages,
-								title: conv.title,
+								title: (await summarize(newPrompt)) ?? conv.title,
 								updatedAt: new Date(),
 							},
 						}
@@ -303,7 +285,7 @@ export async function POST({ request, fetch, locals, params, getClientAddress })
 				{
 					$set: {
 						messages,
-						title: conv.title,
+						title: (await summarize(newPrompt)) ?? conv.title,
 						updatedAt: new Date(),
 					},
 				}
@@ -360,4 +342,4 @@ export async function PATCH({ request, locals, params }) {
 	);
 
 	return new Response();
-}
+}