Add conditional external vision projector discovery to profiles
This commit is contained in:
1 parent
0dbca8cb46
commit
6255f0b8ea
5 files changed
+72
-6
No files matched your search
@@ -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:
|
||||
|
||||
Reference in new issue
Block a user