Add conditional external vision projector discovery to profiles

This commit is contained in:
Mikei386 committed 2026-09-28 22:12:50 +02:00
1 parent 0dbca8cb46
commit 6255f0b8ea
5 files changed
+72 -6

No files matched your search

+6
View File
@@ -41,6 +41,12 @@ class CatalogTests(unittest.TestCase):
with tempfile.TemporaryDirectory() as d,patch('catalog.metadata',side_effect=[[],[item]]) as fetch:
result=Catalog(d).search('model.gguf','chat')
self.assertEqual(result['models'][0]['repo'],item['id']);self.assertIn('search=model&',fetch.call_args_list[0].args[0]);self.assertIn('filter=gguf',fetch.call_args.args[0])
def test_projector_search_includes_vision_repositories(self):
with tempfile.TemporaryDirectory() as d,patch('catalog.metadata',return_value=[{'id':'maker/vision','pipeline_tag':'image-text-to-text'}]) as fetch:
result=Catalog(d).search('vision','chat',purpose='projector')
self.assertEqual(result['models'][0]['repo'],'maker/vision')
self.assertIn('filter=gguf',fetch.call_args.args[0])
with self.assertRaises(ValueError):Catalog(d).search('vision','chat',purpose='invalid')
def test_download_integrity_and_library(self):
for valid in (True,False):
with self.subTest(valid=valid),tempfile.TemporaryDirectory() as directory: