Merge pull request #87 from goldlabelapps/staging

Add PATCH /prompt/drop endpoint

Author: goldlabelapps

app/api/prompt/__init__.py

@@ -1,4 +1,6 @@
 """Prompt Routes"""
 
+
 from .prompt import router as prompt_router
 from .linkedin import router as linkedin_router
+from .drop import router as drop_router

app/api/prompt/drop.py

@@ -0,0 +1,24 @@
+import os
+
+from fastapi import APIRouter, Depends, HTTPException
+
+from app.utils.api_key_auth import get_api_key
+from app.utils.db import get_db_connection_direct
+from app.utils.make_meta import make_meta
+
+router = APIRouter()
+
+# PATCH /prompt/drop: empties the prompt table
+@router.patch("/prompt/drop")
+def drop_prompt_table(api_key: str = Depends(get_api_key)) -> dict:
+    """PATCH /prompt/drop: empties the prompt table."""
+    try:
+        conn = get_db_connection_direct()
+        cur = conn.cursor()
+        cur.execute("TRUNCATE TABLE prompt RESTART IDENTITY CASCADE;")
+        conn.commit()
+        cur.close()
+        conn.close()
+        return {"meta": make_meta("success", "Prompt table emptied"), "data": {}}
+    except Exception as e:
+        return {"meta": make_meta("error", f"Failed to empty prompt table: {str(e)}"), "data": {}}

app/api/routes.py

@@ -10,6 +10,7 @@
 from app.utils.notify.resend import router as resend_router
 from app.api.prompt.prompt import router as prompt_router
 from app.api.prompt.linkedin import router as linkedin_router
+from app.api.prompt.drop import router as drop_router
 from app.api.prospects.prospects import router as prospects_router
 from app.api.orders.orders import router as orders_router
 
@@ -18,5 +19,6 @@
 router.include_router(health_router)
 router.include_router(prompt_router)
 router.include_router(linkedin_router)
+router.include_router(drop_router)
 router.include_router(prospects_router)
 router.include_router(orders_router)