接入 Google Agent Platform 并开放已验证的 AI Studio embedding
This commit is contained in:
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user