Feat: Replace the link of the old version of the agent module #3221 (#9130)

### What problem does this PR solve?
Feat: Automatically save agent canvas content
Feat: Replace the link of the old version of the agent module #3221
### Type of change


- [x] New Feature (non-breaking change which adds functionality)
This commit is contained in:
balibabu
2025-07-31 12:14:00 +08:00
committed by GitHub
parent e9c5c7bc7c
commit 0327fd848e
13 changed files with 77 additions and 44 deletions

View File

@ -21,6 +21,7 @@ import React, { ReactNode, useEffect, useState } from 'react';
import { ThemeProvider, useTheme } from './components/theme-provider'; import { ThemeProvider, useTheme } from './components/theme-provider';
import { SidebarProvider } from './components/ui/sidebar'; import { SidebarProvider } from './components/ui/sidebar';
import { TooltipProvider } from './components/ui/tooltip'; import { TooltipProvider } from './components/ui/tooltip';
import { ThemeEnum } from './constants/common';
import storage from './utils/authorization-util'; import storage from './utils/authorization-util';
dayjs.extend(customParseFormat); dayjs.extend(customParseFormat);
@ -101,7 +102,10 @@ const RootProvider = ({ children }: React.PropsWithChildren) => {
return ( return (
<TooltipProvider> <TooltipProvider>
<QueryClientProvider client={queryClient}> <QueryClientProvider client={queryClient}>
<ThemeProvider defaultTheme="light" storageKey="ragflow-ui-theme"> <ThemeProvider
defaultTheme={ThemeEnum.Light}
storageKey="ragflow-ui-theme"
>
<Root>{children}</Root> <Root>{children}</Root>
</ThemeProvider> </ThemeProvider>
</QueryClientProvider> </QueryClientProvider>

View File

@ -1,20 +1,19 @@
import { ThemeEnum } from '@/constants/common';
import React, { createContext, useContext, useEffect, useState } from 'react'; import React, { createContext, useContext, useEffect, useState } from 'react';
type Theme = 'dark' | 'light' | 'system';
type ThemeProviderProps = { type ThemeProviderProps = {
children: React.ReactNode; children: React.ReactNode;
defaultTheme?: Theme; defaultTheme?: ThemeEnum;
storageKey?: string; storageKey?: string;
}; };
type ThemeProviderState = { type ThemeProviderState = {
theme: Theme; theme: ThemeEnum;
setTheme: (theme: Theme) => void; setTheme: (theme: ThemeEnum) => void;
}; };
const initialState: ThemeProviderState = { const initialState: ThemeProviderState = {
theme: 'light', theme: ThemeEnum.Light,
setTheme: () => null, setTheme: () => null,
}; };
@ -22,17 +21,17 @@ const ThemeProviderContext = createContext<ThemeProviderState>(initialState);
export function ThemeProvider({ export function ThemeProvider({
children, children,
defaultTheme = 'light', defaultTheme = ThemeEnum.Light,
storageKey = 'vite-ui-theme', storageKey = 'vite-ui-theme',
...props ...props
}: ThemeProviderProps) { }: ThemeProviderProps) {
const [theme, setTheme] = useState<Theme>( const [theme, setTheme] = useState<ThemeEnum>(
() => (localStorage.getItem(storageKey) as Theme) || defaultTheme, () => (localStorage.getItem(storageKey) as ThemeEnum) || defaultTheme,
); );
useEffect(() => { useEffect(() => {
const root = window.document.documentElement; const root = window.document.documentElement;
root.classList.remove('light', 'dark'); root.classList.remove(ThemeEnum.Light, ThemeEnum.Dark);
localStorage.setItem(storageKey, theme); localStorage.setItem(storageKey, theme);
root.classList.add(theme); root.classList.add(theme);
}, [storageKey, theme]); }, [storageKey, theme]);
@ -62,5 +61,13 @@ export const useTheme = () => {
export const useIsDarkTheme = () => { export const useIsDarkTheme = () => {
const { theme } = useTheme(); const { theme } = useTheme();
return theme === 'dark'; return theme === ThemeEnum.Dark;
}; };
export function useSwitchToDarkThemeOnMount() {
const { setTheme } = useTheme();
useEffect(() => {
setTheme(ThemeEnum.Dark);
}, [setTheme]);
}

View File

@ -150,3 +150,9 @@ export enum Platform {
FastGPT = 'FastGPT', FastGPT = 'FastGPT',
Coze = 'Coze', Coze = 'Coze',
} }
export enum ThemeEnum {
Dark = 'dark',
Light = 'light',
System = 'system',
}

View File

@ -266,7 +266,7 @@ export const useResetAgent = () => {
return { data, loading, resetAgent: mutateAsync }; return { data, loading, resetAgent: mutateAsync };
}; };
export const useSetAgent = () => { export const useSetAgent = (showMessage: boolean = true) => {
const queryClient = useQueryClient(); const queryClient = useQueryClient();
const { const {
data, data,
@ -282,9 +282,11 @@ export const useSetAgent = () => {
}) => { }) => {
const { data = {} } = await agentService.setCanvas(params); const { data = {} } = await agentService.setCanvas(params);
if (data.code === 0) { if (data.code === 0) {
message.success( if (showMessage) {
i18n.t(`message.${params?.id ? 'modified' : 'created'}`), message.success(
); i18n.t(`message.${params?.id ? 'modified' : 'created'}`),
);
}
queryClient.invalidateQueries({ queryClient.invalidateQueries({
queryKey: [AgentApiAction.FetchAgentList], queryKey: [AgentApiAction.FetchAgentList],
}); });

View File

@ -29,7 +29,7 @@ const RagHeader = () => {
{ path: '/knowledge', name: t('knowledgeBase'), icon: KnowledgeBaseIcon }, { path: '/knowledge', name: t('knowledgeBase'), icon: KnowledgeBaseIcon },
{ path: '/chat', name: t('chat'), icon: MessageOutlined }, { path: '/chat', name: t('chat'), icon: MessageOutlined },
{ path: '/search', name: t('search'), icon: SearchOutlined }, { path: '/search', name: t('search'), icon: SearchOutlined },
{ path: '/flow', name: t('flow'), icon: GraphIcon }, { path: '/agent-list', name: t('flow'), icon: GraphIcon },
{ path: '/file', name: t('fileManager'), icon: FileIcon }, { path: '/file', name: t('fileManager'), icon: FileIcon },
], ],
[t], [t],

View File

@ -6,7 +6,7 @@ import React, { useCallback, useMemo } from 'react';
import User from '../user'; import User from '../user';
import { useTheme } from '@/components/theme-provider'; import { useTheme } from '@/components/theme-provider';
import { LanguageList, LanguageMap } from '@/constants/common'; import { LanguageList, LanguageMap, ThemeEnum } from '@/constants/common';
import { useChangeLanguage } from '@/hooks/logic-hooks'; import { useChangeLanguage } from '@/hooks/logic-hooks';
import { useFetchUserInfo, useListTenant } from '@/hooks/user-setting-hooks'; import { useFetchUserInfo, useListTenant } from '@/hooks/user-setting-hooks';
import { TenantRole } from '@/pages/user-setting/constants'; import { TenantRole } from '@/pages/user-setting/constants';
@ -58,10 +58,10 @@ const RightToolBar = () => {
}, []); }, []);
const onMoonClick = React.useCallback(() => { const onMoonClick = React.useCallback(() => {
setTheme('light'); setTheme(ThemeEnum.Light);
}, [setTheme]); }, [setTheme]);
const onSunClick = React.useCallback(() => { const onSunClick = React.useCallback(() => {
setTheme('dark'); setTheme(ThemeEnum.Dark);
}, [setTheme]); }, [setTheme]);
const handleBellClick = useCallback(() => { const handleBellClick = useCallback(() => {

View File

@ -9,7 +9,7 @@ import {
DropdownMenuTrigger, DropdownMenuTrigger,
} from '@/components/ui/dropdown-menu'; } from '@/components/ui/dropdown-menu';
import { Segmented, SegmentedValue } from '@/components/ui/segmented'; import { Segmented, SegmentedValue } from '@/components/ui/segmented';
import { LanguageList, LanguageMap } from '@/constants/common'; import { LanguageList, LanguageMap, ThemeEnum } from '@/constants/common';
import { useChangeLanguage } from '@/hooks/logic-hooks'; import { useChangeLanguage } from '@/hooks/logic-hooks';
import { useNavigatePage } from '@/hooks/logic-hooks/navigate-hooks'; import { useNavigatePage } from '@/hooks/logic-hooks/navigate-hooks';
import { useNavigateWithFromState } from '@/hooks/route-hook'; import { useNavigateWithFromState } from '@/hooks/route-hook';
@ -67,7 +67,7 @@ export function Header() {
})); }));
const onThemeClick = React.useCallback(() => { const onThemeClick = React.useCallback(() => {
setTheme(theme === 'dark' ? 'light' : 'dark'); setTheme(theme === ThemeEnum.Dark ? ThemeEnum.Light : ThemeEnum.Dark);
}, [setTheme, theme]); }, [setTheme, theme]);
const handleBellClick = useCallback(() => { const handleBellClick = useCallback(() => {

View File

@ -4,16 +4,16 @@ import {
useSetAgent, useSetAgent,
} from '@/hooks/use-agent-request'; } from '@/hooks/use-agent-request';
import { RAGFlowNodeType } from '@/interfaces/database/flow'; import { RAGFlowNodeType } from '@/interfaces/database/flow';
import { formatDate } from '@/utils/date';
import { useDebounceEffect } from 'ahooks'; import { useDebounceEffect } from 'ahooks';
import dayjs from 'dayjs';
import { useCallback, useEffect, useState } from 'react'; import { useCallback, useEffect, useState } from 'react';
import { useParams } from 'umi'; import { useParams } from 'umi';
import useGraphStore from '../store'; import useGraphStore from '../store';
import { useBuildDslData } from './use-build-dsl'; import { useBuildDslData } from './use-build-dsl';
export const useSaveGraph = () => { export const useSaveGraph = (showMessage: boolean = true) => {
const { data } = useFetchAgent(); const { data } = useFetchAgent();
const { setAgent, loading } = useSetAgent(); const { setAgent, loading } = useSetAgent(showMessage);
const { id } = useParams(); const { id } = useParams();
const { buildDslData } = useBuildDslData(); const { buildDslData } = useBuildDslData();
@ -57,11 +57,11 @@ export const useWatchAgentChange = (chatDrawerVisible: boolean) => {
const [time, setTime] = useState<string>(); const [time, setTime] = useState<string>();
const nodes = useGraphStore((state) => state.nodes); const nodes = useGraphStore((state) => state.nodes);
const edges = useGraphStore((state) => state.edges); const edges = useGraphStore((state) => state.edges);
const { saveGraph } = useSaveGraph(); const { saveGraph } = useSaveGraph(false);
const { data: flowDetail } = useFetchAgent(); const { data: flowDetail } = useFetchAgent();
const setSaveTime = useCallback((updateTime: number) => { const setSaveTime = useCallback((updateTime: number) => {
setTime(dayjs(updateTime).format('YYYY-MM-DD HH:mm:ss')); setTime(formatDate(updateTime));
}, []); }, []);
useEffect(() => { useEffect(() => {
@ -77,7 +77,7 @@ export const useWatchAgentChange = (chatDrawerVisible: boolean) => {
useDebounceEffect( useDebounceEffect(
() => { () => {
// saveAgent(); saveAgent();
}, },
[nodes, edges], [nodes, edges],
{ {

View File

@ -1,4 +1,5 @@
import { PageHeader } from '@/components/page-header'; import { PageHeader } from '@/components/page-header';
import { useSwitchToDarkThemeOnMount } from '@/components/theme-provider';
import { import {
Breadcrumb, Breadcrumb,
BreadcrumbItem, BreadcrumbItem,
@ -40,6 +41,7 @@ import { useGetBeginNodeDataInputs } from './hooks/use-get-begin-query';
import { import {
useSaveGraph, useSaveGraph,
useSaveGraphBeforeOpeningDebugDrawer, useSaveGraphBeforeOpeningDebugDrawer,
useWatchAgentChange,
} from './hooks/use-save-graph'; } from './hooks/use-save-graph';
import { useShowEmbedModal } from './hooks/use-show-dialog'; import { useShowEmbedModal } from './hooks/use-show-dialog';
import { UploadAgentDialog } from './upload-agent-dialog'; import { UploadAgentDialog } from './upload-agent-dialog';
@ -94,23 +96,31 @@ export default function Agent() {
const { showEmbedModal, hideEmbedModal, embedVisible, beta } = const { showEmbedModal, hideEmbedModal, embedVisible, beta } =
useShowEmbedModal(); useShowEmbedModal();
const { navigateToAgentLogs } = useNavigatePage(); const { navigateToAgentLogs } = useNavigatePage();
const time = useWatchAgentChange(chatDrawerVisible);
useSwitchToDarkThemeOnMount();
return ( return (
<section className="h-full"> <section className="h-full">
<PageHeader> <PageHeader>
<Breadcrumb> <section>
<BreadcrumbList> <Breadcrumb>
<BreadcrumbItem> <BreadcrumbList>
<BreadcrumbLink onClick={navigateToAgentList}> <BreadcrumbItem>
Agent <BreadcrumbLink onClick={navigateToAgentList}>
</BreadcrumbLink> Agent
</BreadcrumbItem> </BreadcrumbLink>
<BreadcrumbSeparator /> </BreadcrumbItem>
<BreadcrumbItem> <BreadcrumbSeparator />
<BreadcrumbPage>{agentDetail.title}</BreadcrumbPage> <BreadcrumbItem>
</BreadcrumbItem> <BreadcrumbPage>{agentDetail.title}</BreadcrumbPage>
</BreadcrumbList> </BreadcrumbItem>
</Breadcrumb> </BreadcrumbList>
</Breadcrumb>
<div className="text-xs text-text-sub-title translate-y-3">
{t('flow.autosaved')} {time}
</div>
</section>
<div className="flex items-center gap-5"> <div className="flex items-center gap-5">
<ButtonLoading <ButtonLoading
variant={'secondary'} variant={'secondary'}

View File

@ -3,6 +3,7 @@ import { NextMessageInput } from '@/components/message-input/next';
import MessageItem from '@/components/next-message-item'; import MessageItem from '@/components/next-message-item';
import PdfDrawer from '@/components/pdf-drawer'; import PdfDrawer from '@/components/pdf-drawer';
import { useClickDrawer } from '@/components/pdf-drawer/hooks'; import { useClickDrawer } from '@/components/pdf-drawer/hooks';
import { useSwitchToDarkThemeOnMount } from '@/components/theme-provider';
import { MessageType } from '@/constants/chat'; import { MessageType } from '@/constants/chat';
import { import {
useFetchExternalAgentInputs, useFetchExternalAgentInputs,
@ -91,6 +92,8 @@ const ChatContainer = () => {
} }
}, [data, showParameterDialog]); }, [data, showParameterDialog]);
useSwitchToDarkThemeOnMount();
if (!conversationId) { if (!conversationId) {
return <div>empty</div>; return <div>empty</div>;
} }

View File

@ -32,7 +32,7 @@ export default function McpServer() {
); );
return ( return (
<section className="p-4"> <section className="p-4 w-full">
<div className="text-text-title text-2xl">MCP Servers</div> <div className="text-text-title text-2xl">MCP Servers</div>
<section className="flex items-center justify-between pb-5"> <section className="flex items-center justify-between pb-5">
<div className="text-text-sub-title"> <div className="text-text-sub-title">

View File

@ -2,6 +2,7 @@ import { useIsDarkTheme, useTheme } from '@/components/theme-provider';
import { Button } from '@/components/ui/button'; import { Button } from '@/components/ui/button';
import { Label } from '@/components/ui/label'; import { Label } from '@/components/ui/label';
import { Switch } from '@/components/ui/switch'; import { Switch } from '@/components/ui/switch';
import { ThemeEnum } from '@/constants/common';
import { useLogout } from '@/hooks/login-hooks'; import { useLogout } from '@/hooks/login-hooks';
import { useSecondPathName } from '@/hooks/route-hook'; import { useSecondPathName } from '@/hooks/route-hook';
import { cn } from '@/lib/utils'; import { cn } from '@/lib/utils';
@ -60,7 +61,7 @@ export function SideBar() {
const handleThemeChange = useCallback( const handleThemeChange = useCallback(
(checked: boolean) => { (checked: boolean) => {
setTheme(checked ? 'dark' : 'light'); setTheme(checked ? ThemeEnum.Dark : ThemeEnum.Light);
}, },
[setTheme], [setTheme],
); );

View File

@ -81,7 +81,7 @@
--text-sub-title: rgba(151, 154, 171, 1); --text-sub-title: rgba(151, 154, 171, 1);
--text-sub-title-invert: rgba(91, 93, 106, 1); --text-sub-title-invert: rgba(91, 93, 106, 1);
--background-header-bar: rgba(11, 11, 12, 1); --background-header-bar: rgba(0, 0, 0, 0.05);
--text-title-invert: rgba(255, 255, 255, 1); --text-title-invert: rgba(255, 255, 255, 1);
--background-card: rgba(22, 22, 24, 0.05); --background-card: rgba(22, 22, 24, 0.05);