Merge branch 'develop' of https://github.com/bluewave-labs/verifywise into develop

Author: Harsh

.claude/settings.local.json

@@ -4,7 +4,14 @@
       "Bash(npx playwright test:*)",
       "Bash(find /Users/harsh/Data/vw/verifywise/Clients/src -type f -name *login* -o -name *Login*)",
       "Bash(grep -E \"\\\\.\\(ts|js\\)$\")",
-      "Bash(npm run:*)"
+      "Bash(npm run:*)",
+      "Bash(ls:*)",
+      "Bash(npx tsc:*)",
+      "Bash(python3:*)",
+      "Bash(find /Users/harsh/Data/vw/verifywise/Clients -type f \\\\\\(-name *.yaml -o -name *.yml -o -name docker-compose* \\\\\\))",
+      "Bash(find /Users/harsh/Data/vw/verifywise -type f \\\\\\(-name *.yaml -o -name *.yml -o -name docker-compose* \\\\\\) -not -path */node_modules/*)",
+      "Bash(grep -r \"superadmin\\\\|super.admin\\\\|SuperAdmin\" /Users/harsh/Data/vw/verifywise/Servers/src --include=*.ts -i)",
+      "Bash(grep:*)"
     ]
   }
 }

AIGateway/Dockerfile

@@ -3,4 +3,6 @@ WORKDIR /app
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 COPY src/ .
-CMD ["sh", "-c", "alembic upgrade head && uvicorn app:app --host 0.0.0.0 --port 8100"]
+COPY start.sh .
+RUN chmod +x start.sh
+CMD ["./start.sh"]

AIGateway/src/routers/prompts.py

@@ -1,11 +1,15 @@
+import json as _json
 import re
 from typing import Any, Optional
 
 from fastapi import APIRouter, HTTPException, Request
+from fastapi.responses import StreamingResponse
 from pydantic import BaseModel, field_validator
 
 from crud import prompts as crud
 from middlewares.auth import verify_internal_key
+from services.llm_service import stream_chat_completion
+from services.proxy_service import resolve_endpoint_for_key
 from utils.auth import get_org_id, get_user_id, require_admin
 from utils.notifications import notify_config_change
 
@@ -377,27 +381,42 @@ async def delete_test_dataset(
     return {"deleted": True}
 
 
-@router.post("/test", status_code=501)
+@router.post("/test")
 async def test_prompt(request: Request, body: TestPromptRequest):
     """
-    Test a prompt by resolving variables and proxying to the LLM completion
-    endpoint. The LLM proxy integration lives in the Express backend for now;
-    this stub returns 501 until the proxy is co-located in the AIGateway
-    service.
+    Test a prompt by resolving variables and streaming the LLM response.
+    Returns an SSE stream compatible with the frontend's streamPromptTest().
     """
     verify_internal_key(request)
+    org_id = get_org_id(request)
 
-    # Resolve variables so callers can at least validate substitution locally
+    # Resolve variables in the prompt content
     resolved_content = crud.resolve_variables(
         body.content,
         body.variables or {},
     )
 
-    raise HTTPException(
-        status_code=501,
-        detail=(
-            "test-prompt requires proxy integration — "
-            "LLM proxy currently runs in the Express backend. "
-            "Resolved content is available but cannot be forwarded yet."
-        ),
-    )
+    # Resolve the endpoint to get provider, model, and API key
+    try:
+        endpoint = await resolve_endpoint_for_key(
+            organization_id=org_id,
+            endpoint_slug=body.endpoint_slug,
+            allowed_endpoint_ids=[],
+        )
+    except ValueError as e:
+        raise HTTPException(status_code=404, detail=str(e))
+
+    # Stream the LLM response
+    async def _stream():
+        try:
+            async for chunk_str in stream_chat_completion(
+                model=endpoint["model"],
+                messages=resolved_content,
+                api_key=endpoint["decrypted_key"],
+            ):
+                yield chunk_str
+        except Exception as e:
+            yield f"data: {_json.dumps({'error': str(e)})}\n\n"
+            yield "data: [DONE]\n\n"
+
+    return StreamingResponse(_stream(), media_type="text/event-stream")

AIGateway/start.sh

@@ -0,0 +1,43 @@
+#!/bin/sh
+set -e
+
+echo "Waiting for main server to create required tables..."
+
+DELAY=2
+MAX_DELAY=30
+ATTEMPT=1
+
+until python -c "
+import asyncio, asyncpg, os
+
+async def check():
+    conn = await asyncpg.connect(
+        host=os.environ['DB_HOST'],
+        port=int(os.environ.get('DB_PORT', 5432)),
+        user=os.environ['DB_USER'],
+        password=os.environ['DB_PASSWORD'],
+        database=os.environ['DB_NAME']
+    )
+    exists = await conn.fetchval(
+        \"SELECT EXISTS(SELECT 1 FROM information_schema.tables WHERE table_schema='verifywise' AND table_name='organizations')\"
+    )
+    await conn.close()
+    if not exists:
+        raise SystemExit(1)
+
+asyncio.run(check())
+" 2>/dev/null; do
+  echo "Attempt $ATTEMPT: organizations table not ready, retrying in ${DELAY}s..."
+  sleep $DELAY
+  ATTEMPT=$((ATTEMPT + 1))
+  DELAY=$((DELAY * 2))
+  if [ $DELAY -gt $MAX_DELAY ]; then
+    DELAY=$MAX_DELAY
+  fi
+done
+
+echo "Tables ready! Running AI Gateway migrations..."
+alembic upgrade head
+
+echo "Starting AI Gateway..."
+uvicorn app:app --host 0.0.0.0 --port 8100

Clients/src/application/hooks/__tests__/useAuth.test.ts

@@ -32,12 +32,14 @@ function createWrapper(authToken: string) {
         expirationDate: null,
         onboardingStatus: "completed",
         isOrgCreator: false,
+        isSuperAdmin: false,
+        activeOrganizationId: null,
       },
     },
   });
 
   return ({ children }: { children: React.ReactNode }) =>
