Find direct Hub repositories and untagged conversational GGUF models

test_catalog.py

@@ -30,6 +30,17 @@ class CatalogTests(unittest.TestCase):
    self.assertEqual([x['repo'] for x in value['models']],['maker/base']);self.assertEqual(value['scanned'],21)
    self.assertIn('limit=100',fetch.call_args.args[0]);self.assertIn('expand=cardData',fetch.call_args.args[0])
    with self.assertRaises(ValueError):Catalog(d).search('','video','downloads','true')
+ def test_direct_repository_bypasses_discovery_filters_explicitly(self):
+  item={'id':'jpetrina/example','tags':['gguf']}
+  with tempfile.TemporaryDirectory() as d,patch('catalog.metadata',return_value=item):
+   for q in ('jpetrina/example','https://huggingface.co/jpetrina/example'):
+    result=Catalog(d).search(q,'chat','downloads',True)
+    self.assertEqual(result['models'][0]['repo'],item['id']);self.assertIn('nicht angewendet',result['notice'])
+ def test_gguf_filename_fallback_finds_conversational_without_pipeline_tag(self):
+  item={'id':'maker/model-GGUF','tags':['gguf','conversational']}
+  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_download_integrity_and_library(self):
   for valid in (True,False):
    with self.subTest(valid=valid),tempfile.TemporaryDirectory() as directory: