From 75e23de4ba2b94474b75163c3b187ed5a4f7472a Mon Sep 17 00:00:00 2001 From: itqop Date: Sat, 19 Jul 2025 17:35:09 +0300 Subject: [PATCH] Reformat code --- src/graph/nodes/guardrails.py | 2 +- src/graph/nodes/parse_output.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/graph/nodes/guardrails.py b/src/graph/nodes/guardrails.py index eab5d43..64ded1d 100644 --- a/src/graph/nodes/guardrails.py +++ b/src/graph/nodes/guardrails.py @@ -43,7 +43,7 @@ def guardrails_node(state: EmailGenerationState) -> EmailGenerationState: meta = { "locale": lead_model.locale if lead_model else "ru", - "lead_normalized": lead_model.dict() if lead_model else {}, + "lead_normalized": lead_model.model_dump() if lead_model else {}, "used_chunks": email_draft.used_chunks, "model": llm_output.model if llm_output else "unknown", "tokens_prompt": llm_output.tokens_prompt if llm_output else 0, diff --git a/src/graph/nodes/parse_output.py b/src/graph/nodes/parse_output.py index 82fd752..48a3eaf 100644 --- a/src/graph/nodes/parse_output.py +++ b/src/graph/nodes/parse_output.py @@ -41,7 +41,7 @@ def parse_output_node(state: EmailGenerationState) -> EmailGenerationState: ranked_context = state.get("ranked_context") real_chunk_ids = [] - + if isinstance(used_chunks, list) and ranked_context: for chunk_idx in used_chunks: try: