cleanup code

src/lib/components/inference-playground/provider-select.svelte

@@ -1,7 +1,7 @@
 <script lang="ts">
 	import type { ConversationClass } from "$lib/state/conversations.svelte";
 	import { pricing } from "$lib/state/pricing.svelte";
-	import type { Model } from "$lib/types.js";
+	import type { AutoPolicy, Model } from "$lib/types.js";
 	import { cn } from "$lib/utils/cn.js";
 	import { Select } from "melt/builders";
 	import { run } from "svelte/legacy";
@@ -36,13 +36,6 @@
 		},
 	});
 
-	const autoPolicySelect = new Select<"default" | "fastest" | "cheapest", false>({
-		value: () => conversation.data.autoPolicy ?? "default",
-		onValueChange(v) {
-			conversation.update({ autoPolicy: v });
-		},
-	});
-
 	const nameMap: Record<string, string> = {
 		"sambanova": "SambaNova",
 		"fal": "fal",
@@ -87,27 +80,26 @@
 		return pricing.formatPricing(pd);
 	}
 
-	function getAutoPolicyLabel(policy: "default" | "fastest" | "cheapest") {
-		switch (policy) {
-			case "default":
-				return "Default";
-			case "fastest":
-				return "Fastest";
-			case "cheapest":
-				return "Cheapest";
-		}
-	}
+	const autoPolicyLabels: Record<AutoPolicy, string> = {
+		default: "Default",
+		fastest: "Fastest",
+		cheapest: "Cheapest",
+	};
 
-	function getAutoPolicyDescription(policy: "default" | "fastest" | "cheapest") {
-		switch (policy) {
-			case "default":
-				return "Uses your preference order from Inference Provider settings";
-			case "fastest":
-				return "Selects the provider with highest throughput";
-			case "cheapest":
-				return "Selects the provider with lowest price per output token";
-		}
-	}
+	const autoPolicyDescriptions: Record<AutoPolicy, string> = {
+		default: "Uses your preference order from Inference Provider settings",
+		fastest: "Selects the provider with highest throughput",
+		cheapest: "Selects the provider with lowest price per output token",
+	};
+
+	const autoPolicyValue = $derived(conversation.data.autoPolicy ?? "default");
+
+	const autoPolicySelect = new Select<AutoPolicy, false>({
+		value: () => autoPolicyValue,
+		onValueChange(v) {
+			conversation.update({ autoPolicy: v });
+		},
+	});
 </script>
 
 {#snippet providerDisplay(provider: string)}
@@ -168,7 +160,7 @@
 							<IconInfo class="size-3" />
 						</button>
 					{/snippet}
-					{getAutoPolicyDescription(conversation.data.autoPolicy ?? "default")}
+					{autoPolicyDescriptions[autoPolicyValue]}
 				</Tooltip>
 			</div>
 			<button
@@ -178,7 +170,7 @@
 					"hover:brightness-95 dark:border-gray-700 dark:bg-gray-800 dark:hover:brightness-110",
 				)}
 			>
-				{getAutoPolicyLabel(conversation.data.autoPolicy ?? "default")}
+				{autoPolicyLabels[autoPolicyValue]}
 				<div
 					class="absolute right-2 grid size-4 flex-none place-items-center rounded-sm bg-gray-100 text-xs dark:bg-gray-600"
 				>
@@ -195,7 +187,7 @@
 							<div class="flex flex-col items-start gap-0.5">
 								<span>{label}</span>
 								<span class="text-xs text-gray-500 dark:text-gray-400">
-									{getAutoPolicyDescription(policy)}
+									{autoPolicyDescriptions[policy]}
 								</span>
 							</div>
 						</div>

src/lib/state/conversations.svelte.ts

@@ -4,7 +4,13 @@ import {
 } from "$lib/components/inference-playground/generation-config-settings.js";
 import { addToast } from "$lib/components/toaster.svelte.js";
 import { AbortManager } from "$lib/spells/abort-manager.svelte";
-import { PipelineTag, type ConversationMessage, type GenerationStatistics, type Model } from "$lib/types.js";
+import {
+	PipelineTag,
+	type AutoPolicy,
+	type ConversationMessage,
+	type GenerationStatistics,
+	type Model,
+} from "$lib/types.js";
 import { handleNonStreamingResponse, handleStreamingResponse, estimateTokens } from "$lib/utils/business.svelte.js";
 import { omit, snapshot } from "$lib/utils/object.svelte";
 import { models } from "./models.svelte";
@@ -52,7 +58,7 @@ export class ConversationEntity {
 	provider?: string;
 
 	@Fields.string()
-	autoPolicy?: "default" | "fastest" | "cheapest";
+	autoPolicy?: AutoPolicy;
 
 	@Fields.string()
 	projectId!: string;

src/lib/types.ts

@@ -232,3 +232,5 @@ export interface SearchContextCostPerQuery {
 	search_context_size_medium: number;
 	search_context_size_high: number;
 }
+
+export type AutoPolicy = "default" | "fastest" | "cheapest";