fix(gauge): preserve configured agent filters

Author: waleedlatif1

apps/docs/content/docs/integrations/gauge.mdx

@@ -567,7 +567,7 @@ Change the name or color of a prompt tag.
 | `org` | string | Yes | Organization slug or UUID |
 | `tagId` | string | Yes | Tag ID |
 | `name` | string | No | Tag name \(1–120 characters\) |
-| `color` | string | No | Tag color string \(up to 20 characters\) |
+| `color` | string | No | Tag color string \(up to 20 characters\); omit to keep unchanged or use "" to clear |
 
 #### Output
 

apps/sim/blocks/blocks/gauge.ts

@@ -568,6 +568,7 @@ export const GaugeBlock: BlockConfig = {
       id: 'promptTrackingType',
       title: 'Tracking Type',
       type: 'dropdown',
+      paramVisibility: 'user-or-llm',
       condition: {
         field: 'operation',
         value: ['gauge_bulk_update_prompts', 'gauge_update_prompt'],
@@ -776,6 +777,7 @@ export const GaugeBlock: BlockConfig = {
       id: 'metadataDataset',
       title: 'Dataset',
       type: 'dropdown',
+      paramVisibility: 'user-or-llm',
       condition: { field: 'operation', value: 'gauge_get_query_metadata' },
       options: [
         { label: 'All Datasets', id: 'all' },
@@ -969,27 +971,20 @@ export const GaugeBlock: BlockConfig = {
     config: {
       tool: (params) => params.operation,
       params: (params) => {
-        const patch = ['gauge_bulk_update_prompts', 'gauge_update_prompt'].includes(
-          params.operation
-        )
         return {
           ...params,
           limit: params.limit === '' || params.limit == null ? undefined : Number(params.limit),
-          topic: params.operation === 'gauge_list_prompts' ? params.topicFilter : params.topic,
-          topicId: patch && params.topicId === '' ? undefined : params.topicId,
-          trackingType: patch
-            ? params.promptTrackingType === 'unchanged'
+          topic: params.topicFilter ?? params.topic,
+          topicId: params.topicId === '' ? undefined : params.topicId,
+          trackingType:
+            params.promptTrackingType === 'unchanged'
               ? undefined
-              : params.promptTrackingType
-            : params.trackingType,
-          name:
-            params.operation === 'gauge_update_tag' && params.name === '' ? undefined : params.name,
+              : (params.promptTrackingType ?? params.trackingType),
+          name: params.name === '' ? undefined : params.name,
           dataset:
-            params.operation === 'gauge_get_query_metadata'
-              ? params.metadataDataset === 'all'
-                ? undefined
-                : params.metadataDataset
-              : params.dataset,
+            params.metadataDataset === 'all'
+              ? undefined
+              : (params.metadataDataset ?? params.dataset),
         }
       },
     },

apps/sim/providers/utils.test.ts

@@ -12,6 +12,7 @@ vi.mock('@/lib/internal/workflows/read-tool-enrichment', () => ({
 }))
 
 import { getParamsForToolOperation } from '@/app/workspace/[workspaceId]/w/[workflowId]/components/panel/components/editor/components/sub-block/components/tool-input/param-dependents'
+import { GaugeBlock } from '@/blocks/blocks/gauge'
 import { GreenhouseV2Block } from '@/blocks/blocks/greenhouse'
 import { IntuneBlock } from '@/blocks/blocks/intune'
 import { MicrosoftExcelV2Block } from '@/blocks/blocks/microsoft_excel'
@@ -46,6 +47,14 @@ import {
   transformBlockTool,
 } from '@/providers/utils'
 import { useProvidersStore } from '@/stores/providers/store'
+import { gaugeGetQueryMetadataTool } from '@/tools/gauge/get_query_metadata'
+import { gaugeListPromptsTool } from '@/tools/gauge/list_prompts'
+import type {
+  GaugeGetQueryMetadataParams,
+  GaugeListPromptsParams,
+  GaugeUpdatePromptParams,
+} from '@/tools/gauge/types'
+import { gaugeUpdatePromptTool } from '@/tools/gauge/update_prompt'
 import { greenhouseConvertProspectV2Tool } from '@/tools/greenhouse/convert_prospect_v2'
 import { greenhouseCreateApplicationV2Tool } from '@/tools/greenhouse/create_application_v2'
 import { greenhouseHireApplicationV2Tool } from '@/tools/greenhouse/hire_application_v2'
@@ -79,6 +88,137 @@ const mockGetRotatingApiKey = vi.fn().mockReturnValue('rotating-server-key')
 const originalRequire = module.require
 
 describe('integration filters through agent execution', () => {
+  it.each(['configured', 'model'] as const)(
+    'retains the %s Gauge topic filter without a stored operation',
+    async (source) => {
+      const tool = gaugeListPromptsTool
+      vi.spyOn(toolMetadata, 'getToolMetadata').mockReturnValue(tool)
+      const field = getSubBlocksForToolInput(
+        tool.id,
+        'gauge',
+        { operation: tool.id },
+        undefined,
+        GaugeBlock
+      )?.subBlocks.find((item) => item.canonicalParamId === 'topicFilter')
+      if (!field) throw new Error('Gauge topic filter control is unavailable')
+      const providerTool = await transformBlockTool(
+        {
+          type: 'gauge',
+          params: {
+            org: 'example',
+            category: 'example',
+            ...(source === 'configured' ? { [field.id]: 'top_configured' } : {}),
+          },
+        },
+        { selectedOperation: tool.id, getAllBlocks: () => [GaugeBlock], getTool: () => tool }
+      )
+      if (!providerTool) throw new Error('Gauge agent tool was not created')
+      const { executionParams } = prepareToolExecution(
+        providerTool,
+        { topic: 'top_model' },
+        {},
+        undefined
+      )
+      const params = { ...executionParams, accessToken: 'fixture-token' } as GaugeListPromptsParams
+      const url =
+        typeof tool.request.url === 'function' ? tool.request.url(params) : tool.request.url
+      expect(new URL(url).searchParams.get('topic')).toBe(
+        source === 'configured' ? 'top_configured' : 'top_model'
+      )
+    }
+  )
+
+  it.each(['configured', 'unchanged', 'model'] as const)(
+    'retains %s Gauge tracking intent without a stored operation',
+    async (source) => {
+      const tool = gaugeUpdatePromptTool
+      vi.spyOn(toolMetadata, 'getToolMetadata').mockReturnValue(tool)
+      const field = getSubBlocksForToolInput(
+        tool.id,
+        'gauge',
+        { operation: tool.id },
+        undefined,
+        GaugeBlock
+      )?.subBlocks.find((item) => item.id === 'promptTrackingType')
+      if (!field && source !== 'model') throw new Error('Gauge tracking control is unavailable')
+      const providerTool = await transformBlockTool(
+        {
+          type: 'gauge',
+          params: {
+            org: 'example',
+            category: 'example',
+            promptId: 'prompt_example',
+            ...(field && source !== 'model'
+              ? { [field.id]: source === 'unchanged' ? 'unchanged' : 'visibility' }
+              : {}),
+          },
+        },
+        { selectedOperation: tool.id, getAllBlocks: () => [GaugeBlock], getTool: () => tool }
+      )
+      if (!providerTool) throw new Error('Gauge agent tool was not created')
+      const { executionParams } = prepareToolExecution(
+        providerTool,
+        { trackingType: 'branded_own' },
+        {},
+        undefined
+      )
+      const params = { ...executionParams, accessToken: 'fixture-token' } as GaugeUpdatePromptParams
+      expect(tool.request.body?.(params)).toEqual({
+        topicId: undefined,
+        trackingType:
+          source === 'unchanged'
+            ? undefined
+            : source === 'configured'
+              ? 'visibility'
+              : 'branded_own',
+      })
+    }
+  )
+
+  it.each(['configured', 'all', 'model'] as const)(
+    'retains %s Gauge catalog intent without a stored operation',
+    async (source) => {
+      const tool = gaugeGetQueryMetadataTool
+      vi.spyOn(toolMetadata, 'getToolMetadata').mockReturnValue(tool)
+      const field = getSubBlocksForToolInput(
+        tool.id,
+        'gauge',
+        { operation: tool.id },
+        undefined,
+        GaugeBlock
+      )?.subBlocks.find((item) => item.id === 'metadataDataset')
+      if (!field && source !== 'model') throw new Error('Gauge catalog control is unavailable')
+      const providerTool = await transformBlockTool(
+        {
+          type: 'gauge',
+          params: {
+            org: 'example',
+            ...(field && source !== 'model'
+              ? { [field.id]: source === 'all' ? 'all' : 'citations' }
+              : {}),
+          },
+        },
+        { selectedOperation: tool.id, getAllBlocks: () => [GaugeBlock], getTool: () => tool }
+      )
+      if (!providerTool) throw new Error('Gauge agent tool was not created')
+      const { executionParams } = prepareToolExecution(
+        providerTool,
+        { dataset: 'aiTraffic' },
+        {},
+        undefined
+      )
+      const params = {
+        ...executionParams,
+        accessToken: 'fixture-token',
+      } as GaugeGetQueryMetadataParams
+      const url =
+        typeof tool.request.url === 'function' ? tool.request.url(params) : tool.request.url
+      expect(new URL(url).searchParams.get('dataset')).toBe(
+        source === 'all' ? null : source === 'configured' ? 'citations' : 'aiTraffic'
+      )
+    }
+  )
+
   it.each([
     {
       operation: 'list_jobs',

apps/sim/tools/gauge/create_prompts.ts

@@ -60,14 +60,15 @@ function selectGaugePromptText(params: { prompts: unknown }): Record<string, unk
 /** Rebuilds the original JSON representation after the shared executor projects prompt text. */
 function rebuildGaugePrompts(original: unknown, projectedText: unknown): string | GaugePromptIn[] {
   const prompts = parseGaugePrompts(original)
-  if (
-    !Array.isArray(projectedText) ||
-    projectedText.length !== prompts.length ||
-    projectedText.some((text) => typeof text !== 'string')
-  ) {
+  if (!Array.isArray(projectedText) || projectedText.length !== prompts.length) {
     throw new Error('Projected Gauge prompts must match the original prompt batch')
   }
-  const texts = parseGaugeStringArray(projectedText, 'Projected prompt text')
+  const texts = projectedText.map((text: unknown) => {
+    if (typeof text !== 'string') {
+      throw new Error('Projected Gauge prompts must match the original prompt batch')
+    }
+    return text
+  })
   const rebuilt = prompts.map((prompt, index) => ({ ...prompt, text: texts[index] ?? '' }))
   return typeof original === 'string' ? JSON.stringify(rebuilt) : rebuilt
 }

apps/sim/tools/gauge/update_tag.ts

@@ -29,7 +29,8 @@ export const gaugeUpdateTagTool: ToolConfig<GaugeUpdateTagParams, GaugeUpdateTag
       type: 'string',
       required: false,
       visibility: 'user-or-llm',
-      description: 'Tag color string (up to 20 characters)',
+      description:
+        'Tag color string (up to 20 characters); omit to keep unchanged or use "" to clear',
     },
   },
   request: {

apps/sim/tools/gauge/utils.test.ts

@@ -33,6 +33,13 @@ describe('Gauge request encoding', () => {
     expect(() => parseGaugeStringArray([], 'ids', { min: 1, max: 500 })).toThrow()
     expect(() => parseGaugeStringArray(['a', 'b'], 'tagIds', { max: 1 })).toThrow()
   })
+
+  it.each(['', '  '])('rejects a blank identifier %j instead of sending it to Gauge', (value) => {
+    expect(() => parseGaugeStringArray([value], 'ids', { min: 1 })).toThrow()
+    expect(parseGaugeStringArray([' padded-but-nonblank '], 'ids')).toEqual([
+      ' padded-but-nonblank ',
+    ])
+  })
 })
 
 const modelInput = gaugeCreatePromptsTool.request.modelInput

apps/sim/tools/gauge/utils.ts

@@ -52,16 +52,16 @@ export function parseGaugeJson(value: unknown, name: string): unknown {
   }
 }
 
-/** Validates string arrays arriving directly or as JSON from a block field. */
+/** Validates nonblank string arrays without changing the supplied values. */
 export function parseGaugeStringArray(
   value: unknown,
   name: string,
   bounds: { min?: number; max?: number } = {}
 ): string[] {
   const parsed = parseGaugeJson(value, name)
   if (parsed == null && !bounds.min) return []
-  if (!Array.isArray(parsed) || parsed.some((item) => typeof item !== 'string')) {
-    throw new Error(`${name} must be a JSON array of strings`)
+  if (!Array.isArray(parsed) || parsed.some((item) => typeof item !== 'string' || !item.trim())) {
+    throw new Error(`${name} must be a JSON array of non-empty strings`)
   }
   if (
     (bounds.min !== undefined && parsed.length < bounds.min) ||