fix UI bug

Signed-off-by: Grant Linville <[email protected]>

ui/admin/app/components/auth-and-model-providers/ProviderConfigure.tsx

@@ -49,11 +49,6 @@ export function ProviderConfigure({ provider }: ProviderConfigureProps) {
 	useEffect(() => {
 		if (!loadingProviderId) return;
 
-		if (provider.type === "authprovider") {
-			setDialogIsOpen(false);
-			return;
-		}
-
 		const { isLoading, data } = getLoadingModelProviderModels;
 		if (isLoading) return;
 
@@ -63,7 +58,7 @@ export function ProviderConfigure({ provider }: ProviderConfigureProps) {
 			// revalidate models to get back populated models
 			mutate(ModelApiService.getModels.key());
 		}
-	}, [getLoadingModelProviderModels, loadingProviderId, provider.type]);
+	}, [getLoadingModelProviderModels, loadingProviderId]);
 
 	const handleDone = () => {
 		setDialogIsOpen(false);
@@ -110,7 +105,11 @@ export function ProviderConfigure({ provider }: ProviderConfigureProps) {
 				) : (
 					<ProviderConfigureContent
 						provider={provider}
-						onSuccess={() => setLoadingProviderId(provider.id)}
+						onSuccess={() =>
+							provider.type === "modelprovider"
+								? setLoadingProviderId(provider.id)
+								: setDialogIsOpen(false)
+						}
 					/>
 				)}
 			</DialogContent>

ui/admin/app/components/auth-and-model-providers/ProviderForm.tsx

@@ -126,9 +126,9 @@ export function ProviderForm({
 		AuthProviderApiService.configureAuthProviderById,
 		{
 			onSuccess: async () => {
-				mutate(AuthProviderApiService.revealAuthProviderById.key(provider.id));
-				mutate(AuthProviderApiService.getAuthProviders.key());
 				onSuccess();
+				mutate(AuthProviderApiService.getAuthProviders.key());
+				mutate(AuthProviderApiService.revealAuthProviderById.key(provider.id));
 			},
 		}
 	);