From 686273048eb404e1948739f5bfebabf6c4a6c904 Mon Sep 17 00:00:00 2001
From: Andrew Kingston <andrew@kingston.dev>
Date: Tue, 10 Dec 2024 12:52:53 +0000
Subject: [PATCH] Update flag endpoints

---
 packages/frontend-core/src/api/flags.ts | 14 +++++++++-----
 1 file changed, 9 insertions(+), 5 deletions(-)

diff --git a/packages/frontend-core/src/api/flags.ts b/packages/frontend-core/src/api/flags.ts
index 55632182e9..1c4596787b 100644
--- a/packages/frontend-core/src/api/flags.ts
+++ b/packages/frontend-core/src/api/flags.ts
@@ -1,12 +1,16 @@
-import { Flags, SetUserFlagRequest, SetUserFlagResponse } from "@budibase/types"
+import {
+  Flags,
+  GetUserFlagsResponse,
+  SetUserFlagRequest,
+  SetUserFlagResponse,
+  ToggleBetaFeatureResponse,
+} from "@budibase/types"
 import { BaseAPIClient } from "./types"
 
 export interface FlagEndpoints {
-  getFlags: () => Promise<Flags>
-
-  // Missing request or response types
+  getFlags: () => Promise<GetUserFlagsResponse>
   updateFlag: (flag: string, value: any) => Promise<SetUserFlagResponse>
-  toggleUiFeature: (value: string) => Promise<{ message: string }>
+  toggleUiFeature: (value: string) => Promise<ToggleBetaFeatureResponse>
 }
 
 export const buildFlagEndpoints = (API: BaseAPIClient): FlagEndpoints => ({