refactor: keep origin customJwtFetcherGuard

logto-io · Mar 21, 2024 · b2e00df · b2e00df
1 parent 45a7ee1
commit b2e00df
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/packages/schemas/src/types/jwt-customizer.ts b/packages/schemas/src/types/jwt-customizer.ts
@@ -1,7 +1,7 @@
 import { z } from 'zod';
 
 import { Roles, UserSsoIdentities, Organizations } from '../db-entries/index.js';
-import { mfaFactorsGuard } from '../foundations/index.js';
+import { jsonObjectGuard, mfaFactorsGuard } from '../foundations/index.js';
 
 import { jwtCustomizerGuard } from './logto-config/index.js';
 import { scopeResponseGuard } from './scope.js';
@@ -41,8 +41,8 @@ export const customJwtFetcherGuard = jwtCustomizerGuard
   .pick({ script: true, envVars: true })
   .required({ script: true })
   .extend({
-    token: z.record(z.unknown()),
-    context: z.record(z.unknown()).optional(),
+    token: jsonObjectGuard,
+    context: jsonObjectGuard.optional(),
   });
 
 export type CustomJwtFetcher = z.infer<typeof customJwtFetcherGuard>;