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.
# See the License for the specific language governing permissions and
# limitations under the License.
from __future__ import annotations
import contextlib
@ -179,7 +181,9 @@ class Gemini(BaseLlm):
api_version = 'v1beta1'
# use default api version for vertex
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:
# use v1alpha for ml_dev