12
12
from browser_use_sdk .types import (
13
13
TaskView ,
14
14
TaskListResponse ,
15
+ TaskCreateResponse ,
15
16
TaskGetLogsResponse ,
16
17
TaskGetOutputFileResponse ,
17
18
TaskGetUserUploadedFileResponse ,
@@ -29,7 +30,7 @@ def test_method_create(self, client: BrowserUse) -> None:
29
30
task = client .tasks .create (
30
31
task = "x" ,
31
32
)
32
- assert_matches_type (TaskView , task , path = ["response" ])
33
+ assert_matches_type (TaskCreateResponse , task , path = ["response" ])
33
34
34
35
@pytest .mark .skip (reason = "Prism tests are disabled" )
35
36
@parametrize
@@ -50,7 +51,7 @@ def test_method_create_with_all_params(self, client: BrowserUse) -> None:
50
51
secrets = {"foo" : "string" },
51
52
structured_output_json = "structuredOutputJson" ,
52
53
)
53
- assert_matches_type (TaskView , task , path = ["response" ])
54
+ assert_matches_type (TaskCreateResponse , task , path = ["response" ])
54
55
55
56
@pytest .mark .skip (reason = "Prism tests are disabled" )
56
57
@parametrize
@@ -62,7 +63,7 @@ def test_raw_response_create(self, client: BrowserUse) -> None:
62
63
assert response .is_closed is True
63
64
assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
64
65
task = response .parse ()
65
- assert_matches_type (TaskView , task , path = ["response" ])
66
+ assert_matches_type (TaskCreateResponse , task , path = ["response" ])
66
67
67
68
@pytest .mark .skip (reason = "Prism tests are disabled" )
68
69
@parametrize
@@ -74,7 +75,7 @@ def test_streaming_response_create(self, client: BrowserUse) -> None:
74
75
assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
75
76
76
77
task = response .parse ()
77
- assert_matches_type (TaskView , task , path = ["response" ])
78
+ assert_matches_type (TaskCreateResponse , task , path = ["response" ])
78
79
79
80
assert cast (Any , response .is_closed ) is True
80
81
@@ -366,7 +367,7 @@ async def test_method_create(self, async_client: AsyncBrowserUse) -> None:
366
367
task = await async_client .tasks .create (
367
368
task = "x" ,
368
369
)
369
- assert_matches_type (TaskView , task , path = ["response" ])
370
+ assert_matches_type (TaskCreateResponse , task , path = ["response" ])
370
371
371
372
@pytest .mark .skip (reason = "Prism tests are disabled" )
372
373
@parametrize
@@ -387,7 +388,7 @@ async def test_method_create_with_all_params(self, async_client: AsyncBrowserUse
387
388
secrets = {"foo" : "string" },
388
389
structured_output_json = "structuredOutputJson" ,
389
390
)
390
- assert_matches_type (TaskView , task , path = ["response" ])
391
+ assert_matches_type (TaskCreateResponse , task , path = ["response" ])
391
392
392
393
@pytest .mark .skip (reason = "Prism tests are disabled" )
393
394
@parametrize
@@ -399,7 +400,7 @@ async def test_raw_response_create(self, async_client: AsyncBrowserUse) -> None:
399
400
assert response .is_closed is True
400
401
assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
401
402
task = await response .parse ()
402
- assert_matches_type (TaskView , task , path = ["response" ])
403
+ assert_matches_type (TaskCreateResponse , task , path = ["response" ])
403
404
404
405
@pytest .mark .skip (reason = "Prism tests are disabled" )
405
406
@parametrize
@@ -411,7 +412,7 @@ async def test_streaming_response_create(self, async_client: AsyncBrowserUse) ->
411
412
assert response .http_request .headers .get ("X-Stainless-Lang" ) == "python"
412
413
413
414
task = await response .parse ()
414
- assert_matches_type (TaskView , task , path = ["response" ])
415
+ assert_matches_type (TaskCreateResponse , task , path = ["response" ])
415
416
416
417
assert cast (Any , response .is_closed ) is True
417
418
0 commit comments