refactor: Move API config to api-config.ts and update imports

Author: ericc-ch

src/lib/constants.ts src/lib/api-config.tssrc/lib/constants.ts renamed to src/lib/api-config.ts

@@ -8,11 +8,15 @@ export const COPILOT_API_CONFIG = {
 } as const
 
 export const COPILOT_API_BASE_URL = "https://api.individual.githubcopilot.com"
-export const COPILOT_API_HEADERS = {
+const COPILOT_API_SPOOF_HEADERS = {
   "copilot-integration-id": "vscode-chat",
-  "editor-version": "vscode/1.98.0-insider",
 }
 
+export const buildCopilotHeaders = (token: string) => ({
+  Authorization: `token ${token}`,
+  ...COPILOT_API_SPOOF_HEADERS,
+})
+
 export const GITHUB_API_BASE_URL = "https://api.github.com"
 
 export const GITHUB_BASE_URL = "https://github.com"

src/routes/embeddings/route.ts

@@ -4,13 +4,13 @@ import { FetchError } from "ofetch"
 
 import type { EmbeddingRequest } from "~/services/copilot/embedding/types"
 
-import { embedding } from "~/services/copilot/embedding/service"
+import { createEmbeddings } from "~/services/copilot/embedding/service"
 
 export const embeddingRoutes = new Hono()
 
 embeddingRoutes.post("/", async (c) => {
   try {
-    const embeddings = await embedding(await c.req.json<EmbeddingRequest>())
+    const embeddings = await createEmbeddings(await c.req.json<EmbeddingRequest>())
     return c.json(embeddings)
   } catch (error) {
     if (error instanceof FetchError) {

src/services/api-instance.ts

@@ -1,7 +1,7 @@
 import consola from "consola"
 import { FetchError, ofetch } from "ofetch"
 
-import { COPILOT_API_CONFIG } from "~/lib/constants"
+import { COPILOT_API_CONFIG } from "~/lib/api-config"
 import { modelsCache } from "~/lib/models"
 import { tokenService } from "~/lib/token"
 

src/services/copilot/chat-completions/service-streaming.ts

@@ -1,6 +1,6 @@
 import { stream } from "fetch-event-stream"
 
-import { COPILOT_API_CONFIG } from "~/lib/constants"
+import { COPILOT_API_CONFIG } from "~/lib/api-config"
 import { tokenService } from "~/lib/token"
 
 import type { ChatCompletionsPayload } from "./types"

src/services/copilot/embedding/service.ts

@@ -1,11 +1,42 @@
+import { COPILOT_API_BASE_URL } from "~/lib/api-config"
+import { state } from "~/lib/state"
+
 import type { EmbeddingRequest, EmbeddingResponse } from "./types"
 
 import { copilot } from "../../api-instance"
 
-export const embedding = (payload: EmbeddingRequest) =>
-  copilot<EmbeddingResponse>("/embeddings", {
-    method: "POST",
-    body: {
-      ...payload,
+export const createEmbeddings = (payload: EmbeddingRequest) => {
+  const response = await fetch(`${COPILOT_API_BASE_URL}/embeddings`, {
+    headers: {
+      authorization: `token ${state.copilotToken}`,
     },
   })
+}
+
+copilot<EmbeddingResponse>("/embeddings", {
+  method: "POST",
+  body: {
+    ...payload,
+  },
+})
+
+export interface EmbeddingRequest {
+  input: string | Array<string>
+  model: string
+}
+
+export interface Embedding {
+  object: string
+  embedding: Array<number>
+  index: number
+}
+
+export interface EmbeddingResponse {
+  object: string
+  data: Array<Embedding>
+  model: string
+  usage: {
+    prompt_tokens: number
+    total_tokens: number
+  }
+}

src/services/copilot/get-copilot-token.ts

@@ -1,4 +1,4 @@
-import { GITHUB_API_BASE_URL } from "~/lib/constants"
+import { GITHUB_API_BASE_URL } from "~/lib/api-config"
 import { state } from "~/lib/state"
 
 export const getCopilotToken = async () => {

src/services/copilot/get-models.ts

@@ -1,4 +1,4 @@
-import { COPILOT_API_BASE_URL } from "~/lib/constants"
+import { COPILOT_API_BASE_URL } from "~/lib/api-config"
 import { state } from "~/lib/state"
 
 export const getModels = async () => {

src/services/github/get-device-code.ts

@@ -1,4 +1,4 @@
-import { GITHUB_BASE_URL, GITHUB_CLIENT_ID } from "~/lib/constants"
+import { GITHUB_BASE_URL, GITHUB_CLIENT_ID } from "~/lib/api-config"
 
 export async function getDeviceCode(): Promise<DeviceCodeResponse> {
   const response = await fetch(`${GITHUB_BASE_URL}/login/device/code`, {

src/services/github/get-user.ts

@@ -1,4 +1,4 @@
-import { GITHUB_API_BASE_URL } from "~/lib/constants"
+import { GITHUB_API_BASE_URL } from "~/lib/api-config"
 import { state } from "~/lib/state"
 
 export async function getGitHubUser() {

src/services/github/poll-access-token.ts

@@ -1,4 +1,4 @@
-import { GITHUB_BASE_URL, GITHUB_CLIENT_ID } from "~/lib/constants"
+import { GITHUB_BASE_URL, GITHUB_CLIENT_ID } from "~/lib/api-config"
 import { sleep } from "~/lib/sleep"
 
 import type { DeviceCodeResponse } from "./get-device-code"