licos-dev-sdk 0.4.7__tar.gz → 0.4.8__tar.gz

{licos_dev_sdk-0.4.7 → licos_dev_sdk-0.4.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: licos-dev-sdk
-Version: 0.4.7
+Version: 0.4.8
 Summary: LICOS Dev SDK - file generation and model capability clients
 Requires-Python: >=3.10
 Requires-Dist: beautifulsoup4>=4.12

{licos_dev_sdk-0.4.7 → licos_dev_sdk-0.4.8}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "licos-dev-sdk"
-version = "0.4.7"
+version = "0.4.8"
 description = "LICOS Dev SDK - file generation and model capability clients"
 requires-python = ">=3.10"
 dependencies = [

{licos_dev_sdk-0.4.7 → licos_dev_sdk-0.4.8}/src/licos_dev_sdk/_usage_identity.py

@@ -14,6 +14,7 @@ IDENTITY_HEADERS = (
     "X-AIOS-Usage-Context-Signature",
     "X-AIOS-Provider-Credential-Id",
     "X-AIOS-Provider-Workspace-Id",
+    "X-Licos-Runtime-Env",
 )
 REFRESH_LEAD_SECS = 30
 

{licos_dev_sdk-0.4.7 → licos_dev_sdk-0.4.8}/tests/test_usage_identity.py

@@ -26,6 +26,7 @@ def detail(context: str, expires_at: float) -> dict:
             IDENTITY_HEADERS[1]: f"{context}-signature",
             IDENTITY_HEADERS[2]: f"{context}-credential",
             IDENTITY_HEADERS[3]: f"{context}-workspace",
+            IDENTITY_HEADERS[4]: "prod",
             "X-Provider-Unrelated": "must-not-merge",
             "X-Licos-Owner-User-Id": "must-not-replace",
         },
@@ -49,6 +50,7 @@ class UsageIdentityTests(unittest.TestCase):
         self.runtime = model.ModelRuntime(
             base_url="http://platform.example", token="user-token", workspace_id="workspace-1",
             user_id="owner-1", current_user_id="caller-1", owner_project_id="project-1",
+            runtime_env="deploy",
         )
 
     def test_valid_lease_reuses_signed_snapshot(self) -> None:
@@ -97,7 +99,7 @@ class UsageIdentityTests(unittest.TestCase):
                     model._post_model_json(selected, self.runtime, {"model": "qwen-test"})
                 request.assert_not_called()
 
-    def test_submit_replaces_only_four_headers_and_retains_owner_and_caller(self) -> None:
+    def test_submit_replaces_trusted_headers_and_retains_owner_and_caller(self) -> None:
         with mock.patch("time.time", return_value=100):
             selected = endpoint(120)
             with mock.patch.object(model, "_fetch_model_detail", return_value=detail("renewed", 200)) as fetch, \
@@ -107,6 +109,7 @@ class UsageIdentityTests(unittest.TestCase):
                 fetch.assert_called_once_with(self.runtime, "qwen-test", refresh=True)
                 headers = request.call_args.kwargs["headers"]
                 self.assertEqual(headers[IDENTITY_HEADERS[0]], "renewed")
+                self.assertEqual(headers[IDENTITY_HEADERS[4]], "prod")
                 self.assertNotIn(IDENTITY_HEADERS[0].lower(), headers)
                 self.assertEqual(headers["X-Licos-Owner-User-Id"], "owner-1")
                 self.assertEqual(headers["X-Licos-Current-User-Id"], "caller-1")