geminiへのリクエストに失敗した場合proxyを使う

Author: na-trium-144

app/actions/chatActions.ts

@@ -1,7 +1,7 @@
 'use server';
 
-import { GoogleGenerativeAI } from "@google/generative-ai";
 import { z } from "zod";
+import { generateContent } from "./gemini";
 
 interface FormState {
   response: string;
@@ -13,8 +13,6 @@ const ChatSchema = z.object({
   documentContent: z.string().min(1, { message: "コンテキストとなるドキュメントがありません。"}),
 });
 
-const genAI = new GoogleGenerativeAI(process.env.API_KEY!);
-
 type ChatParams = z.input<typeof ChatSchema>;
 
 export async function askAI(params: ChatParams): Promise<FormState> {
@@ -30,7 +28,7 @@ export async function askAI(params: ChatParams): Promise<FormState> {
   const { userQuestion, documentContent } = parseResult.data;
 
   try {
-    const model = genAI.getGenerativeModel({ model: "gemini-1.5-flash" });
+    
     const prompt = `
 以下のPythonチュートリアルのドキュメントの内容を正確に理解し、ユーザーからの質問に対して、初心者にも分かりやすく、丁寧な解説を提供してください。
 
@@ -48,7 +46,7 @@ ${userQuestion}
 - 
 
 `;
-    const result = await model.generateContent(prompt);
+    const result = await generateContent(prompt);
     const response = result.response;
     const text = response.text();
     return { response: text, error: null };

app/actions/gemini.ts

@@ -0,0 +1,27 @@
+"use server";
+
+import { GoogleGenerativeAI, ModelParams } from "@google/generative-ai";
+
+export async function generateContent(prompt: string) {
+  const params: ModelParams = {
+    model: "gemini-1.5-flash",
+  };
+
+  const genAI = new GoogleGenerativeAI(process.env.API_KEY!);
+  try {
+    const model = genAI.getGenerativeModel(params);
+    return await model.generateContent(prompt);
+  } catch (e: unknown) {
+    if (String(e).includes("User location is not supported")) {
+      const model = genAI.getGenerativeModel(params, {
+        baseUrl: "https://gemini-proxy.natrium144.org",
+        customHeaders: {
+          "x-goog-api-key": process.env.API_KEY!,
+        },
+      });
+      return await model.generateContent(prompt);
+    } else {
+      throw e;
+    }
+  }
+}

app/actions/questionExample.ts

@@ -1,15 +1,13 @@
 "use server";
 
-import { GoogleGenerativeAI } from "@google/generative-ai";
 import { z } from "zod";
+import { generateContent } from "./gemini";
 
 const QuestionExampleSchema = z.object({
   lang: z.string().min(1),
   documentContent: z.string().min(1),
 });
 
-const genAI = new GoogleGenerativeAI(process.env.API_KEY!);
-
 type QuestionExampleParams = z.input<typeof QuestionExampleSchema>;
 
 export async function getQuestionExample(
@@ -28,15 +26,14 @@ export async function getQuestionExample(
 
   const { lang, documentContent } = parseResult.data;
 
-  const model = genAI.getGenerativeModel({ model: "gemini-1.5-flash" });
   const prompt = `
 以下の${lang}チュートリアルのドキュメントに対して、想定される初心者のユーザーからの質問の例を箇条書きで複数挙げてください。
 強調などはせずテキストだけで1行ごとに1つ出力してください。
 
 # ドキュメント
 ${documentContent}
 `;
-  const result = await model.generateContent(prompt);
+  const result = await generateContent(prompt);
   const response = result.response;
   const text = response.text();
   return text.trim().split("\n");