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

This commit is contained in:
Codex
2026-10-09 09:02:00 -07:00
parent db09881b2d
commit d3a487f86a
5 changed files with 27 additions and 0 deletions
+10
View File
@@ -9,6 +9,7 @@ class ConfigTests(unittest.TestCase):
'zenmux': {'claude-sonnet-latest': 'anthropic/claude-sonnet-5.5'},
'openrouter': {'glm-latest': 'z-ai/glm-5.3-prime'},
'google_ai_studio': {'gemini-latest': 'gemini-test'},
'google_vertex': {'gemini-vertex': 'gemini-vertex-test'},
'chatgpt': {'codex-latest': 'gpt-test'},
'xai': {'grok-latest': 'grok-test'},
}
@@ -44,6 +45,15 @@ class ConfigTests(unittest.TestCase):
path.write_text('{"access_token": "fixture-only"}')
self.assertTrue(oauth_ready(path))
def test_vertex_requires_explicit_enable_and_project(self):
for env in ({'VERTEXAI_PROJECT': 'test-project'}, {'VERTEXAI_ENABLED': 'true'}):
self.assertEqual(build_config(self.catalog, env, lambda _: False)['model_list'], [])
routes = build_config(self.catalog, {'VERTEXAI_ENABLED': 'true',
'VERTEXAI_PROJECT': 'test-project'}, lambda _: False)['model_list']
self.assertEqual(routes[0]['litellm_params']['model'], 'vertex_ai/gemini-vertex-test')
self.assertEqual(routes[0]['litellm_params']['vertex_project'], 'os.environ/VERTEXAI_PROJECT')
self.assertNotIn('api_key', routes[0]['litellm_params'])
if __name__ == '__main__':
unittest.main()