add userId if available to enable session affinity

sourcegraph · Nov 14, 2024 · 79306bd · 79306bd
1 parent 390dd4a
commit 79306bd
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 0 deletions.
diff --git a/vscode/src/autoedits/adapters/cody-gateway.ts b/vscode/src/autoedits/adapters/cody-gateway.ts
@@ -57,6 +57,7 @@ export class CodyGatewayAdapter implements AutoeditsModelAdapter {
                     type: 'text',
                 },
                 speculation: option.codeToRewrite,
+                user: option.userId,
             }
             const response = await getModelResponse(
                 option.url,

diff --git a/vscode/src/autoedits/adapters/fireworks.ts b/vscode/src/autoedits/adapters/fireworks.ts
@@ -58,6 +58,7 @@ export class FireworksAdapter implements AutoeditsModelAdapter {
                         type: 'text',
                     },
                     speculation: option.codeToRewrite,
+                    user: option.userId,
                 }),
                 option.apiKey
             )

diff --git a/vscode/src/autoedits/autoedits-provider.ts b/vscode/src/autoedits/autoedits-provider.ts
@@ -206,6 +206,7 @@ export class AutoeditsProvider implements vscode.Disposable {
             apiKey,
             prompt,
             codeToRewrite: codeToReplace.codeToRewrite,
+            userId: (await currentResolvedConfig()).clientState.anonymousUserID,
         })
         const postProcessedResponse = this.config.provider.postProcessResponse(codeToReplace, response)
 

diff --git a/vscode/src/autoedits/prompt-provider.ts b/vscode/src/autoedits/prompt-provider.ts
@@ -17,6 +17,7 @@ export interface AutoeditModelOptions {
     apiKey: string
     prompt: ChatPrompt
     codeToRewrite: string
+    userId: string | null
 }
 
 export interface PromptResponseData {