Update tools.py

tools.py

@@ -25,6 +25,43 @@ import wikipedia
 from bs4 import BeautifulSoup
 from langchain_core.tools import tool
 
+@tool
+def detect_non_commutative_subset(table_text: str) -> str:
+    """
+    Given a multiplication table in markdown format, return the subset of elements
+    involved in at least one non-commutative pair.
+
+    Args:
+        table_text (str): Markdown-style table defining the operation.
+
+    Returns:
+        str: Comma-separated list of elements in alphabetical order that are involved in non-commutative cases.
+    """
+    try:
+        import re
+        from collections import defaultdict
+
+        lines = [line.strip() for line in table_text.strip().splitlines() if '|' in line]
+        headers = [h.strip() for h in lines[0].split('|')[1:-1]]  # skip empty edges
+        operation = {}
+
+        for row in lines[2:]:  # skip header and separator
+            cells = [c.strip() for c in row.split('|')[1:-1]]
+            row_label = cells[0]
+            for col_index, result in enumerate(cells[1:]):
+                col_label = headers[col_index]
+                operation[(row_label, col_label)] = result
+
+        non_commutative_elements = set()
+        for a in headers:
+            for b in headers:
+                if operation.get((a, b)) != operation.get((b, a)):
+                    non_commutative_elements.update([a, b])
+
+        return ", ".join(sorted(non_commutative_elements)) if non_commutative_elements else "The operation is commutative."
+    except Exception as e:
+        return f"Error analyzing table: {str(e)}"
+
 @tool
 def extract_wikipedia_section_html(page_title: str, section_title: str, mode: str = "html") -> str:
     """
@@ -120,6 +157,7 @@ def filter_vegetables(items: list[str]) -> list[str]:
 
 # List of all tools
 all_tools = [
+    detect_non_commutative_subset,
     reverse_sentence,
     filter_vegetables,
     extract_wikipedia_section_html