feat(models): add transferTo property on old models (#1453)

* feat(models): add transferTo property on old models

used to transfer assistants to new models if specified

* make sure transferTo is an existing model

Author: nsarrazin

src/lib/migrations/routines/02-update-assistants-models.ts

@@ -7,17 +7,36 @@ const updateAssistantsModels: Migration = {
 	name: "Update deprecated models in assistants with the default model",
 	up: async () => {
 		const models = (await import("$lib/server/models")).models;
-
+		const oldModels = (await import("$lib/server/models")).oldModels;
 		const { assistants } = collections;
 
-		const modelIds = models.map((el) => el.id); // string[]
+		const modelIds = models.map((el) => el.id);
 		const defaultModelId = models[0].id;
 
-		// Find all assistants whose modelId is not in modelIds, and update it to use defaultModelId
-		await assistants.updateMany(
-			{ modelId: { $nin: modelIds } },
-			{ $set: { modelId: defaultModelId } }
-		);
+		// Find all assistants whose modelId is not in modelIds, and update it
+		const bulkOps = await assistants
+			.find({ modelId: { $nin: modelIds } })
+			.map((assistant) => {
+				// has an old model
+				let newModelId = defaultModelId;
+
+				const oldModel = oldModels.find((m) => m.id === assistant.modelId);
+				if (oldModel && oldModel.transferTo && !!models.find((m) => m.id === oldModel.transferTo)) {
+					newModelId = oldModel.transferTo;
+				}
+
+				return {
+					updateOne: {
+						filter: { _id: assistant._id },
+						update: { $set: { modelId: newModelId } },
+					},
+				};
+			})
+			.toArray();
+
+		if (bulkOps.length > 0) {
+			await assistants.bulkWrite(bulkOps);
+		}
 
 		return true;
 	},

src/lib/server/models.ts

@@ -320,6 +320,9 @@ export const models: ProcessedModel[] = await Promise.all(
 	modelsRaw.map((e) => processModel(e).then(addEndpoint))
 );
 
+// super ugly but not sure how to make typescript happier
+const validModelIdSchema = z.enum(models.map((m) => m.id) as [string, ...string[]]);
+
 export const defaultModel = models[0];
 
 // Models that have been deprecated
@@ -330,6 +333,7 @@ export const oldModels = env.OLD_MODELS
 					id: z.string().optional(),
 					name: z.string().min(1),
 					displayName: z.string().min(1).optional(),
+					transferTo: validModelIdSchema.optional(),
 				})
 			)
 			.parse(JSON5.parse(env.OLD_MODELS))