include python documentation getter as a builtin

chatlab/builtins/__init__.py

@@ -3,7 +3,7 @@
 from deprecation import deprecated
 
 from .files import chat_functions as file_functions
-from .python import run_python
+from .python import get_python_docs, run_python
 from .shell import chat_functions as shell_functions
 
 # To prevent naming confusion, the builtin that isn't really running a cell
@@ -15,6 +15,6 @@
 )(run_python)
 
 # compose all the file, shell, and python functions into one list for ease of use
-os_functions = file_functions + shell_functions + [run_python]
+os_functions = file_functions + shell_functions + [run_python, get_python_docs]
 
-__all__ = ["run_python", "run_cell", "file_functions", "shell_functions", "os_functions"]
+__all__ = ["run_python", "get_python_docs", "run_cell", "file_functions", "shell_functions", "os_functions"]

chatlab/builtins/python.py

@@ -146,6 +146,16 @@ def run_python(code: str):
     return __shell.run_cell(code)
 
 
+@expose_exception_to_llm
+def get_python_docs(module_name: str):
+    """Read docs for a Python package."""
+    import importlib
+    import inspect
+
+    package = importlib.import_module(module_name)
+    return inspect.getdoc(package)
+
+
 __all__ = ["run_python", "ChatLabShell"]