Added pre populate database command

rnd/market/market/routes/admin.py

@@ -7,7 +7,7 @@
 router = fastapi.APIRouter()
 
 
-@router.post("/admin/agent", response_model=market.model.AgentResponse)
+@router.post("/agent", response_model=market.model.AgentResponse)
 async def create_agent_entry(request: market.model.AddAgentRequest):
     """
     A basic endpoint to create a new agent entry in the database.

rnd/market/pyproject.toml

@@ -39,6 +39,8 @@ build-backend = "poetry.core.masonry.api"
 format = "scripts:format"
 lint = "scripts:lint"
 app = "scripts:app"
+setup = "scripts:setup"
+populate = "scripts:populate_database"
 
 [tool.pytest-watcher]
 now = false

rnd/market/scripts.py

@@ -19,6 +19,29 @@ def lint():
         print("Lint failed, try running `poetry run format` to fix the issues: ", e)
         raise e
 
+def populate_database():
+    import requests
+    import pathlib
+    import glob
+    import json
+    import market.model
+
+    templates = pathlib.Path(__file__).parent.parent / "autogpt_server" / "graph_templates"
+
+    all_files = glob.glob(str(templates / "*.json"))
+
+    for file in all_files:
+        with open(file, "r") as f:
+            data = f.read()
+            req = market.model.AddAgentRequest(
+                graph=json.loads(data),
+                author="Populate DB",
+                categories=["Pre-Populated"],
+                keywords=["test"]
+            )
+            response = requests.post("http://localhost:8001/market/admin/agent", json=req.model_dump())
+            print(response.text)
+
 
 def format():
     run("ruff", "check", "--fix", ".")
@@ -28,4 +51,9 @@ def format():
 
 
 def app():
-    run("uvicorn", "market.app:app", "--reload" , "--port", "8001")
+    run("uvicorn", "market.app:app", "--reload", "--port", "8001")
+
+
+def setup():
+    run("prisma", "generate")
+    run("prisma", "migrate", "deploy")