Model capabilities 734 (#842)

internal/config,server: implement model capabilities

- define the capabilities of a model using a simple config block on the
model
- v1/models renders out capabilities to be compatible with openrouter,
huggingface chat, and mistral formats for broader compatibility
- add support for capabilities in UI

Fixes #734
This commit is contained in:
Benson Wong
2026-06-13 23:23:19 -07:00
committed by GitHub
parent 62aea0e83d
commit 92b90447e8
16 changed files with 868 additions and 35 deletions
+28 -2
View File
@@ -2,14 +2,40 @@ import type { Model } from "./types";
export interface GroupedModels {
local: Model[];
localMatching: Model[];
peersByProvider: Record<string, Model[]>;
}
export function groupModels(models: Model[]): GroupedModels {
export function matchesCapabilities(model: Model, required: string[], matchAny = false): boolean {
if (!required.length) return true;
if (!model.capabilities) return false;
const caps = model.capabilities as Record<string, boolean>;
if (matchAny) {
return required.some((cap) => caps[cap] === true);
}
return required.every((cap) => caps[cap] === true);
}
export function groupModels(models: Model[], capabilities?: string[], matchAny = false): GroupedModels {
const available = models.filter((m) => !m.unlisted);
const local = available.filter((m) => !m.peerID);
const peerModels = available.filter((m) => m.peerID);
let localMatching: Model[] = [];
let localRest: Model[] = [];
if (capabilities && capabilities.length > 0) {
for (const model of local) {
if (matchesCapabilities(model, capabilities, matchAny)) {
localMatching.push(model);
} else {
localRest.push(model);
}
}
} else {
localRest = local;
}
const peersByProvider = peerModels.reduce(
(acc, model) => {
const peerId = model.peerID || "unknown";
@@ -20,5 +46,5 @@ export function groupModels(models: Model[]): GroupedModels {
{} as Record<string, Model[]>
);
return { local, peersByProvider };
return { local: localRest, localMatching, peersByProvider };
}