| # Copyright 2025 Google LLC | |
| # | |
| # Licensed under the Apache License, Version 2.0 (the "License"); | |
| # you may not use this file except in compliance with the License. | |
| # You may obtain a copy of the License at | |
| # | |
| # http://www.apache.org/licenses/LICENSE-2.0 | |
| # | |
| # Unless required by applicable law or agreed to in writing, software | |
| # distributed under the License is distributed on an "AS IS" BASIS, | |
| # 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. | |
| import os | |
| # Only use Google Cloud auth if Vertex AI is enabled | |
| use_vertex_ai = os.environ.get("GOOGLE_GENAI_USE_VERTEXAI", "TRUE").upper() == "TRUE" | |
| if use_vertex_ai: | |
| try: | |
| import google.auth | |
| _, project_id = google.auth.default() | |
| os.environ.setdefault("GOOGLE_CLOUD_PROJECT", project_id) | |
| os.environ["GOOGLE_CLOUD_LOCATION"] = "global" | |
| os.environ.setdefault("GOOGLE_GENAI_USE_VERTEXAI", "True") | |
| except Exception: | |
| # If Google Cloud auth fails, fall back to API key mode | |
| os.environ.setdefault("GOOGLE_GENAI_USE_VERTEXAI", "False") | |
| else: | |
| # Using Google AI Studio API key, no Google Cloud auth needed | |
| os.environ.setdefault("GOOGLE_GENAI_USE_VERTEXAI", "False") | |
| # Only import agent if not in standalone mode (e.g., when using hybrid_agent directly) | |
| if not os.environ.get("SKIP_AGENT_IMPORT", ""): | |
| from . import agent # noqa: E402, F401 | |