Skip to content

Commit f413add

Browse files
michaeljs-cmichaeljs-c
committed
Update UI generated code
1 parent ebdbf31 commit f413add

File tree

7 files changed

+664
-288
lines changed

7 files changed

+664
-288
lines changed

‎airflow/ui/openapi-gen/queries/common.ts

Lines changed: 66 additions & 33 deletions
Original file line numberDiff line numberDiff line change
@@ -1393,6 +1393,72 @@ export const UseProviderServiceGetProvidersKeyFn = (
13931393
} = {},
13941394
queryKey?: Array<unknown>,
13951395
) => [useProviderServiceGetProvidersKey, ...(queryKey ?? [{ limit, offset }])];
1396+
export type XcomServiceGetXcomEntryDefaultResponse = Awaited<
1397+
ReturnType<typeof XcomService.getXcomEntry>
1398+
>;
1399+
export type XcomServiceGetXcomEntryQueryResult<
1400+
TData = XcomServiceGetXcomEntryDefaultResponse,
1401+
TError = unknown,
1402+
> = UseQueryResult<TData, TError>;
1403+
export const useXcomServiceGetXcomEntryKey = "XcomServiceGetXcomEntry";
1404+
export const UseXcomServiceGetXcomEntryKeyFn = (
1405+
{
1406+
dagId,
1407+
dagRunId,
1408+
deserialize,
1409+
mapIndex,
1410+
stringify,
1411+
taskId,
1412+
xcomKey,
1413+
}: {
1414+
dagId: string;
1415+
dagRunId: string;
1416+
deserialize?: boolean;
1417+
mapIndex?: number;
1418+
stringify?: boolean;
1419+
taskId: string;
1420+
xcomKey: string;
1421+
},
1422+
queryKey?: Array<unknown>,
1423+
) => [
1424+
useXcomServiceGetXcomEntryKey,
1425+
...(queryKey ?? [
1426+
{ dagId, dagRunId, deserialize, mapIndex, stringify, taskId, xcomKey },
1427+
]),
1428+
];
1429+
export type XcomServiceGetXcomEntriesDefaultResponse = Awaited<
1430+
ReturnType<typeof XcomService.getXcomEntries>
1431+
>;
1432+
export type XcomServiceGetXcomEntriesQueryResult<
1433+
TData = XcomServiceGetXcomEntriesDefaultResponse,
1434+
TError = unknown,
1435+
> = UseQueryResult<TData, TError>;
1436+
export const useXcomServiceGetXcomEntriesKey = "XcomServiceGetXcomEntries";
1437+
export const UseXcomServiceGetXcomEntriesKeyFn = (
1438+
{
1439+
dagId,
1440+
dagRunId,
1441+
limit,
1442+
mapIndex,
1443+
offset,
1444+
taskId,
1445+
xcomKey,
1446+
}: {
1447+
dagId: string;
1448+
dagRunId: string;
1449+
limit?: number;
1450+
mapIndex?: number;
1451+
offset?: number;
1452+
taskId: string;
1453+
xcomKey?: string;
1454+
},
1455+
queryKey?: Array<unknown>,
1456+
) => [
1457+
useXcomServiceGetXcomEntriesKey,
1458+
...(queryKey ?? [
1459+
{ dagId, dagRunId, limit, mapIndex, offset, taskId, xcomKey },
1460+
]),
1461+
];
13961462
export type TaskServiceGetTasksDefaultResponse = Awaited<
13971463
ReturnType<typeof TaskService.getTasks>
13981464
>;
@@ -1468,39 +1534,6 @@ export const UseVariableServiceGetVariablesKeyFn = (
14681534
useVariableServiceGetVariablesKey,
14691535
...(queryKey ?? [{ limit, offset, orderBy }]),
14701536
];
1471-
export type XcomServiceGetXcomEntryDefaultResponse = Awaited<
1472-
ReturnType<typeof XcomService.getXcomEntry>
1473-
>;
1474-
export type XcomServiceGetXcomEntryQueryResult<
1475-
TData = XcomServiceGetXcomEntryDefaultResponse,
1476-
TError = unknown,
1477-
> = UseQueryResult<TData, TError>;
1478-
export const useXcomServiceGetXcomEntryKey = "XcomServiceGetXcomEntry";
1479-
export const UseXcomServiceGetXcomEntryKeyFn = (
1480-
{
1481-
dagId,
1482-
dagRunId,
1483-
deserialize,
1484-
mapIndex,
1485-
stringify,
1486-
taskId,
1487-
xcomKey,
1488-
}: {
1489-
dagId: string;
1490-
dagRunId: string;
1491-
deserialize?: boolean;
1492-
mapIndex?: number;
1493-
stringify?: boolean;
1494-
taskId: string;
1495-
xcomKey: string;
1496-
},
1497-
queryKey?: Array<unknown>,
1498-
) => [
1499-
useXcomServiceGetXcomEntryKey,
1500-
...(queryKey ?? [
1501-
{ dagId, dagRunId, deserialize, mapIndex, stringify, taskId, xcomKey },
1502-
]),
1503-
];
15041537
export type MonitorServiceGetHealthDefaultResponse = Awaited<
15051538
ReturnType<typeof MonitorService.getHealth>
15061539
>;

‎airflow/ui/openapi-gen/queries/prefetch.ts

