Skip to content

Commit 8523fa8

Browse files
cjermaincjermain
committed
Updating AWS Bedrock tests with cassettes
1 parent f6eec04 commit 8523fa8

File tree

4 files changed

+353
-60
lines changed

4 files changed

+353
-60
lines changed

‎tests/models/cassettes/test_bedrock/test_bedrock_server_error.yaml‎

Lines changed: 112 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,112 @@
1+
interactions:
2+
- request:
3+
body: '{"messages": [{"role": "user", "content": [{"text": "hello"}]}], "system": [], "inferenceConfig": {}}'
4+
headers:
5+
content-length:
6+
- '101'
7+
content-type:
8+
- !!binary |
9+
YXBwbGljYXRpb24vanNvbg==
10+
method: POST
11+
uri: https://bedrock-runtime.us-east-1.amazonaws.com/model/us.amazon.nova-micro-v1%3A0/converse
12+
response:
13+
headers:
14+
connection:
15+
- keep-alive
16+
content-type:
17+
- application/json
18+
parsed_body:
19+
__type: InternalServerError
20+
message: Internal error
21+
status:
22+
code: 500
23+
message: Internal Server Error
24+
- request:
25+
body: '{"messages": [{"role": "user", "content": [{"text": "hello"}]}], "system": [], "inferenceConfig": {}}'
26+
headers:
27+
content-length:
28+
- '101'
29+
content-type:
30+
- !!binary |
31+
YXBwbGljYXRpb24vanNvbg==
32+
method: POST
33+
uri: https://bedrock-runtime.us-east-1.amazonaws.com/model/us.amazon.nova-micro-v1%3A0/converse
34+
response:
35+
headers:
36+
connection:
37+
- keep-alive
38+
content-type:
39+
- application/json
40+
parsed_body:
41+
__type: InternalServerError
42+
message: Internal error
43+
status:
44+
code: 500
45+
message: Internal Server Error
46+
- request:
47+
body: '{"messages": [{"role": "user", "content": [{"text": "hello"}]}], "system": [], "inferenceConfig": {}}'
48+
headers:
49+
content-length:
50+
- '101'
51+
content-type:
52+
- !!binary |
53+
YXBwbGljYXRpb24vanNvbg==
54+
method: POST
55+
uri: https://bedrock-runtime.us-east-1.amazonaws.com/model/us.amazon.nova-micro-v1%3A0/converse
56+
response:
57+
headers:
58+
connection:
59+
- keep-alive
60+
content-type:
61+
- application/json
62+
parsed_body:
63+
__type: InternalServerError
64+
message: Internal error
65+
status:
66+
code: 500
67+
message: Internal Server Error
68+
- request:
69+
body: '{"messages": [{"role": "user", "content": [{"text": "hello"}]}], "system": [], "inferenceConfig": {}}'
70+
headers:
71+
content-length:
72+
- '101'
73+
content-type:
74+
- !!binary |
75+
YXBwbGljYXRpb24vanNvbg==
76+
method: POST
77+
uri: https://bedrock-runtime.us-east-1.amazonaws.com/model/us.amazon.nova-micro-v1%3A0/converse
78+
response:
79+
headers:
80+
connection:
81+
- keep-alive
82+
content-type:
83+
- application/json
84+
parsed_body:
85+
__type: InternalServerError
86+
message: Internal error
87+
status:
88+
code: 500
89+
message: Internal Server Error
90+
- request:
91+
body: '{"messages": [{"role": "user", "content": [{"text": "hello"}]}], "system": [], "inferenceConfig": {}}'
92+
headers:
93+
content-length:
94+
- '101'
95+
content-type:
96+
- !!binary |
97+
YXBwbGljYXRpb24vanNvbg==
98+
method: POST
99+
uri: https://bedrock-runtime.us-east-1.amazonaws.com/model/us.amazon.nova-micro-v1%3A0/converse
100+
response:
101+
headers:
102+
connection:
103+
- keep-alive
104+
content-type:
105+
- application/json
106+
parsed_body:
107+
__type: InternalServerError
108+
message: Internal error
109+
status:
110+
code: 500
111+
message: Internal Server Error
112+
version: 1

