Update gaia_agent.py

gaia_agent.py

@@ -28,6 +28,8 @@ from customtools import (
     multiplication_tool,
     transcribe_audio,
     modulus_tool,
+    power_tool,
+    
 )
 from config import (
     OPENROUTER_API_KEY,
@@ -89,6 +91,7 @@ TOOLS = {
     "wikisearch": wikisearch,
     "transcribe_audio": transcribe_audio,
     "modulus_tool": modulus_tool,
+    "power_tool": power_tool,
 }
 
 
@@ -120,6 +123,7 @@ class Step(BaseModel):
         "subtraction_tool",
         "multiplication_tool",
         "modulus_tool",
+        "power_tool",
         "none",
     ]
     tool_input: str