diff --git a/tools/__init__.py b/tools/__init__.py
index 1c1e383..901f13c 100644
--- a/tools/__init__.py
+++ b/tools/__init__.py
@@ -1,2 +1,4 @@
 from .parse_markdown import parse_markdown
-from .scan_doc_dir import *
\ No newline at end of file
+from .scan_doc_dir import *
+from .json_example import generate_example_json
+from .model import get_model_name
\ No newline at end of file
diff --git a/tools/model.py b/tools/model.py
new file mode 100644
index 0000000..4c84d05
--- /dev/null
+++ b/tools/model.py
@@ -0,0 +1,4 @@
+import os
+
+def get_model_name(model):
+    return os.path.basename(model.name_or_path)
\ No newline at end of file