‎tests/models/cassettes/test_bedrock/test_bedrock_streaming_error.yaml‎

Lines changed: 112 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,112 @@
1+
interactions:
2+
- request:
3+
body: '{"messages": [{"role": "user", "content": [{"text": "hello"}]}], "system": [], "inferenceConfig": {}}'
4+
headers:
5+
content-length:
6+
- '101'
7+
content-type:
8+
- !!binary |
9+
YXBwbGljYXRpb24vanNvbg==
10+
method: POST
11+
uri: https://bedrock-runtime.us-east-1.amazonaws.com/model/us.amazon.nova-micro-v1%3A0/converse-stream
12+
response:
13+
headers:
14+
connection:
15+
- keep-alive
16+
content-type:
17+
- application/json
18+
parsed_body:
19+
__type: ThrottlingException
20+
message: Rate exceeded
21+
status:
22+
code: 429
23+
message: Too Many Requests
24+
- request:
25+
body: '{"messages": [{"role": "user", "content": [{"text": "hello"}]}], "system": [], "inferenceConfig": {}}'
26+
headers:
27+
content-length:
28+
- '101'
29+
content-type:
30+
- !!binary |
31+
YXBwbGljYXRpb24vanNvbg==
32+
method: POST
33+
uri: https://bedrock-runtime.us-east-1.amazonaws.com/model/us.amazon.nova-micro-v1%3A0/converse-stream
34+
response:
35+
headers:
36+
connection:
37+
- keep-alive
38+
content-type:
39+
- application/json
40+
parsed_body:
41+
__type: ThrottlingException
42+
message: Rate exceeded
43+
status:
44+
code: 429
45+
message: Too Many Requests
46+
- request:
47+
body: '{"messages": [{"role": "user", "content": [{"text": "hello"}]}], "system": [], "inferenceConfig": {}}'
48+
headers:
49+
content-length:
50+
- '101'
51+
content-type:
52+
- !!binary |
53+
YXBwbGljYXRpb24vanNvbg==
54+
method: POST
55+
uri: https://bedrock-runtime.us-east-1.amazonaws.com/model/us.amazon.nova-micro-v1%3A0/converse-stream
56+
response:
57+
headers:
58+
connection:
59+
- keep-alive
60+
content-type:
61+
- application/json
62+
parsed_body:
63+
__type: ThrottlingException
64+
message: Rate exceeded
65+
status:
66+
code: 429
67+
message: Too Many Requests
68+
- request:
69+
body: '{"messages": [{"role": "user", "content": [{"text": "hello"}]}], "system": [], "inferenceConfig": {}}'
70+
headers:
71+
content-length:
72+
- '101'
73+
content-type:
74+
- !!binary |
75+
YXBwbGljYXRpb24vanNvbg==
76+
method: POST
77+
uri: https://bedrock-runtime.us-east-1.amazonaws.com/model/us.amazon.nova-micro-v1%3A0/converse-stream
78+
response:
79+
headers:
80+
connection:
81+
- keep-alive
82+
content-type:
83+
- application/json
84+
parsed_body:
85+
__type: ThrottlingException
86+
message: Rate exceeded
87+
status:
88+
code: 429
89+
message: Too Many Requests
90+
- request:
91+
body: '{"messages": [{"role": "user", "content": [{"text": "hello"}]}], "system": [], "inferenceConfig": {}}'
92+
headers:
93+
content-length:
94+
- '101'
95+
content-type:
96+
- !!binary |
97+
YXBwbGljYXRpb24vanNvbg==
98+
method: POST
99+
uri: https://bedrock-runtime.us-east-1.amazonaws.com/model/us.amazon.nova-micro-v1%3A0/converse-stream
100+
response:
101+
headers:
102+
connection:
103+
- keep-alive
104+
content-type:
105+
- application/json
106+
parsed_body:
107+
__type: ThrottlingException
108+
message: Rate exceeded
109+
status:
110+
code: 429
111+
message: Too Many Requests
112+
version: 1

