99
1010from agentex import Agentex , AsyncAgentex
1111from tests .utils import assert_matches_type
12- from agentex .types import Task , TaskListResponse , TaskDeleteResponse , TaskDeleteByNameResponse
12+ from agentex .types import Task , TaskListResponse , TaskDeleteByNameResponse
1313
1414base_url = os .environ .get ("TEST_API_BASE_URL" , "http://127.0.0.1:4010" )
1515
@@ -93,7 +93,7 @@ def test_method_delete(self, client: Agentex) -> None:
9393 task = client .tasks .delete (
9494 "task_id" ,
9595 )
96- assert_matches_type (TaskDeleteResponse , task , path = ["response" ])
96+ assert_matches_type (Task , task , path = ["response" ])
9797
9898 @pytest .mark .skip ()
9999 @parametrize
@@ -105,7 +105,7 @@ def test_raw_response_delete(self, client: Agentex) -> None:
105105 assert response .is_closed is True
106106 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
107107 task = response .parse ()
108- assert_matches_type (TaskDeleteResponse , task , path = ["response" ])
108+ assert_matches_type (Task , task , path = ["response" ])
109109
110110 @pytest .mark .skip ()
111111 @parametrize
@@ -117,7 +117,7 @@ def test_streaming_response_delete(self, client: Agentex) -> None:
117117 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
118118
119119 task = response .parse ()
120- assert_matches_type (TaskDeleteResponse , task , path = ["response" ])
120+ assert_matches_type (Task , task , path = ["response" ])
121121
122122 assert cast (Any , response .is_closed ) is True
123123
@@ -377,7 +377,7 @@ async def test_method_delete(self, async_client: AsyncAgentex) -> None:
377377 task = await async_client .tasks .delete (
378378 "task_id" ,
379379 )
380- assert_matches_type (TaskDeleteResponse , task , path = ["response" ])
380+ assert_matches_type (Task , task , path = ["response" ])
381381
382382 @pytest .mark .skip ()
383383 @parametrize
@@ -389,7 +389,7 @@ async def test_raw_response_delete(self, async_client: AsyncAgentex) -> None:
389389 assert response .is_closed is True
390390 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
391391 task = await response .parse ()
392- assert_matches_type (TaskDeleteResponse , task , path = ["response" ])
392+ assert_matches_type (Task , task , path = ["response" ])
393393
394394 @pytest .mark .skip ()
395395 @parametrize
@@ -401,7 +401,7 @@ async def test_streaming_response_delete(self, async_client: AsyncAgentex) -> No
401401 assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
402402
403403 task = await response .parse ()
404- assert_matches_type (TaskDeleteResponse , task , path = ["response" ])
404+ assert_matches_type (Task , task , path = ["response" ])
405405
406406 assert cast (Any , response .is_closed ) is True
407407
0 commit comments