nexaai 1.0.17rc3__cp310-cp310-macosx_14_0_universal2.whl → 1.0.17rc5__cp310-cp310-macosx_14_0_universal2.whl

nexaai/_version.py

@@ -1,4 +1,4 @@
 # This file is generated by CMake from _version.py.in
 # Do not modify this file manually - it will be overwritten
 
-__version__ = "1.0.17-rc3" 
+__version__ = "1.0.17-rc5" 

nexaai/utils/manifest_utils.py

@@ -35,15 +35,8 @@ def process_manifest_metadata(manifest: Dict[str, Any], repo_id: str) -> Dict[st
     # Handle download_time - keep as null if missing
     download_time = manifest.get('download_time')
     
-    # Handle avatar_url - fetch on-the-fly if missing/null
+    # Handle avatar_url - leave it null if missing/null
     avatar_url = manifest.get('avatar_url')
-    if not avatar_url:
-        try:
-            from .avatar_fetcher import get_avatar_url_for_repo
-            avatar_url = get_avatar_url_for_repo(repo_id)
-        except Exception:
-            # If fetching fails, leave as None
-            avatar_url = None
     
     # Return processed metadata
     processed_manifest = manifest.copy()
@@ -171,9 +164,9 @@ def create_gguf_manifest(repo_id: str, files: List[str], directory_path: str, ol
         },
         "ExtraFiles": None,
         # Preserve old metadata fields
-        "pipeline_tag": old_metadata.get('pipeline_tag'),
-        "download_time": old_metadata.get('download_time'),
-        "avatar_url": old_metadata.get('avatar_url')
+        "pipeline_tag": old_metadata.get('pipeline_tag') if old_metadata.get('pipeline_tag') else existing_manifest.get('pipeline_tag'),
+        "download_time": old_metadata.get('download_time') if old_metadata.get('download_time') else existing_manifest.get('download_time'),
+        "avatar_url": old_metadata.get('avatar_url') if old_metadata.get('avatar_url') else existing_manifest.get('avatar_url')
     }
     
     return manifest
@@ -182,6 +175,9 @@ def create_gguf_manifest(repo_id: str, files: List[str], directory_path: str, ol
 def create_mlx_manifest(repo_id: str, files: List[str], directory_path: str, old_metadata: Dict[str, Any], is_mmproj: bool = False, file_name: Optional[Union[str, List[str]]] = None) -> Dict[str, Any]:
     """Create MLX format manifest."""
     
+    # Load existing manifest to merge MLX files if it exists
+    existing_manifest = load_nexa_manifest(directory_path)
+    
     model_files = {}
     extra_files = []
     
@@ -250,9 +246,9 @@ def create_mlx_manifest(repo_id: str, files: List[str], directory_path: str, old
         },
         "ExtraFiles": extra_files if extra_files else None,
         # Preserve old metadata fields
-        "pipeline_tag": old_metadata.get('pipeline_tag'),
-        "download_time": old_metadata.get('download_time'),
-        "avatar_url": old_metadata.get('avatar_url')
+        "pipeline_tag": old_metadata.get('pipeline_tag') if old_metadata.get('pipeline_tag') else existing_manifest.get('pipeline_tag'),
+        "download_time": old_metadata.get('download_time') if old_metadata.get('download_time') else existing_manifest.get('download_time'),
+        "avatar_url": old_metadata.get('avatar_url') if old_metadata.get('avatar_url') else existing_manifest.get('avatar_url')
     }
     
     return manifest

nexaai/utils/model_manager.py

@@ -11,7 +11,6 @@ from huggingface_hub import HfApi
 from huggingface_hub.utils import HfHubHTTPError, RepositoryNotFoundError
 
 from .progress_tracker import CustomProgressTqdm, DownloadProgressTracker
