tensorcost 1.3.0__tar.gz → 1.3.2__tar.gz

{tensorcost-1.3.0 → tensorcost-1.3.2}/PKG-INFO

@@ -1,12 +1,12 @@
 Metadata-Version: 2.5
 Name: tensorcost
-Version: 1.3.0
+Version: 1.3.2
 Summary: One-line wrapper SDK for TensorCost — LLM observability and control-layer inference routing for OpenAI, Anthropic, Bedrock, and Vertex clients.
 Project-URL: Homepage, https://tensorcost.com
 Project-URL: Documentation, https://docs.tensorcost.com
-Project-URL: Source, https://github.com/tensorcost/agent/tree/main/sdks/python
-Project-URL: Changelog, https://github.com/tensorcost/agent/blob/main/sdks/python/CHANGELOG.md
-Project-URL: Bug Tracker, https://github.com/tensorcost/agent/issues
+Project-URL: Source, https://github.com/vaadhlabs/tensorcost-agent/tree/main/sdks/python
+Project-URL: Changelog, https://github.com/vaadhlabs/tensorcost-agent/blob/main/sdks/python/CHANGELOG.md
+Project-URL: Bug Tracker, https://github.com/vaadhlabs/tensorcost-agent/issues
 Author-email: Vaadh Labs / TensorCost <engineering@tensorcost.com>
 License: Apache-2.0
 License-File: LICENSE

{tensorcost-1.3.0 → tensorcost-1.3.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "tensorcost"
-version = "1.3.0"
+version = "1.3.2"
 description = "One-line wrapper SDK for TensorCost — LLM observability and control-layer inference routing for OpenAI, Anthropic, Bedrock, and Vertex clients."
 readme = "README.md"
 requires-python = ">=3.9"
@@ -39,9 +39,9 @@ dev = [
 [project.urls]
 Homepage = "https://tensorcost.com"
 Documentation = "https://docs.tensorcost.com"
-Source = "https://github.com/tensorcost/agent/tree/main/sdks/python"
-Changelog = "https://github.com/tensorcost/agent/blob/main/sdks/python/CHANGELOG.md"
-"Bug Tracker" = "https://github.com/tensorcost/agent/issues"
+Source = "https://github.com/vaadhlabs/tensorcost-agent/tree/main/sdks/python"
+Changelog = "https://github.com/vaadhlabs/tensorcost-agent/blob/main/sdks/python/CHANGELOG.md"
+"Bug Tracker" = "https://github.com/vaadhlabs/tensorcost-agent/issues"
 
 [tool.hatch.build.targets.wheel]
 packages = ["src/tensorcost"]

{tensorcost-1.3.0 → tensorcost-1.3.2}/src/tensorcost/_config.py

@@ -19,6 +19,7 @@ loudly so the problem surfaces in development, not in production.
 from __future__ import annotations
 
 import os
+import uuid
 from dataclasses import dataclass, field
 from typing import Callable, Optional
 
@@ -119,6 +120,7 @@ def resolve_config(
     feature: Optional[str] = None,
     agent_id: Optional[str] = None,
     workflow_id: Optional[str] = None,
+    auto_workflow_id: Optional[bool] = None,
     team_id: Optional[str] = None,
 ) -> ResolvedConfig:
     """Resolve final config values, env vars filling in any blanks."""
@@ -217,6 +219,11 @@ def resolve_config(
         workflow_id or os.environ.get("TENSORCOST_WORKFLOW_ID"),
         "workflow_id",
     )
+    auto_workflow_enabled = auto_workflow_id is True or (
+        os.environ.get("TENSORCOST_AUTO_WORKFLOW_ID", "").strip() == "1"
+    )
+    if final_workflow_id is None and final_agent_id and auto_workflow_enabled:
+        final_workflow_id = str(uuid.uuid4())
     raw_team = team_id or os.environ.get("TENSORCOST_TEAM_ID")
     final_team_id: Optional[str] = None
     if raw_team is not None:

{tensorcost-1.3.0 → tensorcost-1.3.2}/tests/test_agent_run.py

@@ -26,6 +26,25 @@ class TestAgentRunConfig:
         assert cfg.agent_id == "env-agent"
         assert cfg.workflow_id == "env-workflow"
 
+    def test_auto_workflow_id_off_by_default(self, monkeypatch):
+        monkeypatch.setenv("TENSORCOST_API_KEY", "k")
+        monkeypatch.delenv("TENSORCOST_WORKFLOW_ID", raising=False)
+        monkeypatch.delenv("TENSORCOST_AUTO_WORKFLOW_ID", raising=False)
+        cfg = resolve_config(agent_id="support-bot")
+        assert cfg.workflow_id is None
+
+    def test_auto_workflow_id_when_enabled(self, monkeypatch):
+        monkeypatch.setenv("TENSORCOST_API_KEY", "k")
+        cfg = resolve_config(agent_id="support-bot", auto_workflow_id=True)
+        assert cfg.workflow_id is not None
+        assert len(cfg.workflow_id) >= 32
+
+    def test_auto_workflow_id_env_flag(self, monkeypatch):
+        monkeypatch.setenv("TENSORCOST_API_KEY", "k")
+        monkeypatch.setenv("TENSORCOST_AUTO_WORKFLOW_ID", "1")
+        cfg = resolve_config(agent_id="support-bot")
+        assert cfg.workflow_id is not None
+
     def test_max_length_128(self, monkeypatch):
         monkeypatch.setenv("TENSORCOST_API_KEY", "k")
         with pytest.raises(MissingConfigError, match="agent_id"):