feat: add embedding and rerank support

frontend/src/views/user/model-catalog-helpers.ts

@@ -0,0 +1,41 @@
+import type { PublicGlobalModel } from '@/api/public-models'
+
+export function supportsEmbedding(model: PublicGlobalModel): boolean {
+  return model.supports_embedding === true
+    || model.supported_capabilities?.includes('embedding') === true
+    || model.config?.embedding === true
+    || model.config?.model_type === 'embedding'
+    || (Array.isArray(model.config?.api_formats) && model.config.api_formats.some((format) => String(format).endsWith(':embedding')))
+}
+
+export function supportsRerank(model: PublicGlobalModel): boolean {
+  return model.supported_capabilities?.includes('rerank') === true
+    || model.config?.rerank === true
+    || model.config?.model_type === 'rerank'
+    || (Array.isArray(model.config?.api_formats) && model.config.api_formats.some((format) => String(format).endsWith(':rerank')))
+}
+
+export function hasVideoPricing(model: PublicGlobalModel): boolean {
+  const billing = model.config?.billing
+  const video = billing && typeof billing === 'object' && !Array.isArray(billing)
+    ? (billing as Record<string, unknown>).video
+    : null
+  const priceByResolution = video && typeof video === 'object' && !Array.isArray(video)
+    ? (video as Record<string, unknown>).price_per_second_by_resolution
+    : null
+  return !!priceByResolution && typeof priceByResolution === 'object' && Object.keys(priceByResolution).length > 0
+}
+
+export function getModelCapabilityLabels(model: PublicGlobalModel): string[] {
+  const labels: string[] = []
+  if (supportsRerank(model)) {
+    labels.push('Rerank')
+  } else if (supportsEmbedding(model)) {
+    labels.push('Embedding')
+  } else {
+    labels.push('Chat')
+  }
+  if (model.config?.image_generation === true) labels.push('Image')
+  if (hasVideoPricing(model)) labels.push('Video')
+  return labels
+}