Update session.py (#441)

Format formatted_cost to be outside of the token_cost if else condition.
AgentOps-AI · Oct 14, 2024 · f5e8ec3 · f5e8ec3
1 parent 142e50b
commit f5e8ec3
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/agentops/session.py b/agentops/session.py
@@ -142,13 +142,14 @@ def format_duration(start_time, end_time):
             token_cost_d = Decimal(0)
         else:
             token_cost_d = Decimal(token_cost)
-            formatted_cost = (
-                "{:.2f}".format(token_cost_d)
-                if token_cost_d == 0
-                else "{:.6f}".format(
-                    token_cost_d.quantize(Decimal("0.000001"), rounding=ROUND_HALF_UP)
-                )
+
+        formatted_cost = (
+            "{:.2f}".format(token_cost_d)
+            if token_cost_d == 0
+            else "{:.6f}".format(
+                token_cost_d.quantize(Decimal("0.000001"), rounding=ROUND_HALF_UP)
             )
+        )
 
         analytics = (
             f"Session Stats - "