pateas commited on
Commit
dc70c2d
·
unverified ·
1 Parent(s): 67d6957

Update agent.py: refine final answer instructions and adjust BasicAgent model parameters

Browse files
Files changed (1) hide show
  1. agent.py +7 -7
agent.py CHANGED
@@ -31,8 +31,9 @@ If you are asked for a number only answer with the number itself, without any ad
31
  If you are asked for a string only answer with the string itself, without any additional text or formatting.
32
  If you are asked for a list only answer with the list itself, without any additional text or formatting.
33
 
34
- Finish your Answer with the following template:
35
- FINAL ANSWER: [YOUR FINAL ANSWER]
 
36
 
37
  For example, if the question is "What is the capital of France?", you should answer:
38
  FINAL ANSWER: Paris
@@ -43,8 +44,8 @@ FINAL ANSWER: 4
43
  If the question is "What is 1 divided by 2, answer with 2 digits after the decimal point?", you should answer:
44
  FINAL ANSWER: 0.50
45
 
46
- Think step by step, and use the tools provided to gather information if necessary.
47
- Report your thoughts, and finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER].
48
  """
49
 
50
 
@@ -111,7 +112,7 @@ def wikipedia_page(title: str) -> str:
111
  class BasicAgent:
112
  def __init__(self):
113
  model = OpenAIServerModel(
114
- model_id="gpt-4.1",
115
  api_key=os.getenv("OPENAI_API_KEY"),
116
  temperature=0.0,
117
  )
@@ -120,7 +121,7 @@ class BasicAgent:
120
  visitor = VisitWebpageTool(max_output_length=4000)
121
  wiki_search = WikipediaSearchTool()
122
  self.agent = CodeAgent(
123
- max_steps=5,
124
  verbosity_level=0,
125
  tools=[
126
  search,
@@ -135,7 +136,6 @@ class BasicAgent:
135
  additional_authorized_imports=["pandas", "numpy"],
136
  use_structured_outputs_internally=True,
137
  add_base_tools=True,
138
- # final_answer_checks=[is_correct_format],
139
  )
140
  logging.info(
141
  f"System prompt set for BasicAgent: {self.agent.memory.system_prompt}"
 
31
  If you are asked for a string only answer with the string itself, without any additional text or formatting.
32
  If you are asked for a list only answer with the list itself, without any additional text or formatting.
33
 
34
+ Think step by step. Report your thoughts.
35
+ Finish your answer with the following template:
36
+ FINAL ANSWER: [YOUR FINAL ANSWER].
37
 
38
  For example, if the question is "What is the capital of France?", you should answer:
39
  FINAL ANSWER: Paris
 
44
  If the question is "What is 1 divided by 2, answer with 2 digits after the decimal point?", you should answer:
45
  FINAL ANSWER: 0.50
46
 
47
+ If the quesion is "What is 10 * 10 with four digits after the decimal point?", you should answer:
48
+ FINAL ANSWER: 100.0000
49
  """
50
 
51
 
 
112
  class BasicAgent:
113
  def __init__(self):
114
  model = OpenAIServerModel(
115
+ model_id="gpt-4o-mini",
116
  api_key=os.getenv("OPENAI_API_KEY"),
117
  temperature=0.0,
118
  )
 
121
  visitor = VisitWebpageTool(max_output_length=4000)
122
  wiki_search = WikipediaSearchTool()
123
  self.agent = CodeAgent(
124
+ max_steps=10,
125
  verbosity_level=0,
126
  tools=[
127
  search,
 
136
  additional_authorized_imports=["pandas", "numpy"],
137
  use_structured_outputs_internally=True,
138
  add_base_tools=True,
 
139
  )
140
  logging.info(
141
  f"System prompt set for BasicAgent: {self.agent.memory.system_prompt}"