fix(ui) Live update resources in resource main view

apps/sim/app/workspace/[workspaceId]/home/components/mothership-view/components/resource-registry/resource-registry.tsx

@@ -123,19 +123,20 @@ const RESOURCE_INVALIDATORS: Record<
  MothershipResourceType,
  (qc: QueryClient, workspaceId: string, resourceId: string) => void
 > = {
- table: (qc, wId, id) => {
- qc.invalidateQueries({ queryKey: tableKeys.list(wId) })
+ table: (qc, _wId, id) => {
+ qc.invalidateQueries({ queryKey: tableKeys.lists() })
  qc.invalidateQueries({ queryKey: tableKeys.detail(id) })
  },
  file: (qc, wId, id) => {
- qc.invalidateQueries({ queryKey: workspaceFilesKeys.list(wId) })
+ qc.invalidateQueries({ queryKey: workspaceFilesKeys.lists() })
  qc.invalidateQueries({ queryKey: workspaceFilesKeys.content(wId, id) })
+ qc.invalidateQueries({ queryKey: workspaceFilesKeys.storageInfo() })
  },
- workflow: (qc, wId) => {
- qc.invalidateQueries({ queryKey: workflowKeys.list(wId) })
+ workflow: (qc, _wId) => {
+ qc.invalidateQueries({ queryKey: workflowKeys.lists() })
  },
- knowledgebase: (qc, wId, id) => {
- qc.invalidateQueries({ queryKey: knowledgeKeys.list(wId) })
+ knowledgebase: (qc, _wId, id) => {
+ qc.invalidateQueries({ queryKey: knowledgeKeys.lists() })
  qc.invalidateQueries({ queryKey: knowledgeKeys.detail(id) })
  },
 }

apps/sim/app/workspace/[workspaceId]/home/hooks/use-chat.ts

@@ -8,6 +8,10 @@ import {
  reportManualRunToolStop,
 } from '@/lib/copilot/client-sse/run-tool-execution'
 import { MOTHERSHIP_CHAT_API_PATH } from '@/lib/copilot/constants'
+import {
+ extractResourcesFromToolResult,
+ isResourceToolName,
+} from '@/lib/copilot/resource-extraction'
 import { VFS_DIR_TO_RESOURCE } from '@/lib/copilot/resource-types'
 import { isWorkflowToolName } from '@/lib/copilot/workflow-tools'
 import { getNextWorkflowColor } from '@/lib/workflows/colors'
@@ -621,7 +625,7 @@ export function useChat(
  calledBy: activeSubagent,
  },
  })
- if (name === 'read') {
+ if (name === 'read' || isResourceToolName(name)) {
  const args = (data?.arguments ?? data?.input) as
  | Record<string, unknown>
  | undefined
@@ -720,6 +724,17 @@ export function useChat(
  })
  }
  }
+
+ if (tc.status === 'success' && isResourceToolName(tc.name)) {
+ const resources = extractResourcesFromToolResult(
+ tc.name,
+ toolArgsMap.get(id) as Record<string, unknown> | undefined,
+ tc.result?.output
+ )
+ for (const resource of resources) {
+ invalidateResourceQueries(queryClient, workspaceId, resource.type, resource.id)
+ }
+ }
  }
 
  break

apps/sim/lib/copilot/resource-extraction.ts

