mirror of
https://github.com/HeyPuter/puter.git
synced 2026-05-03 08:00:32 +00:00
puter models list fix (#2869)
Docker Image CI / build-and-push-image (push) Has been cancelled
Maintain Release Merge PR / update-release-pr (push) Has been cancelled
Notify HeyPuter / notify (push) Has been cancelled
release-please / release-please (push) Has been cancelled
test / test-backend (24.x) (push) Has been cancelled
test / API tests (node env, api-test) (24.x) (push) Has been cancelled
test / puterjs (node env, vitest) (24.x) (push) Has been cancelled
Docker Image CI / build-and-push-image (push) Has been cancelled
Maintain Release Merge PR / update-release-pr (push) Has been cancelled
Notify HeyPuter / notify (push) Has been cancelled
release-please / release-please (push) Has been cancelled
test / test-backend (24.x) (push) Has been cancelled
test / API tests (node env, api-test) (24.x) (push) Has been cancelled
test / puterjs (node env, vitest) (24.x) (push) Has been cancelled
This commit is contained in:
@@ -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)),
|
||||
});
|
||||
};
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user