Browse Source

feat: allow valve pipelines edit

Timothy J. Baek 11 months ago
parent
commit
943baad689
2 changed files with 10 additions and 2 deletions
  1. 2 2
      backend/main.py
  2. 8 0
      src/lib/components/admin/Settings/Pipelines.svelte

+ 2 - 2
backend/main.py

@@ -250,8 +250,8 @@ class PipelineMiddleware(BaseHTTPMiddleware):
                 and (
                     model["pipeline"]["pipelines"] == ["*"]
                     or any(
-                        model_id == target_model["id"]
-                        for target_model in model["pipeline"]["pipelines"]
+                        model_id == target_model_id
+                        for target_model_id in model["pipeline"]["pipelines"]
                     )
                 )
             ]

+ 8 - 0
src/lib/components/admin/Settings/Pipelines.svelte

@@ -32,6 +32,10 @@
 		const pipeline = pipelines[selectedPipelineIdx];
 
 		if (pipeline && (pipeline?.pipeline?.valves ?? false)) {
+			if (valves?.pipelines ?? false) {
+				valves.pipelines = valves.pipelines.split(',').map((v) => v.trim());
+			}
+
 			const res = await updatePipelineValves(localStorage.token, pipeline.id, valves).catch(
 				(error) => {
 					toast.error(error);
@@ -52,6 +56,10 @@
 	const getValves = async (idx) => {
 		valves_spec = await getPipelineValvesSpec(localStorage.token, pipelines[idx].id);
 		valves = await getPipelineValves(localStorage.token, pipelines[idx].id);
+
+		if (valves?.pipelines ?? false) {
+			valves.pipelines = valves.pipelines.join(',');
+		}
 	};
 
 	const setPipelines = async () => {