michael/Athena-Deck
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Show endpoint profile checkboxes and enforce exclusive image selection

Browse Source
This commit is contained in:
Mikei386
2026-09-28 21:58:03 +02:00
parent 37a757fd9f
commit a36a151c94
5 changed files with 27 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test_endpoint.py
+11
View File
@@ -55,6 +55,17 @@ class EndpointTests(unittest.TestCase):
with self.assertRaises(ValueError):self.enable('audio')
self.assertEqual(self.request('/v1/audio/speech',{})[0],501)
self.record['api_token_hash']=hashlib.sha256(b'B'*32).hexdigest();self.assertEqual(self.request()[0],401);self.assertEqual(self.request(token='B'*32)[0],200)
def test_single_image_selection_preserves_multiple_llms(self):
self.rows.append(dict(self.rows[2],id='image2',name='image2'))
self.enable('alpha');self.enable('beta');self.enable('image');self.enable('image2')
self.assertEqual(set(self.ep.config['enabled_profiles']),{'alpha','beta','image2'})
self.assertEqual({r['id'] for r in self.request()[1]['data']},{'alpha','beta','image2'})
self.ep.enable(dict(id='image2',enabled=False));self.assertEqual(set(self.ep.config['enabled_profiles']),{'alpha','beta'})
def test_invalid_image_selection_keeps_previous(self):
self.rows.append(dict(self.rows[2],id='badimage',name='badimage',runnable=False,blockers=['missing']))
self.enable('image')
with self.assertRaises(ValueError):self.enable('badimage')
self.assertEqual(self.ep.config['enabled_profiles'],['image'])
def test_profile_switch_and_sampling_defaults(self):
for name in ('alpha','beta'):self.enable(name)
for name in ('alpha','beta'):