File tree Expand file tree Collapse file tree 1 file changed +4
-3
lines changed Expand file tree Collapse file tree 1 file changed +4
-3
lines changed Original file line number Diff line number Diff line change 55 mainlandZAiDefaultModelId ,
66 type InternationalZAiModelId ,
77 type MainlandZAiModelId ,
8+ type ModelInfo ,
89 ZAI_DEFAULT_TEMPERATURE ,
910 zaiApiLineConfigs ,
1011} from "@roo-code/types"
@@ -13,11 +14,11 @@ import type { ApiHandlerOptions } from "../../shared/api"
1314
1415import { BaseOpenAiCompatibleProvider } from "./base-openai-compatible-provider"
1516
16- export class ZAiHandler extends BaseOpenAiCompatibleProvider < InternationalZAiModelId | MainlandZAiModelId > {
17+ export class ZAiHandler extends BaseOpenAiCompatibleProvider < string > {
1718 constructor ( options : ApiHandlerOptions ) {
1819 const isChina = zaiApiLineConfigs [ options . zaiApiLine ?? "international_coding" ] . isChina
19- const models = isChina ? mainlandZAiModels : internationalZAiModels
20- const defaultModelId = isChina ? mainlandZAiDefaultModelId : internationalZAiDefaultModelId
20+ const models = ( isChina ? mainlandZAiModels : internationalZAiModels ) as unknown as Record < string , ModelInfo >
21+ const defaultModelId = ( isChina ? mainlandZAiDefaultModelId : internationalZAiDefaultModelId ) as string
2122
2223 super ( {
2324 ...options ,
You can’t perform that action at this time.
0 commit comments