Skip to content

Commit 907f390

Browse files
committed
standardize google provider
1 parent 85b7535 commit 907f390

File tree

2 files changed

+4
-81
lines changed

2 files changed

+4
-81
lines changed

stagehand/llm/client.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -94,6 +94,10 @@ def create_response(
9494
"No model specified for chat completion (neither default_model nor model argument)."
9595
)
9696

97+
# Standardize gemini provider to google
98+
if completion_model.startswith("google/"):
99+
completion_model = completion_model.replace("google/", "gemini/")
100+
97101
# Prepare arguments directly from kwargs
98102
params = {
99103
"model": completion_model,

tests/functional/test_sync_client.py

Lines changed: 0 additions & 81 deletions
This file was deleted.

0 commit comments

Comments
 (0)