chore: silence pyright noise on embedding test files (OPEN-10480)

Adds the same file-level pragma already used by test_portkey_integration.py
to suppress reportUnknown* and reportMissingParameterType — these come from
openlayer.lib.integrations being in pyright's ignore list, which causes
imports from there to be typed as Unknown.

Per-line pyright ignores added on direct imports of botocore.response and
openai, which are not present in the lint job's environment.

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>

Author: viniciusdsmello

tests/test_async_openai_embedding_integration.py

@@ -1,5 +1,9 @@
 """Test OpenAI embedding integration (async)."""
 
+# openlayer.lib.integrations is in pyright's ignore list, so imports get
+# unknown/partially unknown types; disable these diagnostics for this test file only.
+# pyright: reportUnknownMemberType=false, reportUnknownVariableType=false, reportUnknownArgumentType=false, reportUnknownParameterType=false, reportMissingParameterType=false
+
 import asyncio
 from unittest.mock import Mock, AsyncMock, MagicMock, patch
 
@@ -88,7 +92,7 @@ def test_handle_embedding_async_failure_does_not_break_client(self) -> None:
         assert result is fake
 
     def test_trace_async_openai_patches_embeddings_create(self) -> None:
-        import openai
+        import openai  # pyright: ignore[reportMissingImports]
 
         from openlayer.lib.integrations.async_openai_tracer import trace_async_openai
 

tests/test_bedrock_integration.py

@@ -1,5 +1,9 @@
 """Test AWS Bedrock integration."""
 
+# openlayer.lib.integrations is in pyright's ignore list, so imports from bedrock_tracer
+# get unknown/partially unknown types; disable these diagnostics for this test file only.
+# pyright: reportUnknownMemberType=false, reportUnknownVariableType=false, reportUnknownArgumentType=false, reportUnknownParameterType=false, reportMissingParameterType=false
+
 import io
 import json
 from unittest.mock import MagicMock, patch
@@ -10,7 +14,7 @@ class TestBedrockChatRegression:
 
     def _make_anthropic_response(self, body_dict):
         """Build a response dict mimicking what bedrock-runtime returns."""
-        from botocore.response import StreamingBody
+        from botocore.response import StreamingBody  # pyright: ignore[reportMissingImports]
 
         body_bytes = json.dumps(body_dict).encode("utf-8")
         return {"body": StreamingBody(io.BytesIO(body_bytes), len(body_bytes))}
@@ -88,7 +92,7 @@ def test_is_embedding_model_handles_empty_string(self) -> None:
 
     def test_traced_invoke_routes_embedding_to_new_handler(self) -> None:
         """An embedding modelId must call handle_embedding_invoke, not the chat handler."""
-        from botocore.response import StreamingBody
+        from botocore.response import StreamingBody  # pyright: ignore[reportMissingImports]
 
         from openlayer.lib.integrations.bedrock_tracer import trace_bedrock
 
@@ -120,7 +124,7 @@ class TestBedrockTitanEmbedding:
     """Titan v1 and v2 embedding requests produce well-formed embedding steps."""
 
     def _titan_response(self, embedding, token_count):
