From e8fbb88102e492ccddaeaf8d7f0b9c1bd26f0c2d Mon Sep 17 00:00:00 2001
From: Daniel Salazar <daniel.salazar@puter.com>
Date: Fri, 1 May 2026 01:41:56 -0700
Subject: [PATCH] puter models list fix (#2869)

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

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