Skip to content

Commit 21231bf

Browse files
authored
Remove async from _get_model (#893)
1 parent 9012980 commit 21231bf

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

pydantic_ai_slim/pydantic_ai/agent.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -275,7 +275,7 @@ async def main():
275275
"""
276276
if infer_name and self.name is None:
277277
self._infer_name(inspect.currentframe())
278-
model_used = await self._get_model(model)
278+
model_used = self._get_model(model)
279279

280280
deps = self._get_deps(deps)
281281
new_message_index = len(message_history) if message_history else 0
@@ -520,7 +520,7 @@ async def main():
520520
# f_back because `asynccontextmanager` adds one frame
521521
if frame := inspect.currentframe(): # pragma: no branch
522522
self._infer_name(frame.f_back)
523-
model_used = await self._get_model(model)
523+
model_used = self._get_model(model)
524524

525525
deps = self._get_deps(deps)
526526
new_message_index = len(message_history) if message_history else 0
@@ -971,7 +971,7 @@ def _register_tool(self, tool: Tool[AgentDepsT]) -> None:
971971

972972
self._function_tools[tool.name] = tool
973973

974-
async def _get_model(self, model: models.Model | models.KnownModelName | None) -> models.Model:
974+
def _get_model(self, model: models.Model | models.KnownModelName | None) -> models.Model:
975975
"""Create a model configured for this agent.
976976
977977
Args:

0 commit comments

Comments
 (0)