HCho commited on
Commit
0e3134d
·
verified ·
1 Parent(s): 7aac577

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -3
agent.py CHANGED
@@ -90,8 +90,7 @@ system_prompt = """You are a general AI assistant. I will ask you a question. Us
90
  FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings.
91
  If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise.
92
  If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise.
93
- If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.
94
- Remember: Quality over speed. Take time to find accurate information rather than guessing."""
95
  #Using your tools to 추가하니 툴컬링 하게됨
96
  system_message = SystemMessage(content=system_prompt)
97
 
@@ -101,7 +100,7 @@ tools = [
101
 
102
 
103
  def build_graph(provider: str = "google"):
104
- llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0, api_key=google_api_key)
105
 
106
  llm_with_tools = llm.bind_tools(tools)
107
 
 
90
  FINAL ANSWER: [YOUR FINAL ANSWER]. YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings.
91
  If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise.
92
  If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise.
93
+ If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string."""
 
94
  #Using your tools to 추가하니 툴컬링 하게됨
95
  system_message = SystemMessage(content=system_prompt)
96
 
 
100
 
101
 
102
  def build_graph(provider: str = "google"):
103
+ llm = ChatGoogleGenerativeAI(model="gemini-2.5-flash", temperature=0, api_key=google_api_key)
104
 
105
  llm_with_tools = llm.bind_tools(tools)
106