iQuentin commited on
Commit
473ea0b
·
verified ·
1 Parent(s): 1514b19

Trying with prompt_templates

Browse files
Files changed (1) hide show
  1. agent.py +4 -2
agent.py CHANGED
@@ -108,13 +108,15 @@ class QAgent:
108
 
109
  # Setup imports
110
  self.imports = ["pandas", "numpy", "datetime", "json", "re", "math", "os", "requests", "csv", "urllib"]
111
-
 
 
112
  # Create CodeAgent
113
  print(f"Begin creating CodeAgent")
114
  self.agent = CodeAgent(
115
  tools=self.tools,
116
  model=self.model,
117
- system_prompt=self.system_prompt,
118
  # additional_authorized_imports=self.imports,
119
  # executor_type=executor_type,
120
  # executor_kwargs=executor_kwargs,
 
108
 
109
  # Setup imports
110
  self.imports = ["pandas", "numpy", "datetime", "json", "re", "math", "os", "requests", "csv", "urllib"]
111
+ custom_templates = PromptTemplates(
112
+ system_prompt="Votre system prompt personnalisé ici"
113
+ )
114
  # Create CodeAgent
115
  print(f"Begin creating CodeAgent")
116
  self.agent = CodeAgent(
117
  tools=self.tools,
118
  model=self.model,
119
+ prompt_templates=custom_templates,
120
  # additional_authorized_imports=self.imports,
121
  # executor_type=executor_type,
122
  # executor_kwargs=executor_kwargs,