Add ability to define custom model/dataset URLs (#347)

* Add ability to define custom model/dataset URLs

* lint

---------

Co-authored-by: Nathan Sarrazin <[email protected]>

src/lib/components/ModelCardMetadata.svelte

@@ -3,7 +3,7 @@
 	import CarbonArrowUpRight from "~icons/carbon/arrow-up-right";
 	import type { Model } from "$lib/types/Model";
 
-	export let model: Pick<Model, "name" | "datasetName" | "websiteUrl">;
+	export let model: Pick<Model, "name" | "datasetName" | "websiteUrl" | "modelUrl" | "datasetUrl">;
 
 	export let variant: "light" | "dark" = "light";
 </script>
@@ -15,7 +15,7 @@
 		: 'text-gray-800 dark:bg-gray-100 dark:text-gray-600'}"
 >
 	<a
-		href="https://huggingface.co/{model.name}"
+		href={model.modelUrl || "https://huggingface.co/" + model.name}
 		target="_blank"
 		rel="noreferrer"
 		class="flex items-center hover:underline"
@@ -23,9 +23,9 @@
 		Model
 		<div class="max-sm:hidden">&nbsp;page</div></a
 	>
-	{#if model.datasetName}
+	{#if model.datasetName || model.datasetUrl}
 		<a
-			href="https://huggingface.co/datasets/{model.datasetName}"
+			href={model.datasetUrl || "https://huggingface.co/datasets/" + model.datasetName}
 			target="_blank"
 			rel="noreferrer"
 			class="flex items-center hover:underline"

src/lib/server/models.ts

@@ -11,7 +11,9 @@ const modelsRaw = z
 			displayName: z.string().min(1).optional(),
 			description: z.string().min(1).optional(),
 			websiteUrl: z.string().url().optional(),
+			modelUrl: z.string().url().optional(),
 			datasetName: z.string().min(1).optional(),
+			datasetUrl: z.string().url().optional(),
 			userMessageToken: z.string().min(1),
 			assistantMessageToken: z.string().min(1),
 			messageEndToken: z.string().min(1).optional(),

src/lib/types/Model.ts

@@ -10,4 +10,6 @@ export type Model = Pick<
 	| "promptExamples"
 	| "parameters"
 	| "description"
+	| "modelUrl"
+	| "datasetUrl"
 >;

src/routes/+layout.server.ts

@@ -67,7 +67,9 @@ export const load: LayoutServerLoad = async ({ locals, depends, url }) => {
 			id: model.id,
 			name: model.name,
 			websiteUrl: model.websiteUrl,
+			modelUrl: model.modelUrl,
 			datasetName: model.datasetName,
+			datasetUrl: model.datasetUrl,
 			displayName: model.displayName,
 			description: model.description,
 			promptExamples: model.promptExamples,