-    React.createElement(Provider, { store }, children);
+    React.createElement(Provider, { store, children });
 }
 
 describe("useAuth", () => {

Clients/src/application/hooks/__tests__/useIsAdmin.test.ts

@@ -27,12 +27,14 @@ function createWrapper(authToken: string) {
         expirationDate: null,
         onboardingStatus: "completed",
         isOrgCreator: false,
+        isSuperAdmin: false,
+        activeOrganizationId: null,
       },
     },
   });
 
   return ({ children }: { children: React.ReactNode }) =>
-    React.createElement(Provider, { store }, children);
+    React.createElement(Provider, { store, children });
 }
 
 describe("useIsAdmin", () => {

Clients/src/application/hooks/__tests__/useLogout.test.ts

@@ -29,15 +29,16 @@ function createWrapper() {
         expirationDate: Date.now() + 3600000,
         onboardingStatus: "completed",
         isOrgCreator: false,
+        isSuperAdmin: false,
+        activeOrganizationId: null,
       },
     },
   });
 
   const Wrapper = ({ children }: { children: React.ReactNode }) =>
     React.createElement(
       Provider,
-      { store },
-      React.createElement(MemoryRouter, null, children)
+      { store, children: React.createElement(MemoryRouter, null, children) },
     );
 
   return { Wrapper, store };

Clients/src/application/redux/ui/__tests__/uiSlice.test.ts

@@ -65,7 +65,7 @@ describe("uiSlice", () => {
         stateWithTable,
         setRowsPerPage({ table: "vendors", value: 25 })
       );
-      expect(state.vendors.rowsPerPage).toBe(25);
+      expect((state as any).vendors.rowsPerPage).toBe(25);
     });
 
     it("should be a no-op for an unknown table key", () => {

Clients/src/presentation/components/CommandPalette/__tests__/CommandPalette.test.tsx

@@ -1,4 +1,4 @@
-import { screen, fireEvent } from "@testing-library/react";
+import { screen } from "@testing-library/react";
 import { renderWithProviders } from "../../../../test/renderWithProviders";
 import { CommandPalette } from "../index";
 

Clients/src/presentation/components/Inputs/Field/__tests__/Field.test.tsx

@@ -103,7 +103,7 @@ describe("Field Component", () => {
   it("does not display error when error prop is empty", () => {
     renderWithProviders(<Field label="Email" />);
 
-    expect(screen.queryByClassName?.("input-error")).not.toBeTruthy();
+    expect(screen.queryByText(/error/i)).not.toBeTruthy();
   });
 
   it("renders disabled input", () => {