fix

backend/api/demo.py

@@ -48,7 +48,7 @@ from rl.types import RepairAction, REPAIR_ACTION_NAMES, REPAIR_ACTION_BY_NAME
 from rl.error_classifier import classify_error, extract_offending_token
 from rl.grader import GraderInput, compute_reward, compute_episode_reward
 from rl.types import RLState, EpisodeStep, featurize, ERROR_CLASS_NAMES
-from gepa.optimizer import get_gepa, QueryResult
+from gepa.optimizer import get_gepa, QueryResult, GEPA_OPTIMIZE_EVERY
 
 router = APIRouter()
 
@@ -136,12 +136,14 @@ async def get_prompt_history():
         }
         for c in sorted(pareto, key=lambda x: x.generation)
     ]
+    query_count = len(gepa.get_history())
     return {
         "prompt": gepa.get_current_prompt(),
         "generation": gepa.current_generation,
         "history": history,
-        "queryCount": len(gepa.get_history()),
-        "nextOptimizeAt": (len(gepa.get_history()) // 4 + 1) * 4,
+        "queryCount": query_count,
+        "optimizeEvery": GEPA_OPTIMIZE_EVERY,
+        "cycleProgress": query_count % GEPA_OPTIMIZE_EVERY,
     }
 
 

backend/gepa/optimizer.py

@@ -26,6 +26,10 @@ GEPA_PATH = _DATA_DIR / "gepa_prompt.json"
 
 _MODEL = os.environ.get("MODEL_NAME", "gpt-4o-mini")
 
+# How many queries between each GEPA optimization cycle.
+# Override with the GEPA_OPTIMIZE_EVERY environment variable.
+GEPA_OPTIMIZE_EVERY: int = int(os.environ.get("GEPA_OPTIMIZE_EVERY", "4"))
+
 SEED_SYSTEM_PROMPT = """You are a SQL expert. Given a natural language question and a SQLite database schema, write a correct SQL query.
 
 Rules:
@@ -165,7 +169,7 @@ class GEPAOptimizer:
         return max(c.generation for c in self._pareto_front)
 
     def should_optimize(self) -> bool:
-        return len(self._history) > 0 and len(self._history) % 4 == 0
+        return len(self._history) > 0 and len(self._history) % GEPA_OPTIMIZE_EVERY == 0
 
     def reset(self) -> None:
         self._history.clear()

frontend/src/components/PromptEvolution.tsx

@@ -22,17 +22,18 @@ export function PromptEvolution() {
   const generation = promptGeneration
 
   const [queryCount, setQueryCount] = useState(0)
-  const [nextAt, setNextAt] = useState(4)
+  const [optimizeEvery, setOptimizeEvery] = useState(4)
+  const [cycleProgress, setCycleProgress] = useState(0)
 
   const loadHistory = async () => {
     setLoading(true)
     try {
       const data = await fetchPromptHistory()
       setPromptData(data)
-      if ((data as Record<string, unknown>).queryCount !== undefined) {
-        setQueryCount((data as Record<string, unknown>).queryCount as number)
-        setNextAt((data as Record<string, unknown>).nextOptimizeAt as number)
-      }
+      const d = data as Record<string, unknown>
+      if (d.queryCount !== undefined) setQueryCount(d.queryCount as number)
+      if (d.optimizeEvery !== undefined) setOptimizeEvery(d.optimizeEvery as number)
+      if (d.cycleProgress !== undefined) setCycleProgress(d.cycleProgress as number)
     } catch {
       // noop
     } finally {
@@ -80,12 +81,14 @@ export function PromptEvolution() {
         <div className="flex flex-col gap-1">
           <div className="flex items-center justify-between text-[9px] text-gray-600">
             <span>{queryCount} queries processed</span>
-            <span>{nextAt - queryCount} until next optimization</span>
+            <span className="text-gray-700">
+              {cycleProgress}/{optimizeEvery} · optimizes every {optimizeEvery}
+            </span>
           </div>
           <div className="h-1 bg-white/5 rounded-full overflow-hidden">
             <div
               className="h-full rounded-full bg-violet-500/50 transition-all duration-500"
-              style={{ width: `${((queryCount % 4) / 4) * 100}%` }}
+              style={{ width: `${(cycleProgress / optimizeEvery) * 100}%` }}
             />
           </div>
         </div>