From d91167e0939a0fb25b605290aad29bcc3fa4890a Mon Sep 17 00:00:00 2001
From: Zach Kirsch <zachkirsch@users.noreply.github.com>
Date: Sat, 10 Dec 2022 11:39:06 -0800
Subject: [PATCH] Add missing generators to IR migrations (#975)

---
 .../ir-migrations/src/migrations/ir-v1/migrateFromV2ToV1.ts   | 4 +++-
 .../cli/generation/ir-migrations/src/types/GeneratorName.ts   | 4 +++-
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/packages/cli/generation/ir-migrations/src/migrations/ir-v1/migrateFromV2ToV1.ts b/packages/cli/generation/ir-migrations/src/migrations/ir-v1/migrateFromV2ToV1.ts
index 3282f7cf312..9496c01065c 100644
--- a/packages/cli/generation/ir-migrations/src/migrations/ir-v1/migrateFromV2ToV1.ts
+++ b/packages/cli/generation/ir-migrations/src/migrations/ir-v1/migrateFromV2ToV1.ts
@@ -8,8 +8,10 @@ export const V2_TO_V1_MIGRATION: IrMigration<V2.ir.IntermediateRepresentation, V
         [GeneratorName.TYPESCRIPT]: AlwaysRunMigration,
         [GeneratorName.TYPESCRIPT_SDK]: "0.0.245",
         [GeneratorName.JAVA]: AlwaysRunMigration,
+        [GeneratorName.JAVA_MODEL]: AlwaysRunMigration,
         [GeneratorName.JAVA_SDK]: AlwaysRunMigration,
-        [GeneratorName.PYTHON]: AlwaysRunMigration,
+        [GeneratorName.PYTHON_FASTAPI]: AlwaysRunMigration,
+        [GeneratorName.PYTHON_PYDANTIC]: AlwaysRunMigration,
         [GeneratorName.OPENAPI]: AlwaysRunMigration,
         [GeneratorName.POSTMAN]: AlwaysRunMigration,
     },
diff --git a/packages/cli/generation/ir-migrations/src/types/GeneratorName.ts b/packages/cli/generation/ir-migrations/src/types/GeneratorName.ts
index 2cee8015b10..1df773e9b3c 100644
--- a/packages/cli/generation/ir-migrations/src/types/GeneratorName.ts
+++ b/packages/cli/generation/ir-migrations/src/types/GeneratorName.ts
@@ -2,8 +2,10 @@ export const GeneratorName = {
     TYPESCRIPT: "fernapi/fern-typescript",
     TYPESCRIPT_SDK: "fernapi/fern-typescript-sdk",
     JAVA: "fernapi/fern-java",
+    JAVA_MODEL: "fernapi/java-model",
     JAVA_SDK: "fernapi/fern-java-sdk",
-    PYTHON: "fernapi/fern-python",
+    PYTHON_FASTAPI: "fernapi/fern-fastapi-server",
+    PYTHON_PYDANTIC: "fernapi/fern-pydantic-model",
     OPENAPI: "fernapi/fern-openapi",
     POSTMAN: "fernapi/fern-postman",
 } as const;