diff --git a/packages/core/src/data_sources/types.ts b/packages/core/src/data_sources/types.ts
index cc65b76c8..1eaee9825 100644
--- a/packages/core/src/data_sources/types.ts
+++ b/packages/core/src/data_sources/types.ts
@@ -114,9 +114,13 @@ export type DataFieldSchemas =
   | DataFieldSchemaJSON
   | DataFieldSchemaRelation;
 
+export type DataSourceSchema<DR extends DataRecordProps = DataRecordProps> = {
+  [K in keyof DR]?: DataFieldSchemas;
+};
+
 export interface DataSourceType<DR extends DataRecordProps> extends BaseDataSource {
   records: DataRecords<DR>;
-  schema: Partial<Record<keyof DR, DataFieldSchemas | undefined>>;
+  schema: DataSourceSchema<DR>;
 }
 export interface DataSourceProps<DR extends DataRecordProps> extends BaseDataSource {
   records?: DataRecords<DR> | DataRecord<DR>[] | DR[];