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

Route stable athena-image alias to selected image profile

Browse Source
This commit is contained in:
Mikei386
2026-09-29 09:36:56 +02:00
parent baad2b11a5
commit f337ade65a
4 changed files with 27 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test_endpoint.py
+12 -1
View File
@@ -59,7 +59,7 @@ class EndpointTests(unittest.TestCase):
self.rows[-1].update(runnable=True,blockers=[])
self.rows.append(dict(id='stt',name='stt',kind='stt',runnable=True,blockers=[],parameters={},updated_at=1))
for name in ('alpha','image','audio','stt'):self.enable(name)
for route,expected in [('/v1/models','alpha'),('/v1/images/models','image'),('/v1/audio/speech/models','audio'),('/v1/audio/transcriptions/models','stt')]:
for route,expected in [('/v1/models','alpha'),('/v1/images/models','athena-image'),('/v1/audio/speech/models','audio'),('/v1/audio/transcriptions/models','stt')]:
self.assertEqual([p['id'] for p in self.request(route)[1]['data']],[expected])
self.assertEqual(self.request(route,token='bad')[0],401)
self.rows[-1]['runnable']=False
@@ -97,6 +97,17 @@ class EndpointTests(unittest.TestCase):
self.assertEqual(set(self.ep.config['enabled_profiles']),{'alpha','beta','image2'})
self.assertEqual({r['id'] for r in self.request()[1]['data']},{'alpha','beta'})
self.ep.enable(dict(id='image2',enabled=False));self.assertEqual(set(self.ep.config['enabled_profiles']),{'alpha','beta'})
def test_image_alias_follows_selection_and_profile_size(self):
self.rows[2]['parameters']['height']=512
self.rows.append(dict(self.rows[2],id='image2',name='image2'))
self.ep.images.start.return_value={'id':'job'};self.ep.images.status.return_value={'job':{'id':'job','state':'complete'}};self.ep.images.image.return_value=b'png'
self.assertEqual(self.request('/v1/images/generations',dict(model='athena-image',prompt='synthetic'))[0],404)
for name in ('image','image2'):
self.enable(name)
status,result=self.request('/v1/images/generations',dict(model='athena-image',prompt='synthetic',size='1536x1024'))
self.assertEqual(status,200);self.assertEqual(result['athena_deck']['size'],'512x512');self.assertEqual(self.ep.images.start.call_args.args[0],name)
self.assertEqual([m['id'] for m in self.request('/v1/images/models')[1]['data']],['athena-image'])
self.assertEqual(self.request('/v1/images/generations',dict(model='athena-image',prompt='synthetic',size=[]))[0],400)
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')