Upload 15 files

inference.py

@@ -31,7 +31,10 @@ except ImportError:
 # Defaults are set only for API_BASE_URL and MODEL_NAME 
 # (and should reflect your active inference setup):
 API_BASE_URL = os.getenv("API_BASE_URL", "https://router.huggingface.co/v1")
-MODEL_NAME = os.getenv("MODEL_NAME", "meta-llama/Llama-3.3-70B-Instruct")
+MODEL_NAME = os.getenv("MODEL_NAME")
+if MODEL_NAME is None:
+    print("MODEL_NAME environment variable is required", file=sys.stderr, flush=True)
+    sys.exit(1)
 HF_TOKEN = os.getenv("HF_TOKEN")
 LOCAL_IMAGE_NAME = os.getenv("LOCAL_IMAGE_NAME")
 
@@ -201,7 +204,7 @@ Generate your next action (SQL or action name):"""
                 return 'QUERY'  # Default fallback
                 
         except Exception as e:
-            print(f"[DEBUG] LLM generation failed: {e}", flush=True)
+            print(f"[DEBUG] LLM generation failed: {e}", flush=True, file=sys.stderr)
             return 'SELECT 1;'  # Safe fallback
     
     def run_task(self, task_id: str, seed: int = None):
@@ -244,6 +247,10 @@ Generate your next action (SQL or action name):"""
                     # Extract column name from response
                     column_match = re.search(r'PROFILE\s+(\w+)', action_text, re.IGNORECASE)
                     column_name = column_match.group(1) if column_match else None
+                elif re.match(r'^\s*(UPDATE|INSERT|DELETE|ALTER)\b', action_text, re.IGNORECASE):
+                    action_type = 'fix'
+                    sql = action_text
+                    column_name = None
                 else:
                     # Assume SQL query
                     action_type = 'query'
@@ -282,7 +289,7 @@ Generate your next action (SQL or action name):"""
                     break
             
         except Exception as e:
-            print(f"[DEBUG] Task execution failed: {e}", flush=True)
+            print(f"[DEBUG] Task execution failed: {e}", flush=True, file=sys.stderr)
             
         return success, steps_taken, score, rewards
 
@@ -297,7 +304,7 @@ def main():
     if not HF_TOKEN:
         print("Error: HF_TOKEN environment variable is required", flush=True)
         print("Please set HF_TOKEN or ensure .env file exists with the token", flush=True)
-        return
+        sys.exit(1)
     
     # Initialize agent
     agent = SQLQualityAgent(API_BASE_URL, MODEL_NAME, HF_TOKEN, ENV_BASE_URL)

server/main.py

@@ -190,14 +190,10 @@ def load_static_html() -> str:
 
 # Validate LLM configuration at startup
 runtime = _resolve_llm_runtime()
-if not runtime["configured"]:
+LLM_AVAILABLE = runtime["configured"]
+if not LLM_AVAILABLE:
     missing_vars = ", ".join(runtime["missing"])
-    print(f"FATAL: Missing required LLM environment variables: {missing_vars}")
-    print("Please set the following environment variables:")
-    for var in runtime["missing"]:
-        print(f"  - {var}")
-    import sys
-    sys.exit(1)
+    print(f"[WARNING] LLM env vars not set: {missing_vars}. Server starting in mock mode.", flush=True)
 
 
 @app.get("/", response_class=HTMLResponse)