mirzagitov03/brojs-task-6a22c713fd30e81cf315ea04
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Solution published: update main.py

Browse Source
This commit is contained in:
Ринар Мирзагитов
2026-06-16 10:42:28 +00:00
parent ea87ac85f0
commit 3187fb5d2f
1 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
main.py
+4 -5
View File
@@ -76,7 +76,7 @@ def reflect(state: CodeReviewState) -> CodeReviewState:
response = llm.invoke([HumanMessage(content=prompt)]) response = llm.invoke([HumanMessage(content=prompt)])
try: try:
data = ReflectOutput.model_validate_json(response.content) data = ReflectOutput.model_validate_json(response.content)
except Exception as e: except Exception:
# Fallback simple parsing # Fallback simple parsing
data = ReflectOutput.model_validate_json("{\"scores\":{\"pep8\":5,\"type_hints\":5,\"edge_cases\":5,\"naming\":5},\"weakest_criterion\":\"pep8\",\"verdict\":\"needs_revision\"}") data = ReflectOutput.model_validate_json("{\"scores\":{\"pep8\":5,\"type_hints\":5,\"edge_cases\":5,\"naming\":5},\"weakest_criterion\":\"pep8\",\"verdict\":\"needs_revision\"}")
state["criteria_scores"] = data.scores state["criteria_scores"] = data.scores
@@ -111,13 +111,12 @@ builder.add_conditional_edges(
lambda state: state["verdict"] == "ok", lambda state: state["verdict"] == "ok",
{"ok": END, "needs_revision": "rewrite"}, {"ok": END, "needs_revision": "rewrite"},
) )
# After rewrite, decide to reflect again or end if max rounds reached
builder.add_conditional_edges( builder.add_conditional_edges(
"rewrite", "rewrite",
lambda state: state["round"] < state["max_rounds"], lambda state: "reflect" if state["round"] < state["max_rounds"] else END,
{"rewrite": "reflect", "maxed": END}, {"reflect": "reflect", "END": END}
) )
# If maxed, go to END
builder.add_edge("rewrite", "maxed")
graph = builder.compile() graph = builder.compile()