dsh-audiogen 0.4.19 → 0.4.22

package/src/routes.ts

@@ -11,7 +11,9 @@ import { randomUUID } from 'node:crypto'
 import type { WebRoute } from '@deepseek-ai/dsh-host-webserver'
 import { SettingsConflictError, settingsNamespace, type SettingsDescriptor } from '@deepseek-ai/dsh-settings'
 import { generateAudio, AudioGenError, type AudioChannel } from './audio-engine.ts'
-import { listVendorVoices, deleteVendorVoice } from './voice-manager.ts'
+import { listVendorVoices, listVendorVoicesWithFallback, deleteVendorVoice, type ListVoicesOptions, type VendorVoiceEntry } from './voice-manager.ts'
+import { recommendVoices, type VoiceRecommendation } from './voice-recommend.ts'
+import { appendVoiceRecommendRecord, listVoiceRecommendRecords, removeVoiceRecommendRecord } from './voice-recommend.ts'
 import type { GenerationBudget } from './audio-scheduler.ts'
 import { discoverAudioModels } from './audio-models.ts'
 import { AUDIO_PRESETS } from './audio-presets.ts'
@@ -50,6 +52,8 @@ export interface AudiogenRoutesDeps {
   budget: GenerationBudget
   /** 提示词增强：调用 Agent 默认模型，返回增强后的文本。 */
   enhance: (prompt: string, mode: GenerateAudioRequest['mode']) => Promise<string>
+  /** 音色推荐：需求描述 + 候选池 → top-k 推荐（复用 Agent 默认模型）。 */
+  recommend: (requirement: string, candidates: VendorVoiceEntry[], topK: number) => Promise<VoiceRecommendation[]>
   /** 「设置 → 模型」提供方列表 + 各自可广播模型（增强模型下拉候选）。 */
   llmModelOptions: () => Promise<LlmModelOption[]>
 }
@@ -302,9 +306,52 @@ async function mergeLibraryProvenance(
   }
 }
 
+/** 从请求体解析音色列表/推荐共用的筛选选项。 */
+function voiceListOptionsOf(body: Record<string, unknown> | undefined): ListVoicesOptions {
+  const str = (key: string): string | undefined => {
+    const value = body?.[key]
+    return typeof value === 'string' && value.trim() !== '' ? value.trim() : undefined
+  }
+  const filters = {
+    ...(str('search') === undefined ? {} : { search: str('search')! }),
+    ...(str('use_case') === undefined ? {} : { use_case: str('use_case')! }),
+    ...(str('accent') === undefined ? {} : { accent: str('accent')! }),
+    ...(str('gender') === undefined ? {} : { gender: str('gender')! }),
+    ...(str('age') === undefined ? {} : { age: str('age')! }),
+    ...(str('locale') === undefined ? {} : { locale: str('locale')! }),
+    ...(str('category') === undefined ? {} : { category: str('category')! }),
+    ...(str('sort') === undefined ? {} : { sort: str('sort')! }),
+    ...(body?.featured === true ? { featured: true } : {}),
+    ...(body?.free_users_allowed === true ? { free_users_allowed: true } : {}),
+    ...(body?.descriptive === true ? { descriptive: true } : {}),
+  }
+  return {
+    ...(str('language') === undefined ? {} : { language: str('language')! }),
+    ...(str('keyword') === undefined ? {} : { keyword: str('keyword')! }),
+    ...(str('source') === undefined ? {} : { source: str('source')! }),
+    ...(typeof body?.limit === 'number' && Number.isFinite(body.limit) ? { limit: Math.floor(body.limit) } : {}),
+    ...(Object.keys(filters).length === 0 ? {} : { serverFilters: filters }),
+  }
+}
+
+/** 推荐记录的筛选条件快照（面板展示用，不含密钥）。 */
+function recommendFiltersOf(body: Record<string, unknown> | undefined): Record<string, string | boolean> {
+  const str = (key: string): string | undefined => {
+    const value = body?.[key]
+    return typeof value === 'string' && value.trim() !== '' ? value.trim() : undefined
+  }
+  const out: Record<string, string | boolean> = {}
+  for (const key of ['language', 'keyword', 'source', 'search', 'use_case', 'accent', 'gender', 'age', 'locale', 'category', 'sort'] as const) {
+    const value = str(key)
+    if (value !== undefined) out[key] = value
+  }
+  if (body?.featured === true) out.featured = true
+  if (body?.free_users_allowed === true) out.free_users_allowed = true
+  return out
+}
+
 /** Build every /api/dsh-audiogen route. */
-export function makeRoutes(deps: AudiogenRoutesDeps): WebRoute[] {
-  const guard = (req: IncomingMessage, res: ServerResponse, method: string): boolean => {
+export function makeRoutes(deps: AudiogenRoutesDeps): WebRoute[] {  const guard = (req: IncomingMessage, res: ServerResponse, method: string): boolean => {
     if (!isLoopbackRequest(req)) {
       writeJson(res, 403, { error: 'forbidden: loopback-only' })
       return false
@@ -404,31 +451,8 @@ export function makeRoutes(deps: AudiogenRoutesDeps): WebRoute[] {
           writeJson(res, 200, { ok: false, code: 'channel-not-configured', message: '没有可用的音频渠道（需要已配置 API 地址与密钥），请先在设置中添加。' })
           return
         }
-        const str = (key: string): string | undefined => {
-          const value = body?.[key]
-          return typeof value === 'string' && value.trim() !== '' ? value.trim() : undefined
-        }
-        const filters = {
-          ...(str('search') === undefined ? {} : { search: str('search')! }),
-          ...(str('use_case') === undefined ? {} : { use_case: str('use_case')! }),
-          ...(str('accent') === undefined ? {} : { accent: str('accent')! }),
-          ...(str('gender') === undefined ? {} : { gender: str('gender')! }),
-          ...(str('age') === undefined ? {} : { age: str('age')! }),
-          ...(str('locale') === undefined ? {} : { locale: str('locale')! }),
-          ...(str('category') === undefined ? {} : { category: str('category')! }),
-          ...(str('sort') === undefined ? {} : { sort: str('sort')! }),
-          ...(body?.featured === true ? { featured: true } : {}),
-          ...(body?.free_users_allowed === true ? { free_users_allowed: true } : {}),
-          ...(body?.descriptive === true ? { descriptive: true } : {}),
-        }
         try {
-          const result = await listVendorVoices(channel, {
-            ...(str('language') === undefined ? {} : { language: str('language')! }),
-            ...(str('keyword') === undefined ? {} : { keyword: str('keyword')! }),
-            ...(str('source') === undefined ? {} : { source: str('source')! }),
-            ...(typeof body?.limit === 'number' && Number.isFinite(body.limit) ? { limit: Math.floor(body.limit) } : {}),
-            ...(Object.keys(filters).length === 0 ? {} : { serverFilters: filters }),
-          })
+          const result = await listVendorVoicesWithFallback(channel, voiceListOptionsOf(body))
           writeJson(res, 200, {
             ok: true,
             vendor: result.vendor,
@@ -471,6 +495,112 @@ export function makeRoutes(deps: AudiogenRoutesDeps): WebRoute[] {
         }
       },
     },
+    // ------------------------------------------ vendor voice recommend (LLM)
+    {
+      kind: 'exact',
+      path: VOICES_API.recommend,
+      handler: async (req, res) => {
+        if (!guard(req, res, 'POST')) return
+        const body = await readJsonBody(req) as Record<string, unknown> | undefined
+        const requirement = typeof body?.requirement === 'string' ? body.requirement.trim() : ''
+        if (requirement === '') {
+          writeJson(res, 200, { ok: false, code: 'recommend-requirement-required', message: '需求描述（requirement）不能为空。' })
+          return
+        }
+        const channel = channelOf(deps.resolveChannels(), body?.channel)
+        if (channel === undefined) {
+          writeJson(res, 200, { ok: false, code: 'channel-not-configured', message: '没有可用的音频渠道（需要已配置 API 地址与密钥），请先在设置中添加。' })
+          return
+        }
+        const rawTopK = body?.top_k
+        const topK = typeof rawTopK === 'number' && Number.isFinite(rawTopK)
+          ? Math.max(1, Math.min(10, Math.floor(rawTopK)))
+          : 5
+        try {
+          // 推荐面向足够宽的候选池：不传 limit 用默认上限，避免默认 100 太窄。
+          const result = await listVendorVoicesWithFallback(channel, {
+            ...voiceListOptionsOf(body),
+            limit: 500,
+          })
+          const recommendations = await deps.recommend(requirement, result.voices, topK)
+          // 记录本次推荐（最近 50 条，先记录后返回；失败不影响结果）。
+          void appendVoiceRecommendRecord({
+            channel: channel.name,
+            vendor: result.vendor,
+            requirement,
+            candidate_count: result.voices.length,
+            top_k: topK,
+            channel_id: channel.id,
+            filters: recommendFiltersOf(body),
+            recommendations: recommendations.map(item => ({
+              voice_id: item.voice_id,
+              name: item.name,
+              source: item.source,
+              deletable: item.deletable,
+              ...(item.language === undefined ? {} : { language: item.language }),
+              ...(item.locale === undefined ? {} : { locale: item.locale }),
+              ...(item.accent === undefined ? {} : { accent: item.accent }),
+              ...(item.gender === undefined ? {} : { gender: item.gender }),
+              ...(item.age === undefined ? {} : { age: item.age }),
+              ...(item.use_case === undefined ? {} : { use_case: item.use_case }),
+              ...(item.category === undefined ? {} : { category: item.category }),
+              ...(item.labels === undefined ? {} : { labels: item.labels }),
+              ...(item.descriptive === undefined ? {} : { descriptive: item.descriptive }),
+              ...(item.description === undefined ? {} : { description: item.description }),
+              ...(item.preview_url === undefined ? {} : { preview_url: item.preview_url }),
+              reason: item.reason,
+            })),
+          }).catch(() => { /* best-effort */ })
+          writeJson(res, 200, {
+            ok: true,
+            vendor: result.vendor,
+            channel: channel.name,
+            requirement,
+            candidate_count: result.voices.length,
+            top_k: topK,
+            recommendations,
+            recorded: true,
+            ...(result.note === undefined ? {} : { note: result.note }),
+          })
+        } catch (error) {
+          writeJson(res, 200, { ok: false, code: 'voice-recommend-failed', message: messageOf(error) })
+        }
+      },
+    },
+    // ------------------------------------------ voice recommend history (records)
+    {
+      kind: 'exact',
+      path: VOICES_API.recommendHistory.list,
+      handler: async (req, res) => {
+        if (!guard(req, res, 'POST')) return
+        const body = await readJsonBody(req) as Record<string, unknown> | undefined
+        const rawLimit = body?.limit
+        const limit = typeof rawLimit === 'number' && Number.isFinite(rawLimit)
+          ? Math.max(1, Math.min(50, Math.floor(rawLimit)))
+          : 20
+        const entries = await listVoiceRecommendRecords(limit)
+        writeJson(res, 200, { ok: true, count: entries.length, entries })
+      },
+    },
+    {
+      kind: 'exact',
+      path: VOICES_API.recommendHistory.remove,
+      handler: async (req, res) => {
+        if (!guard(req, res, 'POST')) return
+        const body = await readJsonBody(req) as Record<string, unknown> | undefined
+        const id = typeof body?.id === 'string' ? body.id.trim() : ''
+        if (id === '') {
+          writeJson(res, 200, { ok: false, code: 'record-id-required', message: 'record id 不能为空。' })
+          return
+        }
+        try {
+          await removeVoiceRecommendRecord(id)
+          writeJson(res, 200, { ok: true, removed: id })
+        } catch (error) {
+          writeJson(res, 200, { ok: false, code: 'record-remove-failed', message: messageOf(error) })
+        }
+      },
+    },
     // -------------------------------------------------- settings describe
     {
       kind: 'exact',