Add Hermes auxiliary and subscription model routes
This commit is contained in:
@@ -83,6 +83,14 @@ class ConfigTests(unittest.TestCase):
|
||||
build_config(catalog, {'ZENMUX_SUBSCRIPTION_API_KEY': 'a',
|
||||
'OPENROUTER_API_KEY': 'b'}, lambda _: False)
|
||||
|
||||
def test_zenmux_chat_keeps_existing_data_destination(self):
|
||||
catalog = {**self.catalog, 'zenmux_chat': {'gemini-flash-zenmux': 'google/gemini-3.8-flash'}}
|
||||
config = build_config(catalog, {'ZENMUX_SUBSCRIPTION_API_KEY': 'a'}, lambda _: False)
|
||||
route = next(r for r in config['model_list'] if r['model_name'] == 'gemini-flash-zenmux')
|
||||
self.assertEqual(route['litellm_params']['model'], 'openai/google/gemini-3.8-flash')
|
||||
self.assertEqual(route['litellm_params']['api_base'], 'https://zenmux.ai/api/v1')
|
||||
self.assertNotIn('gemini-flash-zenmux-payg', [r['model_name'] for r in config['model_list']])
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
||||
|
||||
Reference in New Issue
Block a user