Skip to content

chore: Update function calling examples to use gemini-3-pro-preview for go and python #1799

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
copybara-service wants to merge 1 commit into
base: main
Choose a base branch
from
Open

chore: Update function calling examples to use gemini-3-pro-preview for go and python #1799

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
180 changes: 90 additions & 90 deletions google/genai/tests/models/test_function_call_streaming.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -154,7 +154,7 @@ def test_streaming_with_python_native_no_afc_config(client):
return
with pytest.raises(ValueError) as e:
for chunk in client.models.generate_content_stream(
model='gemini-2.5-pro',
model='gemini-3-pro-preview',
contents=generate_content_prompt,
config=types.GenerateContentConfig(
tools=[
Expand All @@ -179,55 +179,36 @@ def test_streaming_with_python_afc_disabled_false(client):
return
with pytest.raises(ValueError) as e:
for chunk in client.models.generate_content_stream(
model='gemini-2.5-pro',
contents='What is the price of GOOG? And what is the weather in Boston?',
config=types.GenerateContentConfig(
tools=[
test_generate_content_tools.get_weather,
test_generate_content_tools.get_stock_price,
],
automatic_function_calling=types.AutomaticFunctionCallingConfig(
disable=False,
),
tool_config=types.ToolConfig(
function_calling_config={
'stream_function_call_arguments': True,
}
),
),
):
pass
assert 'not compatible with automatic function calling (AFC)' in str(e.value)


def test_streaming_with_json_parameters_without_history(client):
"""Tests streaming function calls with FunctionDeclaration withJSON parameters."""

with pytest_helper.exception_if_mldev(client, ValueError):
for chunk in client.models.generate_content_stream(
model='gemini-2.5-pro',
contents=generate_content_prompt,
model='gemini-3-pro-preview',
contents=(
'What is the price of GOOG? And what is the weather in Boston?'
),
config=types.GenerateContentConfig(
tools=[{'function_declarations': json_function_declarations}],
tools=[
test_generate_content_tools.get_weather,
test_generate_content_tools.get_stock_price,
],
automatic_function_calling=types.AutomaticFunctionCallingConfig(
disable=False,
),
tool_config=types.ToolConfig(
function_calling_config={
'stream_function_call_arguments': True,
}
),
),
):
assert chunk is not None
assert chunk.candidates is not None
assert chunk.candidates[0].content is not None
assert chunk.candidates[0].content.parts is not None
pass
assert 'not compatible with automatic function calling (AFC)' in str(e.value)


def test_streaming_with_json_parameters_with_history(client):
def test_streaming_with_json_parameters_without_history(client):
"""Tests streaming function calls with FunctionDeclaration withJSON parameters."""

with pytest_helper.exception_if_mldev(client, ValueError):
for chunk in client.models.generate_content_stream(
model='gemini-2.5-pro',
contents=previous_generate_content_history,
model='gemini-3-pro-preview',
contents=generate_content_prompt,
config=types.GenerateContentConfig(
tools=[{'function_declarations': json_function_declarations}],
tool_config=types.ToolConfig(
Expand All @@ -242,12 +223,13 @@ def test_streaming_with_json_parameters_with_history(client):
assert chunk.candidates[0].content is not None
assert chunk.candidates[0].content.parts is not None


@pytest.mark.asyncio
async def test_streaming_with_json_parameters_async(client):
"""Tests streaming function calls with FunctionDeclaration withJSON parameters."""
with pytest_helper.exception_if_mldev(client, ValueError):
async for chunk in await client.aio.models.generate_content_stream(
model='gemini-2.5-pro',
model='gemini-3-pro-preview',
contents=generate_content_prompt,
config=types.GenerateContentConfig(
tools=[{'function_declarations': json_function_declarations}],
Expand All @@ -268,7 +250,7 @@ def test_streaming_with_gemini_parameters_without_history(client):
"""Tests streaming function calls with FunctionDeclaration withJSON parameters."""
with pytest_helper.exception_if_mldev(client, ValueError):
for chunk in client.models.generate_content_stream(
model='gemini-2.5-pro',
model='gemini-3-pro-preview',
contents=generate_content_prompt,
config=types.GenerateContentConfig(
tools=[{
Expand All @@ -291,7 +273,7 @@ def test_streaming_with_gemini_parameters_with_response(client):
with pytest_helper.exception_if_mldev(client, ValueError):
streaming_function_call_content = []
for chunk in client.models.generate_content_stream(
model='gemini-2.5-pro',
model='gemini-3-pro-preview',
contents=[
types.Content(
role='user',
Expand Down Expand Up @@ -335,7 +317,7 @@ def test_streaming_with_gemini_parameters_with_response(client):
)

for chunk in client.models.generate_content_stream(
model='gemini-2.5-pro',
model='gemini-3-pro-preview',
contents=streaming_function_call_content,
config=types.GenerateContentConfig(
tools=[{'function_declarations': json_function_declarations}],
Expand All @@ -348,41 +330,41 @@ def test_streaming_with_gemini_parameters_with_response(client):
):
pass

def test_streaming_with_gemini_parameters_with_history(client):
"""Tests streaming function calls with FunctionDeclaration withJSON parameters."""
with pytest_helper.exception_if_mldev(client, ValueError):
for chunk in client.models.generate_content_stream(
model='gemini-2.5-pro',
contents=previous_generate_content_history,
config=types.GenerateContentConfig(
tools=[{
'function_declarations': gemini_function_declarations
}],
tool_config=types.ToolConfig(
function_calling_config={
'stream_function_call_arguments': True,
}
),
),
):
assert chunk is not None
assert chunk.candidates is not None
assert chunk.candidates[0].content is not None
assert chunk.candidates[0].content.parts is not None

def test_chat_streaming_with_json_parameters_with_history(client):
"""Tests streaming function calls with FunctionDeclaration withJSON parameters."""
with pytest_helper.exception_if_mldev(client, ValueError):
messages = [
"""
get the current weather in boston in celsius, the country is U.S., the purpose is to know what to wear today.
""",
"""
get the current weather in new brunswick in celsius, the country is U.S., the purpose is to know what to prepare an event today.
""",
test_parts = [
types.Part(
text=(
'get the current weather in boston in celsius, the'
' country should be US, the purpose is to know'
' what to wear today?'
)
),
types.Part.from_function_response(
name='get_current_weather',
response={
'temperature': 21,
'unit': 'C',
},
),
types.Part(
text=(
'get the current weather in new brunswick in celsius, the'
' country should be US, the purpose is to know'
' what to prepare an event today?'
)
),
types.Part.from_function_response(
name='get_current_weather',
response={
'temperature': 21,
'unit': 'C',
},
),
]
chat = client.chats.create(
model='gemini-2.5-pro',
model='gemini-3-pro-preview',
history=previous_generate_content_history,
config=types.GenerateContentConfig(
tools=[{
Expand All @@ -395,7 +377,7 @@ def test_chat_streaming_with_json_parameters_with_history(client):
),
),
)
for message in messages:
for message in test_parts:
result = chat.send_message_stream(message)
for chunk in result:
assert chunk is not None
Expand All @@ -409,17 +391,39 @@ def test_chat_streaming_with_json_parameters_with_history(client):
@pytest.mark.asyncio
async def test_chat_streaming_with_json_parameters_with_history_async(client):
"""Tests streaming function calls with FunctionDeclaration withJSON parameters."""
messages = [
"""
get the current weather in boston in celsius, the country is U.S., the purpose is to know what to wear today.
""",
"""
get the current weather in new brunswick in celsius, the country is U.S., the purpose is to know what to prepare an event today.
""",
test_parts = [
types.Part(
text=(
'get the current weather in boston in celsius, the'
' country should be US, the purpose is to know'
' what to wear today?'
)
),
types.Part.from_function_response(
name='get_current_weather',
response={
'temperature': 21,
'unit': 'C',
},
),
types.Part(
text=(
'get the current weather in new brunswick in celsius, the'
' country should be US, the purpose is to know'
' what to prepare an event today?'
)
),
types.Part.from_function_response(
name='get_current_weather',
response={
'temperature': 21,
'unit': 'C',
},
),
]
with pytest_helper.exception_if_mldev(client, ValueError):
chat = client.aio.chats.create(
model='gemini-2.5-pro',
model='gemini-3-pro-preview',
history=previous_generate_content_history,
config=types.GenerateContentConfig(
tools=[{'function_declarations': gemini_function_declarations}],
Expand All @@ -430,13 +434,9 @@ async def test_chat_streaming_with_json_parameters_with_history_async(client):
),
),
)
async for chunk in await chat.send_message_stream(messages[0]):
assert chunk is not None
assert chunk.candidates is not None
assert chunk.candidates[0].content is not None
assert chunk.candidates[0].content.parts is not None
async for chunk in await chat.send_message_stream(messages[1]):
assert chunk is not None
assert chunk.candidates is not None
assert chunk.candidates[0].content is not None
assert chunk.candidates[0].content.parts is not None
for message in test_parts:
async for chunk in await chat.send_message_stream(message):
assert chunk is not None
assert chunk.candidates is not None
assert chunk.candidates[0].content is not None
assert chunk.candidates[0].content.parts is not None