Merge pull request #40 from hgaogao/main

MichaelYuhe · Dec 11, 2023 · 01f9b7e · 01f9b7e
2 parents 671e79e + 63a9e24
commit 01f9b7e
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 4 deletions.
diff --git a/src/popup.tsx b/src/popup.tsx
@@ -20,6 +20,7 @@ const Popup = () => {
     getStorage<string[]>("types").then((types) => {
       if (!types) {
         setTypes(DEFAULT_GROUP);
+        setStorage<string[]>("types", DEFAULT_GROUP);
         return;
       }
       setTypes(types);

diff --git a/src/services.ts b/src/services.ts
@@ -26,17 +26,20 @@ export async function batchGroupTabs(
   });
 
   const model = (await getStorage("model")) || "gpt-4";
-  const apiURL = (await getStorage("apiURL")) || "https://api.openai.com";
+  const apiURL =
+    (await getStorage("apiURL")) ||
+    "https://api.openai.com/v1/chat/completions";
 
   try {
     await Promise.all(
       tabInfoList.map(async (tab) => {
         if (!tab.url) return;
-        const response = await fetch(`${apiURL}/v1/chat/completions`, {
+        const response = await fetch(apiURL, {
           method: "POST",
           headers: {
             "Content-Type": "application/json",
             Authorization: `Bearer ${openAIKey}`,
+            "Api-Key": openAIKey,
           },
           body: JSON.stringify({
             messages: [
@@ -79,13 +82,16 @@ export async function handleOneTab(
 ) {
   try {
     const model = (await getStorage("model")) || "gpt-4";
-    const apiURL = (await getStorage("apiURL")) || "https://api.openai.com";
+    const apiURL =
+      (await getStorage("apiURL")) ||
+      "https://api.openai.com/v1/chat/completions";
 
-    const response = await fetch(`${apiURL}/v1/chat/completions`, {
+    const response = await fetch(apiURL, {
       method: "POST",
       headers: {
         "Content-Type": "application/json",
         Authorization: `Bearer ${openAIKey}`,
+        "Api-Key": openAIKey,
       },
       body: JSON.stringify({
         messages: [