Update agent.py

agent.py

@@ -1,6 +1,6 @@
 from tools import all_tools
 from dotenv import load_dotenv
-from smolagents import InferenceClientModel, CodeAgent
+from smolagents import InferenceClientModel, CodeAgent, PromptTemplates
 load_dotenv()
 
 model = InferenceClientModel(
@@ -11,12 +11,12 @@ model = InferenceClientModel(
 
 # Loads system prompt from the file
 with open("system_prompt.txt", "r", encoding="utf-8") as f:
-    system_prompt = f.read()
+    prompts = PromptTemplates(system_prompt = f.read())
 
 agent = CodeAgent(
     model=model,
     tools=all_tools,
-    prompt_templates=system_prompt,
+    prompt_templates=prompts,
     additional_authorized_imports=['numpy', 'pandas'],
     add_base_tools=True
 )