diff --git a/gptme/prompts.py b/gptme/prompts.py
index 542be443..26576314 100644
--- a/gptme/prompts.py
+++ b/gptme/prompts.py
@@ -13,7 +13,7 @@
 from .config import get_config
 from .message import Message
 from .tools import init_tools, loaded_tools
-from .util import _document_prompt_function
+from .util import document_prompt_function
 
 PromptType = Literal["full", "short"]
 
@@ -202,7 +202,7 @@ def prompt_tools(examples: bool = True) -> Generator[Message, None, None]:
     yield Message("system", prompt.strip() + "\n\n")
 
 
-_document_prompt_function(interactive=True)(prompt_gptme)
-_document_prompt_function()(prompt_user)
-_document_prompt_function()(prompt_project)
-_document_prompt_function()(prompt_tools)
+document_prompt_function(interactive=True)(prompt_gptme)
+document_prompt_function()(prompt_user)
+document_prompt_function()(prompt_project)
+document_prompt_function()(prompt_tools)
diff --git a/gptme/util.py b/gptme/util.py
index e82ad0a6..92cb3e93 100644
--- a/gptme/util.py
+++ b/gptme/util.py
@@ -184,7 +184,7 @@ def _is_sphinx_build() -> bool:
     return is_sphinx
 
 
-def _document_prompt_function(*args, **kwargs):
+def document_prompt_function(*args, **kwargs):
     """Decorator for adding example output of prompts to docstrings in rst format"""
 
     def decorator(func):  # pragma: no cover