Profile manager migration fixes (#77)

Profile maanager migration fixes
Zipstack · Mar 10, 2024 · 80d4870 · 80d4870
1 parent 40e6148
commit 80d4870
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/backend/prompt_studio/prompt_profile_manager/migrations/0008_profilemanager_migration.py b/backend/prompt_studio/prompt_profile_manager/migrations/0008_profilemanager_migration.py
@@ -30,6 +30,10 @@ def MigrateProfileManager(apps: Any, schema_editor: Any) -> None:
             which has distinct profile manager."""
 
             for index, custom_prompt in enumerate(custom_prompts):
+                """There can be scenario where individual prompts wont have a
+                profile manager attached."""
+                if not custom_prompt.profile_manager:
+                    continue
                 profile_manager = ProfileManager.objects.get(
                     pk=custom_prompt.profile_manager.profile_id
                 )