added support for tuple, lists and dicts (#52)

src/monkey_patch/assertion_visitor.py

@@ -195,6 +195,12 @@ def eval_expr(self, node):
             return node.n  # Assume it's a number for simplicity
         elif isinstance(node, ast.Str):
             return node.s  # Assume it's a string for simplicity
+        elif isinstance(node, ast.List):
+            return [self.eval_expr(x) for x in node.elts]  # Assume it's a list for simplicity
+        elif isinstance(node, ast.Tuple):
+            return tuple([self.eval_expr(x) for x in node.elts])  # Assume it's a tuple for simplicity
+        elif isinstance(node, ast.Dict):
+            return {self.eval_expr(k): self.eval_expr(v) for k, v in zip(node.keys, node.values)}
         else:
             raise NotImplementedError(f"Node type {type(node).__name__} not handled yet")
 

src/monkey_patch/language_models/language_modeler.py

@@ -84,7 +84,7 @@ def get_generation_case(self, args, kwargs, function_modeler, function_descripti
             return prompt, distilled_model, suitable_for_distillation, True
 
         else:
-            aligns = function_modeler.get_alignments(function_description.__hash__(), max=5)
+            aligns = function_modeler.get_alignments(function_description.__hash__(), max=16)
             examples = "\n".join([f"Inputs:\nArgs: {align['args']}\nKwargs: {align['kwargs']}\nOutput: {align['output']}" for align in aligns])
             prompt = self.construct_prompt(f, args, kwargs, examples)
             examples_token_count = approximate_token_count(examples)

src/monkey_patch/utils.py

@@ -110,7 +110,7 @@ def _deep_tuple(obj):
     :param obj:
     :return:
     """
-    if isinstance(obj, list):
+    if isinstance(obj, list) or isinstance(obj, tuple):
         return tuple(_deep_tuple(e) for e in obj)
     elif isinstance(obj, dict):
         return tuple((k, _deep_tuple(v)) for k, v in sorted(obj.items()))