iii-sdk 0.14.0.dev1__tar.gz → 0.15.0.dev1__tar.gz

{iii_sdk-0.14.0.dev1 → iii_sdk-0.15.0.dev1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: iii-sdk
-Version: 0.14.0.dev1
+Version: 0.15.0.dev1
 Summary: III SDK for Python
 Project-URL: Homepage, https://github.com/iii-hq/iii
 Project-URL: Repository, https://github.com/iii-hq/iii
@@ -14,8 +14,8 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Requires-Python: >=3.10
+Requires-Dist: iii-observability==0.13.0.dev1
 Requires-Dist: opentelemetry-api>=1.25
-Requires-Dist: opentelemetry-sdk>=1.25
 Requires-Dist: pydantic>=2.0
 Requires-Dist: websockets>=12.0
 Provides-Extra: dev

{iii_sdk-0.14.0.dev1 → iii_sdk-0.15.0.dev1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "iii-sdk"
-version = "0.14.0.dev1"
+version = "0.15.0.dev1"
 description = "III SDK for Python"
 authors = [{ name = "III" }]
 license = { text = "Apache-2.0" }
@@ -23,13 +23,16 @@ dependencies = [
     "websockets>=12.0",
     "pydantic>=2.0",
     "opentelemetry-api>=1.25",
-    "opentelemetry-sdk>=1.25",
+    "iii-observability==0.13.0.dev1",
 ]
 
 [project.urls]
 Homepage = "https://github.com/iii-hq/iii"
 Repository = "https://github.com/iii-hq/iii"
 
+[tool.uv.sources]
+iii-observability = { path = "../observability", editable = true }
+
 [project.optional-dependencies]
 dev = [
     "pytest>=8.0",

{iii_sdk-0.14.0.dev1 → iii_sdk-0.15.0.dev1}/src/iii/__init__.py

@@ -1,16 +1,41 @@
 """III SDK for Python."""
 
-from .baggage_span_processor import DEFAULT_ALLOWLIST, BaggageSpanProcessor
+from iii_observability import (
+    DEFAULT_ALLOWLIST,
+    REDACTED_PLACEHOLDER,
+    BaggageSpanProcessor,
+    Logger,
+    OtelConfig,
+    ReconnectionConfig,
+    current_span_id,
+    current_span_is_recording,
+    current_trace_id,
+    execute_traced_request,
+    extract_baggage,
+    extract_traceparent,
+    flush_otel,
+    init_otel,
+    inject_baggage,
+    inject_traceparent,
+    record_span_event,
+    redact,
+    redact_and_truncate,
+    resolve_max_bytes_from_env,
+    set_current_span_attribute,
+    set_current_span_error,
+    shutdown_otel,
+    with_span,
+)
+
 from .channels import ChannelReader, ChannelWriter
 from .errors import IIIForbiddenError, IIIInvocationError, IIITimeoutError
 from .format_utils import extract_request_format, extract_response_format, python_type_to_format
 from .iii import TriggerAction, register_worker
-from .iii_constants import FunctionRef, InitOptions, ReconnectionConfig, TelemetryOptions
+from .iii_constants import FunctionRef, InitOptions, TelemetryOptions
 from .iii_types import (
     AuthInput,
     AuthResult,
     EnqueueResult,
-    FunctionInfo,
     HttpAuthConfig,
     HttpInvocationConfig,
     MessageType,
@@ -23,7 +48,6 @@ from .iii_types import (
     OnTriggerTypeRegistrationResult,
     RegisterFunctionFormat,
     RegisterFunctionMessage,
-    RegisterServiceInput,
     RegisterTriggerInput,
     RegisterTriggerMessage,
     RegisterTriggerTypeInput,
@@ -31,22 +55,7 @@ from .iii_types import (
     StreamChannelRef,
     TriggerActionEnqueue,
     TriggerActionVoid,
-    TriggerInfo,
     TriggerRequest,
-    TriggerTypeInfo,
-)
-from .logger import Logger
-from .payload import (
-    REDACTED_PLACEHOLDER,
-    redact,
-    redact_and_truncate,
-    resolve_max_bytes_from_env,
-)
-from .span_ops import (
-    current_span_is_recording,
-    record_span_event,
-    set_current_span_attribute,
-    set_current_span_error,
 )
 from .stream import (
     IStream,
@@ -57,7 +66,6 @@ from .stream import (
     StreamJoinLeaveTriggerConfig,
     StreamTriggerConfig,
 )
-from .telemetry_types import OtelConfig
 from .triggers import Trigger, TriggerConfig, TriggerHandler, TriggerTypeRef
 from .types import (
     ApiRequest,
@@ -76,13 +84,24 @@ __all__ = [
     "BaggageSpanProcessor",
     "DEFAULT_ALLOWLIST",
     "REDACTED_PLACEHOLDER",
+    "current_span_id",
     "current_span_is_recording",
+    "current_trace_id",
+    "execute_traced_request",
+    "extract_baggage",
+    "extract_traceparent",
+    "flush_otel",
+    "init_otel",
+    "inject_baggage",
+    "inject_traceparent",
     "record_span_event",
-    "set_current_span_attribute",
-    "set_current_span_error",
     "redact",
     "redact_and_truncate",
     "resolve_max_bytes_from_env",
+    "set_current_span_attribute",
+    "set_current_span_error",
+    "shutdown_otel",
+    "with_span",
     # Channels
     "ChannelReader",
     "ChannelWriter",
@@ -110,13 +129,11 @@ __all__ = [
     "OnTriggerTypeRegistrationResult",
     # Message types
     "EnqueueResult",
-    "FunctionInfo",
     "HttpAuthConfig",
     "HttpInvocationConfig",
     "MessageType",
     "RegisterFunctionFormat",
     "RegisterFunctionMessage",
-    "RegisterServiceInput",
     "RegisterTriggerInput",
     "RegisterTriggerMessage",
     "RegisterTriggerTypeInput",
@@ -124,9 +141,7 @@ __all__ = [
     "StreamChannelRef",
     "TriggerActionEnqueue",
     "TriggerActionVoid",
-    "TriggerInfo",
     "TriggerRequest",
-    "TriggerTypeInfo",
     # Logger
     "Logger",
     # Triggers

{iii_sdk-0.14.0.dev1 → iii_sdk-0.15.0.dev1}/src/iii/iii.py

@@ -14,6 +14,7 @@ from importlib.metadata import version
 from typing import Any, Awaitable, Callable, Coroutine, TypeVar, cast
 
 import websockets
+from iii_observability import OtelConfig
 from websockets.asyncio.client import ClientConnection
 
 from .channels import ChannelReader, ChannelWriter
@@ -34,8 +35,6 @@ from .iii_types import (
     RegisterFunctionFormat,
     RegisterFunctionInput,
     RegisterFunctionMessage,
-    RegisterServiceInput,
-    RegisterServiceMessage,
     RegisterTriggerInput,
     RegisterTriggerMessage,
     RegisterTriggerTypeInput,
@@ -56,7 +55,6 @@ from .stream import (
     StreamListInput,
     StreamSetInput,
 )
-from .telemetry_types import OtelConfig
 from .triggers import Trigger, TriggerConfig, TriggerHandler, TriggerTypeRef
 from .types import Channel, RemoteFunctionData, RemoteTriggerTypeData, is_channel_ref
 
@@ -152,7 +150,6 @@ class III:
         self._options = options or InitOptions()
         self._ws: ClientConnection | None = None
         self._functions: dict[str, RemoteFunctionData] = {}
-        self._services: dict[str, RegisterServiceMessage] = {}
         self._pending: dict[str, _PendingInvocation] = {}
         self._triggers: dict[str, RegisterTriggerMessage] = {}
         self._trigger_types: dict[str, RemoteTriggerTypeData] = {}
@@ -227,7 +224,7 @@ class III:
         from an async context.
         """
         self._running = True
-        from .telemetry import attach_event_loop, init_otel
+        from iii_observability.telemetry import attach_event_loop, init_otel
 
         loop = asyncio.get_running_loop()
         otel_cfg: OtelConfig | None = None
@@ -283,7 +280,7 @@ class III:
 
         self._set_connection_state("disconnected")
 
-        from .telemetry import shutdown_otel_async
+        from iii_observability.telemetry import shutdown_otel_async
 
         await shutdown_otel_async()
 
@@ -344,8 +341,6 @@ class III:
         # Re-register all (snapshot to avoid mutation from caller thread)
         for trigger_type_data in list(self._trigger_types.values()):
             await self._send(trigger_type_data.message)
-        for svc in list(self._services.values()):
-            await self._send(svc)
         for function_data in list(self._functions.values()):
             await self._send(function_data.message)
         for trigger in list(self._triggers.values()):
@@ -443,6 +438,8 @@ class III:
             )
         elif msg_type == MessageType.REGISTER_TRIGGER.value:
             asyncio.create_task(self._handle_trigger_registration(data))
+        elif msg_type == MessageType.TRIGGER_REGISTRATION_RESULT.value:
+            self._handle_trigger_registration_result(data)
         elif msg_type == MessageType.WORKER_REGISTERED.value:
             worker_id = data.get("worker_id", "")
             self._worker_id = worker_id
@@ -502,7 +499,7 @@ class III:
         tracer = trace.get_tracer("iii-python-sdk")
         import os
 
-        from .payload import redact_and_truncate, resolve_max_bytes_from_env
+        from iii_observability import redact_and_truncate, resolve_max_bytes_from_env
 
         trace_payloads = os.environ.get("III_DISABLE_TRACE_PAYLOADS", "").lower() not in (
             "1",
@@ -706,6 +703,21 @@ class III:
                 }
             )
 
+    def _handle_trigger_registration_result(self, data: dict[str, Any]) -> None:
+        error = data.get("error")
+        if not error:
+            return
+
+        trigger_id = data.get("id", "")
+        trigger_type = data.get("trigger_type", "")
+        message = error.get("message", "")
+        log.error(
+            "[iii] Trigger registration failed for %r (%s): %s",
+            trigger_id,
+            trigger_type,
+            message,
+        )
+
     # Connection state management
 
     def _set_connection_state(self, state: IIIConnectionState) -> None:
@@ -1014,40 +1026,6 @@ class III:
 
         return FunctionRef(id=func_id, unregister=unregister)
 
-    def register_service(self, service: RegisterServiceInput | dict[str, Any]) -> None:
-        """Register a logical service grouping with the engine.
-
-        Services provide an organisational hierarchy for functions.  A
-        service can optionally reference a ``parent_service_id`` to form
-        a tree visible in the engine dashboard.
-
-        Note:
-            Services are organizational groupings visible in the engine
-            dashboard.  They do not affect function invocation behavior.
-
-        Args:
-            service: A ``RegisterServiceInput`` or dict with ``id`` and
-                optional ``name``, ``description``, ``parent_service_id``.
-
-        Examples:
-            >>> iii.register_service({"id": "payments", "description": "Payment processing"})
-            >>> iii.register_service({
-            ...     "id": "payments::refunds",
-            ...     "description": "Refund sub-service",
-            ...     "parent_service_id": "payments",
-            ... })
-        """
-        if isinstance(service, dict):
-            service = RegisterServiceInput(**service)
-        msg = RegisterServiceMessage(
-            id=service.id,
-            name=service.name or service.id,
-            description=service.description,
-            parent_service_id=service.parent_service_id,
-        )
-        self._services[service.id] = msg
-        self._send_if_connected(msg)
-
     def trigger(self, request: "dict[str, Any] | TriggerRequest") -> Any:
         """Invoke a remote function.
 

{iii_sdk-0.14.0.dev1 → iii_sdk-0.15.0.dev1}/src/iii/iii_constants.py

@@ -3,7 +3,7 @@
 from dataclasses import dataclass
 from typing import Any, Callable, Literal
 
-from .telemetry_types import OtelConfig
+from iii_observability import OtelConfig, ReconnectionConfig
 
 IIIConnectionState = Literal["disconnected", "connecting", "connected", "reconnecting", "failed"]
 
@@ -13,25 +13,6 @@ DEFAULT_INVOCATION_TIMEOUT_MS = 30000
 MAX_QUEUE_SIZE = 1000
 
 
-@dataclass
-class ReconnectionConfig:
-    """Configuration for WebSocket reconnection behavior.
-
-    Attributes:
-        initial_delay_ms: Starting delay in milliseconds. Default ``1000``.
-        max_delay_ms: Maximum delay cap in milliseconds. Default ``30000``.
-        backoff_multiplier: Exponential backoff multiplier. Default ``2.0``.
-        jitter_factor: Random jitter factor (0--1). Default ``0.3``.
-        max_retries: Maximum retry attempts. ``-1`` for infinite. Default ``-1``.
-    """
-
-    initial_delay_ms: int = 1000
-    max_delay_ms: int = 30000
-    backoff_multiplier: float = 2.0
-    jitter_factor: float = 0.3
-    max_retries: int = -1
-
-
 DEFAULT_RECONNECTION_CONFIG = ReconnectionConfig()
 
 

{iii_sdk-0.14.0.dev1 → iii_sdk-0.15.0.dev1}/src/iii/iii_types.py

@@ -148,25 +148,6 @@ class RegisterTriggerInput(BaseModel):
     )
 
 
-class RegisterServiceInput(BaseModel):
-    """Input for registering a service (matches Node SDK's RegisterServiceInput).
-
-    Attributes:
-        id: Unique service identifier.
-        name: Human-readable service name.
-        description: Description of the service.
-        parent_service_id: ID of the parent service for hierarchical grouping.
-    """
-
-    id: str = Field(description="Unique service identifier.")
-    name: str | None = Field(default=None, description="Human-readable service name.")
-    description: str | None = Field(default=None, description="Description of the service.")
-    parent_service_id: str | None = Field(
-        default=None,
-        description="ID of the parent service for hierarchical grouping.",
-    )
-
-
 class RegisterTriggerMessage(BaseModel):
     model_config = ConfigDict(populate_by_name=True)
 
@@ -178,16 +159,6 @@ class RegisterTriggerMessage(BaseModel):
     message_type: MessageType = Field(default=MessageType.REGISTER_TRIGGER, alias="type")
 
 
-class RegisterServiceMessage(BaseModel):
-    model_config = ConfigDict(populate_by_name=True)
-
-    id: str
-    name: str | None = None
-    description: str | None = None
-    parent_service_id: str | None = Field(default=None)
-    message_type: MessageType = Field(default=MessageType.REGISTER_SERVICE, alias="type")
-
-
 class RegisterFunctionFormat(BaseModel):
     """Format definition for function parameters.
 
@@ -533,115 +504,6 @@ class UnregisterFunctionMessage(BaseModel):
     message_type: MessageType = Field(default=MessageType.UNREGISTER_FUNCTION, alias="type")
 
 
-class FunctionInfo(BaseModel):
-    """Information about a registered function.
-
-    Attributes:
-        function_id: Unique identifier of the function.
-        description: Human-readable description.
-        request_format: Schema describing expected input (JSON Schema or custom format).
-        response_format: Schema describing expected output (JSON Schema or custom format).
-        metadata: Arbitrary metadata attached to the function.
-    """
-
-    function_id: str = Field(description="Unique identifier of the function.")
-    description: str | None = Field(default=None, description="Human-readable description.")
-    request_format: dict[str, Any] | None = Field(
-        default=None, description="Schema describing expected input (JSON Schema or custom format)."
-    )
-    response_format: dict[str, Any] | None = Field(
-        default=None, description="Schema describing expected output (JSON Schema or custom format)."
-    )
-    metadata: dict[str, Any] | None = Field(
-        default=None, description="Arbitrary metadata attached to the function."
-    )
-
-
-class TriggerInfo(BaseModel):
-    """Information about a registered trigger.
-
-    Attributes:
-        id: Unique trigger identifier.
-        trigger_type: Type of trigger (e.g. ``http``, ``queue``, ``cron``).
-        function_id: ID of the function this trigger invokes.
-        config: Trigger-type-specific configuration.
-        metadata: Arbitrary metadata attached to the trigger.
-    """
-
-    id: str = Field(description="Unique trigger identifier.")
-    trigger_type: str = Field(description="Type of trigger (e.g. ``http``, ``queue``, ``cron``).")
-    function_id: str = Field(description="ID of the function this trigger invokes.")
-    config: Any = Field(default=None, description="Trigger-type-specific configuration.")
-    metadata: dict[str, Any] | None = Field(
-        default=None, description="Arbitrary metadata attached to the trigger."
-    )
-
-
-class TriggerTypeInfo(BaseModel):
-    """Information about a registered trigger type.
-
-    Attributes:
-        id: Trigger type identifier (e.g. ``http``, ``cron``, ``queue``).
-        description: Human-readable description of the trigger type.
-        trigger_request_format: JSON Schema for the trigger configuration.
-        call_request_format: JSON Schema for the call request payload.
-    """
-
-    id: str = Field(description="Trigger type identifier.")
-    description: str = Field(description="Human-readable description.")
-    trigger_request_format: Any | None = Field(
-        default=None, description="JSON Schema for trigger configuration."
-    )
-    call_request_format: Any | None = Field(
-        default=None, description="JSON Schema for the call request payload."
-    )
-
-
-WorkerStatus = Literal["connected", "available", "busy", "disconnected"]
-
-
-class WorkerInfo(BaseModel):
-    """Information about a connected worker.
-
-    Attributes:
-        id: Engine-assigned unique worker ID.
-        name: Worker name from InitOptions.
-        runtime: SDK runtime (``python``, ``node``, ``rust``).
-        version: SDK version string.
-        os: Operating system identifier.
-        ip_address: Worker's IP address as seen by the engine.
-        status: Current status (``connected``, ``available``, ``busy``, ``disconnected``).
-        connected_at_ms: Connection timestamp in milliseconds since epoch.
-        function_count: Number of registered functions.
-        functions: List of registered function IDs.
-        active_invocations: Number of currently executing invocations.
-    """
-
-    id: str = Field(description="Engine-assigned unique worker ID.")
-    name: str | None = Field(default=None, description="Worker name from InitOptions.")
-    runtime: str | None = Field(
-        default=None,
-        description="SDK runtime (``python``, ``node``, ``rust``).",
-    )
-    version: str | None = Field(default=None, description="SDK version string.")
-    os: str | None = Field(default=None, description="Operating system identifier.")
-    ip_address: str | None = Field(
-        default=None,
-        description="Worker's IP address as seen by the engine.",
-    )
-    status: WorkerStatus = Field(
-        description="Current status (``connected``, ``available``, ``busy``, ``disconnected``)."
-    )
-    connected_at_ms: int = Field(description="Connection timestamp in milliseconds since epoch.")
-    function_count: int = Field(description="Number of registered functions.")
-    functions: list[str] = Field(description="List of registered function IDs.")
-    active_invocations: int = Field(description="Number of currently executing invocations.")
-    isolation: str | None = Field(
-        default=None,
-        description="Self-reported isolation context (e.g. ``libkrun``, ``docker``, ``k8s``).",
-    )
-
-
 class StreamChannelRef(BaseModel):
     """Reference to a streaming channel for worker-to-worker data transfer.
 
@@ -678,7 +540,6 @@ IIIMessage = (
     | UnregisterFunctionMessage
     | InvokeFunctionMessage
     | InvocationResultMessage
-    | RegisterServiceMessage
     | RegisterTriggerMessage
     | RegisterTriggerTypeMessage
     | UnregisterTriggerMessage

{iii_sdk-0.14.0.dev1 → iii_sdk-0.15.0.dev1}/src/iii/types.py

@@ -12,7 +12,6 @@ from pydantic import BaseModel, ConfigDict, Field
 from .iii_types import (
     HttpInvocationConfig,
     RegisterFunctionMessage,
-    RegisterServiceInput,
     RegisterTriggerInput,
     RegisterTriggerTypeInput,
     RegisterTriggerTypeMessage,
@@ -82,8 +81,6 @@ class IIIClient(Protocol):
 
     def register_trigger(self, trigger: RegisterTriggerInput | dict[str, Any]) -> Trigger: ...
 
-    def register_service(self, service: RegisterServiceInput | dict[str, Any]) -> None: ...
-
     def register_function(
         self,
         function_id: str,

{iii_sdk-0.14.0.dev1 → iii_sdk-0.15.0.dev1}/tests/test_baggage_span_processor.py

@@ -10,7 +10,7 @@ from opentelemetry.sdk.trace.export.in_memory_span_exporter import (
 )
 from opentelemetry.sdk.trace.sampling import ALWAYS_OFF
 
-from iii.baggage_span_processor import (
+from iii_observability import (
     DEFAULT_ALLOWLIST,
     BaggageSpanProcessor,
 )

{iii_sdk-0.14.0.dev1 → iii_sdk-0.15.0.dev1}/tests/test_bridge.py

@@ -4,7 +4,7 @@ import asyncio
 
 import pytest
 
-from iii import FunctionInfo, TriggerAction
+from iii import TriggerAction
 from iii.iii import III
 
 
@@ -22,7 +22,7 @@ async def wait_for(condition, timeout=5.0, interval=0.1):
 async def test_connect_successfully(iii_client: III):
     """SDK connects to the engine and can list functions."""
     result = iii_client.trigger({"function_id": "engine::functions::list", "payload": {}})
-    functions = [FunctionInfo(**f) for f in result.get("functions", [])]
+    functions = result.get("functions", [])
     assert isinstance(functions, list)
 
 
@@ -93,8 +93,8 @@ async def test_list_registered_functions(iii_client: III):
 
     try:
         result = iii_client.trigger({"function_id": "engine::functions::list", "payload": {}})
-        functions = [FunctionInfo(**f) for f in result.get("functions", [])]
-        function_ids = [f.function_id for f in functions]
+        functions = result.get("functions", [])
+        function_ids = [f["function_id"] for f in functions]
 
         assert "test.bridge.py.list.func1" in function_ids
         assert "test.bridge.py.list.func2" in function_ids

{iii_sdk-0.14.0.dev1 → iii_sdk-0.15.0.dev1}/tests/test_context_propagation.py

@@ -7,8 +7,7 @@ import pytest
 import iii.iii as iii_module
 from iii.iii import III
 from iii.iii_constants import InitOptions
-from iii.telemetry import init_otel, shutdown_otel
-from iii.telemetry_types import OtelConfig
+from iii_observability import OtelConfig, init_otel, shutdown_otel
 
 
 @pytest.fixture(autouse=True)

{iii_sdk-0.14.0.dev1 → iii_sdk-0.15.0.dev1}/tests/test_hold_process.py

@@ -5,8 +5,8 @@ from iii.iii import III
 
 def test_background_thread_is_not_daemon(monkeypatch) -> None:
     """The background event-loop thread must be non-daemon so it keeps the process alive."""
-    monkeypatch.setattr("iii.telemetry.init_otel", lambda **kwargs: None)
-    monkeypatch.setattr("iii.telemetry.attach_event_loop", lambda loop: None)
+    monkeypatch.setattr("iii_observability.telemetry.init_otel", lambda **kwargs: None)
+    monkeypatch.setattr("iii_observability.telemetry.attach_event_loop", lambda loop: None)
 
     async def fake_do_connect(self: III) -> None:
         return None
@@ -23,8 +23,8 @@ def test_background_thread_is_not_daemon(monkeypatch) -> None:
 
 def test_shutdown_stops_background_thread(monkeypatch) -> None:
     """After shutdown(), the background thread should stop within a reasonable timeout."""
-    monkeypatch.setattr("iii.telemetry.init_otel", lambda **kwargs: None)
-    monkeypatch.setattr("iii.telemetry.attach_event_loop", lambda loop: None)
+    monkeypatch.setattr("iii_observability.telemetry.init_otel", lambda **kwargs: None)
+    monkeypatch.setattr("iii_observability.telemetry.attach_event_loop", lambda loop: None)
 
     async def fake_do_connect(self: III) -> None:
         return None
@@ -43,8 +43,8 @@ def test_shutdown_stops_background_thread(monkeypatch) -> None:
 
 def test_shutdown_async_stops_background_thread(monkeypatch) -> None:
     """After shutdown_async(), the background thread should also stop."""
-    monkeypatch.setattr("iii.telemetry.init_otel", lambda **kwargs: None)
-    monkeypatch.setattr("iii.telemetry.attach_event_loop", lambda loop: None)
+    monkeypatch.setattr("iii_observability.telemetry.init_otel", lambda **kwargs: None)
+    monkeypatch.setattr("iii_observability.telemetry.attach_event_loop", lambda loop: None)
 
     async def fake_do_connect(self: III) -> None:
         return None

{iii_sdk-0.14.0.dev1 → iii_sdk-0.15.0.dev1}/tests/test_http_external_functions_integration.py

@@ -157,8 +157,8 @@ def _make_fake_ws_env(monkeypatch: pytest.MonkeyPatch) -> list[dict[str, Any]]:
         return FakeWs()
 
     monkeypatch.setattr(iii_module.websockets, "connect", fake_connect)
-    monkeypatch.setattr("iii.telemetry.init_otel", lambda **kwargs: None)
-    monkeypatch.setattr("iii.telemetry.attach_event_loop", lambda loop: None)
+    monkeypatch.setattr("iii_observability.telemetry.init_otel", lambda **kwargs: None)
+    monkeypatch.setattr("iii_observability.telemetry.attach_event_loop", lambda loop: None)
     monkeypatch.setattr(iii_module.III, "_register_worker_metadata", lambda self: None)
     return sent
 

{iii_sdk-0.14.0.dev1 → iii_sdk-0.15.0.dev1}/tests/test_iii_registration_dedup.py

@@ -17,7 +17,7 @@ def reset_otel():
     # III.connect() calls init_otel() which sets global providers;
     # reset them so subsequent test files start with a clean slate.
     try:
-        from iii.telemetry import shutdown_otel
+        from iii_observability import shutdown_otel
 
         shutdown_otel()
     except Exception:
@@ -73,8 +73,8 @@ def test_preconnect_registration_sent_once(monkeypatch: pytest.MonkeyPatch) -> N
         return ws
 
     monkeypatch.setattr(iii_module.websockets, "connect", fake_connect)
-    monkeypatch.setattr("iii.telemetry.init_otel", lambda **kwargs: None)
-    monkeypatch.setattr("iii.telemetry.attach_event_loop", lambda loop: None)
+    monkeypatch.setattr("iii_observability.telemetry.init_otel", lambda **kwargs: None)
+    monkeypatch.setattr("iii_observability.telemetry.attach_event_loop", lambda loop: None)
     monkeypatch.setattr(iii_module.III, "_register_worker_metadata", lambda self: None)
 
     client = III("ws://fake")
@@ -110,8 +110,8 @@ def test_reconnect_replays_durable_state_once_per_connection(
         return ws
 
     monkeypatch.setattr(iii_module.websockets, "connect", fake_connect)
-    monkeypatch.setattr("iii.telemetry.init_otel", lambda **kwargs: None)
-    monkeypatch.setattr("iii.telemetry.attach_event_loop", lambda loop: None)
+    monkeypatch.setattr("iii_observability.telemetry.init_otel", lambda **kwargs: None)
+    monkeypatch.setattr("iii_observability.telemetry.attach_event_loop", lambda loop: None)
     monkeypatch.setattr(iii_module.III, "_register_worker_metadata", lambda self: None)
 
     client = III("ws://fake")
@@ -153,8 +153,8 @@ def test_call_void_queued_while_disconnected_flushes_after_connect(
         return ws
 
     monkeypatch.setattr(iii_module.websockets, "connect", fake_connect)
-    monkeypatch.setattr("iii.telemetry.init_otel", lambda **kwargs: None)
-    monkeypatch.setattr("iii.telemetry.attach_event_loop", lambda loop: None)
+    monkeypatch.setattr("iii_observability.telemetry.init_otel", lambda **kwargs: None)
+    monkeypatch.setattr("iii_observability.telemetry.attach_event_loop", lambda loop: None)
     monkeypatch.setattr(iii_module.III, "_register_worker_metadata", lambda self: None)
 
     client = III("ws://fake")