Spaces:

huggingchat
/

chat-ui

Paused

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

nsarrazin HF staff

Update models and add check for assistants model on startup (#998)

29b7d2a unverified 3 months ago

No virus

848 Bytes

	import type { Migration } from ".";
	import { getCollections } from "$lib/server/database";
	import { ObjectId } from "mongodb";

	const updateAssistantsModels: Migration = {
	_id: new ObjectId("5f9f3f3f3f3f3f3f3f3f3f3f"),
	name: "Update deprecated models in assistants with the default model",
	up: async (client) => {
	const models = (await import("$lib/server/models")).models;

	const { assistants } = getCollections(client);

	const modelIds = models.map((el) => el.id); // string[]
	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 } }
	);

	return true;
	},
	runEveryTime: true,
	runForHuggingChat: "only",
	};

	export default updateAssistantsModels;