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

fix: resolve claude-code provider image hang (#5100) (#5105)

Browse source 
Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
Co-authored-by: Daniel <57051444+daniel-lxs@users.noreply.github.com>
This commit is contained in:
Hannes Rudolph 2025-06-25 14:08:35 -06:00 • committed by GitHub
parent a9752e0293
commit 83c19ce2c8
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 80 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

72
webview-ui/src/components/ui/hooks/__tests__/useSelectedModel.spec.ts
View file

@ -369,4 +369,76 @@ describe("useSelectedModel", () => {
expect(result.current.info).toBeUndefined()
})
})
describe("claude-code provider", () => {
it("should return claude-code model with supportsImages disabled", () => {
mockUseRouterModels.mockReturnValue({
data: {
openrouter: {},
requesty: {},
glama: {},
unbound: {},
litellm: {},
},
isLoading: false,
isError: false,
} as any)
mockUseOpenRouterModelProviders.mockReturnValue({
data: {},
isLoading: false,
isError: false,
} as any)
const apiConfiguration: ProviderSettings = {
apiProvider: "claude-code",
apiModelId: "claude-sonnet-4-20250514",
}
const wrapper = createWrapper()
const { result } = renderHook(() => useSelectedModel(apiConfiguration), { wrapper })
expect(result.current.provider).toBe("claude-code")
expect(result.current.id).toBe("claude-sonnet-4-20250514")
expect(result.current.info).toBeDefined()
expect(result.current.info?.supportsImages).toBe(false)
expect(result.current.info?.supportsPromptCache).toBe(false)
// Verify it inherits other properties from anthropic models
expect(result.current.info?.maxTokens).toBe(64_000)
expect(result.current.info?.contextWindow).toBe(200_000)
expect(result.current.info?.supportsComputerUse).toBe(true)
})
it("should use default claude-code model when no modelId is specified", () => {
mockUseRouterModels.mockReturnValue({
data: {
openrouter: {},
requesty: {},
glama: {},
unbound: {},
litellm: {},
},
isLoading: false,
isError: false,
} as any)
mockUseOpenRouterModelProviders.mockReturnValue({
data: {},
isLoading: false,
isError: false,
} as any)
const apiConfiguration: ProviderSettings = {
apiProvider: "claude-code",
}
const wrapper = createWrapper()
const { result } = renderHook(() => useSelectedModel(apiConfiguration), { wrapper })
expect(result.current.provider).toBe("claude-code")
expect(result.current.id).toBe("claude-sonnet-4-20250514") // Default model
expect(result.current.info).toBeDefined()
expect(result.current.info?.supportsImages).toBe(false)
})
})
})

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

@ -30,6 +30,8 @@ import {
glamaDefaultModelId,
unboundDefaultModelId,
litellmDefaultModelId,
claudeCodeDefaultModelId,
claudeCodeModels,
} from "@roo-code/types"
import type { RouterModels } from "@roo/api"
@ -199,6 +201,12 @@ function getSelectedModel({
const info = vscodeLlmModels[modelFamily as keyof typeof vscodeLlmModels]
return { id, info: { ...openAiModelInfoSaneDefaults, ...info, supportsImages: false } } // VSCode LM API currently doesn't support images.
}
case "claude-code": {
// Claude Code models extend anthropic models but with images and prompt caching disabled
const id = apiConfiguration.apiModelId ?? claudeCodeDefaultModelId
const info = claudeCodeModels[id as keyof typeof claudeCodeModels]
return { id, info: { ...openAiModelInfoSaneDefaults, ...info } }
}
// case "anthropic":
// case "human-relay":
// case "fake-ai":