Merge pull request #931 from RooVetGit/fix_model_loading_flicker

Fix issue where Ollama/LMStudio URLs would flicker back to previous while entering
This commit is contained in:
Matt Rubens 2025-02-10 23:50:09 -05:00 committed by GitHub
commit 145819f8e4
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 20 additions and 6 deletions

View file

@ -0,0 +1,5 @@
---
"roo-cline": patch
---
Fix issue where Ollama/LMStudio URLs would flicker back to previous while entering

View file

@ -88,12 +88,21 @@ const ApiOptions = ({ apiErrorMessage, modelIdErrorMessage, fromWelcomeView }: A
)
const handleMessage = useCallback((event: MessageEvent) => {
const message: ExtensionMessage = event.data
if (message.type === "ollamaModels" && message.ollamaModels) {
setOllamaModels(message.ollamaModels)
} else if (message.type === "lmStudioModels" && message.lmStudioModels) {
setLmStudioModels(message.lmStudioModels)
} else if (message.type === "vsCodeLmModels" && message.vsCodeLmModels) {
setVsCodeLmModels(message.vsCodeLmModels)
if (message.type === "ollamaModels" && Array.isArray(message.ollamaModels)) {
const newModels = message.ollamaModels
setOllamaModels((prevModels) => {
return JSON.stringify(prevModels) === JSON.stringify(newModels) ? prevModels : newModels
})
} else if (message.type === "lmStudioModels" && Array.isArray(message.lmStudioModels)) {
const newModels = message.lmStudioModels
setLmStudioModels((prevModels) => {
return JSON.stringify(prevModels) === JSON.stringify(newModels) ? prevModels : newModels
})
} else if (message.type === "vsCodeLmModels" && Array.isArray(message.vsCodeLmModels)) {
const newModels = message.vsCodeLmModels
setVsCodeLmModels((prevModels) => {
return JSON.stringify(prevModels) === JSON.stringify(newModels) ? prevModels : newModels
})
}
}, [])
useEvent("message", handleMessage)