Skip to content

Commit 1934bb0

Browse files
jamesbrazajamesbraza
committed
Added mark vcr decorator to a bunch of tests, with fixtures
1 parent 6716bcc commit 1934bb0

23 files changed

+2494
-24
lines changed

‎tests/cassettes/TestHTTPAgentClient.test_lifecycle.yaml

Lines changed: 713 additions & 0 deletions
Large diffs are not rendered by default.

‎tests/cassettes/TestLLMModel.test_model[claude-3-haiku-20240307].yaml

Lines changed: 61 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,61 @@
1+
interactions:
2+
- request:
3+
body:
4+
'{"messages": [{"role": "user", "content": [{"type": "text", "text": "Hello,
5+
how are you?"}]}], "system": [{"type": "text", "text": "Respond with single
6+
words."}], "max_tokens": 4096, "model": "claude-3-haiku-20240307"}'
7+
headers:
8+
accept:
9+
- application/json
10+
accept-encoding:
11+
- gzip, deflate
12+
anthropic-version:
13+
- "2023-06-01"
14+
connection:
15+
- keep-alive
16+
content-length:
17+
- "218"
18+
content-type:
19+
- application/json
20+
host:
21+
- api.anthropic.com
22+
user-agent:
23+
- litellm/1.44.24
24+
method: POST
25+
uri: https://api.anthropic.com/v1/messages
26+
response:
27+
body:
28+
string: !!binary |
29+
H4sIAAAAAAAAA0yO22rDMBBE/2We5eLcCNFjSEJ+oDRQilHkxRaWV653BS3G/x4cWsjTwJkLMyHU
30+
sOilqcrVx/adxHetnC7j8dzce387XFsY6O9AS4pEXEMwGFNcgBMJoo4VBn2qKcLCR5drKjZF60KX
31+
i3W53pabcg8Dn1iJFfZz+l9U+lm6T7G4BKY3zF8GommoRnKSGBbEdaV5ZPwZQt+Z2BMs5xgN8vOV
32+
nRB4yFpp6ogFdnUwSFlf0W6eHwAAAP//AwBu9B8r8gAAAA==
33+
headers:
34+
CF-Cache-Status:
35+
- DYNAMIC
36+
CF-RAY:
37+
- 8c1adb64febe9800-SJC
38+
Connection:
39+
- keep-alive
40+
Content-Encoding:
41+
- gzip
42+
Content-Type:
43+
- application/json
44+
Date:
45+
- Wed, 11 Sep 2024 21:47:34 GMT
46+
Server:
47+
- cloudflare
48+
Transfer-Encoding:
49+
- chunked
50+
X-Robots-Tag:
51+
- none
52+
request-id:
53+
- req_013F7JJtkufBc3MGej2SFWxA
54+
via:
55+
- 1.1 google
56+
x-cloud-trace-context:
57+
- 5a7bd3586c4b8aa6cf7dc15b4a10a79e
58+
status:
59+
code: 200
60+
message: OK
61+
version: 1

‎tests/cassettes/TestLLMModel.test_model[gpt-3.5-turbo].yaml

