fix: segment paging (#3332)

Flagsmith · Jan 29, 2024 · 8050aed · 8050aed · vercel · Jan 29, 2024
1 parent 351232f
commit 8050aed
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/frontend/common/services/useSegment.ts b/frontend/common/services/useSegment.ts
@@ -2,6 +2,7 @@ import { Res } from 'common/types/responses'
 import { Req } from 'common/types/requests'
 import { service } from 'common/service'
 import Utils from 'common/utils/utils'
+import transformCorePaging from 'common/transformCorePaging'
 
 export const segmentService = service
   .enhanceEndpoints({ addTagTypes: ['Segment'] })
@@ -40,6 +41,13 @@ export const segmentService = service
         query: ({ projectId, ...rest }) => ({
           url: `projects/${projectId}/segments/?${Utils.toParam(rest)}`,
         }),
+        transformResponse: (
+          baseQueryReturnValue: Res['segments'],
+          meta,
+          req,
+        ) => {
+          return transformCorePaging(req, baseQueryReturnValue)
+        },
       }),
       updateSegment: builder.mutation<Res['segment'], Req['updateSegment']>({
         invalidatesTags: (q, e, arg) => [