‎tests/models/cassettes/test_bedrock/test_bedrock_throttling_error.yaml‎

Lines changed: 112 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,112 @@
1+
interactions:
2+
- request:
3+
body: '{"messages": [{"role": "user", "content": [{"text": "hello"}]}], "system": [], "inferenceConfig": {}}'
4+
headers:
5+
content-length:
6+
- '101'
7+
content-type:
8+
- !!binary |
9+
YXBwbGljYXRpb24vanNvbg==
10+
method: POST
11+
uri: https://bedrock-runtime.us-east-1.amazonaws.com/model/us.amazon.nova-micro-v1%3A0/converse
12+
response:
13+
headers:
14+
connection:
15+
- keep-alive
16+
content-type:
17+
- application/json
18+
parsed_body:
19+
__type: ThrottlingException
20+
message: Rate exceeded
21+
status:
22+
code: 429
23+
message: Too Many Requests
24+
- request:
25+
body: '{"messages": [{"role": "user", "content": [{"text": "hello"}]}], "system": [], "inferenceConfig": {}}'
26+
headers:
27+
content-length:
28+
- '101'
29+
content-type:
30+
- !!binary |
31+
YXBwbGljYXRpb24vanNvbg==
32+
method: POST
33+
uri: https://bedrock-runtime.us-east-1.amazonaws.com/model/us.amazon.nova-micro-v1%3A0/converse
34+
response:
35+
headers:
36+
connection:
37+
- keep-alive
38+
content-type:
39+
- application/json
40+
parsed_body:
41+
__type: ThrottlingException
42+
message: Rate exceeded
43+
status:
44+
code: 429
45+
message: Too Many Requests
46+
- request:
47+
body: '{"messages": [{"role": "user", "content": [{"text": "hello"}]}], "system": [], "inferenceConfig": {}}'
48+
headers:
49+
content-length:
50+
- '101'
51+
content-type:
52+
- !!binary |
53+
YXBwbGljYXRpb24vanNvbg==
54+
method: POST
55+
uri: https://bedrock-runtime.us-east-1.amazonaws.com/model/us.amazon.nova-micro-v1%3A0/converse
56+
response:
57+
headers:
58+
connection:
59+
- keep-alive
60+
content-type:
61+
- application/json
62+
parsed_body:
63+
__type: ThrottlingException
64+
message: Rate exceeded
65+
status:
66+
code: 429
67+
message: Too Many Requests
68+
- request:
69+
body: '{"messages": [{"role": "user", "content": [{"text": "hello"}]}], "system": [], "inferenceConfig": {}}'
70+
headers:
71+
content-length:
72+
- '101'
73+
content-type:
74+
- !!binary |
75+
YXBwbGljYXRpb24vanNvbg==
76+
method: POST
77+
uri: https://bedrock-runtime.us-east-1.amazonaws.com/model/us.amazon.nova-micro-v1%3A0/converse
78+
response:
79+
headers:
80+
connection:
81+
- keep-alive
82+
content-type:
83+
- application/json
84+
parsed_body:
85+
__type: ThrottlingException
86+
message: Rate exceeded
87+
status:
88+
code: 429
89+
message: Too Many Requests
90+
- request:
91+
body: '{"messages": [{"role": "user", "content": [{"text": "hello"}]}], "system": [], "inferenceConfig": {}}'
92+
headers:
93+
content-length:
94+
- '101'
95+
content-type:
96+
- !!binary |
97+
YXBwbGljYXRpb24vanNvbg==
98+
method: POST
99+
uri: https://bedrock-runtime.us-east-1.amazonaws.com/model/us.amazon.nova-micro-v1%3A0/converse
100+
response:
101+
headers:
102+
connection:
103+
- keep-alive
104+
content-type:
105+
- application/json
106+
parsed_body:
107+
__type: ThrottlingException
108+
message: Rate exceeded
109+
status:
110+
code: 429
111+
message: Too Many Requests
112+
version: 1

0 commit comments

Comments
 (0)