licos-dev-sdk 0.6.4__tar.gz → 0.6.6__tar.gz

{licos_dev_sdk-0.6.4 → licos_dev_sdk-0.6.6}/PKG-INFO

@@ -1,13 +1,13 @@
 Metadata-Version: 2.4
 Name: licos-dev-sdk
-Version: 0.6.4
+Version: 0.6.6
 Summary: LICOS Dev SDK - file generation and model capability clients
 Requires-Python: >=3.10
 Requires-Dist: beautifulsoup4>=4.12
 Requires-Dist: docxtpl>=0.16
 Requires-Dist: graphviz>=0.20
 Requires-Dist: jinja2>=3.1
-Requires-Dist: licos-platform-sdk>=0.7.4
+Requires-Dist: licos-platform-sdk>=0.7.6
 Requires-Dist: matplotlib>=3.9
 Requires-Dist: mistune>=3.0
 Requires-Dist: openpyxl>=3.1

{licos_dev_sdk-0.6.4 → licos_dev_sdk-0.6.6}/pyproject.toml

@@ -4,12 +4,12 @@ build-backend = "hatchling.build"
 
 [project]
 name = "licos-dev-sdk"
-version = "0.6.4"
+version = "0.6.6"
 description = "LICOS Dev SDK - file generation and model capability clients"
 readme = "README.md"
 requires-python = ">=3.10"
 dependencies = [
-    "licos-platform-sdk>=0.7.4",
+    "licos-platform-sdk>=0.7.6",
     "weasyprint>=62.0",
     "python-docx>=1.1",
     "docxtpl>=0.16",

{licos_dev_sdk-0.6.4 → licos_dev_sdk-0.6.6}/src/licos_dev_sdk/model.py

@@ -1229,6 +1229,7 @@ def _model_runtime(
     token = runtime_user_token or resolve_user_token(
         resolved_base_url,
         owner_user_id,
+        workspace_id=resolved_workspace_id,
     )
     owner_project_id = env("LICOS_PROJECT_ID") or env("AGENT_PROJECT_ID")
     owner_project_name = env("LICOS_OWNER_PROJECT_NAME") or env("AGENT_OWNER_PROJECT_NAME")
@@ -1274,7 +1275,7 @@ def _model_runtime(
     except ApiError as error:
         if explicit_user_token or not should_refresh_user_token(error):
             raise
-        token = resolve_user_token(resolved_base_url, owner_user_id, force_refresh=True)
+        token = resolve_user_token(resolved_base_url, owner_user_id, force_refresh=True, workspace_id=resolved_workspace_id)
         current_user_id, current_user_name, current_org_id, current_org_name = _fetch_current_user_context(
             resolved_base_url, token,
         )
@@ -1332,6 +1333,7 @@ def _refresh_model_runtime(runtime: ModelRuntime) -> ModelRuntime:
         runtime.base_url,
         runtime.user_id,
         force_refresh=True,
+        workspace_id=runtime.workspace_id,
     )
     current_user_id, current_user_name, current_org_id, current_org_name = _fetch_current_user_context(
         runtime.base_url,
@@ -1422,6 +1424,7 @@ def _credit_subject(
             runtime.base_url,
             owner_user_id,
             force_refresh=force_refresh,
+            workspace_id=runtime.workspace_id,
         )
     return subject_user_id, subject_org_id, token
 

{licos_dev_sdk-0.6.4 → licos_dev_sdk-0.6.6}/src/licos_dev_sdk/observability.py

@@ -31,6 +31,7 @@ class ObservabilityRuntime:
     base_url: str
     project_id: str
     token: str
+    explicit_user_token: bool = False
     workspace_id: str | None = None
     user_id: str | None = None
     project_type: str | None = None
@@ -289,7 +290,6 @@ def _observability_runtime(
 ) -> ObservabilityRuntime:
     resolved_base_url = normalize_base_url(base_url) if base_url else platform_base_url()
     owner_user_id = user_id or env("LICOS_USER_ID") or env("AGENT_USER_ID")
-    token = (user_token or "").strip() or resolve_user_token(resolved_base_url, owner_user_id)
     resolved_project_id = project_id or env("LICOS_PROJECT_ID") or env("AGENT_PROJECT_ID")
     if not resolved_project_id:
         raise ConfigurationError("LICOS_PROJECT_ID or AGENT_PROJECT_ID is not configured")
@@ -300,6 +300,7 @@ def _observability_runtime(
         or env("LICOS_OWNER_WORKSPACE_ID")
         or env("AGENT_WORKSPACE_ID")
     )
+    token = (user_token or "").strip() or resolve_user_token(resolved_base_url, owner_user_id, workspace_id=resolved_workspace_id)
     resolved_project_type = (
         project_type or env("LICOS_PROJECT_TYPE") or env("AGENT_PROJECT_TYPE")
     )
@@ -325,7 +326,7 @@ def _observability_runtime(
 
 
 def _refresh_runtime(runtime: ObservabilityRuntime) -> ObservabilityRuntime:
-    token = resolve_user_token(runtime.base_url, runtime.user_id, force_refresh=True)
+    token = resolve_user_token(runtime.base_url, runtime.user_id, force_refresh=True, workspace_id=runtime.workspace_id)
     return replace(runtime, token=token)
 
 

{licos_dev_sdk-0.6.4 → licos_dev_sdk-0.6.6}/tests/test_model.py

@@ -521,7 +521,7 @@ class ModelSdkTests(unittest.TestCase):
 
         self.assertEqual(result.text, "hello")
         self.assertEqual(captured["exchange_headers"]["Authorization"], "Bearer ai-agent-token")
-        self.assertEqual(captured["exchange_body"], {"userId": "user-1"})
+        self.assertEqual(captured["exchange_body"], {"userId": "user-1", "workspaceId": "workspace-1"})
         self.assertEqual(captured["catalog_headers"]["Authorization"], "Bearer user-token")
         self.assertEqual(captured["detail_headers"]["Authorization"], "Bearer user-token")
         self.assertEqual(captured["chat_headers"]["Authorization"], "Bearer user-token")

{licos_dev_sdk-0.6.4 → licos_dev_sdk-0.6.6}/tests/test_observability.py

@@ -1,150 +1,150 @@
-from __future__ import annotations
-
-import json
-import os
-import sys
-import unittest
-from pathlib import Path
-from typing import Any
-from unittest import mock
-from urllib import error as urlerror
-
-sys.path.insert(0, str(Path(__file__).resolve().parents[1] / "src"))
-sys.path.insert(0, str(Path(__file__).resolve().parents[2] / "licos-platform-sdk" / "src"))
-
-from licos_dev_sdk import observability
-from licos_platform_sdk import _runtime
-
-
-class _FakeResponse:
-    status = 200
-
-    def __init__(self, payload: dict[str, Any]) -> None:
-        self._payload = payload
-
-    def __enter__(self) -> "_FakeResponse":
-        return self
-
-    def __exit__(self, *_args: Any) -> None:
-        return None
-
-    def read(self) -> bytes:
-        return json.dumps(self._payload).encode("utf-8")
-
-
-class _FakeErrorBody:
-    def __init__(self, payload: dict[str, Any]) -> None:
-        self._payload = payload
-
-    def read(self) -> bytes:
-        return json.dumps(self._payload).encode("utf-8")
-
-
-class ObservabilitySdkTests(unittest.TestCase):
-    def setUp(self) -> None:
-        self.env = mock.patch.dict(
-            os.environ,
-            {
-                "LICOS_PLATFORM_API_BASE_URL": "http://platform.example/api/v1",
-                "AGENT_USER_ID": "user-1",
-                "AGENT_WORKSPACE_ID": "workspace-1",
-                "AGENT_PROJECT_ID": "project-1",
-                "AGENT_PROJECT_TYPE": "AGENT",
-                "LICOS_AI_AGENT_TOKEN": "ai-agent-token",
-                "LICOS_DEPLOYMENT_VERSION": "v1",
-                "LICOS_DEPLOYMENT_HASH": "hash-1",
-            },
-            clear=True,
-        )
-        self.env.start()
-        self.addCleanup(self.env.stop)
-        _runtime._clear_token_cache_for_tests()
-
-    def test_log_uses_runtime_context_and_owner_token(self) -> None:
-        captured: dict[str, Any] = {}
-
-        def fake_urlopen(req: Any, timeout: int = 0) -> _FakeResponse:
-            if req.full_url == "http://platform.example/api/v1/internal/auth/ai-user-token":
-                captured["exchange_headers"] = dict(req.header_items())
-                captured["exchange_body"] = json.loads(req.data.decode("utf-8"))
-                return _FakeResponse({"code": 0, "success": True, "data": {"accessToken": "user-token"}})
-            if req.full_url == "http://platform.example/api/v1/studio/observability/logs?projectId=project-1":
-                captured["log_headers"] = dict(req.header_items())
-                captured["log_body"] = json.loads(req.data.decode("utf-8"))
-                return _FakeResponse({"code": 0, "success": True, "data": {"id": 1}})
-            raise AssertionError(req.full_url)
-
-        with mock.patch.object(observability.request, "urlopen", fake_urlopen):
-            result = observability.log_info("started", log_time="2026-05-22T10:00:00+08:00")
-
-        self.assertEqual(result, {"id": 1})
-        self.assertEqual(captured["exchange_headers"]["Authorization"], "Bearer ai-agent-token")
-        self.assertEqual(captured["exchange_body"], {"userId": "user-1"})
-        self.assertEqual(captured["log_headers"]["Authorization"], "Bearer user-token")
-        log_headers = {key.lower(): value for key, value in captured["log_headers"].items()}
-        self.assertEqual(log_headers["x-workspace-id"], "workspace-1")
-        self.assertEqual(captured["log_body"]["deploymentVersion"], "v1")
-        self.assertEqual(captured["log_body"]["level"], "info")
-        self.assertEqual(captured["log_body"]["logContent"], "started")
-
-    def test_trace_metric_and_error_payloads(self) -> None:
-        calls: list[tuple[str, dict[str, Any]]] = []
-
-        def fake_urlopen(req: Any, timeout: int = 0) -> _FakeResponse:
-            if req.full_url == "http://platform.example/api/v1/internal/auth/ai-user-token":
-                return _FakeResponse({"code": 0, "success": True, "data": {"accessToken": "user-token"}})
-            calls.append((req.full_url, json.loads(req.data.decode("utf-8"))))
-            return _FakeResponse({"code": 0, "success": True, "data": {"ok": True}})
-
-        with mock.patch.object(observability.request, "urlopen", fake_urlopen):
-            client = observability.ObservabilityClient()
-            client.record_trace(trace_id="trace-1", input={"q": "hi"}, output="ok", input_tokens=3, output_tokens=4, latency_seconds=1.2)
-            client.record_metric(route="/stream_run", method="post", latency_ms=640, status_code=500)
-            client.record_error(route="/stream_run", method="post", error_code="MODEL_TIMEOUT", error_message="timeout", trace_id="trace-1")
-
-        self.assertEqual(calls[0][0], "http://platform.example/api/v1/studio/observability/traces?projectId=project-1")
-        self.assertEqual(calls[0][1]["traceId"], "trace-1")
-        self.assertEqual(calls[0][1]["hashVersion"], "hash-1")
-        self.assertEqual(calls[0][1]["input"], '{"q": "hi"}')
-        self.assertEqual(calls[0][1]["inputTokens"], 3)
-
-        self.assertEqual(calls[1][0], "http://platform.example/api/v1/studio/observability/analysis/metrics?projectId=project-1")
-        self.assertEqual(calls[1][1]["workspaceId"], "workspace-1")
-        self.assertEqual(calls[1][1]["projectType"], "AGENT")
-        self.assertEqual(calls[1][1]["method"], "POST")
-        self.assertEqual(calls[1][1]["errorCount"], 1)
-        self.assertEqual(calls[1][1]["status5xxCount"], 1)
-        self.assertEqual(calls[1][1]["latencyBuckets"], {"500+": 1})
-
-        self.assertEqual(calls[2][0], "http://platform.example/api/v1/studio/observability/analysis/errors?projectId=project-1")
-        self.assertEqual(calls[2][1]["errorCode"], "MODEL_TIMEOUT")
-        self.assertEqual(calls[2][1]["traceId"], "trace-1")
-
-    def test_refreshes_owner_token_once_after_unauthorized(self) -> None:
-        tokens = iter(["old-token", "new-token"])
-        log_tokens: list[str] = []
-
-        def fake_urlopen(req: Any, timeout: int = 0) -> _FakeResponse:
-            if req.full_url == "http://platform.example/api/v1/internal/auth/ai-user-token":
-                return _FakeResponse({"code": 0, "success": True, "data": {"accessToken": next(tokens)}})
-            if req.full_url == "http://platform.example/api/v1/studio/observability/logs?projectId=project-1":
-                log_tokens.append(dict(req.header_items())["Authorization"])
-                if len(log_tokens) == 1:
-                    raise urlerror.HTTPError(
-                        req.full_url,
-                        401,
-                        "Unauthorized",
-                        hdrs=None,
-                        fp=_FakeErrorBody({"code": 10002, "message": "token invalid or expired", "success": False}),
-                    )
-                return _FakeResponse({"code": 0, "success": True, "data": {"id": 1}})
-            raise AssertionError(req.full_url)
-
-        with mock.patch.object(observability.request, "urlopen", fake_urlopen):
-            observability.log_error("failed")
-
-        self.assertEqual(log_tokens, ["Bearer old-token", "Bearer new-token"])
-
-
-if __name__ == "__main__":
-    unittest.main()
+from __future__ import annotations
+
+import json
+import os
+import sys
+import unittest
+from pathlib import Path
+from typing import Any
+from unittest import mock
+from urllib import error as urlerror
+
+sys.path.insert(0, str(Path(__file__).resolve().parents[1] / "src"))
+sys.path.insert(0, str(Path(__file__).resolve().parents[2] / "licos-platform-sdk" / "src"))
+
+from licos_dev_sdk import observability
+from licos_platform_sdk import _runtime
+
+
+class _FakeResponse:
+    status = 200
+
+    def __init__(self, payload: dict[str, Any]) -> None:
+        self._payload = payload
+
+    def __enter__(self) -> "_FakeResponse":
+        return self
+
+    def __exit__(self, *_args: Any) -> None:
+        return None
+
+    def read(self) -> bytes:
+        return json.dumps(self._payload).encode("utf-8")
+
+
+class _FakeErrorBody:
+    def __init__(self, payload: dict[str, Any]) -> None:
+        self._payload = payload
+
+    def read(self) -> bytes:
+        return json.dumps(self._payload).encode("utf-8")
+
+
+class ObservabilitySdkTests(unittest.TestCase):
+    def setUp(self) -> None:
+        self.env = mock.patch.dict(
+            os.environ,
+            {
+                "LICOS_PLATFORM_API_BASE_URL": "http://platform.example/api/v1",
+                "AGENT_USER_ID": "user-1",
+                "AGENT_WORKSPACE_ID": "workspace-1",
+                "AGENT_PROJECT_ID": "project-1",
+                "AGENT_PROJECT_TYPE": "AGENT",
+                "LICOS_AI_AGENT_TOKEN": "ai-agent-token",
+                "LICOS_DEPLOYMENT_VERSION": "v1",
+                "LICOS_DEPLOYMENT_HASH": "hash-1",
+            },
+            clear=True,
+        )
+        self.env.start()
+        self.addCleanup(self.env.stop)
+        _runtime._clear_token_cache_for_tests()
+
+    def test_log_uses_runtime_context_and_owner_token(self) -> None:
+        captured: dict[str, Any] = {}
+
+        def fake_urlopen(req: Any, timeout: int = 0) -> _FakeResponse:
+            if req.full_url == "http://platform.example/api/v1/internal/auth/ai-user-token":
+                captured["exchange_headers"] = dict(req.header_items())
+                captured["exchange_body"] = json.loads(req.data.decode("utf-8"))
+                return _FakeResponse({"code": 0, "success": True, "data": {"accessToken": "user-token"}})
+            if req.full_url == "http://platform.example/api/v1/studio/observability/logs?projectId=project-1":
+                captured["log_headers"] = dict(req.header_items())
+                captured["log_body"] = json.loads(req.data.decode("utf-8"))
+                return _FakeResponse({"code": 0, "success": True, "data": {"id": 1}})
+            raise AssertionError(req.full_url)
+
+        with mock.patch.object(observability.request, "urlopen", fake_urlopen):
+            result = observability.log_info("started", log_time="2026-05-22T10:00:00+08:00")
+
+        self.assertEqual(result, {"id": 1})
+        self.assertEqual(captured["exchange_headers"]["Authorization"], "Bearer ai-agent-token")
+        self.assertEqual(captured["exchange_body"], {"userId": "user-1", "workspaceId": "workspace-1"})
+        self.assertEqual(captured["log_headers"]["Authorization"], "Bearer user-token")
+        log_headers = {key.lower(): value for key, value in captured["log_headers"].items()}
+        self.assertEqual(log_headers["x-workspace-id"], "workspace-1")
+        self.assertEqual(captured["log_body"]["deploymentVersion"], "v1")
+        self.assertEqual(captured["log_body"]["level"], "info")
+        self.assertEqual(captured["log_body"]["logContent"], "started")
+
+    def test_trace_metric_and_error_payloads(self) -> None:
+        calls: list[tuple[str, dict[str, Any]]] = []
+
+        def fake_urlopen(req: Any, timeout: int = 0) -> _FakeResponse:
+            if req.full_url == "http://platform.example/api/v1/internal/auth/ai-user-token":
+                return _FakeResponse({"code": 0, "success": True, "data": {"accessToken": "user-token"}})
+            calls.append((req.full_url, json.loads(req.data.decode("utf-8"))))
+            return _FakeResponse({"code": 0, "success": True, "data": {"ok": True}})
+
+        with mock.patch.object(observability.request, "urlopen", fake_urlopen):
+            client = observability.ObservabilityClient()
+            client.record_trace(trace_id="trace-1", input={"q": "hi"}, output="ok", input_tokens=3, output_tokens=4, latency_seconds=1.2)
+            client.record_metric(route="/stream_run", method="post", latency_ms=640, status_code=500)
+            client.record_error(route="/stream_run", method="post", error_code="MODEL_TIMEOUT", error_message="timeout", trace_id="trace-1")
+
+        self.assertEqual(calls[0][0], "http://platform.example/api/v1/studio/observability/traces?projectId=project-1")
+        self.assertEqual(calls[0][1]["traceId"], "trace-1")
+        self.assertEqual(calls[0][1]["hashVersion"], "hash-1")
+        self.assertEqual(calls[0][1]["input"], '{"q": "hi"}')
+        self.assertEqual(calls[0][1]["inputTokens"], 3)
+
+        self.assertEqual(calls[1][0], "http://platform.example/api/v1/studio/observability/analysis/metrics?projectId=project-1")
+        self.assertEqual(calls[1][1]["workspaceId"], "workspace-1")
+        self.assertEqual(calls[1][1]["projectType"], "AGENT")
+        self.assertEqual(calls[1][1]["method"], "POST")
+        self.assertEqual(calls[1][1]["errorCount"], 1)
+        self.assertEqual(calls[1][1]["status5xxCount"], 1)
+        self.assertEqual(calls[1][1]["latencyBuckets"], {"500+": 1})
+
+        self.assertEqual(calls[2][0], "http://platform.example/api/v1/studio/observability/analysis/errors?projectId=project-1")
+        self.assertEqual(calls[2][1]["errorCode"], "MODEL_TIMEOUT")
+        self.assertEqual(calls[2][1]["traceId"], "trace-1")
+
+    def test_refreshes_owner_token_once_after_unauthorized(self) -> None:
+        tokens = iter(["old-token", "new-token"])
+        log_tokens: list[str] = []
+
+        def fake_urlopen(req: Any, timeout: int = 0) -> _FakeResponse:
+            if req.full_url == "http://platform.example/api/v1/internal/auth/ai-user-token":
+                return _FakeResponse({"code": 0, "success": True, "data": {"accessToken": next(tokens)}})
+            if req.full_url == "http://platform.example/api/v1/studio/observability/logs?projectId=project-1":
+                log_tokens.append(dict(req.header_items())["Authorization"])
+                if len(log_tokens) == 1:
+                    raise urlerror.HTTPError(
+                        req.full_url,
+                        401,
+                        "Unauthorized",
+                        hdrs=None,
+                        fp=_FakeErrorBody({"code": 10002, "message": "token invalid or expired", "success": False}),
+                    )
+                return _FakeResponse({"code": 0, "success": True, "data": {"id": 1}})
+            raise AssertionError(req.full_url)
+
+        with mock.patch.object(observability.request, "urlopen", fake_urlopen):
+            observability.log_error("failed")
+
+        self.assertEqual(log_tokens, ["Bearer old-token", "Bearer new-token"])
+
+
+if __name__ == "__main__":
+    unittest.main()

{licos_dev_sdk-0.6.4 → licos_dev_sdk-0.6.6}/tests/test_request_identity.py

@@ -17,8 +17,8 @@ class ModelIdentityTests(TestCase):
         self.assertEqual(runtime.token, "renewed")
         self.assertEqual(runtime.current_user_id, "owner")
         self.assertEqual(exchange.call_args_list, [
-            mock.call("https://platform.example", "owner"),
-            mock.call("https://platform.example", "owner", force_refresh=True),
+            mock.call("https://platform.example", "owner", workspace_id=None),
+            mock.call("https://platform.example", "owner", force_refresh=True, workspace_id=None),
         ])
 
     def test_business_authentication_error_is_not_a_user_profile(self):