File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -72,7 +72,7 @@ class GeminiClient:
72
72
"max_output_tokens" : "max_output_tokens" ,
73
73
}
74
74
75
- def _initialize_vartexai (self , ** params ):
75
+ def _initialize_vertexai (self , ** params ):
76
76
if "google_application_credentials" in params :
77
77
# Path to JSON Keyfile
78
78
os .environ ["GOOGLE_APPLICATION_CREDENTIALS" ] = params ["google_application_credentials" ]
@@ -106,7 +106,7 @@ def __init__(self, **kwargs):
106
106
self .api_key = os .getenv ("GOOGLE_API_KEY" )
107
107
if self .api_key is None :
108
108
self .use_vertexai = True
109
- self ._initialize_vartexai (** kwargs )
109
+ self ._initialize_vertexai (** kwargs )
110
110
else :
111
111
self .use_vertexai = False
112
112
else :
@@ -142,7 +142,7 @@ def get_usage(response) -> Dict:
142
142
143
143
def create (self , params : Dict ) -> ChatCompletion :
144
144
if self .use_vertexai :
145
- self ._initialize_vartexai (** params )
145
+ self ._initialize_vertexai (** params )
146
146
else :
147
147
assert ("project_id" not in params ) and (
148
148
"location" not in params
You can’t perform that action at this time.
0 commit comments