Scott Cogan commited on
Commit
b7f3749
·
1 Parent(s): a267900

fix: Handle system prompt template separately

Browse files
Files changed (1) hide show
  1. app.py +7 -5
app.py CHANGED
@@ -216,13 +216,15 @@ with open("prompts.yaml", 'r') as stream:
216
  agent_templates[key] = {}
217
  for subkey, subvalue in value.items():
218
  if isinstance(subvalue, str):
219
- # Ensure each template string is properly formatted
220
- if '{{' in subvalue or '{%' in subvalue:
221
- # Keep template syntax as is
222
  agent_templates[key][subkey] = subvalue
223
  else:
224
- # For non-template content, wrap in a simple template expression
225
- agent_templates[key][subkey] = '{{ "' + subvalue.replace('"', '\\"') + '" }}'
 
 
 
226
  else:
227
  agent_templates[key][subkey] = subvalue
228
  else:
 
216
  agent_templates[key] = {}
217
  for subkey, subvalue in value.items():
218
  if isinstance(subvalue, str):
219
+ # For system prompt, keep template syntax as is
220
+ if key == 'system_prompt':
 
221
  agent_templates[key][subkey] = subvalue
222
  else:
223
+ # For other templates, wrap non-template content in a simple template expression
224
+ if '{{' in subvalue or '{%' in subvalue:
225
+ agent_templates[key][subkey] = subvalue
226
+ else:
227
+ agent_templates[key][subkey] = '{{ "' + subvalue.replace('"', '\\"') + '" }}'
228
  else:
229
  agent_templates[key][subkey] = subvalue
230
  else: