fix recursive call in get_root

app.py

@@ -235,9 +235,9 @@ def alter_body(old_code, func_id, funcs_list: list, prompt="", temperature=0.2,
     func_start_idx = line_chr2char(old_code, func_node.start_point[0], func_node.start_point[1])
     identifier_str = func_node.child_by_field_name("type").text.decode() + " " + func_node.child_by_field_name("declarator").text.decode() #func_start_idx:body_start_idx?
     body_node = func_node.child_by_field_name("body")
-    # body_start_idx, body_end_idx = node_str_idx(body_node) #can cause index error, needs better testing!
-    body_start_idx = line_chr2char(old_code, body_node.start_point[0], body_node.start_point[1])
-    body_end_idx = line_chr2char(old_code, body_node.end_point[0], body_node.end_point[1])
+    body_start_idx, body_end_idx = node_str_idx(body_node) #can cause index error, needs better testing!
+    # body_start_idx = line_chr2char(old_code, body_node.start_point[0], body_node.start_point[1])
+    # body_end_idx = line_chr2char(old_code, body_node.end_point[0], body_node.end_point[1])
     print(f"{old_code[body_start_idx:body_end_idx]=}")
     model_context = identifier_str # base case
     # add any comments at the beginning of the function to the model_context

utils/tree_utils.py

@@ -18,11 +18,12 @@ def replace_function(old_func_node, new_func_node):
 
 def get_root(node):
     """
-    returns the root node of a node
+    returns the root node the tree of the given node (recursively)
     """
-    while node.parent != None:
-        node = node.parent
-    return node.parent #still need to call parent here
+    if node.parent is None:
+        return node
+    else:
+        return get_root(node.parent)
 
 def node_str_idx(node):
     """