Send Model ID Override to playbook gen/exp API

ansible · Sep 26, 2024 · 6f71fc9 · 6f71fc9
1 parent 2bceee6
commit 6f71fc9
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git 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
@@ -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 {