Skip to content

Commit a4cc1ca

Browse files
committed
improve api models endpoint
1 parent ab91811 commit a4cc1ca

1 file changed

Lines changed: 14 additions & 0 deletions

File tree

src/routes/api/models/+server.ts

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,8 +20,22 @@ export const GET: RequestHandler = async ({ request }) => {
2020
const key = `nanogpt:${model.id}`;
2121
const enabledModel = enabledModelsMap[key];
2222

23+
// Check capabilities based on architecture if not explicitly set
24+
const outputModalities = model.architecture?.output_modalities || [];
25+
26+
const capabilities = {
27+
vision: model.capabilities?.vision ?? false,
28+
reasoning: model.capabilities?.reasoning ?? false,
29+
// Add explicit image/video generation capabilities
30+
images: outputModalities.includes('image') && outputModalities.length === 1,
31+
video: outputModalities.includes('video'),
32+
};
33+
2334
return {
2435
...model,
36+
capabilities,
37+
// Ensure subscription info is always present with defaults
38+
subscription: model.subscription || { included: false, note: '' },
2539
enabled: !!enabledModel,
2640
pinned: enabledModel?.pinned ?? false,
2741
};

0 commit comments

Comments
 (0)