Scott Cogan commited on
Commit
3f67850
·
1 Parent(s): 0062c3b

Fix prompt templates loading and structure

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -69,7 +69,8 @@ model = OpenAIModel(
69
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
70
 
71
  with open("prompts.yaml", 'r') as stream:
72
- prompt_templates = yaml.safe_load(stream)
 
73
 
74
  agent = CodeAgent(
75
  model=model,
 
69
  image_generation_tool = load_tool("agents-course/text-to-image", trust_remote_code=True)
70
 
71
  with open("prompts.yaml", 'r') as stream:
72
+ yaml_content = yaml.safe_load(stream)
73
+ prompt_templates = yaml_content['prompt_templates'] # Extract the prompt_templates section
74
 
75
  agent = CodeAgent(
76
  model=model,