fix: align signature of wrapped functions

CS-SI · Sep 24, 2024 · fdcb347 · fdcb347
1 parent 4627886
commit fdcb347
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/opentelemetry/instrumentation/eodag/__init__.py b/src/opentelemetry/instrumentation/eodag/__init__.py
@@ -35,6 +35,7 @@
     DEFAULT_DOWNLOAD_TIMEOUT,
     DEFAULT_DOWNLOAD_WAIT,
     ProgressCallback,
+    StreamResponse,
     Unpack,
 )
 from fastapi import Request
@@ -303,7 +304,7 @@ def wrapper_server_download_stac_item(
         provider: Optional[str] = None,
         asset: Optional[str] = None,
         **kwargs: Any,
-    ) -> StreamingResponse:
+    ) -> Response:
         span_name = "core-download"
         attributes = {
             "operation": "download",
@@ -372,7 +373,7 @@ def wrapper_http_HTTPDownload_stream_download_dict(
         wait: int = DEFAULT_DOWNLOAD_WAIT,
         timeout: int = DEFAULT_DOWNLOAD_TIMEOUT,
         **kwargs: Unpack[DownloadConf],
-    ) -> Dict[str, Any]:
+    ) -> StreamResponse:
         span_name = "core-download"
         # Don't use there the provider's product type.
         attributes = {