Skip to content

Commit ba7f1fd

Browse files
sarth6sarth6
committed
Clean up tests
1 parent f8f9cd5 commit ba7f1fd

File tree

4 files changed

+356
-1031
lines changed

4 files changed

+356
-1031
lines changed

‎tests/models/cassettes/test_google/test_google_model_web_fetch_tool[False].yaml‎

Lines changed: 76 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,76 @@
1+
interactions:
2+
- request:
3+
body: '{"contents": [{"parts": [{"text": "What is the first sentence on the page
4+
https://ai.pydantic.dev? Reply with only the sentence."}], "role": "user"}],
5+
"systemInstruction": {"parts": [{"text": "You are a helpful chatbot."}], "role":
6+
"user"}, "tools": [{"urlContext": {}}], "generationConfig": {}}'
7+
headers:
8+
accept:
9+
- '*/*'
10+
accept-encoding:
11+
- gzip, deflate
12+
connection:
13+
- keep-alive
14+
content-length:
15+
- '295'
16+
content-type:
17+
- application/json
18+
host:
19+
- generativelanguage.googleapis.com
20+
method: POST
21+
uri: https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash:generateContent
22+
response:
23+
body:
24+
string: "{\n \"candidates\": [\n {\n \"content\": {\n \"parts\"\
25+
: [\n {\n \"text\": \"Pydantic AI is a Python agent framework\
26+
\ designed to make it less painful to build production grade applications\
27+
\ with Generative AI.\"\n }\n ],\n \"role\": \"model\"\
28+
\n },\n \"finishReason\": \"STOP\",\n \"index\": 0,\n \
29+
\ \"groundingMetadata\": {\n \"groundingChunks\": [\n {\n\
30+
\ \"web\": {\n \"uri\": \"https://ai.pydantic.dev\"\
31+
,\n \"title\": \"Pydantic AI\"\n }\n }\n\
32+
\ ],\n \"groundingSupports\": [\n {\n \"\
33+
segment\": {\n \"endIndex\": 131,\n \"text\": \"\
34+
Pydantic AI is a Python agent framework designed to make it less painful to\
35+
\ build production grade applications with Generative AI.\"\n },\n\
36+
\ \"groundingChunkIndices\": [\n 0\n ]\n\
37+
\ }\n ]\n },\n \"urlContextMetadata\": {\n \
38+
\ \"urlMetadata\": [\n {\n \"retrievedUrl\": \"https://ai.pydantic.dev\"\
39+
,\n \"urlRetrievalStatus\": \"URL_RETRIEVAL_STATUS_SUCCESS\"\n\
40+
\ }\n ]\n }\n }\n ],\n \"usageMetadata\": {\n \
41+
\ \"promptTokenCount\": 32,\n \"candidatesTokenCount\": 41,\n \"totalTokenCount\"\
42+
: 2515,\n \"promptTokensDetails\": [\n {\n \"modality\": \"\
43+
TEXT\",\n \"tokenCount\": 32\n }\n ],\n \"toolUsePromptTokenCount\"\
44+
: 2395,\n \"toolUsePromptTokensDetails\": [\n {\n \"modality\"\
45+
: \"TEXT\",\n \"tokenCount\": 2395\n }\n ],\n \"thoughtsTokenCount\"\
46+
: 47\n },\n \"modelVersion\": \"gemini-2.5-flash\",\n \"responseId\": \"\
47+
qgqkaI-iDLrTjMcP0bP24A4\"\n}\n"
48+
headers:
49+
Alt-Svc:
50+
- h3=":443"; ma=2592000,h3-29=":443"; ma=2592000
51+
Content-Type:
52+
- application/json; charset=UTF-8
53+
Date:
54+
- Tue, 19 Aug 2025 05:24:58 GMT
55+
Server:
56+
- scaffolding on HTTPServer2
57+
Server-Timing:
58+
- gfet4t7; dur=9182
59+
Transfer-Encoding:
60+
- chunked
61+
Vary:
62+
- Origin
63+
- X-Origin
64+
- Referer
65+
X-Content-Type-Options:
66+
- nosniff
67+
X-Frame-Options:
68+
- SAMEORIGIN
69+
X-XSS-Protection:
70+
- '0'
71+
content-length:
72+
- '1627'
73+
status:
74+
code: 200
75+
message: OK
76+
version: 1

