Spaces:
Sleeping
Sleeping
Trying with prompt_templates
Browse files
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 |
-
|
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,
|