-from .avatar_fetcher import get_avatar_url_for_repo
 from .manifest_utils import (
     load_download_metadata,
     save_download_metadata,
@@ -790,7 +789,7 @@ class HuggingFaceDownloader:
         # If no expected size, just check that file is not empty
         return os.path.getsize(file_path) > 0
     
-    def _fetch_and_save_metadata(self, repo_id: str, local_dir: str, is_mmproj: bool = False, file_name: Optional[Union[str, List[str]]] = None) -> None:
+    def _fetch_and_save_metadata(self, repo_id: str, local_dir: str, is_mmproj: bool = False, file_name: Optional[Union[str, List[str]]] = None, **kwargs) -> None:
         """Fetch model info and save metadata after successful download."""
         # Initialize metadata with defaults to ensure manifest is always created
         old_metadata = {
@@ -809,14 +808,9 @@ class HuggingFaceDownloader:
             # Log the error but continue with manifest creation
             print(f"Warning: Could not fetch model info for {repo_id}: {e}")
         
-        try:
-            # Get avatar URL
-            avatar_url = get_avatar_url_for_repo(repo_id, custom_endpoint=self.endpoint)
-            if avatar_url:
-                old_metadata['avatar_url'] = avatar_url
-        except Exception as e:
-            # Log the error but continue with manifest creation
-            print(f"Warning: Could not fetch avatar URL for {repo_id}: {e}")
+        # Use input avater url if provided
+        old_metadata['avatar_url'] = kwargs.get('avatar_url')
+
         
         # CRITICAL: Always create the manifest file, regardless of metadata fetch failures
         try:
@@ -850,7 +844,8 @@ class HuggingFaceDownloader:
         file_name: str,
         local_dir: str,
         progress_tracker: Optional[DownloadProgressTracker],
-        force_download: bool = False
+        force_download: bool = False,
+        **kwargs
     ) -> str:
         """Download a single file from the repository using HuggingFace Hub API."""
         # Create repo-specific directory for the single file
@@ -882,7 +877,7 @@ class HuggingFaceDownloader:
                 progress_tracker.stop_tracking()
             
             # Save metadata after successful download
-            self._fetch_and_save_metadata(repo_id, file_local_dir, self._current_is_mmproj, self._current_file_name)
+            self._fetch_and_save_metadata(repo_id, file_local_dir, self._current_is_mmproj, self._current_file_name, **kwargs)
             
             return downloaded_path
             
@@ -901,7 +896,8 @@ class HuggingFaceDownloader:
         repo_id: str,
         local_dir: str,
         progress_tracker: Optional[DownloadProgressTracker],
-        force_download: bool = False
+        force_download: bool = False,
+        **kwargs
     ) -> str:
         """Download the entire repository."""
         # Create a subdirectory for this specific repo
@@ -927,7 +923,7 @@ class HuggingFaceDownloader:
                 progress_tracker.stop_tracking()
             
             # Save metadata after successful download
-            self._fetch_and_save_metadata(repo_id, repo_local_dir, self._current_is_mmproj, self._current_file_name)
+            self._fetch_and_save_metadata(repo_id, repo_local_dir, self._current_is_mmproj, self._current_file_name, **kwargs)
             
             return downloaded_path
             
@@ -944,7 +940,8 @@ class HuggingFaceDownloader:
         file_names: List[str],
         local_dir: str,
         progress_tracker: Optional[DownloadProgressTracker],
-        force_download: bool = False
+        force_download: bool = False,
+        **kwargs
     ) -> str:
         """Download multiple specific files from HuggingFace Hub."""
         # Create repo-specific directory
@@ -989,7 +986,7 @@ class HuggingFaceDownloader:
                 progress_tracker.stop_tracking()
             
             # Save metadata after successful download
-            self._fetch_and_save_metadata(repo_id, repo_local_dir, self._current_is_mmproj, self._current_file_name)
+            self._fetch_and_save_metadata(repo_id, repo_local_dir, self._current_is_mmproj, self._current_file_name, **kwargs)
             
             return repo_local_dir
             
@@ -1015,7 +1012,8 @@ class HuggingFaceDownloader:
         progress_callback: Optional[Callable[[Dict[str, Any]], None]] = None,
         show_progress: bool = True,
         force_download: bool = False,
