feat: Support GitHub Copilot Business accounts

Author: ericc-ch

src/lib/api-config.ts

@@ -5,7 +5,8 @@ export const standardHeaders = () => ({
   accept: "application/json",
 })
 
-export const COPILOT_API_BASE_URL = "https://api.individual.githubcopilot.com"
+export const copilotBaseUrl = (state: State) =>
+  `https://api.${state.accountType}.githubcopilot.com`
 export const copilotHeaders = (state: State) => ({
   Authorization: `Bearer ${state.copilotToken}`,
   "content-type": standardHeaders()["content-type"],

src/lib/state.ts

@@ -4,8 +4,11 @@ export interface State {
   githubToken?: string
   copilotToken?: string
 
+  accountType: string
   models?: ModelsResponse
   vsCodeVersion?: string
 }
 
-export const state: State = {}
+export const state: State = {
+  accountType: "individual",
+}

src/main.ts

@@ -6,13 +6,15 @@ import { serve, type ServerHandler } from "srvx"
 
 import { cacheModels } from "./lib/models"
 import { ensurePaths } from "./lib/paths"
+import { state } from "./lib/state"
 import { setupCopilotToken, setupGitHubToken } from "./lib/token"
 import { cacheVSCodeVersion } from "./lib/vscode-version"
 import { server } from "./server"
 
 interface RunServerOptions {
   port: number
   verbose: boolean
+  business: boolean
 }
 
 export async function runServer(options: RunServerOptions): Promise<void> {
@@ -21,6 +23,11 @@ export async function runServer(options: RunServerOptions): Promise<void> {
     consola.info("Verbose logging enabled")
   }
 
+  if (options.business) {
+    state.accountType = "business"
+    consola.info("Using business plan GitHub account")
+  }
+
   await ensurePaths()
   await cacheVSCodeVersion()
   await setupGitHubToken()
@@ -50,13 +57,19 @@ const main = defineCommand({
       default: false,
       description: "Enable verbose logging",
     },
+    business: {
+      type: "boolean",
+      default: false,
+      description: "Use a business plan GitHub Account",
+    },
   },
   run({ args }) {
     const port = Number.parseInt(args.port, 10)
 
     return runServer({
       port,
       verbose: args.verbose,
+      business: args.business,
     })
   },
 })

src/services/copilot/create-chat-completions.ts

@@ -1,6 +1,6 @@
 import { events } from "fetch-event-stream"
 
-import { copilotHeaders, COPILOT_API_BASE_URL } from "~/lib/api-config"
+import { copilotHeaders, copilotBaseUrl } from "~/lib/api-config"
 import { HTTPError } from "~/lib/http-error"
 import { state } from "~/lib/state"
 
@@ -9,7 +9,7 @@ export const createChatCompletions = async (
 ) => {
   if (!state.copilotToken) throw new Error("Copilot token not found")
 
-  const response = await fetch(`${COPILOT_API_BASE_URL}/chat/completions`, {
+  const response = await fetch(`${copilotBaseUrl(state)}/chat/completions`, {
     method: "POST",
     headers: copilotHeaders(state),
     body: JSON.stringify(payload),

src/services/copilot/create-embeddings.ts

@@ -1,11 +1,11 @@
-import { copilotHeaders, COPILOT_API_BASE_URL } from "~/lib/api-config"
+import { copilotHeaders, copilotBaseUrl } from "~/lib/api-config"
 import { HTTPError } from "~/lib/http-error"
 import { state } from "~/lib/state"
 
 export const createEmbeddings = async (payload: EmbeddingRequest) => {
   if (!state.copilotToken) throw new Error("Copilot token not found")
 
-  const response = await fetch(`${COPILOT_API_BASE_URL}/embeddings`, {
+  const response = await fetch(`${copilotBaseUrl(state)}/embeddings`, {
     method: "POST",
     headers: copilotHeaders(state),
     body: JSON.stringify(payload),

src/services/copilot/get-models.ts

@@ -1,9 +1,9 @@
-import { COPILOT_API_BASE_URL, copilotHeaders } from "~/lib/api-config"
+import { copilotBaseUrl, copilotHeaders } from "~/lib/api-config"
 import { HTTPError } from "~/lib/http-error"
 import { state } from "~/lib/state"
 
 export const getModels = async () => {
-  const response = await fetch(`${COPILOT_API_BASE_URL}/models`, {
+  const response = await fetch(`${copilotBaseUrl(state)}/models`, {
     headers: copilotHeaders(state),
   })