Skip to content

Commit f9fa0a5

Browse files
committed
routing mode for snippets
1 parent 491aab8 commit f9fa0a5

File tree

2 files changed

+8
-7
lines changed

2 files changed

+8
-7
lines changed

src/lib/components/inference-playground/code-snippets.svelte

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@
4040
let showToken = $state(false);
4141
4242
type GetSnippetArgs = {
43-
tokenStr: string;
43+
tokenStr?: string;
4444
conversation: ConversationClass;
4545
lang: InferenceSnippetLanguage;
4646
};
@@ -53,6 +53,7 @@
5353
max_tokens: data.config.max_tokens,
5454
temperature: data.config.temperature,
5555
top_p: data.config.top_p,
56+
accessToken: tokenStr,
5657
// eslint-disable-next-line @typescript-eslint/no-explicit-any
5758
} as any;
5859
// eslint-disable-next-line @typescript-eslint/no-explicit-any
@@ -61,7 +62,7 @@
6162
}
6263
6364
if (isCustomModel(model)) {
64-
const snippets = getInferenceSnippet(conversation, lang, tokenStr, opts);
65+
const snippets = getInferenceSnippet(conversation, lang, opts);
6566
return snippets
6667
.filter(s => s.client.startsWith("open") || lang === "sh")
6768
.map(s => {
@@ -74,7 +75,7 @@
7475
});
7576
}
7677
77-
return getInferenceSnippet(conversation, lang, tokenStr, opts);
78+
return getInferenceSnippet(conversation, lang, opts);
7879
}
7980
8081
// { javascript: 0, python: 0, http: 0 } at first
@@ -101,10 +102,10 @@
101102
if (isCustomModel(conversation.model)) {
102103
const t = conversation.model.accessToken;
103104
104-
return t && showToken ? t : "YOUR_ACCESS_TOKEN";
105+
return t && showToken ? t : undefined;
105106
}
106107
107-
return token.value && showToken ? token.value : "YOUR_HF_TOKEN";
108+
return token.value && showToken ? token.value : undefined;
108109
});
109110
110111
const snippetsByLang = $derived({

src/lib/utils/business.svelte.ts

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -316,8 +316,8 @@ export type GetInferenceSnippetReturn = InferenceSnippet[];
316316
export function getInferenceSnippet(
317317
conversation: ConversationClass,
318318
language: InferenceSnippetLanguage,
319-
accessToken: string,
320319
opts?: {
320+
accessToken?: string;
321321
messages?: ConversationEntityMembers["messages"];
322322
streaming?: ConversationEntityMembers["streaming"];
323323
max_tokens?: ConversationEntityMembers["config"]["max_tokens"];
@@ -342,7 +342,7 @@ export function getInferenceSnippet(
342342
provider,
343343
// eslint-disable-next-line @typescript-eslint/no-explicit-any
344344
{ ...providerMapping, hfModelId: model.id } as any,
345-
{ ...opts, accessToken },
345+
{ ...opts, directRequest: false },
346346
);
347347

348348
return allSnippets

0 commit comments

Comments
 (0)