69 lines
1.9 KiB
Python
69 lines
1.9 KiB
Python
from docspec_python import ParserOptions
|
|
from docs.pydocs.pydocs_markdown_impl import render_loader
|
|
from pydoc_markdown.contrib.loaders.python import PythonLoader
|
|
from pydoc_markdown.contrib.processors.filter import FilterProcessor
|
|
from docs.pydocs.helpers import write_to_file
|
|
|
|
|
|
export_map = {
|
|
"guardrails/prompt/base_prompt.py": [
|
|
"guardrails.prompt.base_prompt",
|
|
"BasePrompt",
|
|
"__init__",
|
|
"format",
|
|
"substitute_constants",
|
|
"get_prompt_variables",
|
|
"escape",
|
|
],
|
|
"guardrails/prompt/prompt.py": [
|
|
"guardrails.prompt.prompt",
|
|
"Prompt",
|
|
"format",
|
|
],
|
|
"guardrails/prompt/instructions.py": [
|
|
"guardrails.prompt.instructions",
|
|
"Instructions",
|
|
"format",
|
|
],
|
|
"guardrails/llm_providers.py": [
|
|
"guardrails.llm_providers",
|
|
"PromptCallableBase",
|
|
"_invoke_llm",
|
|
"__call__",
|
|
],
|
|
"guardrails/classes/llm/llm_response.py": [
|
|
"guardrails.classes.llm.llm_response",
|
|
"LLMResponse",
|
|
],
|
|
}
|
|
|
|
|
|
conditionals = []
|
|
for k, v in export_map.items():
|
|
conditionals.append(
|
|
f"((name in {v}) if ('{k}' in obj.location.filename) else False)"
|
|
)
|
|
|
|
export_string = " or ".join(conditionals)
|
|
|
|
write_to_file(
|
|
str="# Helpers for LLM Interactions\n\n"
|
|
+ render_loader(
|
|
PythonLoader(
|
|
modules=[
|
|
"guardrails.prompt.base_prompt",
|
|
"guardrails.prompt.prompt",
|
|
"guardrails.prompt.instructions",
|
|
"guardrails.llm_providers",
|
|
"guardrails.classes.llm.llm_response",
|
|
],
|
|
parser=ParserOptions(print_function=False),
|
|
),
|
|
processor=FilterProcessor(
|
|
expression=f"({export_string})", # noqa
|
|
skip_empty_modules=True,
|
|
),
|
|
),
|
|
filename="docs/api_reference/llm_interaction.md",
|
|
)
|