Lines changed: 96 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,96 @@
1+
interactions:
2+
- request:
3+
body:
4+
'{"messages": [{"role": "system", "content": "Respond with single words."},
5+
{"role": "user", "content": "Hello, how are you?"}], "model": "gpt-3.5-turbo"}'
6+
headers:
7+
accept:
8+
- application/json
9+
accept-encoding:
10+
- gzip, deflate
11+
connection:
12+
- keep-alive
13+
content-length:
14+
- "153"
15+
content-type:
16+
- application/json
17+
host:
18+
- api.openai.com
19+
user-agent:
20+
- AsyncOpenAI/Python 1.44.1
21+
x-stainless-arch:
22+
- arm64
23+
x-stainless-async:
24+
- async:asyncio
25+
x-stainless-lang:
26+
- python
27+
x-stainless-os:
28+
- MacOS
29+
x-stainless-package-version:
30+
- 1.44.1
31+
x-stainless-raw-response:
32+
- "true"
33+
x-stainless-runtime:
34+
- CPython
35+
x-stainless-runtime-version:
36+
- 3.12.5
37+
method: POST
38+
uri: https://api.openai.com/v1/chat/completions
39+
response:
40+
body:
41+
string: !!binary |
42+
H4sIAAAAAAAAAwAAAP//VJBBawIxFITv+yvCO7virqzFvQkFD4W2UHtpKRKzzzWa5IXkLa2I/71k
43+
XbW95DBfZjKTUyYE6AZqAWonWVlv8sXsdbWSRdBvy8PyyM9zsh/0+LRf+O+XdxglB232qPjqGiuy
44+
3iBrchesAkrGlFo8lLPJvCirqgeWGjTJ1nrOp+Mq5y5sKJ8UZTU4d6QVRqjFZyaEEKf+TB1dgz9Q
45+
i8noqliMUbYI9e2SEBDIJAVkjDqydAyjO1TkGF1fe0nU/EUBt12UqZrrjBn08+0tQ60PtIkDv+lb
46+
7XTcrQPKSC7lRiYPPT1nQnz1m7p/NcEHsp7XTAd0KbAsL3Fw/8U7LAbGxNL88UyzoR7EY2S06612
47+
LQYfdL+vH3HOfgEAAP//AwBMSqZg3gEAAA==
48+
headers:
49+
CF-Cache-Status:
50+
- DYNAMIC
51+
CF-RAY:
52+
- 8c1adb68dfeb966f-SJC
53+
Connection:
54+
- keep-alive
55+
Content-Encoding:
56+
- gzip
57+
Content-Type:
58+
- application/json
59+
Date:
60+
- Wed, 11 Sep 2024 21:47:35 GMT
61+
Server:
62+
- cloudflare
63+
Transfer-Encoding:
64+
- chunked
65+
X-Content-Type-Options:
66+
- nosniff
67+
access-control-expose-headers:
68+
- X-Request-ID
69+
alt-svc:
70+
- h3=":443"; ma=86400
71+
openai-organization:
72+
- future-house-xr4tdh
73+
openai-processing-ms:
74+
- "94"
75+
openai-version:
76+
- "2020-10-01"
77+
strict-transport-security:
78+
- max-age=15552000; includeSubDomains; preload
79+
x-ratelimit-limit-requests:
80+
- "12000"
81+
x-ratelimit-limit-tokens:
82+
- "1000000"
83+
x-ratelimit-remaining-requests:
84+
- "11999"
85+
x-ratelimit-remaining-tokens:
86+
- "999969"
87+
x-ratelimit-reset-requests:
88+
- 5ms
89+
x-ratelimit-reset-tokens:
90+
- 1ms
91+
x-request-id:
92+
- req_59689a9a4b451bc51b25aa932438cde3
93+
status:
94+
code: 200
95+
message: OK
96+
version: 1

‎tests/cassettes/TestLLMModel.test_output_schema.yaml

Lines changed: 101 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,101 @@
1+
interactions:
2+
- request:
3+
body:
4+
'{"messages": [{"role": "system", "content": "Respond following this JSON
5+
schema:\n\n{\n \"properties\": {\n \"name\": {\n \"title\": \"Name\",\n \"type\":
6+
\"string\"\n },\n \"age\": {\n \"title\": \"Age\",\n \"type\":
7+
\"integer\"\n }\n },\n \"required\": [\n \"name\",\n \"age\"\n ],\n \"title\":
8+
\"DummyOutputSchema\",\n \"type\": \"object\"\n}"}, {"role": "user", "content":
9+
"My name is Claude and I am 1 year old. What is my name and age?"}], "model":
10+
"gpt-3.5-turbo", "response_format": {"type": "json_object"}}'
11+
headers:
12+
accept:
13+
- application/json
14+
accept-encoding:
15+
- gzip, deflate
16+
connection:
17+
- keep-alive
18+
content-length:
19+
- "559"
20+
content-type:
21+
- application/json
22+
host:
23+
- api.openai.com
24+
user-agent:
25+
- AsyncOpenAI/Python 1.44.1
26+
x-stainless-arch:
27+
- arm64
28+
x-stainless-async:
29+
- async:asyncio
30+
x-stainless-lang:
31+
- python
32+
x-stainless-os:
33+
- MacOS
34+
x-stainless-package-version:
35+
- 1.44.1
36+
x-stainless-raw-response:
37+
- "true"
38+
x-stainless-runtime:
39+
- CPython
40+
x-stainless-runtime-version:
41+
- 3.12.5
42+
method: POST
43+
uri: https://api.openai.com/v1/chat/completions
44+
response:
45+
body:
46+
string: !!binary |
47+
H4sIAAAAAAAAA1SQzW7CMBCE73mK1Z4TREghIbeqVXurekFV21TIJEswdWzLdgQI8e6VTfjpxYf5
48+
POMZHyMA5A2WgPWGubrTInmcvRfZ4evjk78U+vU5e9i+LXb73aIxOcsx9g612lLtLq5RrTotyHEl
49+
z7g2xBz51DSfzMbFfD7JA+hUQ8LbWu2SbDRNXG9WKhmnk+ng3Chek8USviMAgGM4fUfZ0B5LGMcX
50+
pSNrWUtYXi8BoFHCK8is5dYx6TC+wVpJRzLUPlYSoELJOqqwhAqfBOsbqjA+A9YGPa3k6T7B0Lq3
51+
zC+QvRCDfrpWEqrVRq3swK/6mktuN0tDzCrpn7dOaQz0FAH8hOn9vzWojeq0Wzr1S9IHpmlxzsPb
52+
b9/RfIBOOSbu9GwaDQ3RHqyjbrnmsiWjDQ8/EXacoj8AAAD//wMANu8s3AgCAAA=
53+
headers:
54+
CF-Cache-Status:
55+
- DYNAMIC
56+
CF-RAY:
57+
- 8c1abafcea3c171a-SJC
58+
Connection:
59+
- keep-alive
60+
Content-Encoding:
61+
- gzip
62+
Content-Type:
63+
- application/json
64+
Date:
65+
- Wed, 11 Sep 2024 21:25:27 GMT
66+
Server:
67+
- cloudflare
68+
Transfer-Encoding:
69+
- chunked
70+
X-Content-Type-Options:
71+
- nosniff
72+
access-control-expose-headers:
73+
- X-Request-ID
74+
alt-svc:
75+
- h3=":443"; ma=86400
76+
openai-organization:
77+
- future-house-xr4tdh
78+
openai-processing-ms:
79+
- "267"
80+
openai-version:
81+
- "2020-10-01"
82+
strict-transport-security:
83+
- max-age=15552000; includeSubDomains; preload
84+
x-ratelimit-limit-requests:
85+
- "12000"
86+
x-ratelimit-limit-tokens:
87+
- "1000000"
88+
x-ratelimit-remaining-requests:
89+
- "11999"
90+
x-ratelimit-remaining-tokens:
91+
- "999838"
92+
x-ratelimit-reset-requests:
93+
- 5ms
94+
x-ratelimit-reset-tokens:
95+
- 9ms
96+
x-request-id:
97+
- req_65355d90c1cb42732d3c85abb2413f94
98+
status:
99+
code: 200
100+
message: OK
101+
version: 1

