Update backend.py

backend.py

@@ -1,4 +1,4 @@
-# backend.py — FINAL HARDENED VERSION
+# backend.py — FINAL HARDENED VERSION v1.1
 import sqlite3
 import os
 import json
@@ -74,7 +74,8 @@ def load_model(model_name):
     if model_name in _MODEL_CACHE: return _MODEL_CACHE[model_name]
     print(f"Loading model: {model_name}...")
     tokenizer = AutoTokenizer.from_pretrained(model_name)
-    model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype="auto", device_map="auto", trust_remote_code=True, attn_implementation="eager")
+    # --- THIS IS THE FIX FOR THE WARNING ---
+    model = AutoModelForCausalLM.from_pretrained(model_name, dtype="auto", device_map="auto", trust_remote_code=True, attn_implementation="eager")
     _MODEL_CACHE[model_name] = (tokenizer, model)
     print(f"Model {model_name} loaded and cached.")
     return tokenizer, model
@@ -140,7 +141,7 @@ def generate_with_model(role: str, prompt: str) -> str:
         print(f"Error during model generation for role {role}: {e}")
         return f'{{"error": "Failed to generate response: {str(e)}"}}'
 
-# ------------------------------ THE AGENT CHAIN EXECUTOR ------------------------------
+# ------------------------------ THE AGENT CHAIN EXECUTOR (REWRITTEN FOR RELIABILITY) ------------------------------
 def run_agent_chain(project_id, user_id, initial_prompt):
     project_dir = get_project_dir(user_id, project_id)
     log_entries = []