chore(api): format code

10 months ago · 8216ebacd3
parent dfeeddd88b
commit 8216ebacd3
1 changed files with 4 additions and 4 deletions
--- 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"),