Add type hint for return

AgentOps-AI · Jun 26, 2024 · 2b745b6 · 2b745b6
1 parent 3c43888
commit 2b745b6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/agentops/client.py b/agentops/client.py
@@ -388,7 +388,7 @@ def end_session(
         end_state_reason: Optional[str] = None,
         video: Optional[str] = None,
         is_auto_end: Optional[bool] = None,
-    ):
+    ) -> Decimal:
         """
         End the current session with the AgentOps service.
 
@@ -419,6 +419,7 @@ def end_session(
 
         if token_cost is None or token_cost == "unknown":
             logger.info("Could not determine cost of run.")
+            token_cost_d = Decimal(0)
         else:
             token_cost_d = Decimal(token_cost)
             logger.info(