handle unknown cost (#148)

AgentOps-AI · Apr 9, 2024 · 3d1b80c · 3d1b80c
1 parent 5c11613
commit 3d1b80c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/agentops/client.py b/agentops/client.py
@@ -240,8 +240,11 @@ def end_session(self,
 
         self._session.video = video
         self._session.end_session(end_state, end_state_reason)
-        token_cost = float(self._worker.end_session(self._session))
-        print('🖇 AgentOps: This run cost ${:.6f}'.format(token_cost))
+        token_cost = self._worker.end_session(self._session)
+        if token_cost is 'unknown':
+            print('🖇 AgentOps: Could not determine cost of run.')
+        else:
+            print('🖇 AgentOps: This run cost ${:.6f}'.format(float(token_cost)))
         self._session = None
         self._worker = None
 

diff --git a/pyproject.toml b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "agentops"
-version = "0.1.2"
+version = "0.1.3"
 authors = [
   { name="Alex Reibman", email="areibman@gmail.com" },
   { name="Shawn Qiu", email="siyangqiu@gmail.com" },