Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
826 changes: 826 additions & 0 deletions packages/stats/app/src/routes/[lab]/[model].tsx

Large diffs are not rendered by default.

427 changes: 427 additions & 0 deletions packages/stats/app/src/routes/[lab]/index.tsx

Large diffs are not rendered by default.

1,130 changes: 952 additions & 178 deletions packages/stats/app/src/routes/index.css

Large diffs are not rendered by default.

492 changes: 20 additions & 472 deletions packages/stats/app/src/routes/index.tsx

Large diffs are not rendered by default.

224 changes: 224 additions & 0 deletions packages/stats/app/src/routes/model-catalog.ts
Original file line number Diff line number Diff line change
@@ -0,0 +1,224 @@
import { query } from "@solidjs/router"

export const modelCatalogSourceUrl = "https://models.dev/models.json"

export type ModelCatalogEntry = {
id: string
lab: string
slug: string
name: string
family?: string
knowledge?: string
releaseDate?: string
lastUpdated?: string
limit?: { context?: number; output?: number }
modalities: { input: string[]; output: string[] }
openWeights: boolean
reasoning: boolean
toolCall: boolean
attachment: boolean
temperature: boolean
weights: { label: string; url: string }[]
benchmarks: ModelCatalogBenchmark[]
}

export type ModelCatalogBenchmark = {
name: string
score: number
metric?: string
harness?: string
variant?: string
dataset?: string
version?: string
source?: string
}

export type ModelCatalogLab = {
id: string
name: string
models: ModelCatalogEntry[]
}

export type ModelCatalog = {
models: ModelCatalogEntry[]
labs: ModelCatalogLab[]
}

export const getModelCatalog = query(async () => {
"use server"
const payload = await fetch(modelCatalogSourceUrl)
.then((response): Promise<unknown> => (response.ok ? (response.json() as Promise<unknown>) : Promise.resolve()))
.catch(() => undefined)
return buildModelCatalog(payload)
}, "getModelCatalog")

export function findModelCatalogEntry(catalog: ModelCatalog, model: string, lab?: string) {
const normalizedId = lab ? `${catalogSlug(lab)}/${catalogSlug(model)}` : model.trim().toLowerCase()
const leaf = catalogSlug(model)
return (
catalog.models.find((entry) => entry.id.toLowerCase() === normalizedId) ??
catalog.models.find((entry) => (lab ? entry.lab === catalogSlug(lab) : true) && entry.slug === leaf) ??
catalog.models.find((entry) => entry.slug === leaf)
)
}

export function findModelCatalogLab(catalog: ModelCatalog, lab: string) {
const id = catalogSlug(lab)
return catalog.labs.find((entry) => entry.id === id)
}

export function formatCatalogLabName(lab: string) {
const known: Record<string, string> = {
alibaba: "Alibaba",
anthropic: "Anthropic",
cohere: "Cohere",
deepseek: "DeepSeek",
google: "Google",
meta: "Meta",
minimax: "MiniMax",
mistral: "Mistral",
moonshotai: "Moonshot",
openai: "OpenAI",
perplexity: "Perplexity",
stepfun: "StepFun",
tencent: "Tencent",
xai: "xAI",
xiaomi: "Xiaomi",
zai: "Z.ai",
zhipuai: "Zhipu",
}
return known[catalogSlug(lab)] ?? lab.replace(/[-_]/g, " ").replace(/\b\w/g, (letter) => letter.toUpperCase())
}

export function catalogSlug(value: string) {
return value
.trim()
.toLowerCase()
.replace(/[^a-z0-9]+/g, "-")
.replace(/^-+|-+$/g, "")
.replace(/-{2,}/g, "-")
}

function buildModelCatalog(payload: unknown): ModelCatalog {
const models = (Array.isArray(payload) ? payload : isRecord(payload) ? Object.values(payload) : [])
.flatMap(readModelCatalogEntry)
.toSorted((a, b) => a.lab.localeCompare(b.lab) || displayDateTime(b.releaseDate) - displayDateTime(a.releaseDate))
return {
models,
labs: Object.values(
models.reduce<Record<string, ModelCatalogLab>>((result, model) => {
result[model.lab] = {
id: model.lab,
name: formatCatalogLabName(model.lab),
models: [...(result[model.lab]?.models ?? []), model],
}
return result
}, {}),
).toSorted((a, b) => a.name.localeCompare(b.name)),
}
}

function readModelCatalogEntry(value: unknown): ModelCatalogEntry[] {
if (!isRecord(value)) return []
const id = stringValue(value.id)
const name = stringValue(value.name)
const lab = id?.split("/")[0]
const slug = id?.split("/").slice(1).join("/")
if (!id || !name || !lab || !slug) return []
return [
{
id,
lab: catalogSlug(lab),
slug: catalogSlug(slug),
name,
family: stringValue(value.family),
knowledge: stringValue(value.knowledge),
releaseDate: stringValue(value.release_date),
lastUpdated: stringValue(value.last_updated),
limit: readCatalogLimit(value.limit),
modalities: readCatalogModalities(value.modalities),
openWeights: booleanValue(value.open_weights),
reasoning: booleanValue(value.reasoning),
toolCall: booleanValue(value.tool_call),
attachment: booleanValue(value.attachment),
temperature: booleanValue(value.temperature),
weights: readCatalogWeights(value.weights),
benchmarks: readCatalogBenchmarks(value.benchmarks),
},
]
}

function readCatalogLimit(value: unknown) {
if (!isRecord(value)) return undefined
return {
context: numberValue(value.context),
output: numberValue(value.output),
}
}

function readCatalogModalities(value: unknown) {
if (!isRecord(value)) return { input: [], output: [] }
return {
input: stringArrayValue(value.input),
output: stringArrayValue(value.output),
}
}

function readCatalogWeights(value: unknown) {
if (!Array.isArray(value)) return []
return value.flatMap((item) => {
if (!isRecord(item)) return []
const label = stringValue(item.label)
const url = stringValue(item.url)
return label && url ? [{ label, url }] : []
})
}

function readCatalogBenchmarks(value: unknown) {
if (!Array.isArray(value)) return []
return value.flatMap((item) => {
if (!isRecord(item)) return []
const name = stringValue(item.name)
const score = numberValue(item.score)
return name && score !== undefined
? [
{
name,
score,
metric: stringValue(item.metric),
harness: stringValue(item.harness),
variant: stringValue(item.variant),
dataset: stringValue(item.dataset),
version: stringValue(item.version),
source: stringValue(item.source),
},
]
: []
})
}

function displayDateTime(value: string | undefined) {
return value ? new Date(value).getTime() || 0 : 0
}

function isRecord(value: unknown): value is Record<string, unknown> {
return typeof value === "object" && value !== null && !Array.isArray(value)
}

function stringValue(value: unknown) {
return typeof value === "string" && value.trim() ? value : undefined
}

function numberValue(value: unknown) {
return typeof value === "number" && Number.isFinite(value) ? value : undefined
}

function booleanValue(value: unknown) {
return value === true
}

function stringArrayValue(value: unknown) {
return Array.isArray(value)
? value.filter((item): item is string => typeof item === "string" && item.trim() !== "")
: []
}
Loading
Loading