@@ -14,7 +14,7 @@ class CityLocation(BaseModel):
14
14
country: str
15
15
16
16
17
- agent = Agent(' gemini-1.5-flash' , result_type = CityLocation)
17
+ agent = Agent(' google-gla: gemini-1.5-flash' , result_type = CityLocation)
18
18
result = agent.run_sync(' Where were the olympics held in 2012?' )
19
19
print (result.data)
20
20
# > city='London' country='United Kingdom'
@@ -129,7 +129,7 @@ class InvalidRequest(BaseModel):
129
129
130
130
Response = Union[Success, InvalidRequest]
131
131
agent: Agent[DatabaseConn, Response] = Agent(
132
- ' gemini-1.5-flash' ,
132
+ ' google-gla: gemini-1.5-flash' ,
133
133
result_type = Response, # type: ignore
134
134
deps_type = DatabaseConn,
135
135
system_prompt = ' Generate PostgreSQL flavored SQL queries based on user input.' ,
@@ -171,7 +171,7 @@ Example of streamed text result:
171
171
``` python {title="streamed_hello_world.py" line_length="120"}
172
172
from pydantic_ai import Agent
173
173
174
- agent = Agent(' gemini-1.5-flash' ) # (1)!
174
+ agent = Agent(' google-gla: gemini-1.5-flash' ) # (1)!
175
175
176
176
177
177
async def main ():
@@ -197,7 +197,7 @@ We can also stream text as deltas rather than the entire text in each item:
197
197
``` python {title="streamed_delta_hello_world.py"}
198
198
from pydantic_ai import Agent
199
199
200
- agent = Agent(' gemini-1.5-flash' )
200
+ agent = Agent(' google-gla: gemini-1.5-flash' )
201
201
202
202
203
203
async def main ():
0 commit comments