Upload 4 files

app/agents/analyzer.py

@@ -1,8 +0,0 @@
-from app.services.llm_service import call_gpt
-
-def analyzer_node(state):
-    response = call_gpt(
-        "You are a software architect. Return structured JSON.",
-        state["prompt"]
-    )
-    return {"structured_prompt": response}

app/agents/generator.py

@@ -0,0 +1,9 @@
+from app.services.llm_service import call_gpt
+
+def generator_node(state):
+    response = call_gpt(
+        "Generate production-ready code.",
+        state["structured_prompt"],
+        temperature=0.2
+    )
+    return {"code": response}

app/agents/refactor.py

@@ -0,0 +1,19 @@
+from app.services.llm_service import call_gpt
+
+def refactor_node(state):
+    response = call_gpt(
+        "Fix and improve code based on review.",
+        f"""
+        Code:
+        {state['code']}
+
+        Review:
+        {state['review']}
+        """,
+        temperature=0.2
+    )
+
+    return {
+        "code": response,
+        "iteration": state["iteration"] + 1
+    }

app/agents/reviewer.py

@@ -0,0 +1,25 @@
+from app.services.llm_service import call_gpt
+from app.services.code_tools import run_pylint, run_bandit
+
+def reviewer_node(state):
+    code = state["code"]
+
+    pylint_report = run_pylint(code)
+    bandit_report = run_bandit(code)
+
+    response = call_gpt(
+        "Review code with lint + security insights. Return JSON.",
+        f"""
+        Code:
+        {code}
+
+        Pylint:
+        {pylint_report}
+
+        Bandit:
+        {bandit_report}
+        """,
+        temperature=0
+    )
+
+    return {"review": response}