23
23
24
24
from arcadepy import Arcade , AsyncArcade , APIResponseValidationError
25
25
from arcadepy ._types import Omit
26
+ from arcadepy ._utils import maybe_transform
26
27
from arcadepy ._models import BaseModel , FinalRequestOptions
27
28
from arcadepy ._constants import RAW_RESPONSE_HEADER
28
29
from arcadepy ._exceptions import ArcadeError , APIStatusError , APITimeoutError , APIResponseValidationError
32
33
BaseClient ,
33
34
make_request_options ,
34
35
)
36
+ from arcadepy .types .chat .completion_create_params import CompletionCreateParams
35
37
36
38
from .utils import update_env
37
39
@@ -717,13 +719,16 @@ def test_retrying_timeout_errors_doesnt_leak(self, respx_mock: MockRouter) -> No
717
719
"/v1/chat/completions" ,
718
720
body = cast (
719
721
object ,
720
- dict (
721
- messages = [
722
- {
723
- "role" : "user" ,
724
- "content" : "Hello, how can I use Arcade AI?" ,
725
- }
726
- ]
722
+ maybe_transform (
723
+ dict (
724
+ messages = [
725
+ {
726
+ "role" : "user" ,
727
+ "content" : "Hello, how can I use Arcade AI?" ,
728
+ }
729
+ ]
730
+ ),
731
+ CompletionCreateParams ,
727
732
),
728
733
),
729
734
cast_to = httpx .Response ,
@@ -742,13 +747,16 @@ def test_retrying_status_errors_doesnt_leak(self, respx_mock: MockRouter) -> Non
742
747
"/v1/chat/completions" ,
743
748
body = cast (
744
749
object ,
745
- dict (
746
- messages = [
747
- {
748
- "role" : "user" ,
749
- "content" : "Hello, how can I use Arcade AI?" ,
750
- }
751
- ]
750
+ maybe_transform (
751
+ dict (
752
+ messages = [
753
+ {
754
+ "role" : "user" ,
755
+ "content" : "Hello, how can I use Arcade AI?" ,
756
+ }
757
+ ]
758
+ ),
759
+ CompletionCreateParams ,
752
760
),
753
761
),
754
762
cast_to = httpx .Response ,
@@ -1513,13 +1521,16 @@ async def test_retrying_timeout_errors_doesnt_leak(self, respx_mock: MockRouter)
1513
1521
"/v1/chat/completions" ,
1514
1522
body = cast (
1515
1523
object ,
1516
- dict (
1517
- messages = [
1518
- {
1519
- "role" : "user" ,
1520
- "content" : "Hello, how can I use Arcade AI?" ,
1521
- }
1522
- ]
1524
+ maybe_transform (
1525
+ dict (
1526
+ messages = [
1527
+ {
1528
+ "role" : "user" ,
1529
+ "content" : "Hello, how can I use Arcade AI?" ,
1530
+ }
1531
+ ]
1532
+ ),
1533
+ CompletionCreateParams ,
1523
1534
),
1524
1535
),
1525
1536
cast_to = httpx .Response ,
@@ -1538,13 +1549,16 @@ async def test_retrying_status_errors_doesnt_leak(self, respx_mock: MockRouter)
1538
1549
"/v1/chat/completions" ,
1539
1550
body = cast (
1540
1551
object ,
1541
- dict (
1542
- messages = [
1543
- {
1544
- "role" : "user" ,
1545
- "content" : "Hello, how can I use Arcade AI?" ,
1546
- }
1547
- ]
1552
+ maybe_transform (
1553
+ dict (
1554
+ messages = [
1555
+ {
1556
+ "role" : "user" ,
1557
+ "content" : "Hello, how can I use Arcade AI?" ,
1558
+ }
1559
+ ]
1560
+ ),
1561
+ CompletionCreateParams ,
1548
1562
),
1549
1563
),
1550
1564
cast_to = httpx .Response ,
0 commit comments