接入 Google Agent Platform 并开放已验证的 AI Studio embedding

2026-10-09 09:02:00 -07:00
parent db09881b2d
commit d3a487f86a
5 changed files with 27 additions and 0 deletions
@@ -34,6 +34,11 @@ def build_config(catalog, env, exists):
    if env.get('GOOGLE_AI_STUDIO_API_KEY'):
        for alias, model in catalog['google_ai_studio'].items():
            add(alias, 'gemini/' + model, 'GOOGLE_AI_STUDIO_API_KEY')
+    if env.get('VERTEXAI_ENABLED', '').lower() == 'true' and env.get('VERTEXAI_PROJECT'):
+        for alias, model in catalog.get('google_vertex', {}).items():
+            add(alias, 'vertex_ai/' + model,
+                vertex_project='os.environ/VERTEXAI_PROJECT',
+                vertex_location='os.environ/VERTEXAI_LOCATION')
    for provider, token_var, default_dir in [
        ('chatgpt', 'CHATGPT_TOKEN_DIR', '/tokens/chatgpt'),
        ('xai', 'XAI_OAUTH_TOKEN_DIR', '/tokens/xai'),