Add LLM DB schema, seed script & rename tests

Create a new llm table SQL schema (vector(1536), prompt, completion, duration, time, data). Add a small psycopg2 seed script (insert_llm_lorem.py) that loads .env, connects to the DB and inserts 5 dummy records with random 1536-d vectors. Rename tests/test_gemini.py to tests/test_llm.py and update tests to use LLM naming and /llm endpoints, adjust skip/failure messages and assertions accordingly (including updated mocked API handling). These changes introduce the DB schema and local seeding helper and align tests to the LLM endpoint/name changes.

Author: goldlabelapps

app/api/llm/sql/create_table.sql

@@ -0,0 +1,10 @@
+
+CREATE TABLE IF NOT EXISTS llm (
+	id SERIAL PRIMARY KEY,
+	vector vector(1536),
+	prompt TEXT NOT NULL,
+	completion TEXT NOT NULL,
+	duration FLOAT,
+	time TIMESTAMPTZ DEFAULT NOW(),
+	data JSONB
+);

app/api/llm/sql/insert_llm_lorem.py

@@ -0,0 +1,25 @@
+import os
+import psycopg2
+from dotenv import load_dotenv
+import random
+
+load_dotenv()
+
+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'),
+)
+cur = conn.cursor()
+lorem = 'Lorem ipsum dolor sit amet, consectetur adipiscing elit.'
+for i in range(5):
+    cur.execute(
+        'INSERT INTO llm (vector, prompt, completion, duration, data) VALUES (%s, %s, %s, %s, %s);',
+        ([random.random() for _ in range(1536)], lorem, lorem, random.uniform(0.1, 2.0), '{}')
+    )
+conn.commit()
+print('Inserted 5 records.')
+cur.close()
+conn.close()

tests/test_gemini.py tests/test_llm.pytests/test_gemini.py renamed to tests/test_llm.py

@@ -1,9 +1,9 @@
 import os
 import pytest
-def test_gemini_real_api():
+def test_llm_real_api():
     api_key = os.getenv("GEMINI_API_KEY")
     if not api_key:
-        pytest.skip("GEMINI_API_KEY not set; skipping real Gemini API test.")
+        pytest.skip("GEMINI_API_KEY not set; skipping real LLM API test.")
     from google import genai
     client = genai.Client(api_key=api_key)
     try:
@@ -14,7 +14,7 @@ def test_gemini_real_api():
         completion = getattr(response, "text", None)
         assert completion is not None and "hello" in completion.lower()
     except Exception as e:
-        pytest.fail(f"Gemini real API call failed: {e}")
+        pytest.fail(f"LLM real API call failed: {e}")
 import sys
 import os
 sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "../")))
@@ -26,16 +26,17 @@ def test_gemini_real_api():
 
 
 
-def test_gemini_get_endpoint():
-    response = client.get("/gemini")
+
+def test_llm_get_endpoint():
+    response = client.get("/llm")
     assert response.status_code == 200
     data = response.json()
     assert "meta" in data
     assert data["meta"]["severity"] == "success"
-    assert "Gemini endpoint says hello" in data["meta"]["title"]
+    assert "LLM endpoint says hello" in data["meta"]["title"]
 
 
-def test_gemini_post_endpoint(monkeypatch):
+def test_llm_post_endpoint(monkeypatch):
     # Mock google-genai SDK to avoid real API call
     class MockGenAIResponse:
         text = "Test completion"
@@ -50,12 +51,12 @@ class MockGenAIClient:
     monkeypatch.setattr("google.genai.Client", lambda *args, **kwargs: MockGenAIClient())
 
     payload = {"prompt": "Test prompt"}
-    response = client.post("/gemini", json=payload)
+    response = client.post("/llm", json=payload)
     assert response.status_code == 200
     data = response.json()
     assert "meta" in data
     assert data["meta"]["severity"] == "success"
-    assert "Gemini completion received" in data["meta"]["title"]
+    assert "completion received" in data["meta"]["title"]
     assert data["data"]["prompt"] == "Test prompt"
     assert data["data"]["completion"] == "Test completion"
     assert "data" in data