Microsoft/vscode
1
0
Fork 0
mirror of https://github.com/microsoft/vscode.git synced 2026-08-19 14:22:19 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix ollam provider (#3858)

Browse Source
This commit is contained in:
Sandeep Somavarapu
2026-02-19 17:15:12 +00:00
committed by GitHub
parent 08e2a3a331
commit 63de200f58
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
extensions/copilot/src/extension/byok/vscode-node/ollamaProvider.ts
+7
View File
@@ -9,6 +9,7 @@ import { IFetcherService } from '../../../platform/networking/common/fetcherServ
import { IExperimentationService } from '../../../platform/telemetry/common/nullExperimentationService';
import { IInstantiationService } from '../../../util/vs/platform/instantiation/common/instantiation';
import { byokKnownModelsToAPIInfo, resolveModelInfo } from '../common/byokProvider';
import { OpenAIEndpoint } from '../node/openAIEndpoint';
import { AbstractOpenAICompatibleLMProvider, LanguageModelChatConfiguration, OpenAICompatibleLanguageModelChatInformation } from './abstractLanguageModelChatProvider';
import { IBYOKStorageService } from './byokStorageService';
@@ -125,6 +126,12 @@ export class OllamaLMProvider extends AbstractOpenAICompatibleLMProvider<OllamaC
return configuration?.url ?? 'http://localhost:11434';
}
protected override async createOpenAIEndPoint(model: OpenAICompatibleLanguageModelChatInformation<OllamaConfig>): Promise<OpenAIEndpoint> {
const modelInfo = this.getModelInfo(model.id, model.url);
const url = `${model.url}/v1/chat/completions`;
return this._instantiationService.createInstance(OpenAIEndpoint, modelInfo, model.configuration?.apiKey ?? '', url);
}
private async _getOllamaModelInfo(ollamaBaseUrl: string, modelId: string): Promise<IChatModelInformation> {
const modelInfo = await this._fetchOllamaModelInformation(ollamaBaseUrl, modelId);
const contextWindow = modelInfo?.model_info?.[`${modelInfo.model_info['general.architecture']}.context_length`] ?? 32768;