fix(gauge): scope inputs to the selected operation

Author: waleedlatif1

apps/sim/blocks/blocks/gauge.ts

@@ -664,6 +664,7 @@ export const GaugeBlock: BlockConfig = {
       id: 'topic',
       title: 'Topic Filters',
       type: 'code',
+      dependsOn: ['operation'],
       condition: { field: 'operation', value: ['gauge_list_answers', 'gauge_list_citations'] },
       language: 'json',
       placeholder: '["top_..."]',
@@ -970,21 +971,29 @@ export const GaugeBlock: BlockConfig = {
     ],
     config: {
       tool: (params) => params.operation,
-      params: (params) => {
+      params: (params, context) => {
+        const operation = context?.operation ?? params.operation
+        const patch = ['gauge_bulk_update_prompts', 'gauge_update_prompt'].includes(operation)
         return {
           ...params,
           limit: params.limit === '' || params.limit == null ? undefined : Number(params.limit),
-          topic: params.topicFilter ?? params.topic,
+          topic:
+            operation === 'gauge_list_prompts'
+              ? (params.topicFilter ?? (context?.operation ? params.topic : undefined))
+              : params.topic,
           topicId: params.topicId === '' ? undefined : params.topicId,
-          trackingType:
-            params.promptTrackingType === 'unchanged'
+          trackingType: patch
+            ? params.promptTrackingType === 'unchanged'
               ? undefined
-              : (params.promptTrackingType ?? params.trackingType),
+              : (params.promptTrackingType ?? params.trackingType)
+            : params.trackingType,
           name: params.name === '' ? undefined : params.name,
           dataset:
-            params.metadataDataset === 'all'
-              ? undefined
-              : (params.metadataDataset ?? params.dataset),
+            operation === 'gauge_get_query_metadata'
+              ? params.metadataDataset === 'all'
+                ? undefined
+                : (params.metadataDataset ?? params.dataset)
+              : params.dataset,
         }
       },
     },

apps/sim/blocks/types.ts

@@ -597,6 +597,12 @@ export type CanvasSentenceClause =
 /** An ordered set of clauses forming one card summary sentence. */
 export type CanvasSentence = readonly CanvasSentenceClause[]
 
+/** Trusted mapper context kept separate from user-configured and model-supplied arguments. */
+export interface BlockToolParamsContext {
+  /** Operation selected by the configuring surface. */
+  operation?: string
+}
+
 export interface BlockConfig {
   type: string
   name: string
@@ -669,7 +675,10 @@ export interface BlockConfig {
     access: string[]
     config?: {
       tool: (params: Record<string, any>) => string
-      params?: (params: Record<string, any>) => Record<string, any>
+      params?: (
+        params: Record<string, any>,
+        context?: BlockToolParamsContext
+      ) => Record<string, any>
     }
   }
   inputs: Record<string, ParamConfig>

apps/sim/executor/handlers/human-in-the-loop/handler.ts

@@ -423,6 +423,9 @@ export class HumanInTheLoopBlockHandler implements BlockHandler {
     ).filter(isCanonicalPair)
 
     const { paramsTransform } = buildBlockToolParamsTransform({
+      operation:
+        toolConfig.operation ??
+        (typeof storedParams.operation === 'string' ? storedParams.operation : undefined),
       blockSubBlocks: subBlocks,
       blockParamsFn: notificationBlock?.tools?.config?.params,
       blockInputDefs: notificationBlock?.inputs,

apps/sim/providers/utils.test.ts

@@ -48,10 +48,15 @@ import {
 } from '@/providers/utils'
 import { useProvidersStore } from '@/stores/providers/store'
 import { gaugeGetQueryMetadataTool } from '@/tools/gauge/get_query_metadata'
+import { gaugeListAnswersTool } from '@/tools/gauge/list_answers'
+import { gaugeListCitationsTool } from '@/tools/gauge/list_citations'
 import { gaugeListPromptsTool } from '@/tools/gauge/list_prompts'
+import { gaugeQueryTool } from '@/tools/gauge/query'
 import type {
   GaugeGetQueryMetadataParams,
+  GaugeListAnswersParams,
   GaugeListPromptsParams,
+  GaugeQueryParams,
   GaugeUpdatePromptParams,
 } from '@/tools/gauge/types'
 import { gaugeUpdatePromptTool } from '@/tools/gauge/update_prompt'
@@ -88,6 +93,84 @@ const mockGetRotatingApiKey = vi.fn().mockReturnValue('rotating-server-key')
 const originalRequire = module.require
 
 describe('integration filters through agent execution', () => {
+  it.each([
+    { tool: gaugeListAnswersTool, field: 'topicFilterSelector' },
+    { tool: gaugeListAnswersTool, field: 'manualTopicFilter' },
+    { tool: gaugeListCitationsTool, field: 'topicFilterSelector' },
+    { tool: gaugeListCitationsTool, field: 'manualTopicFilter' },
+  ])('ignores stale $field aliases for $tool.id', async ({ tool, field }) => {
+    const providerTool = await transformBlockTool(
+      {
+        type: 'gauge',
+        params: {
+          org: 'example',
+          category: 'example',
+          [field]: 'top_stale',
+          topic: '["top_active"]',
+          promptTrackingType: 'branded_own',
+          trackingType: 'all',
+        },
+      },
+      { selectedOperation: tool.id, getAllBlocks: () => [GaugeBlock], getTool: () => tool }
+    )
+    if (!providerTool) throw new Error('Gauge agent tool was not created')
+    const { executionParams } = prepareToolExecution(providerTool, {}, {}, undefined)
+    const params = { ...executionParams, accessToken: 'fixture-token' } as GaugeListAnswersParams
+    const url = typeof tool.request.url === 'function' ? tool.request.url(params) : tool.request.url
+    expect(new URL(url).searchParams.getAll('topic')).toEqual(['top_active'])
+    expect(new URL(url).searchParams.get('trackingType')).toBe('all')
+  })
+
+  it.each(['answers', 'all'])(
+    'ignores a stale %s metadata selection when querying a different dataset',
+    async (metadataDataset) => {
+      const tool = gaugeQueryTool
+      const providerTool = await transformBlockTool(
+        {
+          type: 'gauge',
+          params: {
+            org: 'example',
+            dataset: 'citations',
+            metrics: '["citations"]',
+            metadataDataset,
+          },
+        },
+        { selectedOperation: tool.id, getAllBlocks: () => [GaugeBlock], getTool: () => tool }
+      )
+      if (!providerTool) throw new Error('Gauge agent tool was not created')
+      const { executionParams } = prepareToolExecution(providerTool, {}, {}, undefined)
+      const params = { ...executionParams, accessToken: 'fixture-token' } as GaugeQueryParams
+      expect(tool.request.body?.(params)).toMatchObject({ dataset: 'citations' })
+    }
+  )
+
+  it('clears an incompatible Gauge topic array when switching an agent to List Prompts', async () => {
+    const tool = gaugeListPromptsTool
+    vi.spyOn(blockRegistry, 'getBlock').mockReturnValue(GaugeBlock)
+    const params = getParamsForToolOperation(
+      { type: 'gauge', params: { org: 'example', category: 'example', topic: '["top_stale"]' } },
+      tool
+    )
+    const providerTool = await transformBlockTool(
+      { type: 'gauge', params },
+      { 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 requestParams = {
+      ...executionParams,
+      accessToken: 'fixture-token',
+    } as GaugeListPromptsParams
+    const url =
+      typeof tool.request.url === 'function' ? tool.request.url(requestParams) : tool.request.url
+    expect(new URL(url).searchParams.get('topic')).toBe('top_model')
+  })
+
   it.each(['configured', 'model'] as const)(
     'retains the %s Gauge topic filter without a stored operation',
     async (source) => {

apps/sim/providers/utils.ts

@@ -23,7 +23,7 @@ import {
 } from '@/lib/workflows/subblocks/visibility'
 import { resolveBlockToolId } from '@/lib/workflows/tool-input/identity'
 import { assembleCustomBlockInputMapping, isCustomBlockType } from '@/blocks/custom/build-config'
-import type { SubBlockConfig } from '@/blocks/types'
+import type { BlockToolParamsContext, SubBlockConfig } from '@/blocks/types'
 import { isCustomTool } from '@/executor/constants'
 import {
   findProviderFromModel,
@@ -523,7 +523,10 @@ function buildCustomBlockInputMappingSchema(
   }
 }
 
-type BlockToolParamsFn = (params: Record<string, any>) => Record<string, any>
+type BlockToolParamsFn = (
+  params: Record<string, any>,
+  context?: BlockToolParamsContext
+) => Record<string, any>
 
 /**
  * Builds the transform that turns a tool row's stored sub-block values into the
@@ -547,6 +550,7 @@ type BlockToolParamsFn = (params: Record<string, any>) => Record<string, any>
  * none of it, which is why it needed a new version rather than a fix in place.
  */
 export function buildBlockToolParamsTransform(config: {
+  operation?: string
   blockSubBlocks: SubBlockConfig[] | undefined
   blockParamsFn: BlockToolParamsFn | undefined
   blockInputDefs: Record<string, unknown> | undefined
@@ -558,6 +562,7 @@ export function buildBlockToolParamsTransform(config: {
   jsonShapedParamKeys: string[]
 } {
   const {
+    operation,
     blockSubBlocks,
     blockParamsFn,
     blockInputDefs,
@@ -613,7 +618,7 @@ export function buildBlockToolParamsTransform(config: {
         }
 
         if (blockParamsFn) {
-          const transformed = blockParamsFn(result)
+          const transformed = blockParamsFn(result, { operation })
           result = { ...result, ...transformed }
         }
 
@@ -893,6 +898,9 @@ export async function transformBlockTool(
   }
 
   const { paramsTransform, jsonShapedParamKeys } = buildBlockToolParamsTransform({
+    operation:
+      selectedOperation ??
+      (typeof userProvidedParams.operation === 'string' ? userProvidedParams.operation : undefined),
     blockSubBlocks: blockDef?.subBlocks,
     blockParamsFn: blockDef?.tools?.config?.params as BlockToolParamsFn | undefined,
     blockInputDefs: blockDef?.inputs as Record<string, unknown> | undefined,

apps/sim/serializer/gauge.test.ts

@@ -0,0 +1,41 @@
+import { beforeEach, describe, expect, it, vi } from 'vitest'
+import { getBlock } from '@/blocks'
+import { GaugeBlock } from '@/blocks/blocks/gauge'
+import { extractBlockParams } from '@/serializer/index'
+import { prepareBlockState } from '@/stores/workflows/prepare-block-state'
+import { gaugeListPromptsTool } from '@/tools/gauge/list_prompts'
+import type { GaugeListPromptsParams } from '@/tools/gauge/types'
+import { prepareToolRequest } from '@/tools/request-transport'
+
+describe('Gauge canvas operation changes', () => {
+  beforeEach(() => {
+    vi.mocked(getBlock).mockReturnValue(GaugeBlock)
+  })
+
+  it.each([false, true])(
+    'does not reuse an old topic array after clearing the prompt filter (advanced: %s)',
+    (advancedMode) => {
+      const block = prepareBlockState({
+        id: 'gauge_fixture',
+        type: 'gauge',
+        name: 'Gauge',
+        position: { x: 0, y: 0 },
+      })
+      block.advancedMode = advancedMode
+      block.subBlocks.operation.value = 'gauge_list_prompts'
+      block.subBlocks.manualOrg.value = 'example'
+      block.subBlocks.manualCategory.value = 'example'
+      block.subBlocks.topic.value = '["top_stale"]'
+      const params = extractBlockParams(block)
+      const mapped = GaugeBlock.tools.config?.params?.(params)
+      if (!mapped) throw new Error('Gauge parameter mapping is unavailable')
+      const request = prepareToolRequest(gaugeListPromptsTool, {
+        ...mapped,
+        org: 'example',
+        category: 'example',
+        accessToken: 'fixture-token',
+      } as GaugeListPromptsParams)
+      expect(new URL(request.url).searchParams.has('topic')).toBe(false)
+    }
+  )
+})

scripts/check-test-types.baseline.json

@@ -657,8 +657,7 @@
     "TS18048": 7
   },
   "apps/sim/blocks/blocks/google_drive.test.ts": {
-    "TS18048": 1,
-    "TS2554": 4
+    "TS18048": 1
   },
   "apps/sim/blocks/blocks/instagram.test.ts": {
     "TS18048": 1