Skip to content

Commit adb9c16

Browse files
authored
Fix some issues with openrouter (#417)
1 parent 7da123f commit adb9c16

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

src/extension/byok/vscode-node/baseOpenAICompatibleProvider.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ export abstract class BaseOpenAICompatibleLMProvider implements BYOKModelProvide
2222
public readonly authType: BYOKAuthType,
2323
private readonly _name: string,
2424
private readonly _baseUrl: string,
25-
private _knownModels: BYOKKnownModels | undefined,
25+
protected _knownModels: BYOKKnownModels | undefined,
2626
private readonly _byokStorageService: IBYOKStorageService,
2727
@IFetcherService protected readonly _fetcherService: IFetcherService,
2828
@ILogService protected readonly _logService: ILogService,

src/extension/byok/vscode-node/openRouterProvider.ts

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,11 +38,12 @@ export class OpenRouterLMProvider extends BaseOpenAICompatibleLMProvider {
3838
knownModels[model.id] = {
3939
name: model.name,
4040
toolCalling: model.supported_parameters?.includes('tools') ?? false,
41-
vision: model.input_modalities?.includes('image') ?? false,
41+
vision: model.architecture?.input_modalities?.includes('image') ?? false,
4242
maxInputTokens: model.top_provider.context_length - 16000,
4343
maxOutputTokens: 16000
4444
};
4545
}
46+
this._knownModels = knownModels;
4647
return knownModels;
4748
} catch (error) {
4849
this._logService.error(error, `Error fetching available OpenRouter models`);

0 commit comments

Comments
 (0)