Prompts: add started/completed, remove duration_ms

Migrate prompts to track async timings: remove duration_ms and add started (BIGINT NOT NULL, default epoch) and completed (BIGINT nullable). Update API and Pydantic schemas to return and accept started/completed, initialize response and completed as NULL on create, and set started to provided value or current epoch. Includes SQL and Python migration scripts, bumps package version to 2.0.1, and tweaks app metadata.

Author: goldlabelapps

app/__init__.py

@@ -1,4 +1,4 @@
 """NX AI - FastAPI, Python, Postgres, tsvector"""
 
 # Current Version
-__version__ = "2.0.0"
+__version__ = "2.0.1"

app/api/prompts/prompts.py

@@ -42,9 +42,9 @@ def root() -> dict:
         conn = get_db_connection_direct()
         try:
             with conn.cursor() as cur:
-                cur.execute("SELECT id, prompt, response, duration_ms, llm, timestamp FROM prompts ORDER BY id DESC;")
+                cur.execute("SELECT id, prompt, response, llm, started, completed, timestamp FROM prompts ORDER BY id DESC;")
                 rows = cur.fetchall()
-                keys = ["id", "prompt", "response", "duration_ms", "llm", "timestamp"]
+                keys = ["id", "prompt", "response", "llm", "started", "completed", "timestamp"]
                 for row in rows:
                     data.append(dict(zip(keys, row)))
         finally:
@@ -59,21 +59,24 @@ def create_prompt(prompt_in: PromptCreate = Body(...)):
     """Create a new prompt record in the prompts table."""
     from app.utils.db import get_db_connection_direct
     import psycopg2
+    import time
     conn = get_db_connection_direct()
+    started = prompt_in.started if prompt_in.started is not None else int(time.time())
     try:
         with conn.cursor() as cur:
             cur.execute(
                 """
-                INSERT INTO prompts (prompt, response, duration_ms, llm, timestamp)
-                VALUES (%s, %s, %s, %s, COALESCE(%s, NOW()))
-                RETURNING id, prompt, response, duration_ms, llm, timestamp
+                INSERT INTO prompts (prompt, response, llm, started, completed, timestamp)
+                VALUES (%s, %s, %s, %s, %s, COALESCE(%s, NOW()))
+                RETURNING id, prompt, response, llm, started, completed, timestamp
                 """,
                 (
                     prompt_in.prompt,
-                    prompt_in.response,
-                    prompt_in.duration_ms,
+                    None,  # Always set response to NULL initially
                     prompt_in.llm,
-                    prompt_in.timestamp,
+                    started,
+                    None,  # completed is NULL initially
+                    None,  # timestamp
                 )
             )
             row = cur.fetchone()
@@ -84,7 +87,7 @@ def create_prompt(prompt_in: PromptCreate = Body(...)):
     finally:
         conn.close()
     if row:
-        keys = ["id", "prompt", "response", "duration_ms", "llm", "timestamp"]
+        keys = ["id", "prompt", "response", "llm", "started", "completed", "timestamp"]
         return dict(zip(keys, row))
     return {"error": "Failed to insert prompt."}
 

app/api/prompts/schemas.py

@@ -3,15 +3,14 @@
 
 class PromptCreate(BaseModel):
     prompt: str
-    response: str
-    duration_ms: int
     llm: str
-    timestamp: Optional[str] = None  # ISO format, optional for auto
+    started: Optional[int] = None  # Unix epoch, set by backend if not provided
 
 class PromptOut(BaseModel):
     id: int
     prompt: str
-    response: str
-    duration_ms: int
+    response: Optional[str]
     llm: str
+    started: int
+    completed: Optional[int]
     timestamp: str

app/api/prompts/utils/alter_prompts_table_add_started_completed.sql

@@ -0,0 +1,5 @@
+-- Migration: Update prompts table for async processing
+ALTER TABLE prompts
+    DROP COLUMN IF EXISTS duration_ms,
+    ADD COLUMN IF NOT EXISTS started BIGINT NOT NULL DEFAULT (extract(epoch from now())),
+    ADD COLUMN IF NOT EXISTS completed BIGINT;

app/api/prompts/utils/migrate_prompts_table.py

@@ -0,0 +1,40 @@
+"""
+Python migration script to update prompts table:
+- Remove duration_ms column
+- Add started (BIGINT NOT NULL, default now as epoch)
+- Add completed (BIGINT, nullable)
+"""
+import os
+import psycopg2
+from dotenv import load_dotenv
+
+load_dotenv()
+
+ALTER_SQL = """
+ALTER TABLE prompts
+    DROP COLUMN IF EXISTS duration_ms,
+    ADD COLUMN IF NOT EXISTS started BIGINT NOT NULL DEFAULT (extract(epoch from now())),
+    ADD COLUMN IF NOT EXISTS completed BIGINT;
+"""
+
+def main():
+    conn = psycopg2.connect(
+        host=os.getenv('DB_HOST'),
+        port=os.getenv('DB_PORT', '5432'),
+        dbname=os.getenv('DB_NAME'),
+        user=os.getenv('DB_USER'),
+        password=os.getenv('DB_PASSWORD'),
+    )
+    try:
+        with conn.cursor() as cur:
+            cur.execute(ALTER_SQL)
+            conn.commit()
+            print("Migration applied successfully.")
+    except Exception as e:
+        conn.rollback()
+        print(f"Migration failed: {e}")
+    finally:
+        conn.close()
+
+if __name__ == "__main__":
+    main()

app/main.py

@@ -7,8 +7,8 @@
 from app.api.routes import router
 
 app = FastAPI(
-    title="NX AI",
-    description="NX AI - FastAPI, Python, Postgres, tsvector",
+    title="Python",
+    description="FastAPI, Python, Postgres, tsvector",
     version=__version__,
 )