Fix issue where Ollama/LMStudio URLs would flicker back to previous while entering

This commit is contained in:
Matt Rubens 2025-02-10 23:16:52 -05:00
parent 4aa43d0b5e
commit febf434fc1
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)