From 97b0d8a2991bb8c6f1b3020df8a9e41c086aae92 Mon Sep 17 00:00:00 2001 From: Saumya Date: Wed, 8 Oct 2025 15:02:54 +0530 Subject: [PATCH] add missing tool views --- .../agents/agent-configuration-dialog.tsx | 30 +- .../dashboard/custom-agents-section.tsx | 8 +- .../discover-user-mcp-servers/_utils.ts | 146 ++++++++++ .../discover-user-mcp-servers.tsx | 264 +++++++++++++++++ .../thread/tool-views/update-agent/_utils.ts | 189 ++++++++++++ .../tool-views/update-agent/update-agent.tsx | 274 ++++++++++++++++++ .../src/components/thread/tool-views/utils.ts | 6 + .../tool-views/wrapper/ToolViewRegistry.tsx | 4 + frontend/src/components/thread/utils.ts | 6 +- .../hooks/react-query/agents/use-agents.ts | 2 + frontend/src/hooks/useAgentStream.ts | 51 ++-- .../src/lib/versioning/hooks/use-versions.ts | 4 + 12 files changed, 956 insertions(+), 28 deletions(-) create mode 100644 frontend/src/components/thread/tool-views/discover-user-mcp-servers/_utils.ts create mode 100644 frontend/src/components/thread/tool-views/discover-user-mcp-servers/discover-user-mcp-servers.tsx create mode 100644 frontend/src/components/thread/tool-views/update-agent/_utils.ts create mode 100644 frontend/src/components/thread/tool-views/update-agent/update-agent.tsx diff --git a/frontend/src/components/agents/agent-configuration-dialog.tsx b/frontend/src/components/agents/agent-configuration-dialog.tsx index 430d503a..e7890c18 100644 --- a/frontend/src/components/agents/agent-configuration-dialog.tsx +++ b/frontend/src/components/agents/agent-configuration-dialog.tsx @@ -50,6 +50,7 @@ import { useAgentVersionData } from '@/hooks/use-agent-version-data'; import { useUpdateAgent, useAgents } from '@/hooks/react-query/agents/use-agents'; import { useUpdateAgentMCPs } from '@/hooks/react-query/agents/use-update-agent-mcps'; import { useExportAgent } from '@/hooks/react-query/agents/use-agent-export-import'; +import { agentKeys } from '@/hooks/react-query/agents/keys'; import { ExpandableMarkdownEditor } from '@/components/ui/expandable-markdown-editor'; import { AgentModelSelector } from './config/model-selector'; import { AgentToolsConfiguration } from './agent-tools-configuration'; @@ -82,7 +83,11 @@ export function AgentConfigurationDialog({ const queryClient = useQueryClient(); const { agent, versionData, isViewingOldVersion, isLoading, error } = useAgentVersionData({ agentId }); - const { data: agentsResponse } = useAgents({}, { enabled: !!onAgentChange }); + const { data: agentsResponse, refetch: refetchAgents } = useAgents({}, { + enabled: !!onAgentChange, + refetchOnWindowFocus: true, + refetchOnMount: 'always' + }); const agents = agentsResponse?.agents || []; const updateAgentMutation = useUpdateAgent(); @@ -106,6 +111,29 @@ export function AgentConfigurationDialog({ } }, [open, initialTab]); + // Listen for query invalidations and refetch when agent data changes + useEffect(() => { + const unsubscribe = queryClient.getQueryCache().subscribe((event) => { + // Check if the invalidation is for this agent's data + if (event?.type === 'updated' && event?.query?.queryKey) { + const queryKey = event.query.queryKey; + + // Check if it's an agent-related query for our agentId + if ( + (Array.isArray(queryKey) && queryKey.includes(agentId)) || + (Array.isArray(queryKey) && queryKey.includes('agents') && queryKey.includes('detail')) || + (Array.isArray(queryKey) && queryKey.includes('versions')) + ) { + // Force a re-render by invalidating our specific queries + queryClient.invalidateQueries({ queryKey: agentKeys.detail(agentId) }); + queryClient.invalidateQueries({ queryKey: ['versions', 'list', agentId] }); + } + } + }); + + return () => unsubscribe(); + }, [agentId, queryClient]); + const [formData, setFormData] = useState({ name: '', system_prompt: '', diff --git a/frontend/src/components/dashboard/custom-agents-section.tsx b/frontend/src/components/dashboard/custom-agents-section.tsx index 3aede5f2..9300b9cf 100644 --- a/frontend/src/components/dashboard/custom-agents-section.tsx +++ b/frontend/src/components/dashboard/custom-agents-section.tsx @@ -167,11 +167,9 @@ export function CustomAgentsSection({ onAgentSelect }: CustomAgentsSectionProps)
{[1, 2, 3, 4, 5, 6, 7, 8].map((i) => ( -
- - - - +
+ +
))}
diff --git a/frontend/src/components/thread/tool-views/discover-user-mcp-servers/_utils.ts b/frontend/src/components/thread/tool-views/discover-user-mcp-servers/_utils.ts new file mode 100644 index 00000000..c98632e1 --- /dev/null +++ b/frontend/src/components/thread/tool-views/discover-user-mcp-servers/_utils.ts @@ -0,0 +1,146 @@ +import { parseToolResult } from '../tool-result-parser'; + +export interface McpTool { + name: string; + description: string; + inputSchema?: any; +} + +export interface ProfileInfo { + profile_name: string; + toolkit_name: string; + toolkit_slug: string; + is_connected: boolean; +} + +export interface DiscoverUserMcpServersData { + profile_id: string | null; + message: string | null; + profile_info: ProfileInfo | null; + tools: McpTool[]; + total_tools: number; + success?: boolean; + timestamp?: string; +} + +const parseContent = (content: any): any => { + if (typeof content === 'string') { + try { + return JSON.parse(content); + } catch (e) { + return content; + } + } + return content; +}; + +export function extractDiscoverUserMcpServersData( + assistantContent?: string, + toolContent?: any, + isSuccess?: boolean, + toolTimestamp?: string, + assistantTimestamp?: string +): DiscoverUserMcpServersData & { + actualIsSuccess: boolean; + actualToolTimestamp: string | undefined; + actualAssistantTimestamp: string | undefined; +} { + const defaultResult: DiscoverUserMcpServersData & { + actualIsSuccess: boolean; + actualToolTimestamp: string | undefined; + actualAssistantTimestamp: string | undefined; + } = { + profile_id: null, + message: null, + profile_info: null, + tools: [], + total_tools: 0, + actualIsSuccess: isSuccess || false, + actualToolTimestamp: toolTimestamp, + actualAssistantTimestamp: assistantTimestamp + }; + + try { + if (toolContent) { + let content = toolContent; + + if (typeof toolContent === 'string') { + try { + content = JSON.parse(toolContent); + } catch (e) { + content = toolContent; + } + } + + if (content && typeof content === 'object' && content.content) { + try { + const nestedContent = typeof content.content === 'string' ? JSON.parse(content.content) : content.content; + content = nestedContent; + } catch (e) { + } + } + + if (content && typeof content === 'object' && content.tool_execution) { + const toolExecution = content.tool_execution; + if (toolExecution.result && toolExecution.result.success) { + const args = toolExecution.arguments; + const output = toolExecution.result.output; + + if (args && output) { + return { + ...defaultResult, + profile_id: args.profile_id || null, + message: output.message || null, + profile_info: output.profile_info || null, + tools: output.tools || [], + total_tools: output.total_tools || 0, + actualIsSuccess: true + }; + } + } + } + + if (content && typeof content === 'object' && content.tool === 'discover-user-mcp-servers') { + const parameters = content.parameters; + const output = content.output; + + if (parameters && output) { + return { + ...defaultResult, + profile_id: parameters.profile_id || null, + message: output.message || null, + profile_info: output.profile_info || null, + tools: output.tools || [], + total_tools: output.total_tools || 0, + actualIsSuccess: output.success !== false + }; + } + } + } + + if (assistantContent) { + const parsed = parseToolResult(assistantContent); + if (parsed && parsed.isSuccess) { + const toolOutput = parseContent(parsed.toolOutput); + const args = parsed.arguments; + + if (args && toolOutput) { + return { + ...defaultResult, + profile_id: args.profile_id || null, + message: toolOutput.message || null, + profile_info: toolOutput.profile_info || null, + tools: toolOutput.tools || [], + total_tools: toolOutput.total_tools || 0, + actualIsSuccess: true + }; + } + } + } + + return defaultResult; + } catch (error) { + console.error('Error extracting discover user mcp servers data:', error); + return defaultResult; + } +} diff --git a/frontend/src/components/thread/tool-views/discover-user-mcp-servers/discover-user-mcp-servers.tsx b/frontend/src/components/thread/tool-views/discover-user-mcp-servers/discover-user-mcp-servers.tsx new file mode 100644 index 00000000..80cd8d35 --- /dev/null +++ b/frontend/src/components/thread/tool-views/discover-user-mcp-servers/discover-user-mcp-servers.tsx @@ -0,0 +1,264 @@ +import React from 'react'; +import { + Search, + CheckCircle, + AlertTriangle, + Plug, + Zap, + Package, + Link2, + Wrench, + ChevronRight, + Database +} from 'lucide-react'; +import { ToolViewProps } from '../types'; +import { formatTimestamp, getToolTitle } from '../utils'; +import { cn } from '@/lib/utils'; +import { Card, CardContent, CardHeader, CardTitle } from '@/components/ui/card'; +import { Badge } from '@/components/ui/badge'; +import { ScrollArea } from "@/components/ui/scroll-area"; +import { LoadingState } from '../shared/LoadingState'; +import { Separator } from "@/components/ui/separator"; +import { extractDiscoverUserMcpServersData, McpTool } from './_utils'; + +export function DiscoverUserMcpServersToolView({ + name = 'discover-user-mcp-servers', + assistantContent, + toolContent, + assistantTimestamp, + toolTimestamp, + isSuccess = true, + isStreaming = false, +}: ToolViewProps) { + + const { + profile_id, + message, + profile_info, + tools, + total_tools, + actualIsSuccess, + actualToolTimestamp, + actualAssistantTimestamp + } = extractDiscoverUserMcpServersData( + assistantContent, + toolContent, + isSuccess, + toolTimestamp, + assistantTimestamp + ); + + const toolTitle = getToolTitle(name); + + const formatToolName = (toolName: string): string => { + return toolName + .replace(/^LINEAR_/, '') + .replace(/_/g, ' ') + .split(' ') + .map(word => word.charAt(0) + word.slice(1).toLowerCase()) + .join(' '); + }; + + const getToolCategory = (toolName: string): string => { + if (toolName.includes('CREATE')) return 'Create'; + if (toolName.includes('UPDATE')) return 'Update'; + if (toolName.includes('DELETE') || toolName.includes('REMOVE')) return 'Delete'; + if (toolName.includes('GET') || toolName.includes('LIST')) return 'Read'; + if (toolName.includes('RUN')) return 'Advanced'; + return 'Other'; + }; + + const getCategoryIcon = (category: string) => { + switch (category) { + case 'Create': + return +; + case 'Update': + return ✎; + case 'Delete': + return ×; + case 'Read': + return ⊙; + case 'Advanced': + return ⚡; + default: + return •; + } + }; + + const groupToolsByCategory = (tools: McpTool[]) => { + const grouped: Record = {}; + tools.forEach(tool => { + const category = getToolCategory(tool.name); + if (!grouped[category]) { + grouped[category] = []; + } + grouped[category].push(tool); + }); + return grouped; + }; + + const groupedTools = groupToolsByCategory(tools); + const categoryOrder = ['Read', 'Create', 'Update', 'Delete', 'Advanced', 'Other']; + + return ( + + +
+
+
+ +
+
+ + {toolTitle} + +
+
+ {!isStreaming && ( + + {actualIsSuccess ? ( + + ) : ( + + )} + {actualIsSuccess ? 'Tools discovered' : 'Discovery failed'} + + )} +
+ + + {isStreaming ? ( + + ) : actualIsSuccess && profile_info ? ( + +
+
+
+
+
+ +
+
+

+ {profile_info.profile_name} +

+

+ {profile_info.toolkit_name} Integration +

+
+
+
+ + {profile_info.is_connected ? ( + <> + + Connected + + ) : 'Disconnected'} + +
+
+
+ + {tools.length > 0 && ( +
+
+
+ +

+ Discovered Tools +

+
+ + {total_tools} available + +
+ + + +
+ {categoryOrder.map(category => { + if (!groupedTools[category] || groupedTools[category].length === 0) return null; + + return ( +
+
+ + {getCategoryIcon(category)} + + {category} Operations + + {groupedTools[category].length} + +
+ +
+ {groupedTools[category].map((tool, index) => ( +
+
+
+
+ +

+ {formatToolName(tool.name)} +

+
+

+ {tool.description} +

+
+
+
+ ))} +
+
+ ); + })} +
+
+ )} + + {tools.length === 0 && ( +
+ +

+ No tools discovered for this profile +

+
+ )} +
+ + ) : ( +
+

+ + Failed to discover MCP tools. Please check the profile configuration. +

+
+ )} + + + ); +} diff --git a/frontend/src/components/thread/tool-views/update-agent/_utils.ts b/frontend/src/components/thread/tool-views/update-agent/_utils.ts new file mode 100644 index 00000000..8422bac9 --- /dev/null +++ b/frontend/src/components/thread/tool-views/update-agent/_utils.ts @@ -0,0 +1,189 @@ +import { parseToolResult } from '../tool-result-parser'; + +export interface UpdateAgentData { + name: string | null; + description?: string | null; + system_prompt: string | null; + agentpress_tools: Record | null; + configured_mcps?: any[] | null; + is_default?: boolean; + icon_name?: string | null; + icon_color?: string | null; + icon_background?: string | null; + agent?: { + agent_id: string; + account_id: string; + name: string; + description?: string | null; + is_default: boolean; + created_at: string; + updated_at: string; + is_public: boolean; + tags: string[]; + current_version_id: string; + version_count: number; + metadata: Record; + icon_name: string; + icon_color: string; + icon_background: string; + } | null; + updated_fields?: string[]; + version_created?: boolean; + message?: string; + success?: boolean; + timestamp?: string; +} + +const parseContent = (content: any): any => { + if (typeof content === 'string') { + try { + return JSON.parse(content); + } catch (e) { + return content; + } + } + return content; +}; + +export function extractUpdateAgentData( + assistantContent?: string, + toolContent?: any, + isSuccess?: boolean, + toolTimestamp?: string, + assistantTimestamp?: string +): UpdateAgentData & { + actualIsSuccess: boolean; + actualToolTimestamp: string | undefined; + actualAssistantTimestamp: string | undefined; +} { + const defaultResult: UpdateAgentData & { + actualIsSuccess: boolean; + actualToolTimestamp: string | undefined; + actualAssistantTimestamp: string | undefined; + } = { + name: null, + description: null, + system_prompt: null, + agentpress_tools: null, + configured_mcps: null, + is_default: false, + icon_name: null, + icon_color: null, + icon_background: null, + agent: null, + updated_fields: [], + version_created: false, + message: null, + actualIsSuccess: isSuccess || false, + actualToolTimestamp: toolTimestamp, + actualAssistantTimestamp: assistantTimestamp + }; + + try { + if (toolContent) { + let content = toolContent; + + if (typeof toolContent === 'string') { + try { + content = JSON.parse(toolContent); + } catch (e) { + content = toolContent; + } + } + + if (content && typeof content === 'object' && content.content) { + try { + const nestedContent = typeof content.content === 'string' ? JSON.parse(content.content) : content.content; + content = nestedContent; + } catch (e) { + } + } + + if (content && typeof content === 'object' && content.tool_execution) { + const toolExecution = content.tool_execution; + if (toolExecution.result && toolExecution.result.success) { + const args = toolExecution.arguments; + const output = toolExecution.result.output; + + if (args && output) { + return { + ...defaultResult, + name: args.name || null, + description: args.description || null, + system_prompt: args.system_prompt || null, + agentpress_tools: args.agentpress_tools || null, + configured_mcps: args.configured_mcps || null, + is_default: args.is_default || false, + icon_name: args.icon_name || null, + icon_color: args.icon_color || null, + icon_background: args.icon_background || null, + agent: output.agent || null, + updated_fields: output.updated_fields || [], + version_created: output.version_created || false, + message: output.message || null, + actualIsSuccess: true + }; + } + } + } + + if (content && typeof content === 'object' && content.tool === 'update-agent') { + const parameters = content.parameters; + const output = content.output; + + if (parameters && output) { + return { + ...defaultResult, + name: parameters.name || null, + description: parameters.description || null, + system_prompt: parameters.system_prompt || null, + agentpress_tools: parameters.agentpress_tools || null, + configured_mcps: parameters.configured_mcps || null, + is_default: parameters.is_default || false, + icon_name: parameters.icon_name || null, + icon_color: parameters.icon_color || null, + icon_background: parameters.icon_background || null, + agent: output.agent || null, + updated_fields: output.updated_fields || [], + version_created: output.version_created || false, + message: output.message || null, + actualIsSuccess: output.success !== false + }; + } + } + } + + if (assistantContent) { + const parsed = parseToolResult(assistantContent); + if (parsed && parsed.isSuccess) { + const toolOutput = parseContent(parsed.toolOutput); + const args = parsed.arguments; + + if (args && toolOutput) { + return { + ...defaultResult, + name: args.name || null, + description: args.description || null, + system_prompt: args.system_prompt || null, + agentpress_tools: args.agentpress_tools || null, + configured_mcps: args.configured_mcps || null, + is_default: args.is_default || false, + icon_name: args.icon_name || null, + icon_color: args.icon_color || null, + icon_background: args.icon_background || null, + agent: toolOutput.agent || null, + updated_fields: toolOutput.updated_fields || [], + version_created: toolOutput.version_created || false, + message: toolOutput.message || null, + actualIsSuccess: true + }; + } + } + } + + return defaultResult; + } catch (error) { + console.error('Error extracting update agent data:', error); + return defaultResult; + } +} diff --git a/frontend/src/components/thread/tool-views/update-agent/update-agent.tsx b/frontend/src/components/thread/tool-views/update-agent/update-agent.tsx new file mode 100644 index 00000000..f459c835 --- /dev/null +++ b/frontend/src/components/thread/tool-views/update-agent/update-agent.tsx @@ -0,0 +1,274 @@ +import React from 'react'; +import { + Bot, + CheckCircle, + AlertTriangle, + Calendar, + Sparkles, + User, + RefreshCw, + History, + Edit3 +} from 'lucide-react'; +import { ToolViewProps } from '../types'; +import { formatTimestamp, getToolTitle } from '../utils'; +import { cn } from '@/lib/utils'; +import { Card, CardContent, CardHeader, CardTitle } from '@/components/ui/card'; +import { Badge } from '@/components/ui/badge'; +import { ScrollArea } from "@/components/ui/scroll-area"; +import { LoadingState } from '../shared/LoadingState'; +import { Separator } from "@/components/ui/separator"; +import { extractUpdateAgentData } from './_utils'; +import { AgentAvatar } from '../../content/agent-avatar'; + +export function UpdateAgentToolView({ + name = 'update-agent', + assistantContent, + toolContent, + assistantTimestamp, + toolTimestamp, + isSuccess = true, + isStreaming = false, +}: ToolViewProps) { + + const { + name: agentName, + description, + system_prompt, + icon_name, + icon_color, + icon_background, + agentpress_tools, + configured_mcps, + is_default, + agent, + updated_fields, + version_created, + message, + actualIsSuccess, + actualToolTimestamp, + actualAssistantTimestamp + } = extractUpdateAgentData( + assistantContent, + toolContent, + isSuccess, + toolTimestamp, + assistantTimestamp + ); + + const toolTitle = getToolTitle(name); + + const getEnabledToolsCount = () => { + if (!agentpress_tools) return 0; + return Object.values(agentpress_tools).filter(Boolean).length; + }; + + const formatFieldName = (field: string): string => { + return field + .split('_') + .map(word => word.charAt(0).toUpperCase() + word.slice(1)) + .join(' '); + }; + + const displayName = agent?.name || agentName; + const displayIconName = agent?.icon_name || icon_name; + const displayIconColor = agent?.icon_color || icon_color; + const displayIconBackground = agent?.icon_background || icon_background; + + return ( + + +
+
+
+ +
+
+ + {toolTitle} + +
+
+ {!isStreaming && ( + + {actualIsSuccess ? ( + + ) : ( + + )} + {actualIsSuccess ? 'Agent updated' : 'Update failed'} + + )} +
+ + + {isStreaming ? ( + + ) : actualIsSuccess && agent ? ( + +
+
+
+
+ +
+

+ {displayName} +

+

+ Custom AI Agent +

+
+
+
+ {version_created && ( + + + v{agent.version_count} + + )} + {is_default && ( + + + Default + + )} + + Active + +
+
+ + {description && ( +
+

Description

+

+ {description} +

+
+ )} + + {updated_fields && updated_fields.length > 0 && ( + <> + +
+
+ + Updated Fields +
+
+ {updated_fields.map((field, index) => ( + + {formatFieldName(field)} + + ))} +
+ {version_created && ( +
+
+ + New version created (Version {agent.version_count}) +
+
+ )} +
+ + )} + + + +
+
+
+ + Created +
+

+ {new Date(agent.created_at).toLocaleDateString()} +

+
+
+
+ + Updated +
+

+ {new Date(agent.updated_at).toLocaleDateString()} at{' '} + {new Date(agent.updated_at).toLocaleTimeString()} +

+
+
+
+ + {system_prompt && ( +
+

+ + System Prompt Preview +

+
+ {system_prompt.substring(0, 200)} + {system_prompt.length > 200 && '...'} +
+
+ )} + + {agentpress_tools && ( +
+
+

+ Tool Configuration +

+ + {getEnabledToolsCount()} enabled + +
+
+ Agent has access to {getEnabledToolsCount()} tools out of {Object.keys(agentpress_tools).length} available tools +
+
+ )} + + {message && ( +
+

+ + {message} +

+
+ )} +
+ + ) : ( +
+

+ + Failed to update agent. Please try again. +

+
+ )} + + + ); +} diff --git a/frontend/src/components/thread/tool-views/utils.ts b/frontend/src/components/thread/tool-views/utils.ts index 95006e11..6ef9b8eb 100644 --- a/frontend/src/components/thread/tool-views/utils.ts +++ b/frontend/src/components/thread/tool-views/utils.ts @@ -78,10 +78,12 @@ export function getToolTitle(toolName: string): string { // Agent Creation Tools 'create-new-agent': 'Create New Agent', + 'update-agent': 'Update Agent', 'search-mcp-servers-for-agent': 'Search MCP Servers for Agent', 'get-mcp-server-details': 'Get MCP Server Details', 'create-credential-profile-for-agent': 'Create Credential Profile for Agent', 'discover-mcp-tools-for-agent': 'Discover MCP Tools for Agent', + 'discover-user-mcp-servers': 'Discovering tools', 'configure-agent-integration': 'Configure Agent Integration', 'list-available-integrations': 'List Available Integrations', 'create-agent-scheduled-trigger': 'Create Scheduled Trigger', @@ -1306,6 +1308,10 @@ export function getToolComponent(toolName: string): string { return 'GetCredentialProfilesToolView'; case 'get-current-agent-config': return 'GetCurrentAgentConfigToolView'; + case 'update-agent': + return 'UpdateAgentToolView'; + case 'discover-user-mcp-servers': + return 'DiscoverUserMcpServersToolView'; //Deploy case 'deploy': diff --git a/frontend/src/components/thread/tool-views/wrapper/ToolViewRegistry.tsx b/frontend/src/components/thread/tool-views/wrapper/ToolViewRegistry.tsx index 3d8eec1c..a988c8bb 100644 --- a/frontend/src/components/thread/tool-views/wrapper/ToolViewRegistry.tsx +++ b/frontend/src/components/thread/tool-views/wrapper/ToolViewRegistry.tsx @@ -47,9 +47,11 @@ import { DesignerToolView } from '../designer-tool/DesignerToolView'; import { UploadFileToolView } from '../UploadFileToolView'; import { DocsToolView, ListDocumentsToolView, DeleteDocumentToolView } from '../docs-tool'; import { CreateNewAgentToolView } from '../create-new-agent/create-new-agent'; +import { UpdateAgentToolView } from '../update-agent/update-agent'; import { SearchMcpServersForAgentToolView } from '../search-mcp-servers-for-agent/search-mcp-servers-for-agent'; import { CreateCredentialProfileForAgentToolView } from '../create-credential-profile-for-agent/create-credential-profile-for-agent'; import { DiscoverMcpToolsForAgentToolView } from '../discover-mcp-tools-for-agent/discover-mcp-tools-for-agent'; +import { DiscoverUserMcpServersToolView } from '../discover-user-mcp-servers/discover-user-mcp-servers'; import { ConfigureAgentIntegrationToolView } from '../configure-agent-integration/configure-agent-integration'; import CreateAgentScheduledTriggerToolView from '../create-agent-scheduled-trigger/create-agent-scheduled-trigger'; import { createPresentationViewerToolContent, parsePresentationSlidePath } from '../utils/presentation-utils'; @@ -189,9 +191,11 @@ const defaultRegistry: ToolViewRegistryType = { 'default': GenericToolView, 'create-new-agent': CreateNewAgentToolView, + 'update-agent': UpdateAgentToolView, 'search-mcp-servers-for-agent': SearchMcpServersForAgentToolView, 'create-credential-profile-for-agent': CreateCredentialProfileForAgentToolView, 'discover-mcp-tools-for-agent': DiscoverMcpToolsForAgentToolView, + 'discover-user-mcp-servers': DiscoverUserMcpServersToolView, 'configure-agent-integration': ConfigureAgentIntegrationToolView, 'create-agent-scheduled-trigger': CreateAgentScheduledTriggerToolView, }; diff --git a/frontend/src/components/thread/utils.ts b/frontend/src/components/thread/utils.ts index a9f8080d..d495218b 100644 --- a/frontend/src/components/thread/utils.ts +++ b/frontend/src/components/thread/utils.ts @@ -399,10 +399,12 @@ const TOOL_DISPLAY_NAMES = new Map([ ['update_agent', 'Updating Agent'], ['get_current_agent_config', 'Getting Agent Config'], ['search_mcp_servers', 'Searching MCP Servers'], - ['get_mcp_server_tools', 'Getting MCP Server Tools'], - ['configure_mcp_server', 'Configuring MCP Server'], ['get_popular_mcp_servers', 'Getting Popular MCP Servers'], ['test_mcp_server_connection', 'Testing MCP Server Connection'], + ['discover-user-mcp-servers', 'Discovering tools'], + ['create-credential-profile', 'Creating profile'], + ['get-credential-profiles', 'Getting profiles'], + ['configure-profile-for-agent', 'Adding tools to agent'], ['create-new-agent', 'Creating New Agent'], diff --git a/frontend/src/hooks/react-query/agents/use-agents.ts b/frontend/src/hooks/react-query/agents/use-agents.ts index aaa77873..9ed3fcdd 100644 --- a/frontend/src/hooks/react-query/agents/use-agents.ts +++ b/frontend/src/hooks/react-query/agents/use-agents.ts @@ -32,6 +32,8 @@ export const useAgent = (agentId: string) => { enabled: !!agentId, staleTime: 5 * 60 * 1000, gcTime: 10 * 60 * 1000, + refetchOnMount: 'always', + refetchOnWindowFocus: true, } )(); }; diff --git a/frontend/src/hooks/useAgentStream.ts b/frontend/src/hooks/useAgentStream.ts index c89d14d2..65298bda 100644 --- a/frontend/src/hooks/useAgentStream.ts +++ b/frontend/src/hooks/useAgentStream.ts @@ -263,31 +263,35 @@ export function useAgentStream( }); if (agentId) { + // Core agent data + queryClient.invalidateQueries({ queryKey: agentKeys.all }); queryClient.invalidateQueries({ queryKey: agentKeys.detail(agentId) }); queryClient.invalidateQueries({ queryKey: agentKeys.lists() }); + queryClient.invalidateQueries({ queryKey: agentKeys.details() }); + + // Agent tools and integrations queryClient.invalidateQueries({ queryKey: ['agent-tools', agentId] }); - - queryClient.invalidateQueries({ - queryKey: ['custom-mcp-tools', agentId], - }); + queryClient.invalidateQueries({ queryKey: ['agent-tools'] }); + + // MCP configurations + queryClient.invalidateQueries({ queryKey: ['custom-mcp-tools', agentId] }); + queryClient.invalidateQueries({ queryKey: ['custom-mcp-tools'] }); queryClient.invalidateQueries({ queryKey: composioKeys.mcpServers() }); - - queryClient.invalidateQueries({ - queryKey: composioKeys.profiles.all(), - }); - queryClient.invalidateQueries({ - queryKey: composioKeys.profiles.credentials(), - }); - + queryClient.invalidateQueries({ queryKey: composioKeys.profiles.all() }); + queryClient.invalidateQueries({ queryKey: composioKeys.profiles.credentials() }); + + // Triggers queryClient.invalidateQueries({ queryKey: ['triggers', agentId] }); - - queryClient.invalidateQueries({ - queryKey: knowledgeBaseKeys.agent(agentId), - }); - - // Invalidate versioning queries for agent config page + queryClient.invalidateQueries({ queryKey: ['triggers'] }); + + // Knowledge base + queryClient.invalidateQueries({ queryKey: knowledgeBaseKeys.agent(agentId) }); + queryClient.invalidateQueries({ queryKey: knowledgeBaseKeys.all }); + + queryClient.invalidateQueries({ queryKey: ['versions'] }); + queryClient.invalidateQueries({ queryKey: ['versions', 'list'] }); queryClient.invalidateQueries({ queryKey: ['versions', 'list', agentId] }); - // Invalidate current version details if available + queryClient.invalidateQueries({ queryKey: ['versions', 'detail'] }); queryClient.invalidateQueries({ queryKey: ['versions', 'detail'], predicate: (query) => { @@ -295,7 +299,14 @@ export function useAgentStream( } }); - console.log(`[useAgentStream] Invalidated agent queries for refetch instead of page reload - Agent ID: ${agentId}`); + // Invalidate any version store cache + queryClient.invalidateQueries({ queryKey: ['version-store'] }); + + // Force refetch of agent configuration data + queryClient.refetchQueries({ queryKey: agentKeys.detail(agentId) }); + queryClient.refetchQueries({ queryKey: ['versions', 'list', agentId] }); + + console.log(`[useAgentStream] Comprehensively invalidated and refetched all agent queries for Agent ID: ${agentId}`); } if ( diff --git a/frontend/src/lib/versioning/hooks/use-versions.ts b/frontend/src/lib/versioning/hooks/use-versions.ts index d8a1e23d..14d88240 100644 --- a/frontend/src/lib/versioning/hooks/use-versions.ts +++ b/frontend/src/lib/versioning/hooks/use-versions.ts @@ -36,6 +36,8 @@ export const useAgentVersions = (agentId: string) => { }, enabled: !!agentId, staleTime: 30000, + refetchOnMount: 'always', + refetchOnWindowFocus: true, }); }; @@ -50,6 +52,8 @@ export const useAgentVersion = (agentId: string, versionId: string | null | unde }, enabled: !!agentId && !!versionId, staleTime: 30000, + refetchOnMount: 'always', + refetchOnWindowFocus: true, }); };