Update samples for async session changes and remove exp suffix from gemini model

PiperOrigin-RevId: 759364113
This commit is contained in:
Selcuk Gun
2025-05-15 17:11:32 -07:00
committed by Copybara-Service
parent 04820cb0a7
commit 01cf186299
7 changed files with 8 additions and 112 deletions

View File

@@ -28,7 +28,7 @@ def roll_die(sides: int) -> int:
roll_agent = LlmAgent(
name="roll_agent",
description="Handles rolling dice of different sizes.",
model="gemini-2.0-flash-exp",
model="gemini-2.0-flash",
instruction="""
You are responsible for rolling dice based on the user's request.
When asked to roll a die, you must call the roll_die tool with the number of sides as an integer.
@@ -69,7 +69,7 @@ def check_prime(nums: list[int]) -> str:
prime_agent = LlmAgent(
name="prime_agent",
description="Handles checking if numbers are prime.",
model="gemini-2.0-flash-exp",
model="gemini-2.0-flash",
instruction="""
You are responsible for checking whether numbers are prime.
When asked to check primes, you must call the check_prime tool with a list of integers.