Spaces:
Running
Running
File size: 2,745 Bytes
fc69895 |
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 78 79 80 81 82 83 84 |
import { collections } from "$lib/server/database";
import { MetricsServer } from "$lib/server/metrics";
import { error } from "@sveltejs/kit";
import { ObjectId } from "mongodb";
import { authCondition } from "$lib/server/auth";
/**
* Create a new conversation from a shared conversation ID.
* If the conversation already exists for the user/session, return the existing conversation ID.
* returns the conversation ID.
*/
export async function createConversationFromShare(
fromShareId: string,
locals: App.Locals,
userAgent?: string
): Promise<string> {
const conversation = await collections.sharedConversations.findOne({
_id: fromShareId,
});
if (!conversation) {
error(404, "Conversation not found");
}
// Check if shared conversation exists already for this user/session
const existingConversation = await collections.conversations.findOne({
"meta.fromShareId": fromShareId,
...authCondition(locals),
});
if (existingConversation) {
return existingConversation._id.toString();
}
// Create new conversation from shared conversation
const res = await collections.conversations.insertOne({
_id: new ObjectId(),
title: conversation.title.replace(/<\/?think>/gi, "").trim(),
rootMessageId: conversation.rootMessageId,
messages: conversation.messages,
model: conversation.model,
preprompt: conversation.preprompt,
createdAt: new Date(),
updatedAt: new Date(),
userAgent,
...(locals.user ? { userId: locals.user._id } : { sessionId: locals.sessionId }),
meta: { fromShareId },
});
// Copy files from shared conversation bucket entries to the new conversation
// Shared files are stored with filenames "${sharedId}-${sha}" and metadata.conversation = sharedId
// New conversation expects files to be stored under its own id prefix
const newConvId = res.insertedId.toString();
const sharedId = fromShareId;
const files = await collections.bucket.find({ filename: { $regex: `^${sharedId}-` } }).toArray();
await Promise.all(
files.map(
(file) =>
new Promise<void>((resolve, reject) => {
try {
const newFilename = file.filename.replace(`${sharedId}-`, `${newConvId}-`);
const downloadStream = collections.bucket.openDownloadStream(file._id);
const uploadStream = collections.bucket.openUploadStream(newFilename, {
metadata: { ...file.metadata, conversation: newConvId },
});
downloadStream
.on("error", reject)
.pipe(uploadStream)
.on("error", reject)
.on("finish", () => resolve());
} catch (e) {
reject(e);
}
})
)
);
if (MetricsServer.isEnabled()) {
MetricsServer.getMetrics().model.conversationsTotal.inc({ model: conversation.model });
}
return res.insertedId.toString();
}
|