Merge pull request #1191 from Agenta-AI/reference-fields-migration

Migration - Update odmantic reference to beanie link & migrate old evaluation scenarios to new evaluation scenarios
Agenta-AI · Jan 14, 2024 · 5071ff2 · 5071ff2
2 parents 109fbc8 + f057095
commit 5071ff2
Show file tree

Hide file tree

Showing 9 changed files with 1,231 additions and 225 deletions.
diff --git a/agenta-backend/agenta_backend/migrations/20240110001454_initial_migration.py b/agenta-backend/agenta_backend/migrations/20240110001454_initial_migration.py
@@ -167,9 +167,9 @@ class OldEvaluationDB(Document):
     organization: Link[OrganizationDB]
     user: Link[UserDB]
     status: str
-    evaluation_type: str
-    evaluation_type_settings: OldEvaluationTypeSettings
-    variants: List[PydanticObjectId]
+    evaluation_type: Optional[str]
+    evaluation_type_settings: Optional[OldEvaluationTypeSettings]
+    variants: Optional[List[PydanticObjectId]]
     testsets: Link[TestSetDB]
     created_at: Optional[datetime] = Field(default=datetime.utcnow())
     updated_at: Optional[datetime] = Field(default=datetime.utcnow())
@@ -183,9 +183,9 @@ class EvaluationDB(Document):
     organization: Link[OrganizationDB]
     user: Link[UserDB]
     status: str
-    evaluation_type: str
-    evaluation_type_settings: OldEvaluationTypeSettings
-    variants: List[PydanticObjectId]
+    evaluation_type: Optional[str]
+    evaluation_type_settings: Optional[OldEvaluationTypeSettings]
+    variants: Optional[List[PydanticObjectId]]
     testset: Link[TestSetDB]
     created_at: Optional[datetime] = Field(default=datetime.utcnow())
     updated_at: Optional[datetime] = Field(default=datetime.utcnow())