@@ -4,7 +4,10 @@ import {
44 validateInput ,
55} from "@/app/api/v1/osograph/utils/validation" ;
66import { requireAuthentication } from "@/app/api/v1/osograph/utils/auth" ;
7- import { checkMembershipExists } from "@/app/api/v1/osograph/utils/resolver-helpers" ;
7+ import {
8+ checkMembershipExists ,
9+ getMaterializations ,
10+ } from "@/app/api/v1/osograph/utils/resolver-helpers" ;
811import { createAdminClient } from "@/lib/supabase/admin" ;
912import {
1013 AuthenticationErrors ,
@@ -15,6 +18,8 @@ import { logger } from "@/lib/logger";
1518import z from "zod" ;
1619import { DataIngestionsRow } from "@/lib/types/schema-types" ;
1720import { getModelContext } from "@/app/api/v1/osograph/schema/resolvers/model-context" ;
21+ import type { FilterableConnectionArgs } from "@/app/api/v1/osograph/utils/pagination" ;
22+ import { generateTableId } from "@/app/api/v1/osograph/utils/model" ;
1823
1924export const dataIngestionResolvers = {
2025 Mutation : {
@@ -118,5 +123,19 @@ export const dataIngestionResolvers = {
118123 ) => {
119124 return getModelContext ( parent . dataset_id , args . tableName ) ;
120125 } ,
126+ materializations : async (
127+ parent : DataIngestionsRow ,
128+ args : FilterableConnectionArgs & { tableName : string } ,
129+ context : GraphQLContext ,
130+ ) => {
131+ const { tableName, ...restArgs } = args ;
132+ return getMaterializations (
133+ restArgs ,
134+ context ,
135+ parent . org_id ,
136+ parent . dataset_id ,
137+ generateTableId ( "DATA_INGESTION" , tableName ) ,
138+ ) ;
139+ } ,
121140 } ,
122141} ;
0 commit comments