feat: vision flag

Author: ericc-ch

src/lib/api-config.ts

@@ -26,6 +26,7 @@ export const copilotHeaders = (state: State) => ({
   "x-github-api-version": API_VERSION,
   "x-request-id": randomUUID(),
   "x-vscode-user-agent-library-version": "electron-fetch",
+  "copilot-vision-request": state.visionEnabled ? "true" : "false",
 })
 
 export const GITHUB_API_BASE_URL = "https://api.github.com"

src/lib/state.ts

@@ -10,6 +10,7 @@ export interface State {
 
   manualApprove: boolean
   rateLimitWait: boolean
+  visionEnabled: boolean
 
   // Rate limiting configuration
   rateLimitSeconds?: number
@@ -20,4 +21,5 @@ export const state: State = {
   accountType: "individual",
   manualApprove: false,
   rateLimitWait: false,
+  visionEnabled: false,
 }

src/main.ts

@@ -17,9 +17,10 @@ interface RunServerOptions {
   verbose: boolean
   business: boolean
   manual: boolean
-  rateLimit: number | undefined
+  rateLimit?: number
   rateLimitWait: boolean
   githubToken?: string
+  visionEnabled: boolean
 }
 
 export async function runServer(options: RunServerOptions): Promise<void> {
@@ -36,6 +37,11 @@ export async function runServer(options: RunServerOptions): Promise<void> {
   state.manualApprove = options.manual
   state.rateLimitSeconds = options.rateLimit
   state.rateLimitWait = options.rateLimitWait
+  state.visionEnabled = options.visionEnabled
+
+  if (options.visionEnabled) {
+    consola.info("Vision capability enabled")
+  }
 
   await ensurePaths()
   await cacheVSCodeVersion()
@@ -105,6 +111,12 @@ const start = defineCommand({
       description:
         "Provide GitHub token directly instead of using stored token",
     },
+    vision: {
+      type: "boolean",
+      default: false,
+      description: "Enable vision capabilities",
+      required: false,
+    },
   },
   run({ args }) {
     const rateLimitRaw = args["rate-limit"]
@@ -122,6 +134,7 @@ const start = defineCommand({
       rateLimit,
       rateLimitWait: Boolean(args.wait),
       githubToken: args["github-token"],
+      visionEnabled: args.vision,
     })
   },
 })