diff --git a/api/src/engine/models/result/common/result-union.model.ts b/api/src/engine/models/result/common/result-union.model.ts index 8146c69d275fb1c91eff2ec63a92812ee8086f07..6cbcc444c664466348f68e464c4f78aeb23e7a31 100644 --- a/api/src/engine/models/result/common/result-union.model.ts +++ b/api/src/engine/models/result/common/result-union.model.ts @@ -1,18 +1,24 @@ import { createUnionType } from '@nestjs/graphql'; +import { GroupsResult } from '../groups-result.model'; import { RawResult } from '../raw-result.model'; import { TableResult } from '../table-result.model'; export const ResultUnion = createUnionType({ name: 'ResultUnion', - types: () => [TableResult, RawResult], + types: () => [TableResult, RawResult, GroupsResult], resolveType(value) { if (value.headers) { return TableResult; } + if (value.listMax) { return RawResult; } + if (value.group) { + return GroupsResult; + } + return null; }, });