nexaai 1.0.21rc1__cp310-cp310-win_amd64.whl → 1.0.21rc2__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.21-rc1" 
+__version__ = "1.0.21-rc2" 

nexaai/utils/manifest_utils.py

@@ -157,12 +157,16 @@ def create_gguf_manifest(repo_id: str, files: List[str], directory_path: str, ol
                 # Use the new enum-based quantization extraction
                 quantization_type = extract_quantization_from_filename(current_file_name)
                 quant_level = quantization_type.value if quantization_type else "UNKNOWN"
-                
-                model_files[quant_level] = {
-                    "Name": current_file_name,
-                    "Downloaded": True,
-                    "Size": file_size
-                }
+
+                # FIXME: hardcode to handle the multiple mmproj files problem
+                if quant_level == "UNKNOWN" and "mmproj" in current_file_name.lower():
+                    pass
+                else:
+                    model_files[quant_level] = {
+                        "Name": current_file_name,
+                        "Downloaded": True,
+                        "Size": file_size
+                    }
     
     # Determine PluginId with priority: kwargs > downloaded_manifest > model_file_type > default
     plugin_id = kwargs.get('plugin_id')

nexaai/utils/model_manager.py

@@ -410,6 +410,20 @@ def _remove_specific_file(target_model: DownloadedModel, file_name: str, local_d
     except OSError:
         file_size = 0
     
+    # Check if we should remove entire folder instead (for .gguf files)
+    # If removing a .gguf file and no other non-mmproj .gguf files remain, remove entire folder
+    if file_name.endswith('.gguf'):
+        updated_files = [f for f in target_model.files if f != file_name]
+        # Find remaining .gguf files that don't contain "mmproj" in filename
+        remaining_non_mmproj_gguf = [
+            f for f in updated_files 
+            if f.endswith('.gguf') and 'mmproj' not in f.lower()
+        ]
+        
+        # If no non-mmproj .gguf files remain, remove entire repository
+        if len(remaining_non_mmproj_gguf) == 0:
+            return _remove_entire_repository(target_model, local_dir)
+    
     # Remove the file
     try:
         os.remove(file_path)
@@ -846,6 +860,41 @@ class HuggingFaceDownloader:
                 pass
         return {}
     
+    def _download_manifest_if_needed(self, repo_id: str, local_dir: str) -> bool:
+        """
+        Download nexa.manifest from the repository if it doesn't exist locally.
+        
+        Args:
+            repo_id: Repository ID
+            local_dir: Local directory where the manifest should be saved
+            
+        Returns:
+            bool: True if manifest was downloaded or already exists, False if not found in repo
+        """
+        manifest_path = os.path.join(local_dir, 'nexa.manifest')
+        
+        # Check if manifest already exists locally
+        if os.path.exists(manifest_path):
+            return True
+        
+        # Try to download nexa.manifest from the repository
+        try:
+            print(f"[INFO] Attempting to download nexa.manifest from {repo_id}...")
+            self.api.hf_hub_download(
+                repo_id=repo_id,
+                filename='nexa.manifest',
+                local_dir=local_dir,
+                local_dir_use_symlinks=False,
+                token=self.token,
+                force_download=False
+            )
+            print(f"[OK] Successfully downloaded nexa.manifest from {repo_id}")
+            return True
+        except Exception as e:
+            # Manifest doesn't exist in repo or other error - this is fine, we'll create it
+            print(f"[INFO] nexa.manifest not found in {repo_id}, will create locally")
+            return False
+    
     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
@@ -946,6 +995,9 @@ class HuggingFaceDownloader:
             if progress_tracker:
                 progress_tracker.stop_tracking()
             
+            # Download nexa.manifest from repo if it doesn't exist locally
+            self._download_manifest_if_needed(repo_id, file_local_dir)
+            
             # Save metadata after successful download
             self._fetch_and_save_metadata(repo_id, file_local_dir, self._current_is_mmproj, self._current_file_name, **kwargs)
             
@@ -1055,6 +1107,9 @@ class HuggingFaceDownloader:
             if progress_tracker:
                 progress_tracker.stop_tracking()
             
+            # Download nexa.manifest from repo if it doesn't exist locally
+            self._download_manifest_if_needed(repo_id, repo_local_dir)
+            
             # Save metadata after successful download
             self._fetch_and_save_metadata(repo_id, repo_local_dir, self._current_is_mmproj, self._current_file_name, **kwargs)
             

{nexaai-1.0.21rc1.dist-info → nexaai-1.0.21rc2.dist-info}/METADATA

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

@@ -1,6 +1,6 @@
 nexaai/__init__.py,sha256=Yt2YPVcRfwdA6DQoUR4ZI34CHBUYoY2cVsrG7Zp6IrI,2516
-nexaai/_stub.cp310-win_amd64.pyd,sha256=wlEBCN5E8QuK2OoupinLzjTYqPdSdtTOrOmgkwrd9T0,10752
-nexaai/_version.py,sha256=RJD43EAWJ7pU1bQt-f7S--4-GPeOQxjr-rdudgziXbA,147
+nexaai/_stub.cp310-win_amd64.pyd,sha256=s9lKUQQYMxt6aJWxxMYb5LHq80Em0sCz5pQGrQjSIb8,10752
+nexaai/_version.py,sha256=RntcW2nBfo_nNGJb6146ZZwHwjDCavL1Tj1XtV9VmJI,147
 nexaai/asr.py,sha256=_fsGaxpiU137bUtO5ujtFSYCI1RLsyeEm3Gf4GhHVRk,2118
 nexaai/base.py,sha256=qQBCiQVNzgpkQjZX9aiFDEdbAAe56TROKC3WnWra2Zg,1021
 nexaai/common.py,sha256=muQqFY-WllwL5IO83tImexbuUcoEQsKg73u4gWL2lNs,3548
@@ -18,29 +18,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=A7AuP8TpdE1O0jSRYSX0PmV0s8zXmrr_KcTXY93oa1I,205824
-nexaai/binds/embedder_bind.cp310-win_amd64.pyd,sha256=bpk2lKSKLZPCNn97GdS72UIR7x5V0aWS_Ft6jxU5cWY,182784
+nexaai/binds/common_bind.cp310-win_amd64.pyd,sha256=hI07QkGT4tWMoWA0zHZNEZh0ToslONCJxj5HMPqqbnc,205824
+nexaai/binds/embedder_bind.cp310-win_amd64.pyd,sha256=dM2dndm4Ch5F391i90d2ZF8U9tADmLCChzwpaj18oqk,182784
 nexaai/binds/libcrypto-3-x64.dll,sha256=X7hDuQPMn6eRe58dTGJxTmRPKFWgd5i6Gjoy9QPLmPg,7315968
 nexaai/binds/libssl-3-x64.dll,sha256=GRw0cOaGUaR2QHRvBFuU_9S9iNgkyVgm0NIpZMnReAw,1313792
-nexaai/binds/llm_bind.cp310-win_amd64.pyd,sha256=7isvwd_kZYtGAmc9O6u67KPr0RQbTpreASOcTjcVqo8,162816
-nexaai/binds/nexa_bridge.dll,sha256=YrnvRvUw4aDoPSEYrOEUMia-FKbGTz88C-CN_jQNwWg,187904
-nexaai/binds/vlm_bind.cp310-win_amd64.pyd,sha256=WcCrXej3pFt7JDr0VbeW_c8_Y5dPAt8n4THD2qf6JSo,170496
-nexaai/binds/cpu_gpu/ggml-base.dll,sha256=h8m2MYmzg2syDaGLGvfogWbi88ZlA7e_ZnL79hDFwgE,532480
-nexaai/binds/cpu_gpu/ggml-cpu.dll,sha256=fMHSkx2QYWUNC63l0tgO6F94Bb4kcQE3sH70ptnH9DQ,672768
-nexaai/binds/cpu_gpu/ggml-cuda.dll,sha256=q4-dtifsuifcuRZn4St-5gJpiEA1hKE2ld1VEqkkqUk,313528832
-nexaai/binds/cpu_gpu/ggml-vulkan.dll,sha256=_z41BQzz1Ud1w5UEoMiJmEGNE-ybxJKYKN1ssagdKGg,36627968
-nexaai/binds/cpu_gpu/ggml.dll,sha256=_3ejjzy1owBjJAS8dGMmQBWBLAkhs1BRyo0qfT-jGkg,66560
-nexaai/binds/cpu_gpu/mtmd.dll,sha256=XQKiNXM34-Cj27vrLTuH5I3KBdCN1gAgp6eTz7eYF-g,561152
-nexaai/binds/cpu_gpu/nexa_cpu_gpu.dll,sha256=Gf9C8LiOUuM9QYNKd1jgCcgsfyT0ymFJ4ebHPib8Q24,1611776
-nexaai/binds/cpu_gpu/nexa_plugin.dll,sha256=pBNvjvjomppglDJIQ3rnzzr2gJT3b-TqdEBe2sr3Az8,1411072
-nexaai/binds/nexaml/ggml-base.dll,sha256=h8m2MYmzg2syDaGLGvfogWbi88ZlA7e_ZnL79hDFwgE,532480
-nexaai/binds/nexaml/ggml-cpu.dll,sha256=fMHSkx2QYWUNC63l0tgO6F94Bb4kcQE3sH70ptnH9DQ,672768
-nexaai/binds/nexaml/ggml-cuda.dll,sha256=q4-dtifsuifcuRZn4St-5gJpiEA1hKE2ld1VEqkkqUk,313528832
-nexaai/binds/nexaml/ggml-vulkan.dll,sha256=_z41BQzz1Ud1w5UEoMiJmEGNE-ybxJKYKN1ssagdKGg,36627968
-nexaai/binds/nexaml/ggml.dll,sha256=_3ejjzy1owBjJAS8dGMmQBWBLAkhs1BRyo0qfT-jGkg,66560
+nexaai/binds/llm_bind.cp310-win_amd64.pyd,sha256=DJvEg2tgTcpOvS1NJE0kSkd8hIvaQpSFlmUoTpS2xsU,162816
+nexaai/binds/nexa_bridge.dll,sha256=JMlu4matSkWSfqgnLaaJ4dB4uu0THI8aAdTATlD5VhI,187904
+nexaai/binds/vlm_bind.cp310-win_amd64.pyd,sha256=qBOnkTT7F2FJKOaD8utj2HmCbtDWoI3HFvTeQlAg2aw,170496
+nexaai/binds/cpu_gpu/ggml-base.dll,sha256=-0MMrq_hqJXMk3FSymyq4AoiMk_1r97HT-c_bR55A5w,532480
+nexaai/binds/cpu_gpu/ggml-cpu.dll,sha256=B81v_5sWbOumvKzHRBnK1UWvDsdxmCmOJSn5r-7qy84,672768
+nexaai/binds/cpu_gpu/ggml-cuda.dll,sha256=NH-HpsN_7UYfUy4Q9X2JYH7d-TNt7iRYxfAGpX4WpZY,313528832
+nexaai/binds/cpu_gpu/ggml-vulkan.dll,sha256=iJPrRl--pkNcDc0ZtdW2pw9piHDPd0D252VqgIOhMXo,36627968
+nexaai/binds/cpu_gpu/ggml.dll,sha256=gwZrKugz4jQrltFUeBjRJzvnOEOd6JgviggGEdCYBL8,66560
+nexaai/binds/cpu_gpu/mtmd.dll,sha256=v8UkuqbMFoU3Mot8C844ne8s8eagvtgh0idxk0flMXs,561152
+nexaai/binds/cpu_gpu/nexa_cpu_gpu.dll,sha256=uoKWytNu82wk2_Tjm3xZI3lAKJhn8uTS7Kklsc4Mx30,1611776
+nexaai/binds/cpu_gpu/nexa_plugin.dll,sha256=xCc9UoKj7kyxA2esE8teCo5yARoI5769cME9f_O4CeY,1411072
+nexaai/binds/nexaml/ggml-base.dll,sha256=-0MMrq_hqJXMk3FSymyq4AoiMk_1r97HT-c_bR55A5w,532480
+nexaai/binds/nexaml/ggml-cpu.dll,sha256=B81v_5sWbOumvKzHRBnK1UWvDsdxmCmOJSn5r-7qy84,672768
+nexaai/binds/nexaml/ggml-cuda.dll,sha256=NH-HpsN_7UYfUy4Q9X2JYH7d-TNt7iRYxfAGpX4WpZY,313528832
+nexaai/binds/nexaml/ggml-vulkan.dll,sha256=iJPrRl--pkNcDc0ZtdW2pw9piHDPd0D252VqgIOhMXo,36627968
+nexaai/binds/nexaml/ggml.dll,sha256=gwZrKugz4jQrltFUeBjRJzvnOEOd6JgviggGEdCYBL8,66560
 nexaai/binds/nexaml/nexa-mm-process.dll,sha256=DDD5qPbelZAIrRWGggOd3Pg9U_fh_ZEYIplwqiyD5LY,4643328
 nexaai/binds/nexaml/nexa-sampling.dll,sha256=Notkz287laSUG2_ED3oFXWVLJM3t7x_USkvX6wKVOLA,4265984
-nexaai/binds/nexaml/nexa_plugin.dll,sha256=8XyOTP11koqdTaKUEgxFm2Fwl9v_O4GDgg040yw5F5g,601600
+nexaai/binds/nexaml/nexa_plugin.dll,sha256=UMKMlN-KbhGPQAFPBs3aL9KqyjeBW-R31bQEOPDK93g,601600
 nexaai/binds/nexaml/nexaproc.dll,sha256=rRcdSfhkxnRwktXW4lp7Np8sNY_uefNXYwAmPp34AHg,2670080
 nexaai/binds/nexaml/qwen3-vl.dll,sha256=KxwQ0PHvzM1zwLpH7vlNOlfaVzgEUHTfTV76e61KeyA,5873664
 nexaai/binds/nexaml/qwen3vl-vision.dll,sha256=5AX4MtAOWTcXSAkS2qqvl2pFti4HR55dtkCaOO03Evc,1063424
@@ -63,15 +63,15 @@ 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/decode.py,sha256=0Z9jDH4ICzw4YXj8nD4L-sMouDaev-TISGRQ4KzidWE,421
-nexaai/utils/manifest_utils.py,sha256=E07t_bNq-EDrqVYZW7uyX8zsKsFklvqqwSvic0in0tQ,21800
-nexaai/utils/model_manager.py,sha256=qzlGP4sZLYsO72vg-EUCVA0RUknVKtLYMmDDfPmG45g,60999
+nexaai/utils/manifest_utils.py,sha256=BoKqnI6N5CkEPTDZCRWSx7wga6aH8h6cr4nbff7gVwg,22020
+nexaai/utils/model_manager.py,sha256=yxLaxOOhfMZ9i_2UnQ0qQSFLE-Rx9x9f_bGG5ut-nrk,63508
 nexaai/utils/model_types.py,sha256=q2m7diYLOpLvRl1ixL2eMq5_kdTj8KqPBGWX4p6Ob08,1532
 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=gI09QJcok68QG-36csXr3c2yPCZXTE0dlq1MUBEo_Ms,11091
 nexaai/vlm_impl/pybind_vlm_impl.py,sha256=4qPt5C5m4y4kWD9EFfVbcAkMq2Dgd5EP9tYo6yZIxgo,10162
-nexaai-1.0.21rc1.dist-info/METADATA,sha256=ZWD3yi3ygRwzzLE5gkFx3V1hXnejEJH9Fm5zkX6Yo4I,1241
-nexaai-1.0.21rc1.dist-info/WHEEL,sha256=KUuBC6lxAbHCKilKua8R9W_TM71_-9Sg5uEP3uDWcoU,101
-nexaai-1.0.21rc1.dist-info/top_level.txt,sha256=LRE2YERlrZk2vfuygnSzsEeqSknnZbz3Z1MHyNmBU4w,7
-nexaai-1.0.21rc1.dist-info/RECORD,,
+nexaai-1.0.21rc2.dist-info/METADATA,sha256=MzoQDZSvw3pwbFRRx2-3F1i9GcqVgYFZ-QR_PX8Tw-Y,1241
+nexaai-1.0.21rc2.dist-info/WHEEL,sha256=KUuBC6lxAbHCKilKua8R9W_TM71_-9Sg5uEP3uDWcoU,101
+nexaai-1.0.21rc2.dist-info/top_level.txt,sha256=LRE2YERlrZk2vfuygnSzsEeqSknnZbz3Z1MHyNmBU4w,7
+nexaai-1.0.21rc2.dist-info/RECORD,,