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

fix: Ensure consistent template formatting for all content

Browse files
Files changed (1) hide show
  1. app.py +11 -7
app.py CHANGED
@@ -216,22 +216,26 @@ with open("prompts.yaml", 'r') as stream:
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:
231
  if isinstance(value, str):
232
  if '{{' in value or '{%' in value:
 
 
 
233
  agent_templates[key] = value
234
  else:
 
235
  agent_templates[key] = '{{ "' + value.replace('"', '\\"') + '" }}'
236
  else:
237
  agent_templates[key] = value
 
216
  agent_templates[key] = {}
217
  for subkey, subvalue in value.items():
218
  if isinstance(subvalue, str):
219
+ # For all templates, ensure they are valid Jinja2 templates
220
+ if '{{' in subvalue or '{%' in subvalue:
221
+ # If template syntax exists, ensure proper spacing
222
+ subvalue = re.sub(r'{{([^{}]+)}}', lambda m: '{{ ' + m.group(1).strip() + ' }}', subvalue)
223
+ subvalue = re.sub(r'{%([^{}]+)%}', lambda m: '{% ' + m.group(1).strip() + ' %}', subvalue)
224
  agent_templates[key][subkey] = subvalue
225
  else:
226
+ # For non-template content, wrap in a simple template expression
227
+ agent_templates[key][subkey] = '{{ "' + subvalue.replace('"', '\\"') + '" }}'
 
 
 
228
  else:
229
  agent_templates[key][subkey] = subvalue
230
  else:
231
  if isinstance(value, str):
232
  if '{{' in value or '{%' in value:
233
+ # If template syntax exists, ensure proper spacing
234
+ value = re.sub(r'{{([^{}]+)}}', lambda m: '{{ ' + m.group(1).strip() + ' }}', value)
235
+ value = re.sub(r'{%([^{}]+)%}', lambda m: '{% ' + m.group(1).strip() + ' %}', value)
236
  agent_templates[key] = value
237
  else:
238
+ # For non-template content, wrap in a simple template expression
239
  agent_templates[key] = '{{ "' + value.replace('"', '\\"') + '" }}'
240
  else:
241
  agent_templates[key] = value