list models

Author: github-actions[bot]

src/gemini-api-client/gemini-api-client.ts

@@ -22,9 +22,10 @@ interface Task {
 }
 
 export async function listModels(apiParam: ApiParam | null) {
-  const url = new URL(`${BASE_URL}/v1beta/models`)
-  url.searchParams.append("key", apiParam?.apikey ?? "")
-  const resp = await makeRequest(url, undefined, undefined, "GET")
+  const url = new URL(`${BASE_URL}/v1beta/openai/models`)
+  const resp = await makeRequest(url, undefined, undefined, "GET", {
+    Authorization: `Bearer ${apiParam?.apikey ?? ""}`,
+  })
   return (await resp.json()) as components["schemas"]["ListModelsResponse"]
 }
 export async function* streamGenerateContent(
@@ -74,6 +75,7 @@ async function makeRequest(
   body: string | undefined,
   requestOptions?: RequestOptions,
   requestMethod = "POST",
+  headers: Record<string, string> = {},
 ): Promise<Response> {
   let response: Response
   try {
@@ -82,6 +84,7 @@ async function makeRequest(
       method: requestMethod,
       headers: {
         "Content-Type": "application/json",
+        ...headers,
       },
       body,
     })

test/models_test.ts

@@ -11,7 +11,7 @@ const fetchMocker = new MockFetch()
 bdd.describe("openai model api test", () => {
   bdd.beforeEach(() => {
     fetchMocker.mock(
-      (req) => req.url.includes(`generativelanguage.googleapis.com/v1beta/models`),
+      (req) => req.url.includes(`generativelanguage.googleapis.com/v1beta/openai/models`),
       () => Response.json({ data: modelData }),
     )
   })