Merge pull request #186 from TECH-C-LT/feature/guardians-model-option

Guardian APIにAIのモデルを選択できるオプションを追加

Author: ryu1013-job

apps/api/package.json

@@ -8,6 +8,8 @@
     "lint": "eslint src --ext .ts,.tsx"
   },
   "dependencies": {
+    "@ai-sdk/anthropic": "^0.0.43",
+    "@ai-sdk/google": "^0.0.39",
     "@ai-sdk/openai": "^0.0.46",
     "@hono/zod-validator": "^0.2.2",
     "@peace-net/shared": "workspace:*",

apps/api/src/config/environment.ts

@@ -6,6 +6,8 @@ import { z } from 'zod'
 const envSchema = z.object({
   DOCS_URL: z.string(),
   OPENAI_API_KEY: z.string(),
+  ANTHROPIC_API_KEY: z.string(),
+  GOOGLE_API_KEY: z.string(),
   SUPABASE_URL: z.string(),
   SUPABASE_SERVICE_ROLE_KEY: z.string(),
   ENCRYPTION_KEY: z.string(),

apps/api/src/features/guardians/guardian.route.ts

@@ -15,6 +15,8 @@ import { UsageLogService } from '~/features/usageLogs/usageLog.service'
 import { UsageFacade } from '~/features/usages/usage.facade'
 import { UserPlanRepository } from '~/features/userPlans/userPlan.repository'
 import { UserPlanService } from '~/features/userPlans/userPlan.service'
+import { AnthropicClient } from '~/libs/anthropic'
+import { GoogleClient } from '~/libs/google'
 import { OpenAIClient } from '~/libs/openai'
 import { SupabaseClient } from '~/libs/supabase'
 
@@ -37,7 +39,11 @@ guardianRoutes.post(
   async (c) => {
     return new GuardianController(
       new GuardianUseCase(
-        new GuardianService(OpenAIClient(getEnv(c).OPENAI_API_KEY)),
+        new GuardianService(
+          OpenAIClient(getEnv(c).OPENAI_API_KEY),
+          AnthropicClient(getEnv(c).ANTHROPIC_API_KEY),
+          GoogleClient(getEnv(c).GOOGLE_API_KEY),
+        ),
         new UsageFacade(
           new UserPlanService(
             new UserPlanRepository(
@@ -72,7 +78,11 @@ guardianRoutes.post(
   async (c) => {
     return new GuardianController(
       new GuardianUseCase(
-        new GuardianService(OpenAIClient(getEnv(c).OPENAI_API_KEY)),
+        new GuardianService(
+          OpenAIClient(getEnv(c).OPENAI_API_KEY),
+          AnthropicClient(getEnv(c).ANTHROPIC_API_KEY),
+          GoogleClient(getEnv(c).GOOGLE_API_KEY),
+        ),
         new UsageFacade(
           new UserPlanService(
             new UserPlanRepository(

apps/api/src/features/guardians/guardian.service.ts

@@ -1,6 +1,8 @@
+import { AnthropicProvider } from '@ai-sdk/anthropic'
+import { GoogleGenerativeAIProvider } from '@ai-sdk/google'
 import { OpenAIProvider } from '@ai-sdk/openai'
 import { categoryScoresSchema } from '@peace-net/shared/schemas/guardian'
-import type { CategoryScores } from '@peace-net/shared/types/guardian'
+import type { CategoryScores, Models } from '@peace-net/shared/types/guardian'
 import { generateObject } from 'ai'
 import { z } from 'zod'
 
@@ -55,7 +57,7 @@ const systemPrompt = `
  * @returns 分析結果を含むResultオブジェクト。成功時はGuardianResultを、失敗時はエラーを含みます。
  */
 export interface IGuardianService {
-  guardianText(text: string): Promise<CategoryScores>
+  guardianText(text: string, selectedModel: Models): Promise<CategoryScores>
 }
 
 /**
@@ -71,11 +73,34 @@ export interface IGuardianService {
  * @returns 分析結果を含むResultオブジェクト。成功時はGuardianResultを、失敗時はエラーを含みます。
  */
 export class GuardianService implements IGuardianService {
-  constructor(private openai: OpenAIProvider) {}
+  constructor(
+    private openai: OpenAIProvider,
+    private anthropic: AnthropicProvider,
+    private google: GoogleGenerativeAIProvider,
+  ) {}
+
+  async guardianText(
+    text: string,
+    selectedModel: Models,
+  ): Promise<CategoryScores> {
+    let model
+    switch (selectedModel) {
+      case 'gpt-4o-mini':
+        model = this.openai('gpt-4o-mini')
+        break
+      case 'claude-3-haiku':
+        model = this.anthropic('claude-3-haiku-20240307')
+        break
+      case 'gemini-1.5-flash':
+        model = this.google('models/gemini-1.5-flash')
+        break
+      default:
+        model = this.openai('gpt-4o-mini')
+        break
+    }
 
-  async guardianText(text: string): Promise<CategoryScores> {
     const { object } = await generateObject({
-      model: this.openai('gpt-4o-mini'),
+      model,
       schema: z.object({ category_scores: categoryScoresSchema }),
       messages: [
         { role: 'system', content: systemPrompt },

apps/api/src/features/guardians/guardian.usecase.ts

@@ -48,8 +48,18 @@ export class GuardianUseCase implements IGuardianUseCase {
     input: GuardianTextInput,
   ): Promise<Result<GuardianResult>> {
     try {
-      const { text, score_threshold = 0.5, userId, apiKeyId } = input
-      const categoryScores = await this.guardianService.guardianText(text)
+      const {
+        text,
+        score_threshold = 0.5,
+        model = 'gpt-4o-mini',
+        userId,
+        apiKeyId,
+      } = input
+
+      const categoryScores = await this.guardianService.guardianText(
+        text,
+        model,
+      )
 
       const flagged = checkFlagged(categoryScores, score_threshold)
 

apps/api/src/libs/anthropic.ts

@@ -0,0 +1,13 @@
+import { createAnthropic } from '@ai-sdk/anthropic'
+
+/**
+ * Anthropic APIのクライアントを生成します
+ *
+ * @param apiKey - Anthropic APIのAPIキー
+ * @returns Anthropic APIのクライアント
+ */
+export const AnthropicClient = (apiKey: string) => {
+  return createAnthropic({
+    apiKey,
+  })
+}

apps/api/src/libs/google.ts

@@ -0,0 +1,13 @@
+import { createGoogleGenerativeAI } from '@ai-sdk/google'
+
+/**
+ * Google Generative AI APIのクライアントを生成します
+ *
+ * @param apiKey - Google Generative AI APIのAPIキー
+ * @returns Google Generative AIのクライアント
+ */
+export const GoogleClient = (apiKey: string) => {
+  return createGoogleGenerativeAI({
+    apiKey,
+  })
+}

packages/shared/src/schemas/guardian.ts

@@ -8,9 +8,16 @@ export const categoryScoresSchema = z.object({
   defamation: z.number(),
 })
 
+export const modelsSchema = z.union([
+  z.literal('gpt-4o-mini'),
+  z.literal('claude-3-haiku'),
+  z.literal('gemini-1.5-flash'),
+])
+
 export const guardianTextRequestSchema = z.object({
   text: z.string().max(500),
   score_threshold: z.number().max(1).min(0).optional().default(0.5),
+  model: modelsSchema.optional().default('gpt-4o-mini'),
 })
 
 export const guardianImageRequestSchema = z.object({

packages/shared/src/types/guardian.ts

@@ -4,12 +4,15 @@ import {
   guardianTextRequestSchema,
   guardianImageRequestSchema,
   guardianResultSchema,
+  modelsSchema,
 } from '../schemas/guardian'
 
 export type Category = keyof z.infer<typeof categoryScoresSchema>
 
 export type CategoryScores = z.infer<typeof categoryScoresSchema>
 
+export type Models = z.infer<typeof modelsSchema>
+
 export type GuardianTextDTO = z.infer<typeof guardianTextRequestSchema>
 
 export type GuardianTextInput = GuardianTextDTO & {