iQuentin commited on
Commit
8dd5413
·
verified ·
1 Parent(s): 9c254c5

Essai avec instructions=

Browse files
Files changed (1) hide show
  1. agent.py +8 -7
agent.py CHANGED
@@ -10,7 +10,7 @@ from smolagents import (
10
  # HfApiModel, # import bug from smolagents after adding duckduckgo-search in requirements
11
  tool
12
  )
13
- from smolagents.prompts import CODE_SYSTEM_PROMPT
14
  from typing import List, Dict, Any, Optional
15
 
16
 
@@ -32,7 +32,7 @@ class QAgent:
32
  """
33
  # Enhance system prompt for GAIA test
34
  extra="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]. 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 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), 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 put in the list is a number or a string."
35
- custom_prompt = f"{extra}\n\n{CODE_SYSTEM_PROMPT}"
36
 
37
 
38
  print(f"Begin QAgent init with: ")
@@ -115,16 +115,17 @@ class QAgent:
115
 
116
  # Setup imports
117
  self.imports = ["pandas", "numpy", "datetime", "json", "re", "math", "os", "requests", "csv", "urllib"]
118
- custom_templates = PromptTemplates(
119
- system_prompt=custom_prompt,
120
- planning=PlanningPromptTemplate(template="Remember: reflect deeply."),
121
- )
122
  # Create CodeAgent
123
  print(f"Begin creating CodeAgent")
124
  self.agent = CodeAgent(
125
  tools=self.tools,
126
  model=self.model,
127
- prompt_templates=custom_templates,
 
128
  # additional_authorized_imports=self.imports,
129
  # executor_type=executor_type,
130
  # executor_kwargs=executor_kwargs,
 
10
  # HfApiModel, # import bug from smolagents after adding duckduckgo-search in requirements
11
  tool
12
  )
13
+ # from smolagents.prompts import CODE_SYSTEM_PROMPT
14
  from typing import List, Dict, Any, Optional
15
 
16
 
 
32
  """
33
  # Enhance system prompt for GAIA test
34
  extra="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]. 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 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), 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 put in the list is a number or a string."
35
+ # custom_prompt = f"{extra}\n\n{CODE_SYSTEM_PROMPT}"
36
 
37
 
38
  print(f"Begin QAgent init with: ")
 
115
 
116
  # Setup imports
117
  self.imports = ["pandas", "numpy", "datetime", "json", "re", "math", "os", "requests", "csv", "urllib"]
118
+ # custom_templates = PromptTemplates(
119
+ # system_prompt=custom_prompt,
120
+ # planning=PlanningPromptTemplate(template="Remember: reflect deeply."),
121
+ # )
122
  # Create CodeAgent
123
  print(f"Begin creating CodeAgent")
124
  self.agent = CodeAgent(
125
  tools=self.tools,
126
  model=self.model,
127
+ instructions=extra,
128
+ # prompt_templates=custom_templates,
129
  # additional_authorized_imports=self.imports,
130
  # executor_type=executor_type,
131
  # executor_kwargs=executor_kwargs,