Enhance TFO-Python-SDK with OTEL Autoinstrumentation (Fixing pipeline)

Author: zeroc0d3

src/telemetryflow/__init__.py

@@ -12,6 +12,8 @@
     - Full OpenTelemetry compatibility
 """
 
+from typing import Any
+
 from telemetryflow.builder import TelemetryFlowBuilder
 from telemetryflow.client import TelemetryFlowClient
 from telemetryflow.domain.config import Protocol, SignalType, TelemetryConfig
@@ -66,7 +68,7 @@ def new_simple(
 
 def auto_instrument(
     client: TelemetryFlowClient | None = None,
-    **kwargs: object,
+    **kwargs: Any,
 ) -> dict[str, bool]:
     """Auto-instrument all available libraries.
 

src/telemetryflow/client.py

@@ -335,7 +335,8 @@ def start_span(
             kind=kind,
             attributes=attributes or {},
         )
-        return self._handler.handle(command)
+        result: str = self._handler.handle(command)
+        return result
 
     def end_span(
         self,

src/telemetryflow/infrastructure/handlers.py

@@ -116,7 +116,7 @@ def handle(self, command: Command) -> Any:
         if handler is None:
             raise ValueError(f"Unknown command type: {type(command).__name__}")
 
-        return handler(command)
+        return handler(command)  # type: ignore[operator]
 
     def _handle_initialize(self, command: InitializeSDKCommand) -> None:
         """Initialize the SDK with the given configuration."""

src/telemetryflow/middleware/fastapi.py

@@ -8,7 +8,13 @@
 from telemetryflow.middleware.base import TelemetryMiddleware
 
 if TYPE_CHECKING:
-    from starlette.types import ASGIApp, Message, Receive, Scope, Send
+    from starlette.types import (  # type: ignore[import-not-found]
+        ASGIApp,
+        Message,
+        Receive,
+        Scope,
+        Send,
+    )
 
     from telemetryflow.client import TelemetryFlowClient
 

src/telemetryflow/middleware/flask.py

@@ -7,7 +7,7 @@
 from telemetryflow.middleware.base import TelemetryMiddleware
 
 if TYPE_CHECKING:
-    from flask import Flask, Response
+    from flask import Flask, Response  # type: ignore[import-not-found]
 
     from telemetryflow.client import TelemetryFlowClient