lezaf commited on
Commit
d640a27
·
1 Parent(s): 9d428dc

Change model and improve system prompt

Browse files
Files changed (2) hide show
  1. agent.py +2 -2
  2. system_prompt.txt +2 -3
agent.py CHANGED
@@ -48,8 +48,8 @@ def build_agent(provider: str = "hf"):
48
  elif provider == "google":
49
  # Google Gemini
50
  llm = ChatGoogleGenerativeAI(
51
- # model="gemini-2.0-flash",
52
- model="gemini-2.5-flash-preview-05-20",
53
  # temperature=0,
54
  max_tokens=512,
55
  # timeout=None,
 
48
  elif provider == "google":
49
  # Google Gemini
50
  llm = ChatGoogleGenerativeAI(
51
+ model="gemini-2.0-flash",
52
+ # model="gemini-2.5-flash-preview-05-20",
53
  # temperature=0,
54
  max_tokens=512,
55
  # timeout=None,
system_prompt.txt CHANGED
@@ -1,10 +1,9 @@
1
  You are a world class expert at answering questions. The answers you will provide will be evaluated in an exact match manner to obtain a certificate in AI agents.
2
  So answer the questions with precision to get the certificate.
3
- Use this template for your answers: YOUR_FINAL_ANSWER
4
  Always output ONLY the answer and nothing else.
5
 
6
- For YOUR_FINAL_ANSWER follow strictly the instructions below:
7
- * YOUR_FINAL_ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings.
8
  * If you are asked for a number, don't use comma to write your number neither use units such as: [$, meters (m), centimeters (cm), oz] or any other unit of measurement
9
  or percent sign unless specified otherwise.
10
  * 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.
 
1
  You are a world class expert at answering questions. The answers you will provide will be evaluated in an exact match manner to obtain a certificate in AI agents.
2
  So answer the questions with precision to get the certificate.
 
3
  Always output ONLY the answer and nothing else.
4
 
5
+ For your final answer follow strictly the instructions below:
6
+ * Your final answer should be a number OR as few words as possible OR a comma separated list of numbers and/or strings.
7
  * If you are asked for a number, don't use comma to write your number neither use units such as: [$, meters (m), centimeters (cm), oz] or any other unit of measurement
8
  or percent sign unless specified otherwise.
9
  * 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.