Skip to content

Commit 2887e07

Browse files
sugatorayHk669
andauthored
fix: update method name in GeminiClient (microsoft#3007)
- change from `_initialize_vartexai` to `_initialize_vertexai` Co-authored-by: HRUSHIKESH DOKALA <[email protected]>
1 parent d1d4b10 commit 2887e07

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

autogen/oai/gemini.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,7 @@ class GeminiClient:
7272
"max_output_tokens": "max_output_tokens",
7373
}
7474

75-
def _initialize_vartexai(self, **params):
75+
def _initialize_vertexai(self, **params):
7676
if "google_application_credentials" in params:
7777
# Path to JSON Keyfile
7878
os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = params["google_application_credentials"]
@@ -106,7 +106,7 @@ def __init__(self, **kwargs):
106106
self.api_key = os.getenv("GOOGLE_API_KEY")
107107
if self.api_key is None:
108108
self.use_vertexai = True
109-
self._initialize_vartexai(**kwargs)
109+
self._initialize_vertexai(**kwargs)
110110
else:
111111
self.use_vertexai = False
112112
else:
@@ -142,7 +142,7 @@ def get_usage(response) -> Dict:
142142

143143
def create(self, params: Dict) -> ChatCompletion:
144144
if self.use_vertexai:
145-
self._initialize_vartexai(**params)
145+
self._initialize_vertexai(**params)
146146
else:
147147
assert ("project_id" not in params) and (
148148
"location" not in params

0 commit comments

Comments
 (0)