scriptoriumadmin/Roo-Code
1
0
Fork 0
mirror of https://github.com/RooVetGit/Roo-Code.git synced 2026-09-05 08:10:14 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: add HuggingFace case to useSelectedModel hook

Browse source
This commit is contained in:
Roo Code 2025-07-23 16:32:21 +00:00
parent 8a787d2425
commit 6f3921f764
1 changed files with 8 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
webview-ui/src/components/ui/hooks/useSelectedModel.ts
View file

@ -34,6 +34,8 @@ import {
litellmDefaultModelId,
claudeCodeDefaultModelId,
claudeCodeModels,
huggingFaceDefaultModelId,
huggingFaceModels,
} from "@roo-code/types"
import type { RouterModels } from "@roo/api"
@ -214,11 +216,16 @@ function getSelectedModel({
const info = claudeCodeModels[id as keyof typeof claudeCodeModels]
return { id, info: { ...openAiModelInfoSaneDefaults, ...info } }
}
case "huggingface": {
const id = apiConfiguration.huggingFaceModelId ?? huggingFaceDefaultModelId
const info = huggingFaceModels[id as keyof typeof huggingFaceModels]
return { id, info: info || openAiModelInfoSaneDefaults }
}
// case "anthropic":
// case "human-relay":
// case "fake-ai":
default: {
provider satisfies "anthropic" | "gemini-cli" | "human-relay" | "fake-ai"
provider satisfies "anthropic" | "gemini-cli" | "human-relay" | "fake-ai" | "huggingface"
const id = apiConfiguration.apiModelId ?? anthropicDefaultModelId
const info = anthropicModels[id as keyof typeof anthropicModels]
return { id, info }