nexaai 1.0.17rc4__cp310-cp310-win_amd64.whl → 1.0.17rc6__cp310-cp310-win_amd64.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-rc4" 
+__version__ = "1.0.17-rc6" 

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.17rc4.dist-info → nexaai-1.0.17rc6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nexaai
-Version: 1.0.17rc4
+Version: 1.0.17rc6
 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.17rc4.dist-info → nexaai-1.0.17rc6.dist-info}/RECORD

@@ -1,6 +1,6 @@
 nexaai/__init__.py,sha256=mbzzeXrEHHI_E3BQ0_OukD9wNajKJJVk0ykxT0rz8uM,2267
-nexaai/_stub.cp310-win_amd64.pyd,sha256=EREb35gB2Gyp39GsBxOPF194T3bGTww-tRQOZ8UseDQ,10752
-nexaai/_version.py,sha256=ozf8OHb1zplTmsy6N10b0oiZT3eilYbaSAkUpGBalAg,147
+nexaai/_stub.cp310-win_amd64.pyd,sha256=oTs8w18u97m9hVEYxcr0hJzX8xhXH6E1EgEAYhn0WwU,10752
+nexaai/_version.py,sha256=0ZuRl-kfhiNtwV_aBU4Bece34voIsesZ10rDaGoyLhU,147
 nexaai/asr.py,sha256=_fsGaxpiU137bUtO5ujtFSYCI1RLsyeEm3Gf4GhHVRk,2118
 nexaai/base.py,sha256=qQBCiQVNzgpkQjZX9aiFDEdbAAe56TROKC3WnWra2Zg,1021
 nexaai/common.py,sha256=muQqFY-WllwL5IO83tImexbuUcoEQsKg73u4gWL2lNs,3548
@@ -17,29 +17,29 @@ nexaai/asr_impl/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nexaai/asr_impl/mlx_asr_impl.py,sha256=XwMX3LYMeulp8cDS0TCCYcjvttFHAyDWQ_oMvABwQmI,3349
 nexaai/asr_impl/pybind_asr_impl.py,sha256=20o5SOPzhF9x41ra8L_qIM7YxCkYeLb5csSrNde-dds,1560
 nexaai/binds/__init__.py,sha256=ENl-uoIF9-3XGIXitVgZ2QmJ6p7Yet4h1-X7nUDZ0Hk,108
-nexaai/binds/common_bind.cp310-win_amd64.pyd,sha256=PxUW7Fy9ot3rDXj86MnWt4T9UI_UzJ-Xv5s1CzJCPVk,205824
-nexaai/binds/embedder_bind.cp310-win_amd64.pyd,sha256=4cX1dcMQQ0ZVUaVeq0PTkW6vc8Mdmjub91AIgHZhIoQ,182784
+nexaai/binds/common_bind.cp310-win_amd64.pyd,sha256=Q1BhHyzilfRRsXuoBD7pN1dtm6VGXku4Q133hLC-LlU,205824
+nexaai/binds/embedder_bind.cp310-win_amd64.pyd,sha256=xYDozXpDuvp54QPFWTcz8YcDaThjP697ox6mUKLmOy0,182784
 nexaai/binds/libcrypto-3-x64.dll,sha256=TAWcTdJrWviUingNyWaraOfz0ILUy7ybpI4VaiVHURg,7315968
 nexaai/binds/libssl-3-x64.dll,sha256=ZxawOWlONNM0Neg47w4JbU8kZJldj046H1zV3LiETa0,1313792
-nexaai/binds/llm_bind.cp310-win_amd64.pyd,sha256=9_JTLaEC8h6l0lsNv_f2N2z4rdwTAizHFA-gs-_Rdhw,162816
-nexaai/binds/nexa_bridge.dll,sha256=HEbEJW01XhHiWVMfB7uJf1aA-SF6fyhWc33wfmlefVQ,171008
-nexaai/binds/vlm_bind.cp310-win_amd64.pyd,sha256=oqHxXD0kTZxMl2-7W6ufUolY6bLD9rDzLN1G3QSmGSE,168960
-nexaai/binds/nexa_llama_cpp/ggml-base.dll,sha256=Z-t4lg_0F0NDB9CFZ-yFRm6ffw5XPJExzEi7mB9M58c,532480
-nexaai/binds/nexa_llama_cpp/ggml-cpu.dll,sha256=XJud9mLT_z1bHMk5mZkfX6BIPMqvK4YL6z4fLkp_sqM,672768
-nexaai/binds/nexa_llama_cpp/ggml-cuda.dll,sha256=sctJQjVPBBpYhqa5Ch-ai8uGuT3OO-leGsCGZ0mBxaY,313528832
-nexaai/binds/nexa_llama_cpp/ggml-vulkan.dll,sha256=v1FKLrJaZ8ayNIGNOgpAFYnIJ0eGJotqtvAPyYlMrns,36627456
-nexaai/binds/nexa_llama_cpp/ggml.dll,sha256=aitn26Xt84Cn2vQRRn4Kccxw0ts0tA6lCZJ_FCHkaDU,66560
-nexaai/binds/nexa_llama_cpp/llama.dll,sha256=BZom-3FDr9nYIptVL1jsSua9kHS6oRbNqYsixXTWJe8,1611776
-nexaai/binds/nexa_llama_cpp/mtmd.dll,sha256=yoqOUMxat2Y22UtzJZAD5xcyZZvhsikagpy3ARc2EnQ,561152
-nexaai/binds/nexa_llama_cpp/nexa_plugin.dll,sha256=DQTtwSwripOzWuvp-6-pi2rbNZbwEHshQ0hPfX1oOAE,1413120
-nexaai/binds/nexa_nexaml/ggml-base.dll,sha256=Z-t4lg_0F0NDB9CFZ-yFRm6ffw5XPJExzEi7mB9M58c,532480
-nexaai/binds/nexa_nexaml/ggml-cpu.dll,sha256=XJud9mLT_z1bHMk5mZkfX6BIPMqvK4YL6z4fLkp_sqM,672768
-nexaai/binds/nexa_nexaml/ggml-cuda.dll,sha256=sctJQjVPBBpYhqa5Ch-ai8uGuT3OO-leGsCGZ0mBxaY,313528832
-nexaai/binds/nexa_nexaml/ggml-vulkan.dll,sha256=v1FKLrJaZ8ayNIGNOgpAFYnIJ0eGJotqtvAPyYlMrns,36627456
-nexaai/binds/nexa_nexaml/ggml.dll,sha256=aitn26Xt84Cn2vQRRn4Kccxw0ts0tA6lCZJ_FCHkaDU,66560
+nexaai/binds/llm_bind.cp310-win_amd64.pyd,sha256=LJtUKePsf6aCeFWe8mNT_8F4qdF47iLjtnh_fN7Wnxs,162816
+nexaai/binds/nexa_bridge.dll,sha256=8VKruEstzY6SJArTazvU204Sqjg0juKgPUhYCYF4jiw,171008
+nexaai/binds/vlm_bind.cp310-win_amd64.pyd,sha256=ucRXpKhZyBLriM4ef1VW8WHIBPsyTOcAi0RUoCneQRc,168960
+nexaai/binds/nexa_llama_cpp/ggml-base.dll,sha256=4vJCkQbHq__s5tTFCcKJxxL1tQTaEEDv6HynnUyMkzI,532480
+nexaai/binds/nexa_llama_cpp/ggml-cpu.dll,sha256=E_Y--19NPzp9XzvMiSNj6_4AFQoO8VX3O_JTZQ9mULU,672768
+nexaai/binds/nexa_llama_cpp/ggml-cuda.dll,sha256=lISNt-_dtYWIUIiqVrnxYREqpfNqVYVxOInzXFldu9I,313528832
+nexaai/binds/nexa_llama_cpp/ggml-vulkan.dll,sha256=sPLmpIPEcZESdqMWwpiFF43vNtGy03bGlGQXzTrXKV0,36627456
+nexaai/binds/nexa_llama_cpp/ggml.dll,sha256=4L8A2v49A0-xrazUhw2J0R01oMo8jgjrwBHrxO0-gsc,66560
+nexaai/binds/nexa_llama_cpp/llama.dll,sha256=2pYy177Kvg_ZLyQRFZZgTIzjlpZizA_j_kxKfxlvZRk,1611776
+nexaai/binds/nexa_llama_cpp/mtmd.dll,sha256=dSuvEYtCrelo1g0jV4eRg0u57ZUcx9unhPBiPV8L10E,561152
+nexaai/binds/nexa_llama_cpp/nexa_plugin.dll,sha256=hEBZdCtzzLpIZlIIPiGhO_DY44_4e8xKxhNnCZ6PZA8,1413120
+nexaai/binds/nexa_nexaml/ggml-base.dll,sha256=4vJCkQbHq__s5tTFCcKJxxL1tQTaEEDv6HynnUyMkzI,532480
+nexaai/binds/nexa_nexaml/ggml-cpu.dll,sha256=E_Y--19NPzp9XzvMiSNj6_4AFQoO8VX3O_JTZQ9mULU,672768
+nexaai/binds/nexa_nexaml/ggml-cuda.dll,sha256=lISNt-_dtYWIUIiqVrnxYREqpfNqVYVxOInzXFldu9I,313528832
+nexaai/binds/nexa_nexaml/ggml-vulkan.dll,sha256=sPLmpIPEcZESdqMWwpiFF43vNtGy03bGlGQXzTrXKV0,36627456
+nexaai/binds/nexa_nexaml/ggml.dll,sha256=4L8A2v49A0-xrazUhw2J0R01oMo8jgjrwBHrxO0-gsc,66560
 nexaai/binds/nexa_nexaml/nexa-mm-process.dll,sha256=oDnqTaAUeZrWJePQEDuIYKEOMP-ndu221mMFHht4bSY,4642816
 nexaai/binds/nexa_nexaml/nexa-sampling.dll,sha256=PNtidsRbGO8dTfbf7-VMdhQRbF328yvLnmDjMy2Tb7Y,4265984
-nexaai/binds/nexa_nexaml/nexa_plugin.dll,sha256=gQUPmukEIKMEAStZnGialgpH9_I-JTLddO8HIGoGBsQ,602112
+nexaai/binds/nexa_nexaml/nexa_plugin.dll,sha256=Oic5O_NrtsVoO_NwKyQDh_yLDc5--JRYdXXBQx8AlzI,602112
 nexaai/binds/nexa_nexaml/nexaproc.dll,sha256=9X4TF6p7wOiIXi_SA7uHLoEmrvcTlH2WpfijK7HxZgY,2668544
 nexaai/binds/nexa_nexaml/qwen3-vl.dll,sha256=Es-etUDiMmK_BwOki8V79Nja-NY4kBbrkhtnesXfMMU,5870592
 nexaai/binds/nexa_nexaml/qwen3vl-vision.dll,sha256=HFAk0FxWlf_lko3AQL6uBXTIhm6AzDFi5PjxI0lxONI,1184256
@@ -61,17 +61,16 @@ nexaai/rerank_impl/pybind_rerank_impl.py,sha256=1IW925bYv4FRwZDNzf9epEzdDqR6T3OO
 nexaai/tts_impl/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nexaai/tts_impl/mlx_tts_impl.py,sha256=LcH9bVdIl3Q6lOzSUB_X2s-_nWFmlCl1yL7XSUK0fYI,3195
 nexaai/tts_impl/pybind_tts_impl.py,sha256=n3z4zmPQayQJgAwcvETw0IBUCp8IYROuYFSg0tAy_8Y,1487
-nexaai/utils/avatar_fetcher.py,sha256=D01f8je-37Nd68zGw8MYK2m7y3fvGlC6h0KR-aN9kdU,3925
 nexaai/utils/decode.py,sha256=0Z9jDH4ICzw4YXj8nD4L-sMouDaev-TISGRQ4KzidWE,421
-nexaai/utils/manifest_utils.py,sha256=nPqK24srLX41x3WU4R-OGQR9u0XA7ZTHY1MQXBKXdrM,12652
-nexaai/utils/model_manager.py,sha256=3rDODGClgnM24LFWIoM-TDpXkpV1vyJSdlv7Qk4ZaPE,57434
+nexaai/utils/manifest_utils.py,sha256=OyLi9e4JrOBunswf4FG2KoVvkbv2FulkOf5LqRDVf50,12979
+nexaai/utils/model_manager.py,sha256=bWMVvcBu5dC3YmWtsJK4C3IaXfaDSByAYcH7GVLtoEg,57334
 nexaai/utils/model_types.py,sha256=arIyb9q-1uG0nyUGdWZaxxDJAxv0cfnJEpjCzyELL5Q,1416
 nexaai/utils/progress_tracker.py,sha256=BztrFqtjwNUmeREwZ5m7H6ZcrVzQEbpZfsxndWh4z0A,15778
 nexaai/utils/quantization_utils.py,sha256=FxnZ6-uAE_bl_vQ5jsRXlpU0NBn-U4Y8iN9_O6aCdPA,8070
 nexaai/vlm_impl/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nexaai/vlm_impl/mlx_vlm_impl.py,sha256=MgqJO7OzuPd79gOZZKhSXXMNSP2eBuhhrdCX8XHn6aQ,11090
 nexaai/vlm_impl/pybind_vlm_impl.py,sha256=NuQ_Ep1TnjmGAkjJuUS0Lb6z7iPu3wroLVOx7kiAwlE,8827
-nexaai-1.0.17rc4.dist-info/METADATA,sha256=VFChU80B0lwHtub5O0_L7tFtVtzsUARLarDXNH3mTs4,1233
-nexaai-1.0.17rc4.dist-info/WHEEL,sha256=KUuBC6lxAbHCKilKua8R9W_TM71_-9Sg5uEP3uDWcoU,101
-nexaai-1.0.17rc4.dist-info/top_level.txt,sha256=LRE2YERlrZk2vfuygnSzsEeqSknnZbz3Z1MHyNmBU4w,7
-nexaai-1.0.17rc4.dist-info/RECORD,,
+nexaai-1.0.17rc6.dist-info/METADATA,sha256=9OQOKa68hwxN-v-QCooWoa3XXveuWupaT4DnJQpWaQE,1233
+nexaai-1.0.17rc6.dist-info/WHEEL,sha256=KUuBC6lxAbHCKilKua8R9W_TM71_-9Sg5uEP3uDWcoU,101
+nexaai-1.0.17rc6.dist-info/top_level.txt,sha256=LRE2YERlrZk2vfuygnSzsEeqSknnZbz3Z1MHyNmBU4w,7
+nexaai-1.0.17rc6.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"