Refactor provider health status and update build configurations

- Updated the `ProviderSettings` component to log the current active provider.
- Changed the provider health status to use specific names for MLX and PyTorch backends.
- Removed unnecessary exclusions from the build scripts for both PyTorch CPU and CUDA providers.
- Ensured consistency in the `.spec` files for PyTorch providers by aligning exclusion lists.
This commit is contained in:
Jamie Pine
2026-02-01 02:29:40 -08:00
parent 580179eba3
commit 595747c3d0
7 changed files with 11 additions and 9 deletions
@@ -172,7 +172,8 @@ export function ProviderSettings() {
const installedProviders = providersData?.installed || [];
// Determine current active provider
const currentProvider = activeProvider?.provider || 'auto';
const currentProvider = activeProvider?.provider;
console.log('currentProvider', currentProvider);
const selectedProvider = currentProvider as ProviderType;
return (
+7 -4
View File
@@ -85,22 +85,25 @@ class BundledProvider:
"""Get provider health status."""
backend = self._get_backend()
backend_type = get_backend_type()
model_size = None
if backend.is_loaded():
# Try to get current model size from backend
if hasattr(backend, '_current_model_size') and backend._current_model_size:
model_size = backend._current_model_size
device = None
if backend_type == "mlx":
device = "metal"
elif hasattr(backend, 'device'):
device = backend.device
# Use apple-mlx for MLX backend, pytorch-cpu for PyTorch
provider_name = "apple-mlx" if backend_type == "mlx" else "pytorch-cpu"
return ProviderHealth(
status="healthy",
provider=f"bundled-{backend_type}",
provider=provider_name,
version=None, # Provider versioning not implemented yet
model=model_size,
device=device,
-1
View File
@@ -63,7 +63,6 @@ def build_provider():
'--exclude-module', 'triton',
'--exclude-module', 'torch._dynamo',
'--exclude-module', 'torch._inductor',
'--exclude-module', 'torch.testing',
'--exclude-module', 'torch.utils.benchmark',
'--exclude-module', 'IPython',
'--exclude-module', 'matplotlib',
@@ -20,7 +20,7 @@ a = Analysis(
hookspath=[],
hooksconfig={},
runtime_hooks=[],
excludes=['torch.utils.tensorboard', 'tensorboard', 'triton', 'torch._dynamo', 'torch._inductor', 'torch.testing', 'torch.utils.benchmark', 'IPython', 'matplotlib', 'PIL', 'cv2', 'torchvision', 'torchaudio'],
excludes=['torch.utils.tensorboard', 'tensorboard', 'triton', 'torch._dynamo', 'torch._inductor', 'torch.utils.benchmark', 'IPython', 'matplotlib', 'PIL', 'cv2', 'torchvision', 'torchaudio'],
noarchive=False,
optimize=0,
)
-1
View File
@@ -65,7 +65,6 @@ def build_provider():
'--exclude-module', 'triton',
'--exclude-module', 'torch._dynamo',
'--exclude-module', 'torch._inductor',
'--exclude-module', 'torch.testing',
'--exclude-module', 'torch.utils.benchmark',
'--exclude-module', 'IPython',
'--exclude-module', 'matplotlib',
@@ -20,7 +20,7 @@ a = Analysis(
hookspath=[],
hooksconfig={},
runtime_hooks=[],
excludes=['torch.utils.tensorboard', 'tensorboard', 'triton', 'torch._dynamo', 'torch._inductor', 'torch.testing', 'torch.utils.benchmark', 'IPython', 'matplotlib', 'PIL', 'cv2', 'torchvision', 'torchaudio'],
excludes=['torch.utils.tensorboard', 'tensorboard', 'triton', 'torch._dynamo', 'torch._inductor', 'torch.utils.benchmark', 'IPython', 'matplotlib', 'PIL', 'cv2', 'torchvision', 'torchaudio'],
noarchive=False,
optimize=0,
)
Binary file not shown.