Lines changed: 112 additions & 55 deletions
Original file line numberDiff line numberDiff line change
@@ -1891,6 +1891,118 @@ export const prefetchUseProviderServiceGetProviders = (
18911891
queryKey: Common.UseProviderServiceGetProvidersKeyFn({ limit, offset }),
18921892
queryFn: () => ProviderService.getProviders({ limit, offset }),
18931893
});
1894+
/**
1895+
* Get Xcom Entry
1896+
* Get an XCom entry.
1897+
* @param data The data for the request.
1898+
* @param data.dagId
1899+
* @param data.taskId
1900+
* @param data.dagRunId
1901+
* @param data.xcomKey
1902+
* @param data.mapIndex
1903+
* @param data.deserialize
1904+
* @param data.stringify
1905+
* @returns unknown Successful Response
1906+
* @throws ApiError
1907+
*/
1908+
export const prefetchUseXcomServiceGetXcomEntry = (
1909+
queryClient: QueryClient,
1910+
{
1911+
dagId,
1912+
dagRunId,
1913+
deserialize,
1914+
mapIndex,
1915+
stringify,
1916+
taskId,
1917+
xcomKey,
1918+
}: {
1919+
dagId: string;
1920+
dagRunId: string;
1921+
deserialize?: boolean;
1922+
mapIndex?: number;
1923+
stringify?: boolean;
1924+
taskId: string;
1925+
xcomKey: string;
1926+
},
1927+
) =>
1928+
queryClient.prefetchQuery({
1929+
queryKey: Common.UseXcomServiceGetXcomEntryKeyFn({
1930+
dagId,
1931+
dagRunId,
1932+
deserialize,
1933+
mapIndex,
1934+
stringify,
1935+
taskId,
1936+
xcomKey,
1937+
}),
1938+
queryFn: () =>
1939+
XcomService.getXcomEntry({
1940+
dagId,
1941+
dagRunId,
1942+
deserialize,
1943+
mapIndex,
1944+
stringify,
1945+
taskId,
1946+
xcomKey,
1947+
}),
1948+
});
1949+
/**
1950+
* Get Xcom Entries
1951+
* Get all XCom entries.
1952+
*
1953+
* This endpoint allows specifying `~` as the dag_id, dag_run_id, task_id to retrieve XCom entries for all DAGs.
1954+
* @param data The data for the request.
1955+
* @param data.dagId
1956+
* @param data.dagRunId
1957+
* @param data.taskId
1958+
* @param data.xcomKey
1959+
* @param data.mapIndex
1960+
* @param data.limit
1961+
* @param data.offset
1962+
* @returns XComCollection Successful Response
1963+
* @throws ApiError
1964+
*/
1965+
export const prefetchUseXcomServiceGetXcomEntries = (
1966+
queryClient: QueryClient,
1967+
{
1968+
dagId,
1969+
dagRunId,
1970+
limit,
1971+
mapIndex,
1972+
offset,
1973+
taskId,
1974+
xcomKey,
1975+
}: {
1976+
dagId: string;
1977+
dagRunId: string;
1978+
limit?: number;
1979+
mapIndex?: number;
1980+
offset?: number;
1981+
taskId: string;
1982+
xcomKey?: string;
1983+
},
1984+
) =>
1985+
queryClient.prefetchQuery({
1986+
queryKey: Common.UseXcomServiceGetXcomEntriesKeyFn({
1987+
dagId,
1988+
dagRunId,
1989+
limit,
1990+
mapIndex,
1991+
offset,
1992+
taskId,
1993+
xcomKey,
1994+
}),
1995+
queryFn: () =>
1996+
XcomService.getXcomEntries({
1997+
dagId,
1998+
dagRunId,
1999+
limit,
2000+
mapIndex,
2001+
offset,
2002+
taskId,
2003+
xcomKey,
2004+
}),
2005+
});
18942006
/**
18952007
* Get Tasks
18962008
* Get tasks for DAG.
@@ -1987,61 +2099,6 @@ export const prefetchUseVariableServiceGetVariables = (
19872099
}),
19882100
queryFn: () => VariableService.getVariables({ limit, offset, orderBy }),
19892101
});
1990-
/**
1991-
* Get Xcom Entry
1992-
* Get an XCom entry.
1993-
* @param data The data for the request.
1994-
* @param data.dagId
1995-
* @param data.taskId
1996-
* @param data.dagRunId
1997-
* @param data.xcomKey
1998-
* @param data.mapIndex
1999-
* @param data.deserialize
2000-
* @param data.stringify
2001-
* @returns unknown Successful Response
2002-
* @throws ApiError
2003-
*/
2004-
export const prefetchUseXcomServiceGetXcomEntry = (
2005-
queryClient: QueryClient,
2006-
{
2007-
dagId,
2008-
dagRunId,
2009-
deserialize,
2010-
mapIndex,
2011-
stringify,
2012-
taskId,
2013-
xcomKey,
2014-
}: {
2015-
dagId: string;
2016-
dagRunId: string;
2017-
deserialize?: boolean;
2018-
mapIndex?: number;
2019-
stringify?: boolean;
2020-
taskId: string;
2021-
xcomKey: string;
2022-
},
2023-
) =>
2024-
queryClient.prefetchQuery({
2025-
queryKey: Common.UseXcomServiceGetXcomEntryKeyFn({
2026-
dagId,
2027-
dagRunId,
2028-
deserialize,
2029-
mapIndex,
2030-
stringify,
2031-
taskId,
2032-
xcomKey,
2033-
}),
2034-
queryFn: () =>
2035-
XcomService.getXcomEntry({
2036-
dagId,
2037-
dagRunId,
2038-
deserialize,
2039-
mapIndex,
2040-
stringify,
2041-
taskId,
2042-
xcomKey,
2043-
}),
2044-
});
20452102
/**
20462103
* Get Health
20472104
* @returns HealthInfoSchema Successful Response

0 commit comments

Comments
 (0)