File size: 1,379 Bytes
f52d137 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 |
import { useState, useCallback, useEffect } from "react";
interface UseModelListResult {
availableModels: string[];
loading: boolean;
error: string | null;
fetchModels: (provider: string) => void;
}
export const useModelList = (provider: string): UseModelListResult => {
const [availableModels, setAvailableModels] = useState<string[]>([]);
const [loading, setLoading] = useState(false);
const [error, setError] = useState<string | null>(null);
const fetchModels = useCallback(async (selectedProvider: string) => {
if (!selectedProvider) {
setAvailableModels([]);
setError(null);
return;
}
setLoading(true);
setError(null);
try {
const response = await fetch(`/api/models/${selectedProvider}`);
if (!response.ok) {
throw new Error(`Failed to fetch models: ${response.status}`);
}
const data = await response.json();
setAvailableModels(data.models || []);
} catch (err) {
console.error("Error fetching models:", err);
setError(err instanceof Error ? err.message : "Unknown error");
setAvailableModels([]);
} finally {
setLoading(false);
}
}, []);
// Auto-fetch when provider changes
useEffect(() => {
fetchModels(provider);
}, [provider, fetchModels]);
return {
availableModels,
loading,
error,
fetchModels,
};
};
|