lokeessshhhh commited on
Commit
d9da53f
·
verified ·
1 Parent(s): 5269d9a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -16,8 +16,14 @@ model = OpenAIServerModel(
16
  web_search = WebSearch()
17
  final_answer = FinalAnswer()
18
 
19
- with open(os.path.join(CURRENT_DIR, "prompts.yaml"), 'r') as stream:
20
- prompt_templates = yaml.safe_load(stream)
 
 
 
 
 
 
21
 
22
  agent = CodeAgent(
23
  model=model,
 
16
  web_search = WebSearch()
17
  final_answer = FinalAnswer()
18
 
19
+ # 1) Load your YAML
20
+ with open(os.path.join(CURRENT_DIR, "prompts.yaml"), "r") as f:
21
+ prompt_templates = yaml.safe_load(f)
22
+
23
+ # 2) Flatten EVERYTHING into strings
24
+ for key, val in prompt_templates.items():
25
+ if not isinstance(val, str):
26
+ prompt_templates[key] = yaml.safe_dump(val, default_flow_style=False)
27
 
28
  agent = CodeAgent(
29
  model=model,