-        is_mmproj: bool = False
+        is_mmproj: bool = False,
+        **kwargs
     ) -> str:
         """
         Main download method that handles all download scenarios.
@@ -1062,13 +1060,13 @@ class HuggingFaceDownloader:
             if file_name is None:
                 # Download entire repository
                 return self._download_entire_repository(
-                    repo_id, local_dir, progress_tracker, force_download
+                    repo_id, local_dir, progress_tracker, force_download, **kwargs
                 )
             elif isinstance(file_name, str):
                 # Download specific single file
                 self._validate_file_exists_in_repo(file_name, info, repo_id, progress_tracker)
                 return self._download_single_file(
-                    repo_id, file_name, local_dir, progress_tracker, force_download
+                    repo_id, file_name, local_dir, progress_tracker, force_download, **kwargs
                 )
             else:  # file_name is a list
                 # Download multiple specific files
@@ -1077,7 +1075,7 @@ class HuggingFaceDownloader:
                     self._validate_file_exists_in_repo(fname, info, repo_id, progress_tracker)
                 
                 return self._download_multiple_files_from_hf(
-                    repo_id, file_name, local_dir, progress_tracker, force_download
+                    repo_id, file_name, local_dir, progress_tracker, force_download, **kwargs
                 )
         
         except Exception as e:
@@ -1107,7 +1105,8 @@ def download_from_huggingface(
     token: Union[bool, str, None] = None,
     custom_endpoint: Optional[str] = None,
     force_download: bool = False,
-    is_mmproj: Optional[bool] = None
+    is_mmproj: Optional[bool] = None,
+    **kwargs
 ) -> str:
     """
     Download models or files from HuggingFace Hub or custom mirror endpoints.
@@ -1197,7 +1196,8 @@ def download_from_huggingface(
         progress_callback=progress_callback,
         show_progress=show_progress,
         force_download=force_download,
-        is_mmproj=is_mmproj
+        is_mmproj=is_mmproj,
+        **kwargs
     )
 
 