-        from botocore.response import StreamingBody
+        from botocore.response import StreamingBody  # pyright: ignore[reportMissingImports]
 
         body = json.dumps(
             {"embedding": embedding, "inputTextTokenCount": token_count}
@@ -198,7 +202,7 @@ class TestBedrockCohereEmbedding:
     """Cohere v3 embedding produces a multi-vector batch step."""
 
     def _cohere_response(self, embeddings):
-        from botocore.response import StreamingBody
+        from botocore.response import StreamingBody  # pyright: ignore[reportMissingImports]
 
         body = json.dumps(
             {
@@ -249,7 +253,7 @@ class TestBedrockEmbeddingResilience:
     """Tracing failures must never break the caller; response body must remain usable."""
 
     def test_embedding_failure_does_not_break_client(self) -> None:
-        from botocore.response import StreamingBody
+        from botocore.response import StreamingBody  # pyright: ignore[reportMissingImports]
 
         from openlayer.lib.integrations.bedrock_tracer import trace_bedrock
 
@@ -275,7 +279,7 @@ def test_embedding_failure_does_not_break_client(self) -> None:
         assert response["body"].read() == body_bytes
 
     def test_embedding_response_body_is_replayable(self) -> None:
-        from botocore.response import StreamingBody
+        from botocore.response import StreamingBody  # pyright: ignore[reportMissingImports]
 
         from openlayer.lib.integrations.bedrock_tracer import trace_bedrock
 

tests/test_litellm_integration.py

@@ -300,7 +300,9 @@ def test_trace_litellm_patches_embedding(self, mock_litellm: Mock) -> None:
 
     @patch("openlayer.lib.integrations.litellm_tracer.HAVE_LITELLM", True)
     def test_handle_embedding_single_input(self) -> None:
-        from openlayer.lib.integrations.litellm_tracer import handle_embedding
+        from openlayer.lib.integrations.litellm_tracer import (
+            handle_embedding,  # pyright: ignore[reportUnknownVariableType]
+        )
 
         fake_response = Mock()
         fake_response.model = "text-embedding-3-small"
@@ -338,7 +340,9 @@ def test_handle_embedding_single_input(self) -> None:
 
     @patch("openlayer.lib.integrations.litellm_tracer.HAVE_LITELLM", True)
     def test_handle_embedding_batch_input(self) -> None:
-        from openlayer.lib.integrations.litellm_tracer import handle_embedding
+        from openlayer.lib.integrations.litellm_tracer import (
+            handle_embedding,  # pyright: ignore[reportUnknownVariableType]
+        )
 
         fake_response = Mock()
         fake_response.model = "text-embedding-3-small"
@@ -376,7 +380,9 @@ def test_handle_embedding_batch_input(self) -> None:
 
     @patch("openlayer.lib.integrations.litellm_tracer.HAVE_LITELLM", True)
     def test_handle_embedding_cost_from_hidden_params(self) -> None:
-        from openlayer.lib.integrations.litellm_tracer import handle_embedding
+        from openlayer.lib.integrations.litellm_tracer import (
+            handle_embedding,  # pyright: ignore[reportUnknownVariableType]
+        )
 
         fake_response = Mock()
         fake_response.model = "text-embedding-3-small"
@@ -398,7 +404,9 @@ def test_handle_embedding_cost_from_hidden_params(self) -> None:
 
     @patch("openlayer.lib.integrations.litellm_tracer.HAVE_LITELLM", True)
     def test_handle_embedding_failure_does_not_break_client(self) -> None:
-        from openlayer.lib.integrations.litellm_tracer import handle_embedding
+        from openlayer.lib.integrations.litellm_tracer import (
+            handle_embedding,  # pyright: ignore[reportUnknownVariableType]
+        )
 
         fake_response = Mock()
         fake_response.model = "text-embedding-3-small"

tests/test_openai_embedding_helpers.py

@@ -1,5 +1,9 @@
 """Test shared OpenAI embedding parsers."""
 
+# openlayer.lib.integrations is in pyright's ignore list, so imports get
+# unknown/partially unknown types; disable these diagnostics for this test file only.
+# pyright: reportUnknownMemberType=false, reportUnknownVariableType=false, reportUnknownArgumentType=false, reportUnknownParameterType=false, reportMissingParameterType=false
+
 from unittest.mock import Mock
 
 

tests/test_openai_embedding_integration.py

@@ -1,5 +1,9 @@
 """Test OpenAI embedding integration (sync)."""
 
+# openlayer.lib.integrations is in pyright's ignore list, so imports get
+# unknown/partially unknown types; disable these diagnostics for this test file only.
+# pyright: reportUnknownMemberType=false, reportUnknownVariableType=false, reportUnknownArgumentType=false, reportUnknownParameterType=false, reportMissingParameterType=false
+
 from unittest.mock import Mock, MagicMock, patch
 
 
@@ -87,7 +91,7 @@ def test_handle_embedding_failure_does_not_break_client(self) -> None:
 
     def test_trace_openai_patches_embeddings_create(self) -> None:
         """After trace_openai, client.embeddings.create is replaced."""
-        import openai
+        import openai  # pyright: ignore[reportMissingImports]
 
         from openlayer.lib.integrations.openai_tracer import trace_openai