Merge pull request #195 from TECH-C-LT/feature/sunshine-model-option

Author: ryu1013-job

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/anthropic.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')
+  }
+}

apps/api/src/libs/google.ts

@@ -27,7 +27,7 @@ default値は`0.5`です。
 
 **`model`** string optional <br/>
 指定可能なモデル
-- gpt-4o-mini
+- gpt-4o-mini (default)
 - claude-3-haiku
 - gemini-1.5-flash