Enhance Tool Explorer and OpenAPI Handling

- Update styling logic in the ToolRow and GroupNode components to differentiate selected and source tools, improving visual clarity.
- Modify OpenAPI preparation to always include parameter schemas, ensuring consistent schema handling.
- Add tests to validate input and output hints for OpenAPI tools, ensuring accurate type representation.
- Update OpenAPI cache versioning to reflect recent changes and improvements in tool discovery.
- Refactor workspace tool inventory logic to streamline state management and improve performance.

Author: RhysSullivan

executor/apps/web/src/components/tools/explorer-groups.tsx

@@ -80,7 +80,9 @@ export function GroupNode({
             "sticky bg-background/95 backdrop-blur-sm",
             isExpanded && "border-b border-border/30",
             isGroupSelected
-              ? "bg-primary/10 ring-1 ring-primary/20"
+              ? isSource
+                ? "bg-primary/10 ring-1 ring-primary/20"
+                : "bg-accent/20 ring-1 ring-accent/30"
               : "hover:bg-accent/30",
           )}
           style={{

executor/apps/web/src/components/tools/explorer-rows.tsx

@@ -50,9 +50,11 @@ const ToolRow = memo(function ToolRow({
           className={cn(
             "flex items-center gap-2 px-2 py-1.5 transition-colors cursor-pointer group/tool",
             expanded
-              ? "sticky bg-background/95 backdrop-blur-sm bg-accent/30"
+              ? selected
+                ? "sticky bg-accent ring-1 ring-accent/30"
+                : "sticky bg-card"
               : selected
-                ? "bg-primary/5 ring-1 ring-primary/10"
+                ? "bg-accent ring-1 ring-accent/30"
                 : "hover:bg-accent/20",
           )}
           style={{

executor/packages/core/src/openapi-prepare.ts

@@ -233,7 +233,7 @@ export async function prepareOpenApiSpec(
     {
       includeSchemas: profile === "full",
       includeTypeHints: true,
-      includeParameterSchemas: profile === "full",
+      includeParameterSchemas: true,
       resolveSchemaRefs: profile === "full",
     },
   );

executor/packages/core/src/openapi-typing.test.ts

@@ -100,6 +100,8 @@ describe("OpenAPI schema-first typing", () => {
     expect(getTool!.typing?.outputSchema).toBeDefined();
     expect(getTool!.typing?.requiredInputKeys ?? []).toContain("id");
     expect(getTool!.typing?.previewInputKeys ?? []).toContain("include");
+    expect(getTool!.typing?.inputHint).toContain("id: string");
+    expect(getTool!.typing?.inputHint).toContain("include?:");
     expect(getTool!.typing?.typedRef).toBeDefined();
   });
 

executor/packages/core/src/openapi/real-specs.test.ts

@@ -301,4 +301,34 @@ describe("real-world OpenAPI specs", () => {
     },
     300_000,
   );
+
+  test(
+    "openai: assistants cancel run keeps path parameter types in inventory mode",
+    async () => {
+      const openAiUrl = "https://app.stainless.com/api/spec/documented/openai/openapi.documented.yml";
+      const prepared = await prepareOpenApiSpec(openAiUrl, "openai", {
+        includeDts: false,
+        profile: "inventory",
+      });
+
+      const tools = buildOpenApiToolsFromPrepared(
+        {
+          type: "openapi",
+          name: "openai",
+          spec: openAiUrl,
+          baseUrl: prepared.servers[0] || "https://api.openai.com",
+        },
+        prepared,
+      );
+
+      const tool = tools.find((t) => t.path === "openai.assistants.cancel_run");
+      expect(tool).toBeDefined();
+
+      const inputHint = tool!.typing?.inputHint ?? "";
+      expect(inputHint).toContain("thread_id: string");
+      expect(inputHint).toContain("run_id: string");
+      expect(inputHint.includes("unknown")).toBe(false);
+    },
+    300_000,
+  );
 });

executor/packages/core/src/tool-discovery.test.ts

@@ -412,6 +412,61 @@ test("discover depth 0 keeps full signatures", async () => {
   );
 });
 
+test("discover prefers schema over lossy typed hints", async () => {
+  const tool = createDiscoverTool([
+    {
+      path: "openai.batch.create_batch",
+      description: "Create batch",
+      approval: "required",
+      source: "openapi:openai",
+      typing: {
+        inputHint: "{ input_file_id: ...; endpoint: ... }",
+        outputHint: "{ id: string; errors?: { message?:... } }",
+        inputSchema: {
+          type: "object",
+          properties: {
+            input_file_id: { type: "string" },
+            endpoint: { type: "string" },
+            completion_window: { type: "string" },
+          },
+          required: ["input_file_id", "endpoint", "completion_window"],
+        },
+        outputSchema: {
+          type: "object",
+          properties: {
+            id: { type: "string" },
+            errors: {
+              type: "object",
+              properties: {
+                message: { type: "string" },
+              },
+            },
+          },
+          required: ["id"],
+        },
+      },
+      run: async () => ({ id: "b_123" }),
+    } satisfies ToolDefinition,
+  ]);
+
+  const result = await tool.run(
+    { query: "create batch", depth: 2 },
+    { taskId: "t", workspaceId: TEST_WORKSPACE_ID, isToolAllowed: () => true },
+  ) as {
+    results: Array<{
+      signatureInfo: {
+        input: string;
+        output: string;
+      };
+    }>;
+  };
+
+  expect(result.results[0]?.signatureInfo.input).toContain("input_file_id: string");
+  expect(result.results[0]?.signatureInfo.input.includes("...")).toBe(false);
+  expect(result.results[0]?.signatureInfo.output).toContain("message?: string");
+  expect(result.results[0]?.signatureInfo.output.includes("...")).toBe(false);
+});
+
 test("discover returns null bestPath when there are no matches", async () => {
   const tool = createDiscoverTool([
     {

executor/packages/database/src/runtime/tool_source_loading.ts

@@ -30,7 +30,7 @@ import {
 
 const OPENAPI_SPEC_CACHE_TTL_MS = 5 * 60 * 60_000;
 
-const OPENAPI_PREPARED_CACHE_VERSION = "openapi_v5";
+const OPENAPI_PREPARED_CACHE_VERSION = "openapi_v6";
 
 const openApiAuthModeSchema = z.enum(["static", "account", "workspace", "organization"]);
 

executor/packages/database/src/runtime/workspace_tools.ts

@@ -1504,11 +1504,6 @@ export async function rebuildWorkspaceToolInventoryForContext(
     sources.filter((source) => source.enabled),
   );
 
-  const alreadyReady = Boolean(
-    state?.readyBuildId
-      && !state?.buildingBuildId
-      && state?.signature === registrySignature,
-  );
   const alreadyBuildingTarget = Boolean(
     state?.buildingBuildId
       && state?.buildingSignature === registrySignature,
@@ -1523,7 +1518,7 @@ export async function rebuildWorkspaceToolInventoryForContext(
       && Date.now() - buildingStartedAt > REGISTRY_BUILD_STALE_MS,
   );
 
-  if (alreadyReady || (alreadyBuildingTarget && !staleBuildingTarget)) {
+  if (alreadyBuildingTarget && !staleBuildingTarget) {
     return { rebuilt: false };
   }