jc7k commited on
Commit
8083d45
·
1 Parent(s): 13b2abb

Remove custom prompts

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -26,8 +26,8 @@ class BasicAgent:
26
  )
27
 
28
  final_answer = FinalAnswerTool()
29
- with open("prompts.yaml", "r") as stream:
30
- prompt_templates = yaml.safe_load(stream)
31
 
32
  # agent = ToolCallingAgent(
33
  agent = CodeAgent(
@@ -37,7 +37,8 @@ class BasicAgent:
37
  add_base_tools=True,
38
  max_steps=5,
39
  model=model,
40
- prompt_templates=prompt_templates)
 
41
 
42
  self.agent = agent
43
  print(f"Agent initialized with model ID: {os.environ['MODEL_ID']}")
 
26
  )
27
 
28
  final_answer = FinalAnswerTool()
29
+ # with open("prompts.yaml", "r") as stream:
30
+ # prompt_templates = yaml.safe_load(stream)
31
 
32
  # agent = ToolCallingAgent(
33
  agent = CodeAgent(
 
37
  add_base_tools=True,
38
  max_steps=5,
39
  model=model,
40
+ # prompt_templates=prompt_templates
41
+ )
42
 
43
  self.agent = agent
44
  print(f"Agent initialized with model ID: {os.environ['MODEL_ID']}")