File size: 2,119 Bytes
5da61b4
 
 
0bfcf81
5da61b4
0bfcf81
 
b7b2c8c
1b66f8d
2e28042
 
0bfcf81
 
 
 
06e879d
 
 
cf7ac8d
af58e08
06e879d
 
0bfcf81
af58e08
 
06e879d
 
 
 
0bfcf81
06e879d
 
0bfcf81
06e879d
 
 
 
af58e08
0bfcf81
 
af58e08
41f29a4
 
 
 
 
 
 
 
 
e6e851b
 
af58e08
1b66f8d
 
f249cfc
0bfcf81
06e879d
e6e851b
1b66f8d
 
64d3841
06e879d
1b66f8d
 
 
 
5da61b4
1b66f8d
5da61b4
1b66f8d
 
7b62aec
 
767afa7
7b62aec
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
import type { RequestHandler } from "./$types";
import { collections } from "$lib/server/database";
import { ObjectId } from "mongodb";
import { error, redirect } from "@sveltejs/kit";
import { base } from "$app/paths";
import { z } from "zod";
import type { Message } from "$lib/types/Message";
import { models, validateModel } from "$lib/server/models";

export const POST: RequestHandler = async ({ locals, request }) => {
	const body = await request.text();

	let title = "";
	let messages: Message[] = [];

	const values = z
		.object({
			fromShare: z.string().optional(),
			model: validateModel(models),
			preprompt: z.string().optional(),
		})
		.parse(JSON.parse(body));

	let preprompt = values.preprompt;

	if (values.fromShare) {
		const conversation = await collections.sharedConversations.findOne({
			_id: values.fromShare,
		});

		if (!conversation) {
			throw error(404, "Conversation not found");
		}

		title = conversation.title;
		messages = conversation.messages;
		values.model = conversation.model;
		preprompt = conversation.preprompt;
	}

	const model = models.find((m) => m.name === values.model);

	if (!model) {
		throw error(400, "Invalid model");
	}

	if (model.unlisted) {
		throw error(400, "Can't start a conversation with an unlisted model");
	}

	// Use the model preprompt if there is no conversation/preprompt in the request body
	preprompt = preprompt === undefined ? model?.preprompt : preprompt;

	const res = await collections.conversations.insertOne({
		_id: new ObjectId(),
		title: title || "New Chat",
		messages,
		model: values.model,
		preprompt: preprompt === model?.preprompt ? model?.preprompt : preprompt,
		createdAt: new Date(),
		updatedAt: new Date(),
		...(locals.user ? { userId: locals.user._id } : { sessionId: locals.sessionId }),
		...(values.fromShare ? { meta: { fromShareId: values.fromShare } } : {}),
	});

	return new Response(
		JSON.stringify({
			conversationId: res.insertedId.toString(),
		}),
		{ headers: { "Content-Type": "application/json" } }
	);
};

export const GET: RequestHandler = async () => {
	throw redirect(302, `${base}/`);
};