chore(internal): cast type in mocked test (#326)

anthropics · Jan 31, 2024 · c92896b · c92896b
1 parent 6bd4ba9
commit c92896b
Showing 1 changed file with 28 additions and 16 deletions.
diff --git a/tests/test_client.py b/tests/test_client.py
@@ -808,10 +808,13 @@ def test_retrying_timeout_errors_doesnt_leak(self, respx_mock: MockRouter) -> No
         with pytest.raises(APITimeoutError):
             self.client.post(
                 "/v1/complete",
-                body=dict(
-                    max_tokens_to_sample=300,
-                    model="claude-2.1",
-                    prompt="\n\nHuman:Where can I get a good coffee in my neighbourhood?\n\nAssistant:",
+                body=cast(
+                    object,
+                    dict(
+                        max_tokens_to_sample=300,
+                        model="claude-2.1",
+                        prompt="\n\nHuman:Where can I get a good coffee in my neighbourhood?\n\nAssistant:",
+                    ),
                 ),
                 cast_to=httpx.Response,
                 options={"headers": {RAW_RESPONSE_HEADER: "stream"}},
@@ -827,10 +830,13 @@ def test_retrying_status_errors_doesnt_leak(self, respx_mock: MockRouter) -> Non
         with pytest.raises(APIStatusError):
             self.client.post(
                 "/v1/complete",
-                body=dict(
-                    max_tokens_to_sample=300,
-                    model="claude-2.1",
-                    prompt="\n\nHuman:Where can I get a good coffee in my neighbourhood?\n\nAssistant:",
+                body=cast(
+                    object,
+                    dict(
+                        max_tokens_to_sample=300,
+                        model="claude-2.1",
+                        prompt="\n\nHuman:Where can I get a good coffee in my neighbourhood?\n\nAssistant:",
+                    ),
                 ),
                 cast_to=httpx.Response,
                 options={"headers": {RAW_RESPONSE_HEADER: "stream"}},
@@ -1610,10 +1616,13 @@ async def test_retrying_timeout_errors_doesnt_leak(self, respx_mock: MockRouter)
         with pytest.raises(APITimeoutError):
             await self.client.post(
                 "/v1/complete",
-                body=dict(
-                    max_tokens_to_sample=300,
-                    model="claude-2.1",
-                    prompt="\n\nHuman:Where can I get a good coffee in my neighbourhood?\n\nAssistant:",
+                body=cast(
+                    object,
+                    dict(
+                        max_tokens_to_sample=300,
+                        model="claude-2.1",
+                        prompt="\n\nHuman:Where can I get a good coffee in my neighbourhood?\n\nAssistant:",
+                    ),
                 ),
                 cast_to=httpx.Response,
                 options={"headers": {RAW_RESPONSE_HEADER: "stream"}},
@@ -1629,10 +1638,13 @@ async def test_retrying_status_errors_doesnt_leak(self, respx_mock: MockRouter)
         with pytest.raises(APIStatusError):
             await self.client.post(
                 "/v1/complete",
-                body=dict(
-                    max_tokens_to_sample=300,
-                    model="claude-2.1",
-                    prompt="\n\nHuman:Where can I get a good coffee in my neighbourhood?\n\nAssistant:",
+                body=cast(
+                    object,
+                    dict(
+                        max_tokens_to_sample=300,
+                        model="claude-2.1",
+                        prompt="\n\nHuman:Where can I get a good coffee in my neighbourhood?\n\nAssistant:",
+                    ),
                 ),
                 cast_to=httpx.Response,
                 options={"headers": {RAW_RESPONSE_HEADER: "stream"}},