From 8216ebacd32213f284d0fb6864681ec94b922f2c Mon Sep 17 00:00:00 2001 From: QuantumGhost Date: Tue, 15 Jul 2025 15:19:18 +0800 Subject: [PATCH] chore(api): format code --- .../graph_engine/entities/test_graph_runtime_state.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/api/tests/unit_tests/core/workflow/graph_engine/entities/test_graph_runtime_state.py b/api/tests/unit_tests/core/workflow/graph_engine/entities/test_graph_runtime_state.py index 8e9fcf196b..cf7cee8710 100644 --- a/api/tests/unit_tests/core/workflow/graph_engine/entities/test_graph_runtime_state.py +++ b/api/tests/unit_tests/core/workflow/graph_engine/entities/test_graph_runtime_state.py @@ -30,11 +30,11 @@ def create_test_graph_runtime_state() -> GraphRuntimeState: llm_usage = LLMUsage( prompt_tokens=150, prompt_unit_price=Decimal("0.001"), - prompt_price_unit=Decimal("1000"), + prompt_price_unit=Decimal(1000), prompt_price=Decimal("0.15"), completion_tokens=75, completion_unit_price=Decimal("0.002"), - completion_price_unit=Decimal("1000"), + completion_price_unit=Decimal(1000), completion_price=Decimal("0.15"), total_tokens=225, total_price=Decimal("0.30"), @@ -121,11 +121,11 @@ def test_llm_usage_round_trip(): llm_usage = LLMUsage( prompt_tokens=100, prompt_unit_price=Decimal("0.0015"), - prompt_price_unit=Decimal("1000"), + prompt_price_unit=Decimal(1000), prompt_price=Decimal("0.15"), completion_tokens=50, completion_unit_price=Decimal("0.003"), - completion_price_unit=Decimal("1000"), + completion_price_unit=Decimal(1000), completion_price=Decimal("0.15"), total_tokens=150, total_price=Decimal("0.30"),