feat: add embedding and rerank support

2026-10-12 20:19:51 +08:00 · 2026-05-03 17:32:41 +08:00
parent 3e2eca4fd0
commit 5abe664d65
87 changed files with 5520 additions and 184 deletions
@@ -0,0 +1,79 @@
+import type { ModelCreate, ModelUpdate, TieredPricingConfig } from '@/api/endpoints'
+
+interface EmbeddingMetadataCarrier {
+  supported_capabilities?: string[] | null
+  supports_embedding?: boolean | null
+  effective_supports_embedding?: boolean | null
+  config?: Record<string, unknown> | null
+}
+
+export interface ProviderModelCreatePayloadInput {
+  globalModelId: string
+  providerModelName: string
+  finalTieredPricing: TieredPricingConfig | null
+  tieredPricingModified: boolean
+  pricePerRequest?: number
+  cleanConfig?: Record<string, unknown>
+  configTouched: boolean
+  supportsVision?: boolean
+  supportsFunctionCalling?: boolean
+  supportsStreaming?: boolean
+  supportsExtendedThinking?: boolean
+  supportsImageGeneration?: boolean
+  isActive: boolean
+}
+
+export interface ProviderModelUpdatePayloadInput {
+  finalTieredPricing: TieredPricingConfig | null
+  pricePerRequest?: number
+  cleanConfig?: Record<string, unknown>
+  supportsVision?: boolean
+  supportsFunctionCalling?: boolean
+  supportsStreaming?: boolean
+  supportsExtendedThinking?: boolean
+  supportsImageGeneration?: boolean
+  isActive: boolean
+}
+
+export function modelSupportsEmbedding(model: EmbeddingMetadataCarrier | null | undefined): boolean {
+  if (!model) return false
+  if ('effective_supports_embedding' in model && model.effective_supports_embedding === true) return true
+  if ('supports_embedding' in model && model.supports_embedding === true) return true
+
+  const supportedCapabilities = 'supported_capabilities' in model ? model.supported_capabilities : null
+  const config = model.config || {}
+  return supportedCapabilities?.includes('embedding') === true
+    || config.embedding === true
+    || config.model_type === 'embedding'
+    || (Array.isArray(config.api_formats) && config.api_formats.some((format) => String(format).endsWith(':embedding')))
+}
+
+export function buildProviderModelCreatePayload(input: ProviderModelCreatePayloadInput): ModelCreate {
+  return {
+    global_model_id: input.globalModelId,
+    provider_model_name: input.providerModelName,
+    tiered_pricing: input.tieredPricingModified && input.finalTieredPricing ? input.finalTieredPricing : undefined,
+    price_per_request: input.pricePerRequest,
+    config: input.configTouched ? input.cleanConfig : undefined,
+    supports_vision: input.supportsVision,
+    supports_function_calling: input.supportsFunctionCalling,
+    supports_streaming: input.supportsStreaming,
+    supports_extended_thinking: input.supportsExtendedThinking,
+    supports_image_generation: input.supportsImageGeneration,
+    is_active: input.isActive,
+  }
+}
+
+export function buildProviderModelUpdatePayload(input: ProviderModelUpdatePayloadInput): ModelUpdate {
+  return {
+    tiered_pricing: input.finalTieredPricing,
+    price_per_request: input.pricePerRequest ?? null,
+    config: input.cleanConfig || null,
+    supports_vision: input.supportsVision,
+    supports_function_calling: input.supportsFunctionCalling,
+    supports_streaming: input.supportsStreaming,
+    supports_extended_thinking: input.supportsExtendedThinking,
+    supports_image_generation: input.supportsImageGeneration,
+    is_active: input.isActive,
+  }
+}