Update agent.py

agent.py

@@ -31,6 +31,12 @@ from langchain_community.vectorstores import FAISS
 from langchain_community.embeddings import HuggingFaceEmbeddings
 
 
+from tools import (
+    multiply, add, subtract, divide, modulus,
+    wiki_search, web_search, arvix_search
+)
+
+
 
 load_dotenv()
 
@@ -129,6 +135,20 @@ def arvix_search(query: str) -> str:
 
 
 
+# -----------------------------
+# Load configuration from YAML
+# -----------------------------
+with open("config.yaml", "r") as f:
+    config = yaml.safe_load(f)
+
+provider = config["provider"]
+#prompt_path = config["system_prompt_path"]
+enabled_tool_names = config["tools"]
+
+
+# -----------------------------
+# Load system prompt
+# -----------------------------
 # load the system prompt from the file
 with open("system_prompt.txt", "r", encoding="utf-8") as f:
     system_prompt = f.read()
@@ -137,6 +157,22 @@ with open("system_prompt.txt", "r", encoding="utf-8") as f:
 sys_msg = SystemMessage(content=system_prompt)
 
 
+# -----------------------------
+# Map tool names to functions
+# -----------------------------
+tool_map = {
+    "multiply": multiply,
+    "add": add,
+    "subtract": subtract,
+    "divide": divide,
+    "modulus": modulus,
+    "wiki_search": wiki_search,
+    "web_search": web_search,
+    "arvix_search": arvix_search,
+}
+
+tools = [tool_map[name] for name in enabled_tool_names]
+
 
 # -------------------------------
 # Step 2: Load the JSON file or tasks (Replace this part if you're loading tasks dynamically)