Update tools_agent.py

tools_agent.py

@@ -4,17 +4,32 @@ from huggingface_hub import InferenceClient
 import os
 import json
 
+from tools import convert_currency
+
 
 class Agent:
-    def __init__(self):
+    def __init__(self, tools: list[Tool]):
         """
         For more information on `huggingface_hub` Inference API support, please check the docs: https://huggingface.co/docs/huggingface_hub/v0.22.2/en/guides/inference
+        Inference API: a service that allows you to run accelerated inference on Hugging Face’s infrastructure for free. This service is a fast way to get started, 
+        test different models, and prototype AI products.
         """
         # self.client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
         self.client = InferenceClient("Qwen/Qwen2.5-Coder-32B-Instruct")
 
         """Initialize Agent with empty tool registry."""
         self.tools: Dict[str, Tool] = {}
+
+        for tool in Tools:
+            self.add_tool(tool)
+
+        self.messages = [
+            {"role": "system", "content": self.create_system_prompt()},
+        ]
+
+    def add_message(self, query_message: str) -> None:
+        """Store the new message to the memory/state of the agent."""
+        self.messages.append({"role": "user", "content": query_message})
     
     def add_tool(self, tool: Tool) -> None:
         """Register a new tool with the agent."""
@@ -164,18 +179,17 @@ Remember to use tools only when they are actually needed for the task."""
 
     def plan(self, user_query: str) -> Dict:
         """Use LLM to create a plan for tool usage."""
-        messages = [
-            {"role": "system", "content": self.create_system_prompt()},
-            {"role": "user", "content": user_query}
-        ]
+
+        self.add_message(user_query)
         
         response = self.client.chat_completion(
-            messages,
+            self.messages,
             max_tokens=512,
             temperature=0,
             top_p=0.95,
         )
         print(response.choices[0])
+        
         try:
             return json.loads(response.choices[0].message.content)
         except json.JSONDecodeError:
@@ -206,7 +220,7 @@ Results: {'. '.join(results)}"""
             return f"Error executing plan: {str(e)}"
 
 def main():
-    from tools import convert_currency
+    
     
     agent = Agent()
     agent.add_tool(convert_currency)