passing parameters from vscode to server

microsoft · Aug 16, 2024 · 0b739e6 · 0b739e6
1 parent aa2453e
commit 0b739e6
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/packages/core/src/parameters.ts b/packages/core/src/parameters.ts
@@ -123,3 +123,8 @@ export function parsePromptParameters(
     }
     return Object.freeze(res)
 }
+
+export function parametersToVars(parameters: PromptParameters): string[] {
+    if (!parameters) return undefined
+    return Object.keys(parameters).map((k) => `${k}=${parameters[k]}`)
+}
diff --git a/packages/vscode/src/state.ts b/packages/vscode/src/state.ts
@@ -34,6 +34,7 @@ import {
 } from "../../core/src/util"
 import { CORE_VERSION } from "../../core/src/version"
 import { Fragment, GenerationResult } from "../../core/src/generation"
+import { parametersToVars } from "../../core/src/parameters"
 
 export const FRAGMENTS_CHANGE = "fragmentsChange"
 export const AI_REQUEST_CHANGE = "aiRequestChange"
@@ -331,6 +332,7 @@ tests/
                 partialCb,
                 label,
                 cache: cache && template.cache,
+                vars: parametersToVars(options.parameters),
             }
         )
         r.runId = runId