modify save yaml

app.py

@@ -13,15 +13,15 @@ from metagpt.ext.spo.components.optimizer import PromptOptimizer
 from metagpt.ext.spo.utils.llm_client import SPO_LLM, RequestType
 
 
-
 def get_user_workspace():
     if "user_id" not in st.session_state:
         st.session_state.user_id = str(uuid.uuid4())
-    
+
     workspace_dir = Path("workspace") / st.session_state.user_id
     workspace_dir.mkdir(parents=True, exist_ok=True)
     return workspace_dir
 
+
 def cleanup_workspace(workspace_dir: Path) -> None:
     try:
         if workspace_dir.exists():
@@ -30,6 +30,7 @@ def cleanup_workspace(workspace_dir: Path) -> None:
     except Exception as e:
         _logger.error(f"Error cleaning up workspace: {e}")
 
+
 def get_template_path(template_name: str, is_new_template: bool = False) -> str:
     """
     Get template file path
@@ -48,23 +49,24 @@ def get_template_path(template_name: str, is_new_template: bool = False) -> str:
         # Use root settings path for existing templates
         return f"{template_name}.yaml"
 
+
 def get_all_templates() -> List[str]:
     """
     Get list of all available templates (both default and user-specific)
     :return: List of template names
     """
     settings_path = Path("metagpt/ext/spo/settings")
-    
+
     # Get default templates
     templates = [f.stem for f in settings_path.glob("*.yaml")]
-    
+
     # Get user-specific templates if user_id exists
     if "user_id" in st.session_state:
         user_path = settings_path / st.session_state.user_id
         if user_path.exists():
             user_templates = [f"{st.session_state.user_id}/{f.stem}" for f in user_path.glob("*.yaml")]
             templates.extend(user_templates)
-    
+
     return sorted(list(set(templates)))
 
 
@@ -75,22 +77,25 @@ def load_yaml_template(template_path: Path) -> Dict:
     return {"prompt": "", "requirements": "", "count": None, "qa": [{"question": "", "answer": ""}]}
 
 
-def save_yaml_template(template_path: Path, data: Dict) -> None:
-    template_format = {
-        "prompt": str(data.get("prompt", "")),
-        "requirements": str(data.get("requirements", "")),
-        "count": data.get("count"),
-        "qa": [
-            {"question": str(qa.get("question", "")).strip(), "answer": str(qa.get("answer", "")).strip()}
-            for qa in data.get("qa", [])
-        ],
-    }
-
-    template_path.parent.mkdir(parents=True, exist_ok=True)
-
-    with open(template_path, "w", encoding="utf-8") as f:
-        yaml.dump(template_format, f, allow_unicode=True, sort_keys=False, default_flow_style=False, indent=2)
-
+def save_yaml_template(template_path: Path, data: Dict, is_new: bool) -> None:
+    
+    if is_new:
+        template_format = {
+            "prompt": str(data.get("prompt", "")),
+            "requirements": str(data.get("requirements", "")),
+            "count": data.get("count"),
+            "qa": [
+                {"question": str(qa.get("question", "")).strip(), "answer": str(qa.get("answer", "")).strip()}
+                for qa in data.get("qa", [])
+            ],
+        }
+    
+        template_path.parent.mkdir(parents=True, exist_ok=True)
+    
+        with open(template_path, "w", encoding="utf-8") as f:
+            yaml.dump(template_format, f, allow_unicode=True, sort_keys=False, default_flow_style=False, indent=2)
+    else:
+        pass
 
 def display_optimization_results(result_data):
     for result in result_data:
@@ -148,7 +153,7 @@ def main():
         template_mode = st.radio("Template Mode", ["Use Existing", "Create New"])
 
         existing_templates = get_all_templates()
-    
+
         if template_mode == "Use Existing":
             template_name = st.selectbox("Select Template", existing_templates)
             is_new_template = False
@@ -190,7 +195,7 @@ def main():
         settings_path = Path("metagpt/ext/spo/settings")
 
         template_path = settings_path / template_real_name
-        
+
         template_data = load_yaml_template(template_path)
 
         if "current_template" not in st.session_state or st.session_state.current_template != template_name:
@@ -233,7 +238,7 @@ def main():
         if st.button("Save Template"):
             new_template_data = {"prompt": prompt, "requirements": requirements, "count": None, "qa": new_qas}
 
-            save_yaml_template(template_path, new_template_data)
+            save_yaml_template(template_path, new_template_data, is_new_template)
 
             st.session_state.qas = new_qas
             st.success(f"Template saved to {template_path}")
@@ -271,9 +276,12 @@ def main():
             try:
                 # Initialize LLM
                 SPO_LLM.initialize(
-                    optimize_kwargs={"model": opt_model, "temperature": opt_temp, "base_url": base_url, "api_key": api_key},
-                    evaluate_kwargs={"model": eval_model, "temperature": eval_temp, "base_url": base_url, "api_key": api_key},
-                    execute_kwargs={"model": exec_model, "temperature": exec_temp, "base_url": base_url, "api_key": api_key},
+                    optimize_kwargs={"model": opt_model, "temperature": opt_temp, "base_url": base_url,
+                                     "api_key": api_key},
+                    evaluate_kwargs={"model": eval_model, "temperature": eval_temp, "base_url": base_url,
+                                     "api_key": api_key},
+                    execute_kwargs={"model": exec_model, "temperature": exec_temp, "base_url": base_url,
+                                    "api_key": api_key},
                 )
 
                 # Create optimizer instance