Skip to content

Commit a9f2193

Browse files
GuinersGuiners
Guiners
and
Guiners
authored
feat(genai): add new tools samples (#13592)
* model update * model update * model update * model update * code review fix * code review fix * code review fix --------- Co-authored-by: Guiners <[email protected]>
1 parent e8b1683 commit a9f2193

File tree

4 files changed

+257
-0
lines changed

4 files changed

+257
-0
lines changed

‎genai/tools/test_tools_examples.py‎

Lines changed: 18 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,8 +24,11 @@
2424
import tools_enterprise_web_search_with_txt
2525
import tools_func_def_with_txt
2626
import tools_func_desc_with_txt
27+
import tools_google_maps_coordinates_with_txt
2728
import tools_google_maps_with_txt
29+
import tools_google_search_and_urlcontext_with_txt
2830
import tools_google_search_with_txt
31+
import tools_urlcontext_with_txt
2932
import tools_vais_with_txt
3033

3134
os.environ["GOOGLE_GENAI_USE_VERTEXAI"] = "True"
@@ -77,3 +80,18 @@ def test_tools_vais_with_txt() -> None:
7780
datastore = f"projects/{PROJECT_ID}/locations/global/collections/default_collection/dataStores/grounding-test-datastore"
7881
response = tools_vais_with_txt.generate_content(datastore)
7982
assert response
83+
84+
85+
def test_tools_google_maps_coordinates_with_txt() -> None:
86+
response = tools_google_maps_coordinates_with_txt.generate_content()
87+
assert response
88+
89+
90+
def test_tools_urlcontext_with_txt() -> None:
91+
response = tools_urlcontext_with_txt.generate_content()
92+
assert response
93+
94+
95+
def test_tools_google_search_and_urlcontext_with_txt() -> None:
96+
response = tools_google_search_and_urlcontext_with_txt.generate_content()
97+
assert response

‎genai/tools/tools_google_maps_coordinates_with_txt.py‎

Lines changed: 59 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,59 @@
1+
# Copyright 2025 Google LLC
2+
#
3+
# Licensed under the Apache License, Version 2.0 (the "License");
4+
# you may not use this file except in compliance with the License.
5+
# You may obtain a copy of the License at
6+
#
7+
# https://www.apache.org/licenses/LICENSE-2.0
8+
#
9+
# Unless required by applicable law or agreed to in writing, software
10+
# distributed under the License is distributed on an "AS IS" BASIS,
11+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+
# See the License for the specific language governing permissions and
13+
# limitations under the License.
14+
15+
16+
def generate_content() -> str:
17+
# [START googlegenaisdk_tools_google_maps_coordinates_with_txt]
18+
from google import genai
19+
from google.genai.types import (
20+
GenerateContentConfig,
21+
GoogleMaps,
22+
HttpOptions,
23+
Tool,
24+
ToolConfig,
25+
RetrievalConfig,
26+
LatLng
27+
)
28+
29+
client = genai.Client(http_options=HttpOptions(api_version="v1"))
30+
31+
response = client.models.generate_content(
32+
model="gemini-2.5-flash",
33+
contents="Where can I get the best espresso near me?",
34+
config=GenerateContentConfig(
35+
tools=[
36+
# Use Google Maps Tool
37+
Tool(google_maps=GoogleMaps())
38+
],
39+
tool_config=ToolConfig(
40+
retrieval_config=RetrievalConfig(
41+
lat_lng=LatLng( # Pass coordinates for location-aware grounding
42+
latitude=40.7128,
43+
longitude=-74.006
44+
),
45+
language_code="en_US", # Optional: localize Maps results
46+
),
47+
),
48+
),
49+
)
50+
51+
print(response.text)
52+
# Example response:
53+
# 'Here are some of the top-rated places to get espresso near you: ...'
54+
# [END googlegenaisdk_tools_google_maps_coordinates_with_txt]
55+
return response.text
56+
57+
58+
if __name__ == "__main__":
59+
generate_content()

‎genai/tools/tools_google_search_and_urlcontext_with_txt.py‎

Lines changed: 95 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,95 @@
1+
# Copyright 2025 Google LLC
2+
#
3+
# Licensed under the Apache License, Version 2.0 (the "License");
4+
# you may not use this file except in compliance with the License.
5+
# You may obtain a copy of the License at
6+
#
7+
# https://www.apache.org/licenses/LICENSE-2.0
8+
#
9+
# Unless required by applicable law or agreed to in writing, software
10+
# distributed under the License is distributed on an "AS IS" BASIS,
11+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+
# See the License for the specific language governing permissions and
13+
# limitations under the License.
14+
15+
16+
def generate_content() -> str:
17+
# [START googlegenaisdk_tools_google_search_and_urlcontext_with_txt]
18+
from google import genai
19+
from google.genai.types import Tool, GenerateContentConfig, HttpOptions, UrlContext, GoogleSearch
20+
21+
client = genai.Client(http_options=HttpOptions(api_version="v1beta1"))
22+
model_id = "gemini-2.5-flash"
23+
24+
tools = [
25+
Tool(url_context=UrlContext),
26+
Tool(google_search=GoogleSearch),
27+
]
28+
29+
# TODO(developer): Here put your URLs!
30+
url = 'https://www.google.com/search?q=events+in+New+York'
31+
32+
response = client.models.generate_content(
33+
model=model_id,
34+
contents=f"Give me three day events schedule based on {url}. Also let me know what needs to taken care of considering weather and commute.",
35+
config=GenerateContentConfig(
36+
tools=tools,
37+
response_modalities=["TEXT"],
38+
)
39+
)
40+
41+
for each in response.candidates[0].content.parts:
42+
print(each.text)
43+
# Here is a possible three-day event schedule for New York City, focusing on the dates around October 7-9, 2025, along with weather and commute considerations.
44+
#
45+
# ### Three-Day Event Schedule: New York City (October 7-9, 2025)
46+
#
47+
# **Day 1: Tuesday, October 7, 2025 - Art and Culture**
48+
#
49+
# * **Morning (10:00 AM - 1:00 PM):** Visit "Phillips Visual Language: The Art of Irving Penn" at 432 Park Avenue. This exhibition is scheduled to end on this day, offering a last chance to see it.
50+
# * **Lunch (1:00 PM - 2:00 PM):** Grab a quick lunch near Park Avenue.
51+
# * **Afternoon (2:30 PM - 5:30 PM):** Explore the "Lincoln Center Festival of Firsts" at Lincoln Center. This festival runs until October 23rd, offering various performances or exhibits. Check their specific schedule for the day.
52+
# * **Evening (7:00 PM onwards):** Experience a classic Broadway show. Popular options mentioned for October 2025 include "Six The Musical," "Wicked," "Hadestown," or "MJ - The Musical."
53+
#
54+
# **Day 2: Wednesday, October 8, 2025 - Unique Experiences and SoHo Vibes**
55+
#
56+
# * **Morning (11:00 AM - 1:00 PM):** Head to Brooklyn for the "Secret Room at IKEA Brooklyn" at 1 Beard Street. This unique event is scheduled to end on October 9th.
57+
# * **Lunch (1:00 PM - 2:00 PM):** Enjoy lunch in Brooklyn, perhaps exploring local eateries in the area.
58+
# * **Afternoon (2:30 PM - 5:30 PM):** Immerse yourself in the "The Weeknd & Nespresso Samra Origins Vinyl Cafe" at 579 Broadway in SoHo. This pop-up, curated by The Weeknd, combines coffee and music and runs until October 14th.
59+
# * **Evening (6:00 PM onwards):** Explore the vibrant SoHo neighborhood, known for its shopping and dining. You could also consider a dinner cruise to see the illuminated Manhattan skyline and the Statue of Liberty.
60+
#
61+
# **Day 3: Thursday, October 9, 2025 - Film and Scenic Views**
62+
#
63+
# * **Morning (10:00 AM - 1:00 PM):** Attend a screening at the New York Greek Film Expo, which runs until October 12th in New York City.
64+
# * **Lunch (1:00 PM - 2:00 PM):** Have lunch near the film expo's location.
65+
# * **Afternoon (2:30 PM - 5:30 PM):** Take advantage of the pleasant October weather and enjoy outdoor activities. Consider biking along the rivers or through Central Park to admire the early autumn foliage.
66+
# * **Evening (6:00 PM onwards):** Visit an observation deck like the Empire State Building or Top of the Rock for panoramic city views. Afterwards, enjoy dinner in a neighborhood of your choice.
67+
#
68+
# ### Weather and Commute Considerations:
69+
#
70+
# **Weather in Early October:**
71+
#
72+
# * **Temperatures:** Expect mild to cool temperatures. Average daily temperatures in early October range from 10°C (50°F) to 18°C (64°F), with occasional warmer days reaching the mid-20s°C (mid-70s°F). Evenings can be quite chilly.
73+
# * **Rainfall:** October has a higher chance of rainfall compared to other months, with an average of 33mm and a 32% chance of rain on any given day.
74+
# * **Sunshine:** You can generally expect about 7 hours of sunshine per day.
75+
# * **What to Pack:** Pack layers! Bring a light jacket or sweater for the daytime, and a warmer coat for the evenings. An umbrella or a light raincoat is highly recommended due to the chance of showers. Comfortable walking shoes are a must for exploring the city.
76+
#
77+
# **Commute in New York City:**
78+
#
79+
# * **Public Transportation is Key:** The subway is generally the fastest and most efficient way to get around New York City, especially during the day. Buses are good for East-West travel, but can be slower due to traffic.
80+
# * **Using Apps:** Utilize Google Maps or official MTA apps to plan your routes and check for real-time service updates. The subway runs 24/7, but expect potential delays or changes to routes during nights and weekends due to maintenance.
81+
# * **Rush Hour:** Avoid subway and commuter train travel during peak rush hours (8 AM - 10 AM and 5 PM - 7 PM) if possible, as trains can be extremely crowded.
82+
# * **Subway Etiquette:** When on the subway, stand to the side of the doors to let people exit before boarding, and move to the center of the car to make space. Hold onto a pole or seat, and remove your backpack to free up space.
83+
# * **Transfers:** Subway fare is $2.90 per ride, and you get one free transfer between the subway and bus within a two-hour window.
84+
# * **Walking:** New York City is very walkable. If the weather is pleasant, walking between nearby attractions is an excellent way to see the city.
85+
# * **Taxis/Ride-sharing:** Uber, Lyft, and Curb (for NYC taxis) are available, but driving in the city is generally discouraged due to traffic and parking difficulties.
86+
# * **Allow Extra Time:** Always factor in an additional 20-30 minutes for travel time, as delays can occur.
87+
88+
# get URLs retrieved for context
89+
print(response.candidates[0].url_context_metadata)
90+
# [END googlegenaisdk_tools_google_search_and_urlcontext_with_txt]
91+
return response.text
92+
93+
94+
if __name__ == "__main__":
95+
generate_content()

‎genai/tools/tools_urlcontext_with_txt.py‎

Lines changed: 85 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,85 @@
1+
# Copyright 2025 Google LLC
2+
#
3+
# Licensed under the Apache License, Version 2.0 (the "License");
4+
# you may not use this file except in compliance with the License.
5+
# You may obtain a copy of the License at
6+
#
7+
# https://www.apache.org/licenses/LICENSE-2.0
8+
#
9+
# Unless required by applicable law or agreed to in writing, software
10+
# distributed under the License is distributed on an "AS IS" BASIS,
11+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+
# See the License for the specific language governing permissions and
13+
# limitations under the License.
14+
15+
16+
def generate_content() -> str:
17+
# [START googlegenaisdk_tools_urlcontext_with_txt]
18+
from google import genai
19+
from google.genai.types import Tool, GenerateContentConfig, HttpOptions, UrlContext
20+
21+
client = genai.Client(http_options=HttpOptions(api_version="v1"))
22+
model_id = "gemini-2.5-flash"
23+
24+
url_context_tool = Tool(
25+
url_context=UrlContext
26+
)
27+
28+
# TODO(developer): Here put your URLs
29+
url1 = "https://cloud.google.com/vertex-ai/docs/generative-ai/start"
30+
url2 = "https://cloud.google.com/docs/overview"
31+
32+
response = client.models.generate_content(
33+
model=model_id,
34+
contents=f"Compare the content, purpose, and audiences of {url1} and {url2}.",
35+
config=GenerateContentConfig(
36+
tools=[url_context_tool],
37+
response_modalities=["TEXT"],
38+
)
39+
)
40+
41+
for each in response.candidates[0].content.parts:
42+
print(each.text)
43+
# Gemini 2.5 Pro and Gemini 2.5 Flash are both advanced models offered by Google AI, but they are optimized for different use cases.
44+
#
45+
# Here's a comparison:
46+
#
47+
# **Gemini 2.5 Pro**
48+
# * **Description**: This is Google's most advanced model, described as a "state-of-the-art thinking model". It excels at reasoning over complex problems in areas like code, mathematics, and STEM, and can analyze large datasets, codebases, and documents using a long context window.
49+
# * **Input Data Types**: It supports audio, images, video, text, and PDF inputs.
50+
# * **Output Data Types**: It produces text outputs.
51+
# * **Token Limits**: It has an input token limit of 1,048,576 and an output token limit of 65,536.
52+
# * **Supported Capabilities**: Gemini 2.5 Pro supports Batch API, Caching, Code execution, Function calling, Search grounding, Structured outputs, Thinking, and URL context.
53+
# * **Knowledge Cutoff**: January 2025.
54+
#
55+
# **Gemini 2.5 Flash**
56+
# * **Description**: Positioned as "fast and intelligent," Gemini 2.5 Flash is highlighted as Google's best model in terms of price-performance, offering well-rounded capabilities. It is ideal for large-scale processing, low-latency, high-volume tasks that require thinking, and agentic use cases.
57+
# * **Input Data Types**: It supports text, images, video, and audio inputs.
58+
# * **Output Data Types**: It produces text outputs.
59+
# * **Token Limits**: Similar to Pro, it has an input token limit of 1,048,576 and an output token limit of 65,536.
60+
# * **Supported Capabilities**: Gemini 2.5 Flash supports Batch API, Caching, Code execution, Function calling, Search grounding, Structured outputs, Thinking, and URL context.
61+
# * **Knowledge Cutoff**: January 2025.
62+
#
63+
# **Key Differences and Similarities:**
64+
#
65+
# * **Primary Focus**: Gemini 2.5 Pro is geared towards advanced reasoning and in-depth analysis of complex problems and large documents. Gemini 2.5 Flash, on the other hand, is optimized for efficiency, scale, and high-volume, low-latency applications, making it a strong choice for price-performance sensitive scenarios.
66+
# * **Input Modalities**: Both models handle various input types including text, images, video, and audio. Gemini 2.5 Pro explicitly lists PDF as an input type, while Gemini 2.5 Flash lists text, images, video, audio.
67+
# * **Technical Specifications (for primary stable versions)**: Both models share the same substantial input and output token limits (1,048,576 input and 65,536 output). They also support a very similar set of core capabilities, including code execution, function calling, and URL context. Neither model supports audio generation, image generation, or Live API in their standard stable versions.
68+
# * **Knowledge Cutoff**: Both models have a knowledge cutoff of January 2025.
69+
#
70+
# In essence, while both models are powerful and capable, Gemini 2.5 Pro is designed for maximum performance in complex reasoning tasks, whereas Gemini 2.5 Flash prioritizes cost-effectiveness and speed for broader, high-throughput applications.
71+
# get URLs retrieved for context
72+
print(response.candidates[0].url_context_metadata)
73+
# url_metadata=[UrlMetadata(
74+
# retrieved_url='https://ai.google.dev/gemini-api/docs/models#gemini-2.5-flash',
75+
# url_retrieval_status=<UrlRetrievalStatus.URL_RETRIEVAL_STATUS_SUCCESS: 'URL_RETRIEVAL_STATUS_SUCCESS'>
76+
# ), UrlMetadata(
77+
# retrieved_url='https://ai.google.dev/gemini-api/docs/models#gemini-2.5-pro',
78+
# url_retrieval_status=<UrlRetrievalStatus.URL_RETRIEVAL_STATUS_SUCCESS: 'URL_RETRIEVAL_STATUS_SUCCESS'>
79+
# )]
80+
# [END googlegenaisdk_tools_urlcontext_with_txt]
81+
return response.text
82+
83+
84+
if __name__ == "__main__":
85+
generate_content()

0 commit comments

Comments
 (0)