‎tests/models/cassettes/test_google/test_google_model_web_fetch_tool[True].yaml‎

Lines changed: 76 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,76 @@
1+
interactions:
2+
- request:
3+
body: '{"contents": [{"parts": [{"text": "What is the first sentence on the page
4+
https://ai.pydantic.dev? Reply with only the sentence."}], "role": "user"}],
5+
"systemInstruction": {"parts": [{"text": "You are a helpful chatbot."}], "role":
6+
"user"}, "tools": [{"urlContext": {}}], "generationConfig": {}}'
7+
headers:
8+
accept:
9+
- '*/*'
10+
accept-encoding:
11+
- gzip, deflate
12+
connection:
13+
- keep-alive
14+
content-length:
15+
- '295'
16+
content-type:
17+
- application/json
18+
host:
19+
- generativelanguage.googleapis.com
20+
method: POST
21+
uri: https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash:generateContent
22+
response:
23+
body:
24+
string: "{\n \"candidates\": [\n {\n \"content\": {\n \"parts\"\
25+
: [\n {\n \"text\": \"Pydantic AI is a Python agent framework\
26+
\ designed to make it less painful to build production grade applications\
27+
\ with Generative AI.\"\n }\n ],\n \"role\": \"model\"\
28+
\n },\n \"finishReason\": \"STOP\",\n \"index\": 0,\n \
29+
\ \"groundingMetadata\": {\n \"groundingChunks\": [\n {\n\
30+
\ \"web\": {\n \"uri\": \"https://ai.pydantic.dev\"\
31+
,\n \"title\": \"Pydantic AI\"\n }\n }\n\
32+
\ ],\n \"groundingSupports\": [\n {\n \"\
33+
segment\": {\n \"endIndex\": 131,\n \"text\": \"\
34+
Pydantic AI is a Python agent framework designed to make it less painful to\
35+
\ build production grade applications with Generative AI.\"\n },\n\
36+
\ \"groundingChunkIndices\": [\n 0\n ]\n\
37+
\ }\n ]\n },\n \"urlContextMetadata\": {\n \
38+
\ \"urlMetadata\": [\n {\n \"retrievedUrl\": \"https://ai.pydantic.dev\"\
39+
,\n \"urlRetrievalStatus\": \"URL_RETRIEVAL_STATUS_SUCCESS\"\n\
40+
\ }\n ]\n }\n }\n ],\n \"usageMetadata\": {\n \
41+
\ \"promptTokenCount\": 32,\n \"candidatesTokenCount\": 41,\n \"totalTokenCount\"\
42+
: 2515,\n \"promptTokensDetails\": [\n {\n \"modality\": \"\
43+
TEXT\",\n \"tokenCount\": 32\n }\n ],\n \"toolUsePromptTokenCount\"\
44+
: 2395,\n \"toolUsePromptTokensDetails\": [\n {\n \"modality\"\
45+
: \"TEXT\",\n \"tokenCount\": 2395\n }\n ],\n \"thoughtsTokenCount\"\
46+
: 47\n },\n \"modelVersion\": \"gemini-2.5-flash\",\n \"responseId\": \"\
47+
qgqkaI-iDLrTjMcP0bP24A4\"\n}\n"
48+
headers:
49+
Alt-Svc:
50+
- h3=":443"; ma=2592000,h3-29=":443"; ma=2592000
51+
Content-Type:
52+
- application/json; charset=UTF-8
53+
Date:
54+
- Tue, 19 Aug 2025 05:24:58 GMT
55+
Server:
56+
- scaffolding on HTTPServer2
57+
Server-Timing:
58+
- gfet4t7; dur=9182
59+
Transfer-Encoding:
60+
- chunked
61+
Vary:
62+
- Origin
63+
- X-Origin
64+
- Referer
65+
X-Content-Type-Options:
66+
- nosniff
67+
X-Frame-Options:
68+
- SAMEORIGIN
69+
X-XSS-Protection:
70+
- '0'
71+
content-length:
72+
- '1627'
73+
status:
74+
code: 200
75+
message: OK
76+
version: 1

0 commit comments

Comments
 (0)