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

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

@@ -15,9 +15,7 @@ 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 { AnthropicClient, GoogleClient, OpenAIClient } from '~/libs/models'
 import { SupabaseClient } from '~/libs/supabase'
 
 /**

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

@@ -3,10 +3,13 @@ import { GoogleGenerativeAIProvider } from '@ai-sdk/google'
 import { OpenAIProvider } from '@ai-sdk/openai'
 import { InternalServerError } from '@peace-net/shared/core/error'
 import { categoryScoresSchema } from '@peace-net/shared/schemas/guardian'
-import type { CategoryScores, Models } from '@peace-net/shared/types/guardian'
+import type { CategoryScores } from '@peace-net/shared/types/guardian'
+import { Models } from '@peace-net/shared/types/model'
 import { generateObject } from 'ai'
 import { z } from 'zod'
 
+import { selectAIModel } from '~/libs/models'
+
 const systemPrompt = `
 # 役割
 あなたは日本語のコンテンツモデレーションの専門家です。
@@ -85,21 +88,12 @@ export class GuardianService implements IGuardianService {
     selectedModel: Models,
   ): Promise<CategoryScores> {
     try {
-      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
-      }
+      const model = selectAIModel(
+        selectedModel,
+        this.openai,
+        this.anthropic,
+        this.google,
+      )
 
       const { object } = await generateObject({
         model,

apps/api/src/features/sunshines/sunshine.route.ts

@@ -12,7 +12,7 @@ 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 { OpenAIClient } from '~/libs/openai'
+import { AnthropicClient, GoogleClient, OpenAIClient } from '~/libs/models'
 import { SupabaseClient } from '~/libs/supabase'
 
 /**
@@ -33,7 +33,11 @@ sunshineRoutes.post(
   async (c) => {
     return new SunshineController(
       new SunshineUseCase(
-        new SunshineService(OpenAIClient(getEnv(c).OPENAI_API_KEY)),
+        new SunshineService(
+          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/sunshines/sunshine.service.ts

@@ -1,9 +1,14 @@
+import { AnthropicProvider } from '@ai-sdk/anthropic'
+import { GoogleGenerativeAIProvider } from '@ai-sdk/google'
 import { OpenAIProvider } from '@ai-sdk/openai'
 import { InternalServerError } from '@peace-net/shared/core/error'
+import { Models } from '@peace-net/shared/types/model'
 import { SunshineResult } from '@peace-net/shared/types/sunshine'
 import { generateObject } from 'ai'
 import { z } from 'zod'
 
+import { selectAIModel } from '~/libs/models'
+
 const systemPrompt = `
 あなたは、ネガティブな表現をポジティブで建設的な表現に変換する専門家です。以下の指針に従ってテキストを変換してください：
 
@@ -33,16 +38,30 @@ const systemPrompt = `
 変換したテキストは、'text'オブジェクトに格納してください。
 `
 export interface ISunshineService {
-  sunshineText(text: string): Promise<SunshineResult>
+  sunshineText(text: string, selectedModel: Models): Promise<SunshineResult>
 }
 
 export class SunshineService implements ISunshineService {
-  constructor(private openai: OpenAIProvider) {}
+  constructor(
+    private openai: OpenAIProvider,
+    private anthropic: AnthropicProvider,
+    private google: GoogleGenerativeAIProvider,
+  ) {}
 
-  async sunshineText(text: string): Promise<SunshineResult> {
+  async sunshineText(
+    text: string,
+    selectedModel: Models,
+  ): Promise<SunshineResult> {
     try {
+      const model = selectAIModel(
+        selectedModel,
+        this.openai,
+        this.anthropic,
+        this.google,
+      )
+
       const { object } = await generateObject({
-        model: this.openai('gpt-4o-mini'),
+        model,
         schema: z.object({ text: z.string() }),
         messages: [
           { role: 'system', content: systemPrompt },
@@ -53,7 +72,7 @@ export class SunshineService implements ISunshineService {
       return object
     } catch (error) {
       console.error(error)
-      throw new InternalServerError('Failed to generate object')
+      throw new InternalServerError('Failed to generate text')
     }
   }
 }

apps/api/src/features/sunshines/sunshine.usecase.ts

@@ -21,8 +21,8 @@ export class SunshineUseCase implements ISunshineUseCase {
     input: SunshineTextInput,
   ): Promise<Result<SunshineResult>> {
     try {
-      const { text, userId, apiKeyId } = input
-      const result = await this.SunshineService.sunshineText(text)
+      const { text, model, userId, apiKeyId } = input
+      const result = await this.SunshineService.sunshineText(text, model)
 
       await this.usageFacade.incrementUsage(userId, apiKeyId, 'sunshines')
 

apps/api/src/libs/models.ts

@@ -0,0 +1,62 @@
+import { AnthropicProvider, createAnthropic } from '@ai-sdk/anthropic'
+import {
+  createGoogleGenerativeAI,
+  GoogleGenerativeAIProvider,
+} from '@ai-sdk/google'
+import { createOpenAI, OpenAIProvider } from '@ai-sdk/openai'
+import { Models } from '@peace-net/shared/types/model'
+import { LanguageModel } from 'ai'
+
+/**
+ * OpenAI APIのクライアントを生成します
+ *
+ * @param apiKey - OpenAI APIのAPIキー
+ * @returns OpenAI APIのクライアント
+ */
+export const OpenAIClient = (apiKey: string) => {
+  return createOpenAI({
+    apiKey,
+  })
+}
+
+/**
+ * Google Generative AI APIのクライアントを生成します
+ *
+ * @param apiKey - Google Generative AI APIのAPIキー
+ * @returns Google Generative AIのクライアント
+ */
+export const GoogleClient = (apiKey: string) => {
+  return createGoogleGenerativeAI({
+    apiKey,
+  })
+}
+
+/**
+ * Anthropic APIのクライアントを生成します
+ *
+ * @param apiKey - Anthropic APIのAPIキー
+ * @returns Anthropic APIのクライアント
+ */
+export const AnthropicClient = (apiKey: string) => {
+  return createAnthropic({
+    apiKey,
+  })
+}
+
+export const selectAIModel: (
+  selectedModel: Models,
+  openai: OpenAIProvider,
+  anthropic: AnthropicProvider,
+  google: GoogleGenerativeAIProvider,
+) => LanguageModel = (selectedModel, openai, anthropic, google) => {
+  switch (selectedModel) {
+    case 'gpt-4o-mini':
+      return openai('gpt-4o-mini')
+    case 'claude-3-haiku':
+      return anthropic('claude-3-haiku-20240307')
+    case 'gemini-1.5-flash':
+      return google('models/gemini-1.5-flash')
+    default:
+      return openai('gpt-4o-mini')
+  }
+}

