Update agent.py

agent.py

@@ -52,13 +52,14 @@ def modulus(a: int, b: int) -> int:
     """
     return a % b
 
-agent = CodeAgent(
-    tools=[
-        multiply,
-        add,
-        subtract,
-        divide,
-        modulus
-    ], 
-    model=InferenceClientModel()
-)
+def return_agent(question):
+    agent = CodeAgent(
+        tools=[
+            multiply,
+            add,
+            subtract,
+            divide,
+            modulus
+        ], 
+        model=InferenceClientModel()
+    )