Etape 2 fini

2026-06-17 10:18:55 +02:00
parent bacfe49578
commit 981cfb6f2e
7 changed files with 207 additions and 126 deletions
--- a/backend/app/agents/dev_agent.py
+++ b/backend/app/agents/dev_agent.py
@@ -1,14 +1,69 @@
+import json
+import logging
+import httpx
+from langchain_openai import ChatOpenAI
+from langchain_core.messages import SystemMessage, HumanMessage
+from app.core.config import settings
+from app.schemas.code_output import ProjectCodeOutput, GeneratedFile
+from app.llm.prompts import DEV_AGENT_PROMPT
+
+logger = logging.getLogger(__name__)
+
+# Clients HTTPX configurés pour ignorer les blocages de certificats/révocation du lab
+sync_client = httpx.Client(verify=False)
+async_client = httpx.AsyncClient(verify=False)
+
 async def run_dev_agent(spec: dict, qa_feedback: list = None) -> dict:
    """
-    Agent Dev minimal :
-    - retourne une pseudo arborescence + un code exemple
+    Agent Dev : prend un état/spec validé, génère l'arborescence et le code,
+    et valide techniquement la sortie avant de la transmettre à la QA.
    """
-    return {
-        "tree": [
-            "main.py",
-            "README.md",
-            "app/__init__.py",
-        ],
-        "code": 'print("Hello from ARC generated project")',
-        "spec_title": spec.get("title"),
-    }
+    logger.info(f"[Dev Agent] Début de la génération pour : {spec.get('title', 'Sans titre')}")
+
+    llm = ChatOpenAI(
+        base_url=settings.llm_base_url,
+        api_key=settings.llm_api_key,
+        model=settings.llm_model_dev,
+        temperature=0.2,
+        max_retries=2,
+        http_client=sync_client,
+        http_async_client=async_client,
+        model_kwargs={"response_format": {"type": "json_object"}}
+    )
+    structured_llm = llm.with_structured_output(ProjectCodeOutput, strict=True)
+
+    messages = [
+        SystemMessage(content=DEV_AGENT_PROMPT),
+    ]
+
+    user_content = f"CAHIER DES CHARGES (ProjectSpec) :\n{json.dumps(spec, indent=2, ensure_ascii=False)}\n\n"
+    if qa_feedback:
+        user_content += f"⚠️ RETOURS DE VALIDATION QA (Corrections à appliquer impérativement) :\n{json.dumps(qa_feedback, indent=2, ensure_ascii=False)}\n\n"
+    
+    user_content += "Génère maintenant le JSON complet contenant l'arborescence ('tree') et tous les fichiers ('files') décrits."
+    messages.append(HumanMessage(content=user_content))
+
+    try:
+        validated_code = await structured_llm.ainvoke(messages)
+        logger.info(f"[Dev Agent] ✅ Code généré et validé avec succès ({len(validated_code.files)} fichiers)")
+        return validated_code.model_dump()
+
+    except Exception as e:
+        logger.error(f"[Dev Agent] ❌ Échec de la génération/validation : {type(e).__name__}: {str(e)}")
+        return _generate_fallback_code_output(spec)
+
+def _generate_fallback_code_output(spec: dict) -> dict:
+    """Génère un livrable minimal de secours en cas de crash du LLM."""
+    logger.warning("[Dev Agent] Génération du package de secours (Fallback)")
+    title = spec.get("title", "automation_script")
+    
+    fallback = ProjectCodeOutput(
+        spec_title=title,
+        tree=["main.py", "README.md", "requirements.txt"],
+        files=[
+            GeneratedFile(path="main.py", content="import logging\nlogging.basicConfig(level=logging.INFO)\n\ndef main():\n    logging.error('Le Dev Agent a rencontré une erreur de génération.')\n\nif __name__ == '__main__':\n    main()"),
+            GeneratedFile(path="README.md", content=f"# {title}\nGénération en mode fallback suite à une erreur technique."),
+            GeneratedFile(path="requirements.txt", content="# Aucune dépendance externe définie (Fallback)\n")
+        ]
+    )
+    return fallback.model_dump()