@@ -1211,7 +1211,8 @@ def _download_model_if_needed(
     param_name: str,
     progress_callback: Optional[Callable[[Dict[str, Any]], None]] = None,
     token: Union[bool, str, None] = None,
-    is_mmproj: bool = False
+    is_mmproj: bool = False,
+    **kwargs
 ) -> str:
     """
     Helper function to download a model from HuggingFace if it doesn't exist locally.
@@ -1247,7 +1248,8 @@ def _download_model_if_needed(
             progress_callback=progress_callback,
             show_progress=True,
             token=token,
-            is_mmproj=is_mmproj
+            is_mmproj=is_mmproj,
+            **kwargs
         )
         
         return downloaded_path
@@ -1320,7 +1322,7 @@ def auto_download_model(func: Callable) -> Callable:
         if name_or_path is not None:
             try:
                 downloaded_name_path = _download_model_if_needed(
-                    name_or_path, 'name_or_path', progress_callback, token
+                    name_or_path, 'name_or_path', progress_callback, token, **kwargs
                 )
                 
                 # Replace name_or_path with downloaded path
@@ -1338,7 +1340,7 @@ def auto_download_model(func: Callable) -> Callable:
         if mmproj_path is not None:
             try:
                 downloaded_mmproj_path = _download_model_if_needed(
-                    mmproj_path, 'mmproj_path', progress_callback, token, is_mmproj=True
+                    mmproj_path, 'mmproj_path', progress_callback, token, is_mmproj=True, **kwargs
                 )
                 
                 # Replace mmproj_path with downloaded path

{nexaai-1.0.17rc3.dist-info → nexaai-1.0.17rc5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nexaai
-Version: 1.0.17rc3
+Version: 1.0.17rc5
 Summary: Python bindings for NexaSDK C-lib backend
 Author-email: "Nexa AI, Inc." <dev@nexa.ai>
 Project-URL: Homepage, https://github.com/NexaAI/nexasdk-bridge

{nexaai-1.0.17rc3.dist-info → nexaai-1.0.17rc5.dist-info}/RECORD

@@ -1,6 +1,6 @@
 nexaai/__init__.py,sha256=L8oB7GFZZMGnUpCg0PecDbI_ycKuQak-ZEJ4Y12_QIw,2184
-nexaai/_stub.cpython-310-darwin.so,sha256=EESW_m6rs5Wlme_i2eC5SBJNYxhWpP2ZbxMKx3K8FFg,66768
-nexaai/_version.py,sha256=m8oymb8W-f11Ra9dj3cr9xkH66hL0S5SPIlATwANYmU,143
+nexaai/_stub.cpython-310-darwin.so,sha256=8bcJLd34iGDb2YnRTI5DCDqIeQKR3hWJ8RZVh01uWyE,66768
+nexaai/_version.py,sha256=F_da1d4cWkmF90PfAVUtepv8hmGILywCGDFetA8CeKk,143
 nexaai/asr.py,sha256=NljMXDErwPNMOPaRkJZMEDka9Nk8xyur7L8i924TStY,2054
 nexaai/base.py,sha256=N8PRgDFA-XPku2vWnQIofQ7ipz3pPlO6f8YZGnuhquE,982
 nexaai/common.py,sha256=Y0NJNLTi4Nq4x1WL6PQsSvGUto0eGmWhjpsC6jcekfA,3444
@@ -19,7 +19,7 @@ nexaai/asr_impl/pybind_asr_impl.py,sha256=pE9Hb_hMi5yAc4MF83bLVOb8zDtreCkB3_u7XE
 nexaai/binds/__init__.py,sha256=eYuay_8DDXeOUWz2_R9HFSabohxs6hvZn391t2L0Po0,104
 nexaai/binds/common_bind.cpython-310-darwin.so,sha256=ya8gz7NnUtHlniGnRyExdwWfdHimEHiJ7Dry4I7_y44,235264
 nexaai/binds/embedder_bind.cpython-310-darwin.so,sha256=tPa0c0Dv_GiW66fgmAGWGCHXRGNApznqoQS0eQx9GFM,202064
-nexaai/binds/libnexa_bridge.dylib,sha256=udoZzHIJvexfdqAaBZUIUAXRbmqvqKWolI7hNFwRqoo,251192
+nexaai/binds/libnexa_bridge.dylib,sha256=-PciDq_xpsOLKyIOW0g5RQs9Fd4X8fuK-8wrvm3YGX8,251192
 nexaai/binds/llm_bind.cpython-310-darwin.so,sha256=TAWfa1Hzq00TjtC1xVsiAeLp6hv2LrL5afDz4omUghc,182784
 nexaai/binds/vlm_bind.cpython-310-darwin.so,sha256=nd6eG_m2EiPthzkSZ97hlXWUOZQir4cQfFJZ4p6eR2U,182704
 nexaai/binds/nexa_llama_cpp/libggml-base.dylib,sha256=JM4oOkie1su0ES5hMdtILeQHlRukRzH1vTleTupUXhg,650736
@@ -188,7 +188,7 @@ nexaai/binds/nexa_nexaml/libggml-metal.so,sha256=zfaX7rIBYQazH2lf-vza007BMhPTK1A
 nexaai/binds/nexa_nexaml/libggml.dylib,sha256=aOTj_6RrAMkfDO0ZI28_3nfcC-l4Y3dRCiS3C0d0_eI,58592
 nexaai/binds/nexa_nexaml/libnexa-mm-process.dylib,sha256=3mITty3oYhoi5yc690CxS0tf4LGanqrPCR0meE-h9HQ,8568120
 nexaai/binds/nexa_nexaml/libnexa-sampling.dylib,sha256=OliTBr-r2y22Ebjc1Y33Ax56XpLMFQKtsHgfouj6TQM,7865400
-nexaai/binds/nexa_nexaml/libnexa_plugin.dylib,sha256=OiQtbPidObQaKFdr1nHKQA9PaDv_a1TQeDQ26x07GxM,249848
+nexaai/binds/nexa_nexaml/libnexa_plugin.dylib,sha256=41VEU8yhk2aQlpfoSFeYkhLLhBSbzCcMns0IGyZvpNg,249848
 nexaai/binds/nexa_nexaml/libnexaproc.dylib,sha256=Pnxgm2k29hfedLjNP4t44gvioh-NWbE_3BW-VsCdHp8,880560
 nexaai/binds/nexa_nexaml/libqwen3-vl.dylib,sha256=hd8VPINYTH8mbU3BPxnjrPOKTw-NkWn7zkDB7p074-U,753168
 nexaai/binds/nexa_nexaml/libqwen3vl-vision.dylib,sha256=AmUk-DBtBAp5DOR14ICZcsiAcKjxxAx1y6mHAaETQu8,570024
@@ -541,17 +541,16 @@ nexaai/rerank_impl/pybind_rerank_impl.py,sha256=CtwkG7YrW58GPMDERJSnISGTVCXWNju5
 nexaai/tts_impl/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nexaai/tts_impl/mlx_tts_impl.py,sha256=i_uNPdvlXYtL3e01oKjDlP9jgkWCRt1bBHsExaaiJi8,3101
 nexaai/tts_impl/pybind_tts_impl.py,sha256=mpn44r6pfYLIl-NrEy2dXHjGtWtNCmM7HRyxiANxUI4,1444
-nexaai/utils/avatar_fetcher.py,sha256=bWy8ujgbOiTHFCjFxTwkn3uXbZ84PgEGUkXkR3MH4bI,3821
 nexaai/utils/decode.py,sha256=61n4Zf6c5QLyqGoctEitlI9BX3tPlP2a5aaKNHbw3T4,404
-nexaai/utils/manifest_utils.py,sha256=sR9Nme4GbD3Cb3fMd55yLvGZpqxb71vd6b2XZTsrIGM,12328
-nexaai/utils/model_manager.py,sha256=p2kJKK63Zk-rEUucFsgY0T5PyXi_IvJY0gKewUVcAV4,56081
+nexaai/utils/manifest_utils.py,sha256=PA84obFP7W1dlneURlIHIzJjWIF5dbDHGdNeHouUy68,12659
+nexaai/utils/model_manager.py,sha256=_WKJP7YVk7q587OoOWwDNWVR-8tbKZkmHKjcCZN8Q4M,55979
 nexaai/utils/model_types.py,sha256=-DER8L4lAUR_iLS99F0r57avwqWtuN21ug5pX2p24_E,1369
 nexaai/utils/progress_tracker.py,sha256=jdUqtmPqyhwC9uSKvQcJEYETwSt-OhP4oitdJ94614o,15394
 nexaai/utils/quantization_utils.py,sha256=FYcNSAKGlBqFDUTx3jSKOr2lnq4nyiyC0ZG8oSxFwiU,7825
 nexaai/vlm_impl/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nexaai/vlm_impl/mlx_vlm_impl.py,sha256=pLtWm_ckz8a0U-AtAOMVseFDO4OVPvHyYO2KlfBaGYk,10833
 nexaai/vlm_impl/pybind_vlm_impl.py,sha256=FAbhpRJzHgI78r0mUvKybO97R1szvNhH0aTn_I52oT4,8597
-nexaai-1.0.17rc3.dist-info/METADATA,sha256=qUwlQ7ioHaewOQ--gpTZsFkQDerm78q4okTIbDyTZ2E,1201
-nexaai-1.0.17rc3.dist-info/WHEEL,sha256=T2p57lol9__xkoU6aJTyN1Pm43ZpRU3q6km7mIbrAMs,114
-nexaai-1.0.17rc3.dist-info/top_level.txt,sha256=LRE2YERlrZk2vfuygnSzsEeqSknnZbz3Z1MHyNmBU4w,7
-nexaai-1.0.17rc3.dist-info/RECORD,,
+nexaai-1.0.17rc5.dist-info/METADATA,sha256=UP8ZerQ7zCuBdjENEvuIzgSkgEMylmN6G0UTs6VcifQ,1201
+nexaai-1.0.17rc5.dist-info/WHEEL,sha256=T2p57lol9__xkoU6aJTyN1Pm43ZpRU3q6km7mIbrAMs,114
+nexaai-1.0.17rc5.dist-info/top_level.txt,sha256=LRE2YERlrZk2vfuygnSzsEeqSknnZbz3Z1MHyNmBU4w,7
+nexaai-1.0.17rc5.dist-info/RECORD,,

nexaai/utils/avatar_fetcher.py

@@ -1,104 +0,0 @@
-"""Utility for fetching avatar URLs from HuggingFace."""
-
-import logging
-from typing import Dict, Optional
-import httpx
-
-logger = logging.getLogger(__name__)
-
-
-def fetch_avatar_urls_from_hf_api(query: str, custom_endpoint: Optional[str] = None) -> Dict[str, str]:
-    """
-    Fetch avatar URLs from HuggingFace models-json endpoint.
-    
-    Args:
-        query: Search query to fetch models for
-        custom_endpoint: Optional custom HuggingFace endpoint
-        
-    Returns:
-        Dictionary mapping author names to avatar URLs
-    """
-    avatar_map = {}
-    try:
-        # Use the base URL from the configured endpoint
-        base_url = custom_endpoint if custom_endpoint else "https://huggingface.co"
-        
-        # Build the URL with query parameter
-        url = f"{base_url}/models-json?sort=trending&search={query}&withCount=true"
-        
-        # Make the HTTP request with a timeout
-        with httpx.Client(timeout=2.0) as client:
-            response = client.get(url)
-            
-            if response.status_code == 200:
-                data = response.json()
-                models = data.get("models", [])
-                
-                # Build a map of author names to avatar URLs
-                for model in models:
-                    author = model.get("author")
-                    author_data = model.get("authorData", {})
-                    avatar_url = author_data.get("avatarUrl")
-                    
-                    if author and avatar_url:
-                        # Handle relative URLs by prepending appropriate base URL
-                        if avatar_url.startswith("/"):
-                            avatar_url = f"{base_url}{avatar_url}"
-                        avatar_map[author] = avatar_url
-                
-                logger.debug(f"Fetched {len(avatar_map)} avatar URLs from HuggingFace API")
-            else:
-                logger.warning(f"Failed to fetch avatar URLs: HTTP {response.status_code}")
-                
-    except Exception as e:
-        logger.warning(f"Error fetching avatar URLs from HuggingFace API: {e}")
-        # Return empty map on error - we'll fall back to default behavior
-    
-    return avatar_map
-
-
-def get_avatar_url_for_repo(repo_id: str, search_query: Optional[str] = None, 
-                            custom_endpoint: Optional[str] = None) -> Optional[str]:
-    """
-    Get avatar URL for a repository ID.
-    
-    This method tries multiple strategies:
-    1. If search_query is provided, fetch from HuggingFace API with that query
-    2. Try fetching with the full repo_id as query
-    3. Try fetching with just the organization name as query
-    4. Fall back to CDN URL pattern
-    
-    Args:
-        repo_id: Repository ID in format "owner/repo"
-        search_query: Optional search query to use for fetching avatars
-        custom_endpoint: Optional custom HuggingFace endpoint
-        
-    Returns:
-        Avatar URL or None if not found
-    """
-    if "/" not in repo_id:
-        return None
-        
-    org_name = repo_id.split("/")[0]
-    
-    # Try with search query if provided
-    if search_query:
-        avatar_map = fetch_avatar_urls_from_hf_api(search_query, custom_endpoint)
-        avatar_url = avatar_map.get(org_name)
-        if avatar_url:
-            return avatar_url
-    
-    # Try with full repo_id
-    avatar_map = fetch_avatar_urls_from_hf_api(repo_id, custom_endpoint)
-    avatar_url = avatar_map.get(org_name)
-    if avatar_url:
-        return avatar_url
-    
-    # Try with just organization name
-    avatar_map = fetch_avatar_urls_from_hf_api(org_name, custom_endpoint)
-    avatar_url = avatar_map.get(org_name)
-    if avatar_url:
-        return avatar_url
-    
-    # Fallback to CDN URL pattern
-    return f"https://cdn-thumbnails.huggingface.co/social-thumbnails/{org_name}.png"