diff --git a/src/shared/migration.test.ts b/src/shared/migration.test.ts index 599b4854d..2b136eaba 100644 --- a/src/shared/migration.test.ts +++ b/src/shared/migration.test.ts @@ -468,13 +468,14 @@ describe("migrateAgentConfigToCategory", () => { // #given: Configs for each mapped model const configs = [ { model: "google/gemini-3-pro" }, + { model: "google/gemini-3-flash" }, { model: "openai/gpt-5.2" }, { model: "anthropic/claude-haiku-4-5" }, { model: "anthropic/claude-opus-4-5" }, { model: "anthropic/claude-sonnet-4-5" }, ] - const expectedCategories = ["visual-engineering", "ultrabrain", "quick", "unspecified-high", "unspecified-low"] + const expectedCategories = ["visual-engineering", "writing", "ultrabrain", "quick", "unspecified-high", "unspecified-low"] // #when: Migrate each config const results = configs.map(migrateAgentConfigToCategory) diff --git a/src/shared/migration.ts b/src/shared/migration.ts index a80437c45..173d60154 100644 --- a/src/shared/migration.ts +++ b/src/shared/migration.ts @@ -83,6 +83,7 @@ export const HOOK_NAME_MAP: Record = { */ export const MODEL_TO_CATEGORY_MAP: Record = { "google/gemini-3-pro": "visual-engineering", + "google/gemini-3-flash": "writing", "openai/gpt-5.2": "ultrabrain", "anthropic/claude-haiku-4-5": "quick", "anthropic/claude-opus-4-5": "unspecified-high",