make models description dynamic (#169)

Co-authored-by: Eliott C <[email protected]>

.env

@@ -10,6 +10,7 @@ MODELS=`[
   {
     "name": "OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5",
     "datasetName": "OpenAssistant/oasst1",
+    "description": "A good alternative to ChatGPT",
     "websiteUrl": "https://open-assistant.io",
     "userMessageToken": "<|prompter|>",
     "assistantMessageToken": "<|assistant|>",
@@ -40,6 +41,7 @@ MODELS=`[
     "name":"bigcode/starcoder",
     "displayName":"BigCode/StarCoder",
     "datasetName":"bigcode/the-stack-dedup",
+    "description": "A good model for answering technical questions",
     "websiteUrl":"https://www.bigcode-project.org/",
     "prepromptUrl": "https://huggingface.co/datasets/coyotte508/bigcodeprompt/raw/main/prompt.txt",
     "promptExamples": [

src/lib/components/ModelsModal.svelte

@@ -56,7 +56,9 @@
 							<span class="text-md block font-semibold leading-tight text-gray-800"
 								>{model.displayName}</span
 							>
-							<span class="text-xs text-[#9FA8B5]">A good alternative to ChatGPT</span>
+							{#if model.description}
+								<span class="text-xs text-[#9FA8B5]">{model.description}</span>
+							{/if}
 						</span>
 						<CarbonCheckmark
 							class="-mr-1 -mt-1 ml-auto shrink-0 text-xl {model.name === selectedModelName

src/lib/server/models.ts

@@ -6,6 +6,7 @@ const modelsRaw = z
 		z.object({
 			name: z.string().min(1),
 			displayName: z.string().min(1).optional(),
+			description: z.string().min(1).optional(),
 			websiteUrl: z.string().url().optional(),
 			datasetName: z.string().min(1).optional(),
 			userMessageToken: z.string().min(1),
@@ -47,7 +48,6 @@ export const models = await Promise.all(
 		...m,
 		displayName: m.displayName || m.name,
 		preprompt: m.prepromptUrl ? await fetch(m.prepromptUrl).then((r) => r.text()) : m.preprompt,
-		promptExamples: m.promptExamples || [],
 	}))
 );
 

src/lib/types/Model.ts

@@ -2,5 +2,11 @@ import type { BackendModel } from "$lib/server/models";
 
 export type Model = Pick<
 	BackendModel,
-	"name" | "displayName" | "websiteUrl" | "datasetName" | "promptExamples" | "parameters"
+	| "name"
+	| "displayName"
+	| "websiteUrl"
+	| "datasetName"
+	| "promptExamples"
+	| "parameters"
+	| "description"
 >;

src/routes/+layout.server.ts

@@ -40,6 +40,7 @@ export const load: LayoutServerLoad = async ({ locals, depends, url }) => {
 			websiteUrl: model.websiteUrl,
 			datasetName: model.datasetName,
 			displayName: model.displayName,
+			description: model.description,
 			promptExamples: model.promptExamples,
 			parameters: model.parameters,
 		})),