diff --git a/src/components/ModelPicker.tsx b/src/components/ModelPicker.tsx
index 1054483..26901b3 100644
--- a/src/components/ModelPicker.tsx
+++ b/src/components/ModelPicker.tsx
@@ -22,6 +22,7 @@ import { useLocalLMSModels } from "@/hooks/useLMStudioModels";
import { useLanguageModelsByProviders } from "@/hooks/useLanguageModelsByProviders";
import { ChevronDown } from "lucide-react";
import { LocalModel } from "@/ipc/ipc_types";
+import { useLanguageModelProviders } from "@/hooks/useLanguageModelProviders";
interface ModelPickerProps {
selectedModel: LargeLanguageModel;
onModelSelect: (model: LargeLanguageModel) => void;
@@ -34,9 +35,13 @@ export function ModelPicker({
const [open, setOpen] = useState(false);
// Cloud models from providers
- const { data: modelsByProviders, isLoading: providersLoading } =
+ const { data: modelsByProviders, isLoading: modelsByProvidersLoading } =
useLanguageModelsByProviders();
+ const { data: providers, isLoading: providersLoading } =
+ useLanguageModelProviders();
+
+ const loading = modelsByProvidersLoading || providersLoading;
// Ollama Models Hook
const {
models: ollamaModels,
@@ -96,7 +101,7 @@ export function ModelPicker({
// Get auto provider models (if any)
const autoModels =
- !providersLoading && modelsByProviders && modelsByProviders["auto"]
+ !loading && modelsByProviders && modelsByProviders["auto"]
? modelsByProviders["auto"]
: [];
@@ -126,7 +131,7 @@ export function ModelPicker({