luping85 commited on
Commit
d36e741
·
verified ·
1 Parent(s): 347686c

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -10,12 +10,12 @@ from tools import web_search_tool, repl_tool
10
 
11
  OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")
12
 
13
- chat_model = ChatOpenAI(model='gpt-4o', temperature=0, api_key=OPENAI_API_KEY)
14
  tools = [web_search_tool, repl_tool]
15
  chat_model_with_tools = chat_model.bind_tools(tools)
16
 
17
  SYSTEM_MESSAGE = """
18
- You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER].
19
  YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to
20
  write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities),
21
  and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be
 
10
 
11
  OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")
12
 
13
+ chat_model = ChatOpenAI(model='gpt-4o-mini', temperature=0, api_key=OPENAI_API_KEY)
14
  tools = [web_search_tool, repl_tool]
15
  chat_model_with_tools = chat_model.bind_tools(tools)
16
 
17
  SYSTEM_MESSAGE = """
18
+ You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer with the following template: [YOUR FINAL ANSWER].
19
  YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to
20
  write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities),
21
  and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be