Skip to content

Commit 1cbb62b

Browse files
author
Krish Dholakia
authored
Merge pull request #8 from Chesars/fix/vertex-ai-model-prefix-display
Add vertex_ai/ prefix to model names for clarity
2 parents 4f9ea4f + 2485df1 commit 1cbb62b

1 file changed

Lines changed: 24 additions & 4 deletions

File tree

src/App.svelte

Lines changed: 24 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -149,6 +149,26 @@ We also need to update [${RESOURCE_BACKUP_NAME}](https://github.com/${REPO_FULL_
149149
});
150150
}
151151
152+
/**
153+
* Get the display name for a model, adding provider prefix when needed.
154+
* This helps users understand which auth method is required.
155+
* e.g., "gemini-2.0-flash" with provider "vertex_ai-language-models" -> "vertex_ai/gemini-2.0-flash"
156+
*/
157+
function getDisplayModelName(name: string, litellm_provider: string): string {
158+
// If the model name already has a provider prefix, return as-is
159+
if (name.includes('/')) {
160+
return name;
161+
}
162+
163+
// Add prefix for vertex_ai models so users know they need GCP credentials
164+
// Provider can be "vertex_ai", "vertex_ai-language-models", etc.
165+
if (litellm_provider && litellm_provider.startsWith('vertex_ai')) {
166+
return `vertex_ai/${name}`;
167+
}
168+
169+
return name;
170+
}
171+
152172
function toggleRow(name: string) {
153173
if (expandedRows.has(name)) {
154174
expandedRows.delete(name);
@@ -340,10 +360,10 @@ We also need to update [${RESOURCE_BACKUP_NAME}](https://github.com/${REPO_FULL_
340360
</div>
341361
{/if}
342362
</div>
343-
<span class="model-title">{name}</span>
344-
<button
345-
class="copy-button"
346-
on:click|stopPropagation={() => copyToClipboard(name)}
363+
<span class="model-title">{getDisplayModelName(name, litellm_provider)}</span>
364+
<button
365+
class="copy-button"
366+
on:click|stopPropagation={() => copyToClipboard(getDisplayModelName(name, litellm_provider))}
347367
title="Copy model name"
348368
type="button"
349369
>

0 commit comments

Comments
 (0)