Pārlūkot izejas kodu

Restore image steps setting when Settings modal is presented

Justin Firth 1 gadu atpakaļ
vecāks
revīzija
af1675813f

+ 32 - 0
src/lib/apis/images/index.ts

@@ -198,6 +198,38 @@ export const updateImageSize = async (token: string = '', size: string) => {
 	return res.IMAGE_SIZE;
 };
 
+export const getImageSteps = async (token: string = '') => {
+	let error = null;
+
+	const res = await fetch(`${IMAGES_API_BASE_URL}/steps`, {
+		method: 'GET',
+		headers: {
+			Accept: 'application/json',
+			'Content-Type': 'application/json',
+			...(token && { authorization: `Bearer ${token}` })
+		}
+	})
+		.then(async (res) => {
+			if (!res.ok) throw await res.json();
+			return res.json();
+		})
+		.catch((err) => {
+			console.log(err);
+			if ('detail' in err) {
+				error = err.detail;
+			} else {
+				error = 'Server connection failed';
+			}
+			return null;
+		});
+
+	if (error) {
+		throw error;
+	}
+
+	return res.IMAGE_STEPS;
+};
+
 export const updateImageSteps = async (token: string = '', steps: number) => {
 	let error = null;
 

+ 2 - 0
src/lib/components/chat/Settings/Images.svelte

@@ -13,6 +13,7 @@
 		updateAUTOMATIC1111Url,
 		updateDefaultDiffusionModel,
 		updateImageSize,
+		getImageSteps,
 		updateImageSteps
 	} from '$lib/apis/images';
 	import { getBackendConfig } from '$lib/apis';
@@ -85,6 +86,7 @@
 
 			if (enableImageGeneration && AUTOMATIC1111_BASE_URL) {
 				imageSize = await getImageSize(localStorage.token);
+				steps = await getImageSteps(localStorage.token);
 				getModels();
 			}
 		}