Move UserOnboarding to model.py

Author: kcze

autogpt_platform/backend/backend/data/model.py

@@ -22,7 +22,7 @@
 from urllib.parse import urlparse
 from uuid import uuid4
 
-from prisma.enums import CreditTransactionType
+from prisma.enums import CreditTransactionType, OnboardingStep
 from pydantic import (
     BaseModel,
     ConfigDict,
@@ -855,3 +855,20 @@ class UserExecutionSummaryStats(BaseModel):
     total_execution_time: float = Field(default=0)
     average_execution_time: float = Field(default=0)
     cost_breakdown: dict[str, float] = Field(default_factory=dict)
+
+
+class UserOnboarding(BaseModel):
+    userId: str
+    completedSteps: list[OnboardingStep]
+    walletShown: bool
+    notified: list[OnboardingStep]
+    rewardedFor: list[OnboardingStep]
+    usageReason: Optional[str]
+    integrations: list[str]
+    otherIntegrations: Optional[str]
+    selectedStoreListingVersionId: Optional[str]
+    agentInput: dict[str, Any]
+    onboardingAgentExecutionId: Optional[str]
+    agentRuns: int
+    lastRunAt: Optional[datetime]
+    consecutiveRunDays: int

autogpt_platform/backend/backend/server/routers/v1.py

@@ -45,7 +45,7 @@
     set_auto_top_up,
 )
 from backend.data.execution import UserContext
-from backend.data.model import CredentialsMetaInput
+from backend.data.model import CredentialsMetaInput, UserOnboarding
 from backend.data.notifications import NotificationPreference, NotificationPreferenceDTO
 from backend.data.onboarding import (
     FrontendOnboardingStep,
@@ -118,25 +118,6 @@ def _create_file_size_error(size_bytes: int, max_size_mb: int) -> HTTPException:
 logger = logging.getLogger(__name__)
 
 
-# Needed to avoid including User from UserOnboarding prisma model in router
-# that causes schema generation for prisma and our LibraryAgent in openapi.json
-class UserOnboarding(pydantic.BaseModel):
-    userId: str
-    completedSteps: list[OnboardingStep]
-    walletShown: bool
-    notified: list[OnboardingStep]
-    rewardedFor: list[OnboardingStep]
-    usageReason: Optional[str]
-    integrations: list[str]
-    otherIntegrations: Optional[str]
-    selectedStoreListingVersionId: Optional[str]
-    agentInput: dict[str, Any]
-    onboardingAgentExecutionId: Optional[str]
-    agentRuns: int
-    lastRunAt: Optional[datetime]
-    consecutiveRunDays: int
-
-
 async def hide_activity_summaries_if_disabled(
     executions: list[execution_db.GraphExecutionMeta], user_id: str
 ) -> list[execution_db.GraphExecutionMeta]: