Update app.py

app.py

@@ -179,13 +179,22 @@ with open("prompts.yaml", 'r') as stream:
 
 
 # Add JSON parsing instructions to the agent prompt
-prompt_templates['system'] += """
-When using API tools:
-1. Always parse JSON responses first
-2. Check for "error" keys before proceeding
-3. Use get() for optional fields
-4. Handle empty values gracefully
-"""
+if 'system' in prompt_templates:
+    prompt_templates['system'] += """
+    When using API tools: 
+    1. Always parse JSON responses first 
+    2. Check for "error" keys before proceeding 
+    3. Use get() for optional fields 
+    4. Handle empty values gracefully 
+    """
+else:
+    prompt_templates['system'] = """
+    When using API tools: 
+    1. Always parse JSON responses first 
+    2. Check for "error" keys before proceeding 
+    3. Use get() for optional fields 
+    4. Handle empty values gracefully 
+    """
     
 agent = CodeAgent(
     model=model,