diff --git a/agentops/client.py b/agentops/client.py index 5c1cd4eb..339a2013 100644 --- a/agentops/client.py +++ b/agentops/client.py @@ -241,7 +241,7 @@ def end_session(self, self._session.video = video self._session.end_session(end_state, end_state_reason) token_cost = self._worker.end_session(self._session) - if token_cost is 'unknown': + if token_cost == 'unknown': print('🖇 AgentOps: Could not determine cost of run.') else: print('🖇 AgentOps: This run cost ${:.6f}'.format(float(token_cost))) diff --git a/pyproject.toml b/pyproject.toml index 5b7a34b6..5de5b79b 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta" [project] name = "agentops" -version = "0.1.3" +version = "0.1.4" authors = [ { name="Alex Reibman", email="areibman@gmail.com" }, { name="Shawn Qiu", email="siyangqiu@gmail.com" }, @@ -22,7 +22,7 @@ classifiers = [ dependencies = [ "requests==2.31.0", "psutil==5.9.8", - "packaging==24.0" + "packaging==23.2" ] [project.optional-dependencies] dev = [