@@ -0,0 +1,248 @@
+import type { MothershipResource, MothershipResourceType } from '@/lib/copilot/resource-types'
+
+type ChatResource = MothershipResource
+type ResourceType = MothershipResourceType
+
+const RESOURCE_TOOL_NAMES = new Set([
+ 'user_table',
+ 'workspace_file',
+ 'create_workflow',
+ 'edit_workflow',
+ 'function_execute',
+ 'knowledge_base',
+ 'knowledge',
+])
+
+export function isResourceToolName(toolName: string): boolean {
+ return RESOURCE_TOOL_NAMES.has(toolName)
+}
+
+function asRecord(value: unknown): Record<string, unknown> {
+ return value && typeof value === 'object' ? (value as Record<string, unknown>) : {}
+}
+
+function getOperation(params: Record<string, unknown> | undefined): string | undefined {
+ const args = asRecord(params?.args)
+ return (args.operation ?? params?.operation) as string | undefined
+}
+
+const READ_ONLY_TABLE_OPS = new Set(['get', 'get_schema', 'get_row', 'query_rows'])
+const READ_ONLY_KB_OPS = new Set(['get', 'query', 'list_tags', 'get_tag_usage'])
+const READ_ONLY_KNOWLEDGE_ACTIONS = new Set(['listed', 'queried'])
+
+/**
+ * Extracts resource descriptors from a tool execution result.
+ * Returns one or more resources for tools that create/modify workspace entities.
+ * Read-only operations are excluded to avoid unnecessary cache invalidation.
+ */
+export function extractResourcesFromToolResult(
+ toolName: string,
+ params: Record<string, unknown> | undefined,
+ output: unknown
+): ChatResource[] {
+ if (!isResourceToolName(toolName)) return []
+
+ const result = asRecord(output)
+ const data = asRecord(result.data)
+
+ switch (toolName) {
+ case 'user_table': {
+ if (READ_ONLY_TABLE_OPS.has(getOperation(params) ?? '')) return []
+
+ if (result.tableId) {
+ return [
+ {
+ type: 'table',
+ id: result.tableId as string,
+ title: (result.tableName as string) || 'Table',
+ },
+ ]
+ }
+ if (result.fileId) {
+ return [
+ {
+ type: 'file',
+ id: result.fileId as string,
+ title: (result.fileName as string) || 'File',
+ },
+ ]
+ }
+ const table = asRecord(data.table)
+ if (table.id) {
+ return [{ type: 'table', id: table.id as string, title: (table.name as string) || 'Table' }]
+ }
+ const args = asRecord(params?.args)
+ const tableId =
+ (data.tableId as string) ?? (args.tableId as string) ?? (params?.tableId as string)
+ if (tableId) {
+ return [
+ { type: 'table', id: tableId as string, title: (data.tableName as string) || 'Table' },
+ ]
+ }
+ return []
+ }
+
+ case 'workspace_file': {
+ const file = asRecord(data.file)
+ if (file.id) {
+ return [{ type: 'file', id: file.id as string, title: (file.name as string) || 'File' }]
+ }
+ const fileId = (data.fileId as string) ?? (data.id as string)
+ if (fileId) {
+ const fileName = (data.fileName as string) || (data.name as string) || 'File'
+ return [{ type: 'file', id: fileId, title: fileName }]
+ }
+ return []
+ }
+
+ case 'function_execute': {
+ if (result.tableId) {
+ return [
+ {
+ type: 'table',
+ id: result.tableId as string,
+ title: (result.tableName as string) || 'Table',
+ },
+ ]
+ }
+ if (result.fileId) {
+ return [
+ {
+ type: 'file',
+ id: result.fileId as string,
+ title: (result.fileName as string) || 'File',
+ },
+ ]
+ }
+ return []
+ }
+
+ case 'create_workflow':
+ case 'edit_workflow': {
+ const workflowId =
+ (result.workflowId as string) ??
+ (data.workflowId as string) ??
+ (params?.workflowId as string)
+ if (workflowId) {
+ const workflowName =
+ (result.workflowName as string) ??
+ (data.workflowName as string) ??
+ (params?.workflowName as string) ??
+ 'Workflow'
+ return [{ type: 'workflow', id: workflowId, title: workflowName }]
+ }
+ return []
+ }
+
+ case 'knowledge_base': {
+ if (READ_ONLY_KB_OPS.has(getOperation(params) ?? '')) return []
+
+ const kbId =
+ (data.id as string) ??
+ (result.knowledgeBaseId as string) ??
+ (data.knowledgeBaseId as string) ??
+ (params?.knowledgeBaseId as string)
+ if (kbId) {
+ const kbName =
+ (data.name as string) ?? (result.knowledgeBaseName as string) ?? 'Knowledge Base'
+ return [{ type: 'knowledgebase', id: kbId, title: kbName }]
+ }
+ return []
+ }
+
+ case 'knowledge': {
+ const action = data.action as string | undefined
+ if (READ_ONLY_KNOWLEDGE_ACTIONS.has(action ?? '')) return []
+
+ const kbArray = data.knowledge_bases as Array<Record<string, unknown>> | undefined
+ if (!Array.isArray(kbArray)) return []
+ const resources: ChatResource[] = []
+ for (const kb of kbArray) {
+ const id = kb.id as string | undefined
+ if (id) {
+ resources.push({
+ type: 'knowledgebase',
+ id,
+ title: (kb.name as string) || 'Knowledge Base',
+ })
+ }
+ }
+ return resources
+ }
+
+ default:
+ return []
+ }
+}
+
+const DELETE_CAPABLE_TOOL_RESOURCE_TYPE: Record<string, ResourceType> = {
+ delete_workflow: 'workflow',
+ workspace_file: 'file',
+ user_table: 'table',
+ knowledge_base: 'knowledgebase',
+}
+
+export function hasDeleteCapability(toolName: string): boolean {
+ return toolName in DELETE_CAPABLE_TOOL_RESOURCE_TYPE
+}
+
+/**
+ * Extracts resource descriptors from a tool execution result when the tool
+ * performed a deletion. Returns one or more deleted resources for tools that
+ * destroy workspace entities.
+ */
+export function extractDeletedResourcesFromToolResult(
+ toolName: string,
+ params: Record<string, unknown> | undefined,
+ output: unknown
+): ChatResource[] {
+ const resourceType = DELETE_CAPABLE_TOOL_RESOURCE_TYPE[toolName]
+ if (!resourceType) return []
+
+ const result = asRecord(output)
+ const data = asRecord(result.data)
+ const args = asRecord(params?.args)
+ const operation = (args.operation ?? params?.operation) as string | undefined
+
+ switch (toolName) {
+ case 'delete_workflow': {
+ const workflowId = (result.workflowId as string) ?? (params?.workflowId as string)
+ if (workflowId && result.deleted) {
+ return [
+ { type: resourceType, id: workflowId, title: (result.name as string) || 'Workflow' },
+ ]
+ }
+ return []
+ }
+
+ case 'workspace_file': {
+ if (operation !== 'delete') return []
+ const fileId = (data.id as string) ?? (args.fileId as string)
+ if (fileId) {
+ return [{ type: resourceType, id: fileId, title: (data.name as string) || 'File' }]
+ }
+ return []
+ }
+
+ case 'user_table': {
+ if (operation !== 'delete') return []
+ const tableId = (args.tableId as string) ?? (params?.tableId as string)
+ if (tableId) {
+ return [{ type: resourceType, id: tableId, title: 'Table' }]
+ }
+ return []
+ }
+
+ case 'knowledge_base': {
+ if (operation !== 'delete') return []
+ const kbId = (data.id as string) ?? (args.knowledgeBaseId as string)
+ if (kbId) {
+ return [{ type: resourceType, id: kbId, title: (data.name as string) || 'Knowledge Base' }]
+ }
+ return []
+ }
+
+ default:
+ return []
+ }
+}