mirror of
https://github.com/RooVetGit/Roo-Code.git
synced 2026-09-06 08:18:39 +00:00
save provider to config
This commit is contained in:
parent
921ede741d
commit
f24c3cdf74
2 changed files with 27 additions and 8 deletions
|
|
@ -223,6 +223,7 @@ const groqSchema = apiModelIdProviderModelSchema.extend({
|
|||
const huggingFaceSchema = baseProviderSettingsSchema.extend({
|
||||
huggingFaceApiKey: z.string().optional(),
|
||||
huggingFaceModelId: z.string().optional(),
|
||||
huggingFaceInferenceProvider: z.string().optional(),
|
||||
})
|
||||
|
||||
const chutesSchema = apiModelIdProviderModelSchema.extend({
|
||||
|
|
|
|||
|
|
@ -44,7 +44,9 @@ export const HuggingFace = ({ apiConfiguration, setApiConfigurationField }: Hugg
|
|||
const { t } = useAppTranslation()
|
||||
const [models, setModels] = useState<HuggingFaceModel[]>([])
|
||||
const [loading, setLoading] = useState(false)
|
||||
const [selectedProvider, setSelectedProvider] = useState<string>("")
|
||||
const [selectedProvider, setSelectedProvider] = useState<string>(
|
||||
apiConfiguration?.huggingFaceInferenceProvider || "auto",
|
||||
)
|
||||
|
||||
const handleInputChange = useCallback(
|
||||
<K extends keyof ProviderSettings, E>(
|
||||
|
|
@ -87,23 +89,39 @@ export const HuggingFace = ({ apiConfiguration, setApiConfigurationField }: Hugg
|
|||
// Set default provider when model changes
|
||||
useEffect(() => {
|
||||
if (currentModel && availableProviders.length > 0) {
|
||||
const currentProvider = availableProviders.find((p) => p.provider === selectedProvider)
|
||||
if (!currentProvider) {
|
||||
// Set to first available provider or "auto"
|
||||
setSelectedProvider("auto")
|
||||
const savedProvider = apiConfiguration?.huggingFaceInferenceProvider
|
||||
if (savedProvider) {
|
||||
// Use saved provider if it exists
|
||||
setSelectedProvider(savedProvider)
|
||||
} else {
|
||||
const currentProvider = availableProviders.find((p) => p.provider === selectedProvider)
|
||||
if (!currentProvider) {
|
||||
// Set to "auto" as default
|
||||
const defaultProvider = "auto"
|
||||
setSelectedProvider(defaultProvider)
|
||||
setApiConfigurationField("huggingFaceInferenceProvider", defaultProvider)
|
||||
}
|
||||
}
|
||||
}
|
||||
}, [currentModel, availableProviders, selectedProvider])
|
||||
}, [
|
||||
currentModel,
|
||||
availableProviders,
|
||||
selectedProvider,
|
||||
apiConfiguration?.huggingFaceInferenceProvider,
|
||||
setApiConfigurationField,
|
||||
])
|
||||
|
||||
const handleModelSelect = (modelId: string) => {
|
||||
setApiConfigurationField("huggingFaceModelId", modelId)
|
||||
// Reset provider selection when model changes
|
||||
setSelectedProvider("auto")
|
||||
const defaultProvider = "auto"
|
||||
setSelectedProvider(defaultProvider)
|
||||
setApiConfigurationField("huggingFaceInferenceProvider", defaultProvider)
|
||||
}
|
||||
|
||||
const handleProviderSelect = (provider: string) => {
|
||||
setSelectedProvider(provider)
|
||||
// You could store this in a separate field if needed
|
||||
setApiConfigurationField("huggingFaceInferenceProvider", provider)
|
||||
}
|
||||
|
||||
// Format provider name for display
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue