nexaai 1.0.7__cp310-cp310-macosx_13_0_x86_64.whl → 1.0.9__cp310-cp310-macosx_13_0_x86_64.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.7" 
+__version__ = "1.0.9" 

nexaai/utils/model_manager.py

@@ -14,6 +14,7 @@ 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,
     save_manifest_with_files_metadata,
 )
 
@@ -117,6 +118,13 @@ def _get_directory_size_and_files(directory_path: str) -> tuple[int, List[str]]:
     return total_size, files
 
 
+def _has_valid_metadata(directory_path: str) -> bool:
+    """Check if directory has either nexa.manifest or download_metadata.json (for backward compatibility)."""
+    manifest_path = os.path.join(directory_path, 'nexa.manifest')
+    old_metadata_path = os.path.join(directory_path, 'download_metadata.json')
+    return os.path.exists(manifest_path) or os.path.exists(old_metadata_path)
+
+
 def _scan_for_repo_folders(base_path: str) -> List[DownloadedModel]:
     """Scan a directory for repository folders and return model information."""
     models = []
@@ -142,9 +150,8 @@ def _scan_for_repo_folders(base_path: str) -> List[DownloadedModel]:
                     if os.path.isdir(subitem_path):
                         has_subdirs = True
                         # This looks like owner/repo structure
-                        # Only include if nexa.manifest exists
-                        manifest_path = os.path.join(subitem_path, 'nexa.manifest')
-                        if os.path.exists(manifest_path):
+                        # Only include if nexa.manifest or download_metadata.json exists (backward compatibility)
+                        if _has_valid_metadata(subitem_path):
                             size_bytes, files = _get_directory_size_and_files(subitem_path)
                             if files:  # Only include if there are files
                                 # Check if the download is complete
@@ -172,9 +179,8 @@ def _scan_for_repo_folders(base_path: str) -> List[DownloadedModel]:
             
             # Direct repo folder (no owner structure)
             if not has_subdirs and direct_files:
-                # Only include if nexa.manifest exists
-                manifest_path = os.path.join(item_path, 'nexa.manifest')
-                if os.path.exists(manifest_path):
+                # Only include if nexa.manifest or download_metadata.json exists (backward compatibility)
+                if _has_valid_metadata(item_path):
                     size_bytes, files = _get_directory_size_and_files(item_path)
                     if files:  # Only include if there are files
                         # Check if the download is complete
@@ -723,38 +729,57 @@ class HuggingFaceDownloader:
     
     def _fetch_and_save_metadata(self, repo_id: str, local_dir: str) -> None:
         """Fetch model info and save metadata after successful download."""
+        # Initialize metadata with defaults to ensure manifest is always created
+        old_metadata = {
+            'pipeline_tag': "text-generation", # Default to text-generation pipeline-tag
+            'download_time': datetime.now().isoformat(),
+            'avatar_url': None
+        }
+        
+        # Try to fetch additional metadata, but don't let failures prevent manifest creation
         try:
             # Fetch model info to get pipeline_tag
             info = self.api.model_info(repo_id, token=self.token)
-            pipeline_tag = info.pipeline_tag if hasattr(info, 'pipeline_tag') else None
-            
+            if hasattr(info, 'pipeline_tag'):
+                old_metadata['pipeline_tag'] = info.pipeline_tag
+        except Exception as e:
+            # 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)
-            
-            # Prepare old-style metadata for compatibility
-            old_metadata = {
-                'pipeline_tag': pipeline_tag,
-                'download_time': datetime.now().isoformat(),
-                'avatar_url': avatar_url
-            }
-            
-            # Get list of files in the directory
-            files = []
-            try:
-                for root, dirs, filenames in os.walk(local_dir):
-                    for filename in filenames:
-                        # Store relative path from the directory
-                        rel_path = os.path.relpath(os.path.join(root, filename), local_dir)
-                        files.append(rel_path)
-            except (OSError, IOError):
-                pass
-            
-            # Create and save manifest using the new utility function
+            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}")
+        
+        # CRITICAL: Always create the manifest file, regardless of metadata fetch failures
+        try:
             save_manifest_with_files_metadata(repo_id, local_dir, old_metadata)
-            
-        except Exception:
-            # Don't fail the download if metadata fetch fails
-            pass
+            print(f"[OK] Successfully created nexa.manifest for {repo_id}")
+        except Exception as e:
+            # This is critical - if manifest creation fails, we should know about it
+            print(f"ERROR: Failed to create nexa.manifest for {repo_id}: {e}")
+            # Try a fallback approach - create a minimal manifest
+            try:
+                minimal_manifest = {
+                    "Name": repo_id,
+                    "ModelType": "other",
+                    "PluginId": "unknown",
+                    "ModelFile": {},
+                    "MMProjFile": {"Name": "", "Downloaded": False, "Size": 0},
+                    "TokenizerFile": {"Name": "", "Downloaded": False, "Size": 0},
+                    "ExtraFiles": None,
+                    "pipeline_tag": old_metadata.get('pipeline_tag'),
+                    "download_time": old_metadata.get('download_time'),
+                    "avatar_url": old_metadata.get('avatar_url')
+                }
+                save_download_metadata(local_dir, minimal_manifest)
+                print(f"[OK] Created minimal nexa.manifest for {repo_id} as fallback")
+            except Exception as fallback_error:
+                print(f"CRITICAL ERROR: Could not create even minimal manifest for {repo_id}: {fallback_error}")
     
     def _download_single_file(
         self,
@@ -771,7 +796,7 @@ class HuggingFaceDownloader:
         # Check if file already exists
         local_file_path = os.path.join(file_local_dir, file_name)
         if not force_download and self._check_file_exists_and_valid(local_file_path):
-            print(f"✓ File already exists, skipping: {file_name}")
+            print(f"[SKIP] File already exists: {file_name}")
             # Stop progress tracking
             if progress_tracker:
                 progress_tracker.stop_tracking()
@@ -878,7 +903,7 @@ class HuggingFaceDownloader:
                 # Check if file already exists
                 local_file_path = os.path.join(repo_local_dir, file_name)
                 if not force_download and self._check_file_exists_and_valid(local_file_path):
-                    print(f"✓ File already exists, skipping: {file_name}")
+                    print(f"[SKIP] File already exists: {file_name}")
                     overall_progress.update(1)
                     continue
                 

nexaai/utils/progress_tracker.py

@@ -249,11 +249,11 @@ class DownloadProgressTracker:
         if known_total and total_size_raw > 0:
             # Known total size - show actual progress
             filled_width = int(bar_width * min(percentage, 100) / 100)
-            bar = '█' * filled_width + '░' * (bar_width - filled_width)
+            bar = '#' * filled_width + '-' * (bar_width - filled_width)
         else:
             # Unknown total size - show animated progress
             animation_pos = int(time.time() * 2) % bar_width
-            bar = '░' * animation_pos + '█' + '░' * (bar_width - animation_pos - 1)
+            bar = '-' * animation_pos + '#' + '-' * (bar_width - animation_pos - 1)
         
         # Format the progress line
         status = progress_data.get('status', 'unknown')

{nexaai-1.0.7.dist-info → nexaai-1.0.9.dist-info}/METADATA

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

@@ -1,6 +1,6 @@
 nexaai/__init__.py,sha256=jXdC4vv6DBK1fVewYTYSUhOOYfvf_Mk81UIeMGGIKUg,2029
-nexaai/_stub.cpython-310-darwin.so,sha256=kCIU_fKCs9GKTHji-U-bFCMzrUVmhXsIavpCgiO9otE,49832
-nexaai/_version.py,sha256=HMQ_cuen1UlESzaxkeIlsIDBtPl1Uc9t60FOoMWVLcM,138
+nexaai/_stub.cpython-310-darwin.so,sha256=kLUvt-wsOlpvaFT0rv2wWDEzwfsvIlRkBAX_SpOLavA,49832
+nexaai/_version.py,sha256=X32yXPerNFCcC9vyv0thXa3X8Z3ftJ4GXv2LbN7omGs,138
 nexaai/asr.py,sha256=NljMXDErwPNMOPaRkJZMEDka9Nk8xyur7L8i924TStY,2054
 nexaai/base.py,sha256=N8PRgDFA-XPku2vWnQIofQ7ipz3pPlO6f8YZGnuhquE,982
 nexaai/common.py,sha256=yBnIbqYaQYnfrl7IczOBh6MDibYZVxwaRJEglYcKgGs,3422
@@ -19,16 +19,16 @@ nexaai/binds/__init__.py,sha256=T9Ua7SzHNglSeEqXlfH5ymYXRyXhNKkC9z_y_bWCNMo,80
 nexaai/binds/common_bind.cpython-310-darwin.so,sha256=FF5WuJj0fNCim_HjseBQu38vL-1M5zI_7EVTD7Bs-Bc,233960
 nexaai/binds/embedder_bind.cpython-310-darwin.so,sha256=mU6hP0SyH8vcmPpC2GIr7ioK7539dsg_YbmrBdmj7l0,202032
 nexaai/binds/libcrypto.dylib,sha256=ysW8ydmDPnnNRy3AHESjJwMTFfmGDKU9eLIaiR37ca0,5091432
-nexaai/binds/libnexa_bridge.dylib,sha256=APoxDEOU9vLEoNGV8vFWDXUQxf6XlNu_ue1crcmecek,250712
+nexaai/binds/libnexa_bridge.dylib,sha256=bjoa5-DUnymTvOtIXEDIOI6Ec3FiVZbRcxBMJNuenNQ,250712
 nexaai/binds/libssl.dylib,sha256=JHPTSbRFnImmoWDO9rFdiKb0lJMT3q78VEsx-5-S0sk,889520
 nexaai/binds/llm_bind.cpython-310-darwin.so,sha256=aYqMs5VhC07RNZZgyS9JeYJJgWCl-toZOmt6vXu5yp0,183008
-nexaai/binds/nexa_llama_cpp/libggml-base.dylib,sha256=oikz7Qxzx6A0mPROq7uHTUwWn66LvvOjcdVstG-M8Fw,629528
-nexaai/binds/nexa_llama_cpp/libggml-cpu.so,sha256=WepzOOeElmdOlsoMv7loLHsj8-Qx2O9ZJPlNnX11KJI,1039800
-nexaai/binds/nexa_llama_cpp/libggml-metal.so,sha256=ssn3Bqmnu7YA_FKL513Y18gbxG8WP9Udw71DNKV34eo,713680
-nexaai/binds/nexa_llama_cpp/libggml.dylib,sha256=Z2ZvkyEEpPtHhMYap-44p9Q0M6TXJbLcMy-smR2X5sk,58336
-nexaai/binds/nexa_llama_cpp/libllama.dylib,sha256=QZBn_w32g8NAJLE1unC_qx1BCVM531LeqTUqWipt9ks,1982280
-nexaai/binds/nexa_llama_cpp/libmtmd.dylib,sha256=F1QLNlfjiECRssUtEZeuqNqej-8COYcQjMZKPAB0CGk,701504
-nexaai/binds/nexa_llama_cpp/libnexa_plugin.dylib,sha256=csMdM6l21qpj-3_4z0xGsYM1snOBg4cJPfLXOQ8oTcI,2644752
+nexaai/binds/nexa_llama_cpp/libggml-base.dylib,sha256=THe7d5VNjazmzN_8rM8GRebUFUGM8ZlXLULigOAugCg,626776
+nexaai/binds/nexa_llama_cpp/libggml-cpu.so,sha256=LH1q-tznZg4PAYfP9iAKlbCZdhdtQsgNLwqYloxN3MI,1004304
+nexaai/binds/nexa_llama_cpp/libggml-metal.so,sha256=7m61FqGB4G-I1jJjnReW9k3GK5NTxBzrDOMH3Yu_8v0,713680
+nexaai/binds/nexa_llama_cpp/libggml.dylib,sha256=rj6BFW9-20D2kfPAxmuekucHFal8PCy9BNBf_fhz-G0,58336
+nexaai/binds/nexa_llama_cpp/libllama.dylib,sha256=kTm5roZm7AioxkPMLQ2CRSnr6b9-nTLbjFy-ROXadUk,1958056
+nexaai/binds/nexa_llama_cpp/libmtmd.dylib,sha256=tUUCuTFrY6xuxxcQtG9iN7RxqnQv_rfhLKbHD2kG4e0,682264
+nexaai/binds/nexa_llama_cpp/libnexa_plugin.dylib,sha256=V4k34ZHPfx2HJEmi_XdtFCe9OhbL1PgXTrOAXeqrHTw,2532312
 nexaai/cv_impl/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nexaai/cv_impl/mlx_cv_impl.py,sha256=gKECQOv8iaWwG3bl7xeqVy2NN_9K7tYerIFzfn4eLo4,3228
 nexaai/cv_impl/pybind_cv_impl.py,sha256=uSmwBste4cT7c8DQmXzRLmzwDf773PAbXNYWW1UzVls,1064
@@ -359,14 +359,14 @@ nexaai/tts_impl/pybind_tts_impl.py,sha256=mpn44r6pfYLIl-NrEy2dXHjGtWtNCmM7HRyxiA
 nexaai/utils/avatar_fetcher.py,sha256=bWy8ujgbOiTHFCjFxTwkn3uXbZ84PgEGUkXkR3MH4bI,3821
 nexaai/utils/decode.py,sha256=61n4Zf6c5QLyqGoctEitlI9BX3tPlP2a5aaKNHbw3T4,404
 nexaai/utils/manifest_utils.py,sha256=2waOuQErodNHhoAETQqlQgXdVes-T5A4HMb8pUIN9hg,9765
-nexaai/utils/model_manager.py,sha256=xzerYqXkvRrHEqpEQvhOeg_6XQho2BvYw6ee4dlz69A,48575
+nexaai/utils/model_manager.py,sha256=wngqu5uISyoXlytJXgDldBLMnNyFWbun5g17TqyiLkA,50247
 nexaai/utils/model_types.py,sha256=-DER8L4lAUR_iLS99F0r57avwqWtuN21ug5pX2p24_E,1369
-nexaai/utils/progress_tracker.py,sha256=mTw7kaKH8BkmecYm7iBMqRHd9uUH4Ch0S8CzbpARDCk,15404
+nexaai/utils/progress_tracker.py,sha256=jdUqtmPqyhwC9uSKvQcJEYETwSt-OhP4oitdJ94614o,15394
 nexaai/utils/quantization_utils.py,sha256=4gvp6UQfSO9G1FYBwnFtQspTzH9sDbi1PBXw2t1N69M,7650
 nexaai/vlm_impl/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nexaai/vlm_impl/mlx_vlm_impl.py,sha256=od1R1mRoIgPG3NHC7JiDlcB_YJY8aklX8Em3ZkeHNpE,10734
 nexaai/vlm_impl/pybind_vlm_impl.py,sha256=5ZMFgDATthmMzjrd-vE5KX5ZAMoWPYbF_FTLz8DBKIk,8908
-nexaai-1.0.7.dist-info/METADATA,sha256=DMyi7lxZHVYv62pJQ6SemiNzIqHSGuS4-r5vHO9llJw,1197
-nexaai-1.0.7.dist-info/WHEEL,sha256=0KYp5feZ1CMUhsfFXKpSQTbSmQbXy4mv6yPPVBXg2EM,110
-nexaai-1.0.7.dist-info/top_level.txt,sha256=LRE2YERlrZk2vfuygnSzsEeqSknnZbz3Z1MHyNmBU4w,7
-nexaai-1.0.7.dist-info/RECORD,,
+nexaai-1.0.9.dist-info/METADATA,sha256=uEyssCVhZdf5ZnkVgTHJ-rihcXEH1TYkzzEmKYOyxwo,1197
+nexaai-1.0.9.dist-info/WHEEL,sha256=0KYp5feZ1CMUhsfFXKpSQTbSmQbXy4mv6yPPVBXg2EM,110
+nexaai-1.0.9.dist-info/top_level.txt,sha256=LRE2YERlrZk2vfuygnSzsEeqSknnZbz3Z1MHyNmBU4w,7
+nexaai-1.0.9.dist-info/RECORD,,