add frontend schema test for tools (#7252)

## 📝 Summary

<!--
Provide a concise summary of what this pull request is addressing.

If this PR fixes any issues, list them here by number (e.g., Fixes
#123).
-->

## 🔍 Description of Changes

<!--
Detail the specific changes made in this pull request. Explain the
problem addressed and how it was resolved. If applicable, provide before
and after comparisons, screenshots, or any relevant details to help
reviewers understand the changes easily.
-->

## 📋 Checklist

- [x] I have read the [contributor
guidelines](https://github.com/marimo-team/marimo/blob/main/CONTRIBUTING.md).
- [ ] For large changes, or changes that affect the public API: this
change was discussed or approved through an issue, on
[Discord](https://marimo.io/discord?ref=pr), or the community
[discussions](https://github.com/marimo-team/marimo/discussions) (Please
provide a link if applicable).
- [x] I have added tests for the changes made.
- [x] I have run the code and verified that it works as expected.

Author: Light2Dark

frontend/src/core/ai/tools/__tests__/registry.test.ts

@@ -1,9 +1,14 @@
 /* Copyright 2024 Marimo. All rights reserved. */
 
 import { describe, expect, it } from "vitest";
-import { FrontendToolRegistry } from "../registry";
+import { COPILOT_MODES } from "@/core/config/config-schema";
+import { FRONTEND_TOOL_REGISTRY, FrontendToolRegistry } from "../registry";
 import { TestFrontendTool } from "../sample-tool";
 
+// Tools that have these keys in their parameters are likely not supported by Google
+// So we should be careful to add these types of schemas
+const INVALID_KEYS_IN_SCHEMA_PARAMS = new Set(["anyOf", "oneOf"]);
+
 describe("FrontendToolRegistry", () => {
   it("registers tools via constructor and supports has()", () => {
     const registry = new FrontendToolRegistry([new TestFrontendTool()]);
@@ -87,4 +92,33 @@ describe("FrontendToolRegistry", () => {
     const schemas3 = registry.getToolSchemas("agent");
     expect(schemas3.length).toBe(0);
   });
+
+  it("tool schemas should not contain invalid keys", () => {
+    const registry = FRONTEND_TOOL_REGISTRY;
+
+    function hasInvalidKeys(obj: unknown): boolean {
+      if (obj && typeof obj === "object") {
+        for (const [key, value] of Object.entries(obj)) {
+          if (INVALID_KEYS_IN_SCHEMA_PARAMS.has(key)) {
+            return true;
+          }
+          if (
+            typeof value === "object" &&
+            value !== null &&
+            hasInvalidKeys(value)
+          ) {
+            return true;
+          }
+        }
+      }
+      return false;
+    }
+
+    for (const mode of COPILOT_MODES) {
+      const schemas = registry.getToolSchemas(mode);
+      for (const schema of schemas) {
+        expect(hasInvalidKeys(schema.parameters)).toBe(false);
+      }
+    }
+  });
 });

frontend/src/core/config/config-schema.ts

@@ -2,7 +2,7 @@
 import { z } from "zod";
 import { invariant } from "@/utils/invariant";
 import { Logger } from "@/utils/Logger";
-import type { MarimoConfig } from "../network/types";
+import type { MarimoConfig, schemas } from "../network/types";
 
 // This has to be defined in the same file as the zod schema to satisfy zod
 export const PackageManagerNames = [
@@ -44,8 +44,8 @@ export const DEFAULT_AI_MODEL = "openai/gpt-4o";
  */
 const AUTO_DOWNLOAD_FORMATS = ["html", "markdown", "ipynb"] as const;
 
-const COPILOT_MODES = ["manual", "ask", "agent"] as const;
-export type CopilotMode = (typeof COPILOT_MODES)[number];
+export type CopilotMode = NonNullable<schemas["AiConfig"]["mode"]>;
+export const COPILOT_MODES: CopilotMode[] = ["manual", "ask", "agent"];
 
 const AiConfigSchema = z
   .object({