Skip to content

Commit cd72743

Browse files
committed
set/get model from query params
1 parent 67f0ff8 commit cd72743

File tree

2 files changed

+12
-1
lines changed

2 files changed

+12
-1
lines changed

src/lib/components/InferencePlayground/InferencePlayground.svelte

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
import type { ModelEntryWithTokenizer } from "./types";
33
import { type ChatCompletionInputMessage } from "@huggingface/tasks";
44
5+
import { page } from "$app/stores";
56
import { defaultGenerationConfig } from "./generationConfigSettings";
67
import {
78
createHfInference,
@@ -24,8 +25,11 @@
2425
const startMessageUser: ChatCompletionInputMessage = { role: "user", content: "" };
2526
const startMessageSystem: ChatCompletionInputMessage = { role: "system", content: "" };
2627
28+
const modelIdFromQueryParam = $page.url.searchParams.get("modelId");
29+
const modelFromQueryParam = models.find(model => model.id === modelIdFromQueryParam);
30+
2731
let conversation: Conversation = {
28-
model: models.find(m => FEATUED_MODELS_IDS.includes(m.id)) ?? models[0],
32+
model: modelFromQueryParam ?? models.find(m => FEATUED_MODELS_IDS.includes(m.id)) ?? models[0],
2933
config: defaultGenerationConfig,
3034
messages: [{ ...startMessageUser }],
3135
systemMessage: startMessageSystem,

src/lib/components/InferencePlayground/InferencePlaygroundModelSelector.svelte

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,9 @@
11
<script lang="ts">
22
import type { Conversation, ModelEntryWithTokenizer } from "./types";
33
4+
import { goto } from "$app/navigation";
5+
import { page } from "$app/stores";
6+
47
import IconCaret from "../Icons/IconCaret.svelte";
58
import ModelSelectorModal from "./InferencePlaygroundModelSelectorModal.svelte";
69
@@ -27,6 +30,10 @@
2730
return;
2831
}
2932
conversation.model = model;
33+
34+
const url = new URL($page.url);
35+
url.searchParams.set("modelId", model.id);
36+
goto(url.toString(), { replaceState: true });
3037
}
3138
3239
$: [nameSpace, modelName] = conversation.model.id.split("/");

0 commit comments

Comments
 (0)