feat(ai-quiz): generate-quizzesを共通化

nakasyou · Aug 20, 2024 · 7b80923 · 7b80923
1 parent 59d39b4
commit 7b80923
Show file tree

Hide file tree

Showing 2 changed files with 61 additions and 34 deletions.
diff --git a/src/islands/ai-quiz/components/Quiz.qwik.tsx b/src/islands/ai-quiz/components/Quiz.qwik.tsx
@@ -30,6 +30,7 @@ import {
 import { safeParse } from 'valibot'
 import { Loading } from './Utils.qwik'
 import { QuizDB } from '../storage'
+import { quizzesGenerator } from '../utils/generate-quizzes'
 
 export const QuizScreen = component$(() => {
   const quizState = useStore<QuizState>(
@@ -180,20 +181,10 @@ export const QuizScreen = component$(() => {
     ) {
       return
     }
-    const gemini = getGoogleGenerativeAI()
-    if (!gemini) {
+    const generateQuizzes = quizzesGenerator()
+    if (!generateQuizzes) {
       return alert('AIエラー')
     }
-    const model = gemini.getGenerativeModel({
-      model: 'gemini-1.5-flash',
-      generationConfig: {
-        responseMimeType: 'application/json',
-      },
-      systemInstruction: {
-        role: 'system',
-        parts: [{ text: PROMPT_TO_GENERATE_SELECT_QUIZ }],
-      },
-    })
 
     const sourceNotes = screenState.note!.notes.filter(
       (note) => note.type === 'text',
@@ -210,28 +201,9 @@ export const QuizScreen = component$(() => {
       const randomNote =
         sourceNotes[Math.floor(Math.random() * sourceNotes.length)]!
 
-      const res = await model
-        .startChat()
-        .sendMessage(randomNote?.canToJsonData.html || '')
-
-      let contents: unknown
-      try {
-        contents = JSON.parse(res.response.text())
-      } catch {
-        // Unable to parse
-        continue
-      }
-
-      if (!Array.isArray(contents)) {
-        continue
-      }
       const quizzes = await Promise.all(
-        contents
-          .filter(
-            (content): content is QuizContent =>
-              safeParse(CONTENT_SCHEMA, content).success,
-          )
-          .map(async (content): Promise<Quiz> => {
+        (await generateQuizzes(randomNote)).map(
+          async (content): Promise<Quiz> => {
             const res = await quizDB.quizzesByNote.add({
               noteId: randomNote.id,
               quiz: {
@@ -254,7 +226,8 @@ export const QuizScreen = component$(() => {
               source: randomNote,
               id: res,
             }
-          }),
+          },
+        ),
       )
       const addingQuizzes: typeof quizState.quizzes = []
       for (const quiz of quizzes) {

diff --git a/src/islands/ai-quiz/utils/generate-quizzes.ts b/src/islands/ai-quiz/utils/generate-quizzes.ts
@@ -0,0 +1,54 @@
+import { safeParse } from 'valibot'
+import type { TextNoteData } from '../../note/components/notes/TextNote/types'
+import { getGoogleGenerativeAI } from '../../shared/gemini'
+import {
+  CONTENT_SCHEMA,
+  PROMPT_TO_GENERATE_SELECT_QUIZ,
+  type QuizContent,
+} from '../constants'
+
+export const quizzesGenerator = () => {
+  const gemini = getGoogleGenerativeAI()
+  if (!gemini) {
+    return null
+  }
+
+  const model = gemini.getGenerativeModel({
+    model: 'gemini-1.5-flash',
+    generationConfig: {
+      responseMimeType: 'application/json',
+    },
+    systemInstruction: {
+      role: 'system',
+      parts: [{ text: PROMPT_TO_GENERATE_SELECT_QUIZ }],
+    },
+  })
+
+  return async function generatedQuizzes(
+    note: TextNoteData,
+  ): Promise<QuizContent[]> {
+    const res = await model
+      .startChat()
+      .sendMessage(note?.canToJsonData.html || '')
+
+    let contents: unknown
+    try {
+      contents = JSON.parse(res.response.text())
+    } catch {
+      // Unable to parse
+      return []
+    }
+
+    if (!Array.isArray(contents)) {
+      return []
+    }
+
+    const quizzes = await Promise.all(
+      contents.filter(
+        (content): content is QuizContent =>
+          safeParse(CONTENT_SCHEMA, content).success,
+      ),
+    )
+    return quizzes
+  }
+}