From 208abda841b35cbe1b088d8808feebc3284d13f1 Mon Sep 17 00:00:00 2001
From: Daniel Salazar <daniel.salazar@puter.com>
Date: Thu, 30 Apr 2026 12:38:39 -0700
Subject: [PATCH] fix: bad ai endpoint (#2856)

---
 src/backend/controllers/puterai/PuterAIController.ts | 12 ++++--------
 1 file changed, 4 insertions(+), 8 deletions(-)

diff --git a/src/backend/controllers/puterai/PuterAIController.ts b/src/backend/controllers/puterai/PuterAIController.ts
index 28d676129..4e477012d 100644
--- a/src/backend/controllers/puterai/PuterAIController.ts
+++ b/src/backend/controllers/puterai/PuterAIController.ts
@@ -184,19 +184,15 @@ export class PuterAIController extends PuterController {
         };
     }
 
-    #modelDetails(driverKey: string) {
+    #modelDetails(driverKey: 'aiChat' | 'aiImage' | 'aiVideo') {
         return async (_req: Request, res: Response): Promise<void> => {
-            const driver = (this.drivers as Record<string, unknown>)[
-                driverKey
-            ] as { models?: () => Array<{ id: string }> } | undefined;
+            const driver = this.drivers[driverKey];
             if (!driver?.models)
                 throw new HttpError(501, 'Model details not available');
-            const models = driver.models();
+            const models = await driver.models();
             const HIDDEN = ['costly', 'fake', 'abuse', 'model-fallback-test-1'];
             res.json({
-                models: (models as Array<{ id: string }>).filter(
-                    (m) => !HIDDEN.includes(m.id),
-                ),
+                models: models?.filter((m) => !HIDDEN.includes(m.id)),
             });
         };
     }