‎tests/cassettes/TestLLMModel.test_text_image_message[claude-3-haiku-20240307].yaml

Lines changed: 62 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,62 @@
1+
interactions:
2+
- request:
3+
body:
4+
'{"messages": [{"role": "user", "content": [{"type": "image", "source":
5+
{"type": "base64", "media_type": "image/png", "data": "iVBORw0KGgoAAAANSUhEUgAAACAAAAAgCAIAAAD8GO2jAAAAKElEQVR4nO3NMQEAAAjDMMC/ZzDBvlRA01vZJvwHAAAAAAAAAAAAbx2jxAE/i2AjOgAAAABJRU5ErkJggg=="}},
6+
{"type": "text", "text": "What color is this square? Respond only with the color
7+
name."}]}], "max_tokens": 4096, "model": "claude-3-haiku-20240307"}'
8+
headers:
9+
accept:
10+
- application/json
11+
accept-encoding:
12+
- gzip, deflate
13+
anthropic-version:
14+
- "2023-06-01"
15+
connection:
16+
- keep-alive
17+
content-length:
18+
- "411"
19+
content-type:
20+
- application/json
21+
host:
22+
- api.anthropic.com
23+
user-agent:
24+
- litellm/1.44.24
25+
method: POST
26+
uri: https://api.anthropic.com/v1/messages
27+
response:
28+
body:
29+
string: !!binary |
30+
H4sIAAAAAAAAA0yOT2vCQBDFv8s7byBNBOlehV5KRUS9SAlLdqrBdTZmZool5LuXiIWeHvzeH96I
31+
LsLjKqemfNnFj3U4L982h9V2d/ve1/b+RQMc9KenOUUi4URwGHKaQRDpRAMrHK45UoJHm4JFKuri
32+
HLqLFVVZLcq6XMKhzazECn8c/xaV7nP3IR5bipg+HURz3wwUJDM8iGOjNjCehtDNiFuCZ0vJwR6f
33+
/IiOe9NG84VY4KtXh2z6Hy2m6RcAAP//AwB6WGlZ8AAAAA==
34+
headers:
35+
CF-Cache-Status:
36+
- DYNAMIC
37+
CF-RAY:
38+
- 8c1aba3c6c959664-SJC
39+
Connection:
40+
- keep-alive
41+
Content-Encoding:
42+
- gzip
43+
Content-Type:
44+
- application/json
45+
Date:
46+
- Wed, 11 Sep 2024 21:24:56 GMT
47+
Server:
48+
- cloudflare
49+
Transfer-Encoding:
50+
- chunked
51+
X-Robots-Tag:
52+
- none
53+
request-id:
54+
- req_01HzTrvjM4ny4Fa5TnY8n8pE
55+
via:
56+
- 1.1 google
57+
x-cloud-trace-context:
58+
- 4ab7e744703e1034c4b84f8d53549416
59+
status:
60+
code: 200
61+
message: OK
62+
version: 1

0 commit comments

Comments
 (0)