feat(platform): updated schema to allow featuring of specific creators

autogpt_platform/backend/backend/server/v2/library/db_test.py

@@ -153,6 +153,7 @@ async def test_add_agent_to_library(mocker):
     )
 
     mock_user_agent = mocker.patch("prisma.models.UserAgent.prisma")
+    mock_user_agent.return_value.find_first = mocker.AsyncMock(return_value=None)
     mock_user_agent.return_value.create = mocker.AsyncMock()
 
     # Call function
@@ -162,6 +163,13 @@ async def test_add_agent_to_library(mocker):
     mock_store_listing_version.return_value.find_unique.assert_called_once_with(
         where={"id": "version123"}, include={"Agent": True}
     )
+    mock_user_agent.return_value.find_first.assert_called_once_with(
+        where={
+            "userId": "test-user",
+            "agentId": "agent1",
+            "agentVersion": 1,
+        }
+    )
     mock_user_agent.return_value.create.assert_called_once_with(
         data=prisma.types.UserAgentCreateInput(
             userId="test-user", agentId="agent1", agentVersion=1, isCreatedByUser=False

autogpt_platform/backend/backend/server/v2/store/db.py

@@ -172,6 +172,9 @@ async def get_store_creators(
     # Build where clause with sanitized inputs
     where = {}
 
+    if featured:
+        where["isFeatured"] = featured
+
     # Add search filter if provided, using parameterized queries
     if search_query:
         # Sanitize and validate search query by escaping special characters
@@ -247,6 +250,7 @@ async def get_store_creators(
                 num_agents=creator.num_agents,
                 agent_rating=creator.agent_rating,
                 agent_runs=creator.agent_runs,
+                is_featured=creator.is_featured,
             )
             for creator in creators
         ]

autogpt_platform/backend/backend/server/v2/store/db_test.py

@@ -113,6 +113,7 @@ async def test_get_store_creator_details(mocker):
         agent_rating=4.5,
         agent_runs=10,
         top_categories=["test"],
+        is_featured=False,
     )
 
     # Mock prisma call
@@ -197,6 +198,7 @@ async def test_update_profile(mocker):
         description="Test description",
         links=["link1"],
         avatarUrl="avatar.jpg",
+        isFeatured=False,
         createdAt=datetime.now(),
         updatedAt=datetime.now(),
     )
@@ -215,6 +217,7 @@ async def test_update_profile(mocker):
         description="Test description",
         links=["link1"],
         avatar_url="avatar.jpg",
+        is_featured=False,
     )
 
     # Call function
@@ -239,6 +242,7 @@ async def test_get_user_profile(mocker):
         description="Test description",
         links=["link1", "link2"],
         avatarUrl="avatar.jpg",
+        isFeatured=False,
         createdAt=datetime.now(),
         updatedAt=datetime.now(),
     )

autogpt_platform/backend/backend/server/v2/store/model.py

@@ -74,6 +74,7 @@ class Creator(pydantic.BaseModel):
     num_agents: int
     agent_rating: float
     agent_runs: int
+    is_featured: bool
 
 
 class CreatorsResponse(pydantic.BaseModel):
@@ -98,6 +99,7 @@ class Profile(pydantic.BaseModel):
     description: str
     links: list[str]
     avatar_url: str
+    is_featured: bool
 
 
 class StoreSubmission(pydantic.BaseModel):

autogpt_platform/backend/backend/server/v2/store/model_test.py

@@ -88,6 +88,7 @@ def test_creator():
         description="Test description",
         avatar_url="avatar.jpg",
         num_agents=5,
+        is_featured=False,
     )
     assert creator.name == "Test Creator"
     assert creator.num_agents == 5
@@ -104,6 +105,7 @@ def test_creators_response():
                 description="Test description",
                 avatar_url="avatar.jpg",
                 num_agents=5,
+                is_featured=False,
             )
         ],
         pagination=backend.server.v2.store.model.Pagination(

autogpt_platform/backend/backend/server/v2/store/routes_test.py

@@ -402,6 +402,7 @@ def test_get_creators_pagination(mocker: pytest_mock.MockFixture):
                 num_agents=1,
                 agent_rating=4.5,
                 agent_runs=100,
+                is_featured=False,
             )
             for i in range(5)
         ],

autogpt_platform/backend/migrations/20241212142024_creator_featured_flag/migration.sql

@@ -0,0 +1,2 @@
+-- AlterTable
+ALTER TABLE "Profile" ADD COLUMN     "isFeatured" BOOLEAN NOT NULL DEFAULT false;

autogpt_platform/backend/migrations/20241212150828_agent_store_v2_views/migration.sql

@@ -71,6 +71,7 @@ SELECT
     p.description,
     ARRAY_AGG(DISTINCT c) FILTER (WHERE c IS NOT NULL) as top_categories,
     p.links,
+    p."isFeatured" as is_featured,
     COALESCE(ast.num_agents, 0) as num_agents,
     COALESCE(ast.agent_rating, 0.0) as agent_rating,
     COALESCE(ast.agent_runs, 0) as agent_runs
@@ -84,7 +85,7 @@ LEFT JOIN LATERAL (
     AND sl."isDeleted" = FALSE 
     AND sl."isApproved" = TRUE
 ) cats ON true
-GROUP BY p.username, p.name, p."avatarUrl", p.description, p.links, 
+GROUP BY p.username, p.name, p."avatarUrl", p.description, p.links, p."isFeatured",
          ast.num_agents, ast.agent_rating, ast.agent_runs;
 
 CREATE VIEW "StoreSubmission" AS

autogpt_platform/backend/schema.prisma

@@ -423,6 +423,8 @@ model Profile {
 
   avatarUrl String?
 
+  isFeatured Boolean @default(false)
+
   @@index([username])
   @@index([userId])
 }
@@ -439,6 +441,7 @@ view Creator {
   num_agents   Int
   agent_rating Float
   agent_runs   Int
+  is_featured  Boolean
 }
 
 view StoreAgent {