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,
|