packages/shared/src/schemas/guardian.ts

@@ -1,4 +1,5 @@
 import { z } from 'zod'
+import { modelsSchema } from './model'
 
 export const categoryScoresSchema = z.object({
   sexual: z.number(),
@@ -8,12 +9,6 @@ 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.1).optional().default(0.5),

packages/shared/src/schemas/model.ts

@@ -0,0 +1,7 @@
+import { z } from 'zod'
+
+export const modelsSchema = z.union([
+  z.literal('gpt-4o-mini'),
+  z.literal('claude-3-haiku'),
+  z.literal('gemini-1.5-flash'),
+])

packages/shared/src/schemas/sunshine.ts

@@ -1,7 +1,9 @@
 import { z } from 'zod'
+import { modelsSchema } from './model'
 
 export const sunshineTextRequestSchema = z.object({
   text: z.string().max(500),
+  model: modelsSchema.optional().default('gpt-4o-mini'),
 })
 
 export const sunshineResultSchema = z.object({

packages/shared/src/types/guardian.ts

@@ -4,15 +4,12 @@ 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 & {

packages/shared/src/types/model.ts

@@ -0,0 +1,4 @@
+import { z } from 'zod'
+import { modelsSchema } from '../schemas/model'
+
+export type Models = z.infer<typeof modelsSchema>