diff --git a/src/features/lightspeed/api.ts b/src/features/lightspeed/api.ts index fa7f064cf..eeea1e61a 100644 --- a/src/features/lightspeed/api.ts +++ b/src/features/lightspeed/api.ts @@ -296,6 +296,11 @@ export class LightSpeedAPI { console.error("Ansible Lightspeed instance is not initialized."); return {} as ExplanationResponseParams; } + // If a Model ID Override is specified, send it to Lightspeed with the request. + if (lightSpeedManager.settingsManager.settings.lightSpeedService.model) { + inputData.model = + lightSpeedManager.settingsManager.settings.lightSpeedService.model; + } try { const customPrompt = lightSpeedManager.settingsManager.settings.lightSpeedService @@ -342,6 +347,11 @@ export class LightSpeedAPI { return {} as GenerationResponseParams; } try { + // If a Model ID Override is specified, send it to Lightspeed with the request. + if (lightSpeedManager.settingsManager.settings.lightSpeedService.model) { + inputData.model = + lightSpeedManager.settingsManager.settings.lightSpeedService.model; + } const customPrompt = lightSpeedManager.settingsManager.settings.lightSpeedService .playbookGenerationCustomPrompt; diff --git a/src/interfaces/lightspeed.ts b/src/interfaces/lightspeed.ts index dcb42ddfb..3b084eb40 100644 --- a/src/interfaces/lightspeed.ts +++ b/src/interfaces/lightspeed.ts @@ -147,6 +147,7 @@ export interface GenerationRequestParams { generationId: string; createOutline: boolean; wizardId?: string; + model?: string; } export interface GenerationResponseParams { @@ -159,6 +160,7 @@ export interface ExplanationRequestParams { content: string; customPrompt?: string; explanationId: string; + model?: string; } export interface ExplanationResponseParams {