chore: formats google_llm.py.

PiperOrigin-RevId: 759803357
This commit is contained in:
Wei Sun (Jack) 2025-05-16 16:49:09 -07:00 committed by Copybara-Service
parent 03f167c1d3
commit 186b5de826

View File

@ -11,6 +11,8 @@
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
from __future__ import annotations from __future__ import annotations
import contextlib import contextlib
@ -179,7 +181,9 @@ class Gemini(BaseLlm):
api_version = 'v1beta1' api_version = 'v1beta1'
# use default api version for vertex # use default api version for vertex
return Client( return Client(
http_options=types.HttpOptions(headers=self._tracking_headers,api_version=api_version) http_options=types.HttpOptions(
headers=self._tracking_headers, api_version=api_version
)
) )
else: else:
# use v1alpha for ml_dev # use v1alpha for ml_dev