added custom Levenschtein distance and already defined tools

app.py

@@ -17,6 +17,45 @@ def my_custom_tool(arg1:str, arg2:int)-> str: #it's import to specify the return
         arg2: the second argument
     """
     return "What magic will you build ?"
+    
+@tool
+def levenshtein_distance(str1:str, str2:str) -> str:
+    """
+    Calculate the Levenshtein distance (edit distance) between two strings.
+    
+    The Levenshtein distance is the minimum number of single-character operations
+    (insertions, deletions, or substitutions) required to change one string into another.
+    
+    Args:
+        str1 (str): First string
+        str2 (str): Second string
+    """
+    # Create a matrix of size (len(str1)+1) x (len(str2)+1)
+    m, n = len(str1), len(str2)
+    dp = [[0 for _ in range(n + 1)] for _ in range(m + 1)]
+    
+    # Initialize the first row and column
+    for i in range(m + 1):
+        dp[i][0] = i
+    for j in range(n + 1):
+        dp[0][j] = j
+    
+    # Fill the matrix using dynamic programming
+    for i in range(1, m + 1):
+        for j in range(1, n + 1):
+            # If characters match, no operation needed
+            if str1[i-1] == str2[j-1]:
+                dp[i][j] = dp[i-1][j-1]
+            else:
+                # Take the minimum of three operations:
+                # 1. Insertion: dp[i][j-1] + 1
+                # 2. Deletion: dp[i-1][j] + 1
+                # 3. Substitution: dp[i-1][j-1] + 1
+                dp[i][j] = 1 + min(dp[i-1][j], dp[i][j-1], dp[i-1][j-1])
+    
+    # The bottom-right cell contains the Levenshtein distance
+    return f"The edit distance between {str1} and {str1} is {dp[m][n]} operations"
+
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -55,7 +94,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+    tools=[final_answer, image_generation_tool, levenshtein_distance, get_current_time_in_timezone], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,