utils/__init__.py

from .tree_utils import (parse_functions, get_docstrings, grab_before_comments, line_chr2char, replace_function, get_root, node_str_idx, give_tree)
from .html_utils import (make_iframe, make_script, construct_embed)
from .generation import (combine_generation_kwargs, stream_generation, construct_model_context)

tree_funcs = ["parse_functions", "get_docstrings", "grab_before_comments", "line_chr2char", "replace_function", "get_root", "node_str_idx", "give_tree"]
html_funcs = ["make_iframe", "make_script", "construct_embed"]
gen_funcs = ["combine_generation_kwargs", "stream_generation", "construct_model_context"]

__all__ = tree_funcs + html_funcs + gen_funcs