Skip to content

Commit 9c87a2c

Browse files
create second hook with different key for settings nav
1 parent 99713cf commit 9c87a2c

File tree

6 files changed

+23
-9
lines changed

6 files changed

+23
-9
lines changed

ui/src/api/ampMetadataApi.ts

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -160,10 +160,17 @@ export interface ProjectConfig {
160160
cdp_auth_token?: string;
161161
}
162162

163-
export const useGetAmpConfig = (poll?: boolean) => {
163+
export const useGetAmpConfig = () => {
164164
return useQuery({
165165
queryKey: [QueryKeys.getAmpConfig],
166166
queryFn: getAmpConfig,
167+
});
168+
};
169+
170+
export const useGetPollingAmpConfig = (poll?: boolean) => {
171+
return useQuery({
172+
queryKey: [QueryKeys.getPollingAmpConfig],
173+
queryFn: getAmpConfig,
167174
refetchInterval: () => {
168175
if (poll) {
169176
return 1000;

ui/src/api/utils.ts

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -112,6 +112,7 @@ export enum QueryKeys {
112112
"getSessionsForProject" = "getSessionsForProject",
113113
"getAmpConfig" = "getAmpConfig",
114114
"getTools" = "getTools",
115+
"getPollingAmpConfig" = "getPollingAmpConfig",
115116
}
116117

117118
export const commonHeaders = {

ui/src/layout/Sidebar.tsx

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -59,7 +59,7 @@ import LightbulbIcon from "src/cuix/icons/LightbulbIcon";
5959
import { cdlAmber200, cdlAmber900 } from "src/cuix/variables.ts";
6060
import "./style.css";
6161
import AmpUpdateBanner from "src/components/AmpUpdate/AmpUpdateBanner.tsx";
62-
import { useGetAmpConfig } from "src/api/ampMetadataApi.ts";
62+
import { useGetPollingAmpConfig } from "src/api/ampMetadataApi.ts";
6363
import { getItem } from "./TopNav";
6464

6565
const { Sider } = Layout;
@@ -71,7 +71,7 @@ const Sidebar: React.FC = () => {
7171
const matchRoute = useMatchRoute();
7272
const navigate = useNavigate();
7373
const ref = useRef<HTMLDivElement>(null);
74-
const { data: config } = useGetAmpConfig();
74+
const { data: config } = useGetPollingAmpConfig();
7575

7676
const navToRagApp = () => {
7777
navigate({ to: "/chats" }).catch(() => null);

ui/src/layout/TopNav.tsx

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -50,12 +50,15 @@ import { cdlAmber200, cdlAmber900, cdlSlate800 } from "src/cuix/variables.ts";
5050
import AmpUpdateBanner from "src/components/AmpUpdate/AmpUpdateBanner.tsx";
5151

5252
import "./style.css";
53-
import { ProjectConfig, useGetAmpConfig } from "src/api/ampMetadataApi.ts";
53+
import {
54+
ProjectConfig,
55+
useGetPollingAmpConfig,
56+
} from "src/api/ampMetadataApi.ts";
5457

5558
const TopNav: React.FC = () => {
5659
const matchRoute = useMatchRoute();
5760
const navigate = useNavigate();
58-
const { data: config } = useGetAmpConfig();
61+
const { data: config } = useGetPollingAmpConfig();
5962

6063
const navigateTo = (path: string) => () => {
6164
navigate({ to: path }).catch(() => null);

ui/src/pages/Settings/AmpSettingsPage.tsx

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,10 @@
3636
* DATA.
3737
******************************************************************************/
3838
import { Alert, Button, Flex, Form, Typography } from "antd";
39-
import { ProjectConfig, useGetAmpConfig } from "src/api/ampMetadataApi.ts";
39+
import {
40+
ProjectConfig,
41+
useGetPollingAmpConfig,
42+
} from "src/api/ampMetadataApi.ts";
4043
import { ReactNode, useState } from "react";
4144
import { ModelSource, useGetModelSource } from "src/api/modelsApi.ts";
4245
import messageQueue from "src/utils/messageQueue.ts";
@@ -64,7 +67,7 @@ const AmpSettingsPage = () => {
6467
const [form] = Form.useForm<ProjectConfig>();
6568
const { data: currentModelSource } = useGetModelSource();
6669
const confirmationModal = useModal();
67-
const { data: projectConfig } = useGetAmpConfig();
70+
const { data: projectConfig } = useGetPollingAmpConfig();
6871
const [selectedFileStorage, setSelectedFileStorage] = useState<FileStorage>(
6972
projectConfig?.aws_config.document_bucket_name ? "AWS" : "Local",
7073
);

ui/src/pages/Settings/RestartAppModal.tsx

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@
3838
import { Button, Flex, FormInstance, Modal, Progress, Typography } from "antd";
3939
import {
4040
ProjectConfig,
41-
useGetAmpConfig,
41+
useGetPollingAmpConfig,
4242
useRestartApplication,
4343
useUpdateAmpConfig,
4444
} from "src/api/ampMetadataApi.ts";
@@ -144,7 +144,7 @@ const RestartAppModal = ({
144144
restartApplication.mutate({});
145145
},
146146
});
147-
const { data: config } = useGetAmpConfig(polling);
147+
const { data: config } = useGetPollingAmpConfig(polling);
148148

149149
const isRestarting = !config && polling;
150150

0 commit comments

Comments
 (0)