at1300 commited on
Commit
486800b
·
verified ·
1 Parent(s): 5d7a4c6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -64,9 +64,10 @@ class MasterAgent:
64
  except Exception as e:
65
  print(f"Error instantiating agent: {e}")
66
  return f"Error initializing agent: {e}", None
67
-
68
- with open("prompts.yaml", 'r') as stream:
69
- prompt_templates = yaml.safe_load(stream)
 
70
 
71
  print("Starting instantiation of master")
72
  self.master_agent = CodeAgent(
@@ -78,7 +79,7 @@ class MasterAgent:
78
  add_base_tools=True,
79
  planning_interval=5,
80
  max_steps=15,
81
- prompt_templates=prompt_templates,
82
  managed_agents=[coder.agent],
83
  additional_authorized_imports=[
84
  "geopandas",
 
64
  except Exception as e:
65
  print(f"Error instantiating agent: {e}")
66
  return f"Error initializing agent: {e}", None
67
+
68
+ # Seems like this format was updated, we'll use the new default for now.
69
+ #with open("prompts.yaml", 'r') as stream:
70
+ # prompt_templates = yaml.safe_load(stream)
71
 
72
  print("Starting instantiation of master")
73
  self.master_agent = CodeAgent(
 
79
  add_base_tools=True,
80
  planning_interval=5,
81
  max_steps=15,
82
+ #prompt_templates=prompt_templates,
83
  managed_agents=[coder.agent],
84
  additional_authorized_imports=[
85
  "geopandas",