Update tools.py

tools.py

@@ -14,6 +14,51 @@ import numpy as np
 from scipy.io import wavfile
 import pandas as pd
 
+import numpy as np
+import pandas as pd
+from smolagent.tools import Tool  # à adapter selon ton vrai import
+
+class is_commutative(Tool):
+    name = "commutative"
+    description = "Performs a study on a table set to see if it is commutative."
+    inputs = {
+        "set": {
+            "type": "array",
+            "items": {"type": "string"},
+            "description": "The set defined, for example : {'a','b','c'}."
+        },
+        "table": {
+            "type": "string",
+            "description": "The table in markdown format with rows separated by '\n'. Give only the table after the '|---|---|---|---|---|---|' symbol, starting with the '\n', ending with '\n\n', as a string."
+        }
+    }
+    output_type = "string"
+
+    def forward(self, set, table):
+        set_0=list(sorted(set))
+        table_0=np.empty((len(set_0)+1,len(set_0)+1),dtype='<U5')
+            
+        for i in range(len(set_0)):
+            table_0[0,i+1]=set_0[i]
+        
+        k=1
+        l=0
+        for item in table:
+            if item=='\n' or item=='|':
+                table_0=table_0
+            else:
+                table_0[k,l]=item
+                l=l+1
+                if(l==6):
+                    l=0
+                    k=k+1
+        
+        for i in range(1,len(set_0)+1):
+            for j in range(1,len(set_0)+1):
+                if (table_0[i,j]!=table_0[j,i]):
+                    return f"Not commutative, because of the elements: {set_0[i-1]} and {set_0[j-1]}"
+        return "It is commutative"
+        
 class Web_research(Tool):
     name="web_search"
     description = "Web search on a specific topic."