restore api_url in tracer_factory

Author: rcholic

.gitignore

@@ -40,6 +40,9 @@ ENV/
 htmlcov/
 .tox/
 
+# Traces (runtime and test-generated)
+traces/
+
 # Jupyter
 .ipynb_checkpoints
 

sentience/tracer_factory.py

@@ -21,6 +21,7 @@
 def create_tracer(
     api_key: str | None = None,
     run_id: str | None = None,
+    api_url: str | None = None,
 ) -> Tracer:
     """
     Create tracer with automatic tier detection.
@@ -34,6 +35,7 @@ def create_tracer(
                  - Free tier: None or empty
                  - Pro/Enterprise: Valid API key
         run_id: Unique identifier for this agent run. If not provided, generates UUID.
+        api_url: Sentience API base URL (default: https://api.sentienceapi.com)
 
     Returns:
         Tracer configured with appropriate sink
@@ -55,16 +57,19 @@ def create_tracer(
     if run_id is None:
         run_id = str(uuid.uuid4())
 
+    if api_url is None:
+        api_url = SENTIENCE_API_URL
+
     # 0. Check for orphaned traces from previous crashes (if api_key provided)
     if api_key:
-        _recover_orphaned_traces(api_key, SENTIENCE_API_URL)
+        _recover_orphaned_traces(api_key, api_url)
 
     # 1. Try to initialize Cloud Sink (Pro/Enterprise tier)
     if api_key:
         try:
             # Request pre-signed upload URL from backend
             response = requests.post(
-                f"{SENTIENCE_API_URL}/v1/traces/init",
+                f"{api_url}/v1/traces/init",
                 headers={"Authorization": f"Bearer {api_key}"},
                 json={"run_id": run_id},
                 timeout=10,

tests/test_cloud_tracing.py

@@ -381,6 +381,30 @@ def test_create_tracer_uses_constant_api_url(self):
 
                 tracer.close()
 
+    def test_create_tracer_custom_api_url(self):
+        """Test create_tracer accepts custom api_url parameter."""
+        custom_api_url = "https://custom.api.example.com"
+
+        with patch("sentience.tracer_factory.requests.post") as mock_post:
+            with patch("sentience.cloud_tracing.requests.put") as mock_put:
+                # Mock API response
+                mock_response = Mock()
+                mock_response.status_code = 200
+                mock_response.json.return_value = {"upload_url": "https://storage.com/upload"}
+                mock_post.return_value = mock_response
+                mock_put.return_value = Mock(status_code=200)
+
+                tracer = create_tracer(
+                    api_key="sk_test123", run_id="test-run", api_url=custom_api_url
+                )
+
+                # Verify custom API URL was used
+                assert mock_post.called
+                call_args = mock_post.call_args
+                assert call_args[0][0] == f"{custom_api_url}/v1/traces/init"
+
+                tracer.close()
+
     def test_create_tracer_missing_upload_url_in_response(self, capsys):
         """Test create_tracer handles missing upload_url gracefully."""
         with patch("sentience.tracer_factory.requests.post") as mock_post: