builtin_provider

Class info

Classes

Name	Children	Inherits
BasePromptProvider llmling_agent.prompts.base Base class for external prompt providers.	BuiltinPromptProvider RuntimePromptProvider	ABC
BuiltinPromptProvider llmling_agent.prompts.builtin_provider Default provider using system prompts.		BasePromptProvider
RuntimePromptProvider llmling_agent.prompts.builtin_provider Provider for prompts from RuntimeConfig.		BasePromptProvider

🛈 DocStrings

Builtin prompt provider.

BuiltinPromptProvider

Bases: BasePromptProvider

Default provider using system prompts.

Source code in src/llmling_agent/prompts/builtin_provider.py

class BuiltinPromptProvider(BasePromptProvider):
    """Default provider using system prompts."""

    supports_variables = True
    name = "builtin"

    def __init__(self, manifest_prompts: dict[str, SystemPrompt]):
        from jinja2 import Environment

        from llmling_agent_functional.run import run_agent, run_agent_sync

        self.prompts = manifest_prompts
        self.env = Environment(autoescape=False, enable_async=True)
        self.env.globals["run_agent"] = run_agent
        self.env.globals["run_agent_sync"] = run_agent_sync

    async def get_prompt(
        self,
        identifier: str,
        version: str | None = None,
        variables: dict[str, Any] | None = None,
    ) -> str:
        """Get prompt content as string."""
        from jinja2 import Template, meta

        if identifier not in self.prompts:
            msg = f"Prompt not found: {identifier}"
            raise KeyError(msg)

        prompt = self.prompts[identifier]
        content = prompt.content

        # Parse template to find required variables
        ast = self.env.parse(content)
        required_vars = meta.find_undeclared_variables(ast)

        if variables:
            # Check for unknown variables
            unknown_vars = set(variables) - required_vars
            if unknown_vars:
                _vars = ", ".join(unknown_vars)
                msg = f"Unknown variables for prompt {identifier}: {_vars}"
                raise KeyError(msg)

        if required_vars:
            if not variables:
                req = ", ".join(required_vars)
                msg = f"Prompt {identifier} requires variables: {req}"
                raise KeyError(msg)

            # Check for missing required variables
            missing_vars = required_vars - set(variables or {})
            if missing_vars:
                _vars = ", ".join(missing_vars)
                msg = f"Missing required variables for prompt {identifier}: {_vars}"
                raise KeyError(msg)

            try:
                template = Template(content)
                content = template.render(**variables)
            except Exception as e:
                msg = f"Failed to render prompt {identifier}: {e}"
                raise ValueError(msg) from e

        return content

    async def list_prompts(self) -> list[str]:
        """List available prompt identifiers."""
        return list(self.prompts.keys())

get_prompt async

get_prompt(
    identifier: str, version: str | None = None, variables: dict[str, Any] | None = None
) -> str

Get prompt content as string.

Source code in src/llmling_agent/prompts/builtin_provider.py

async def get_prompt(
    self,
    identifier: str,
    version: str | None = None,
    variables: dict[str, Any] | None = None,
) -> str:
    """Get prompt content as string."""
    from jinja2 import Template, meta

    if identifier not in self.prompts:
        msg = f"Prompt not found: {identifier}"
        raise KeyError(msg)

    prompt = self.prompts[identifier]
    content = prompt.content

    # Parse template to find required variables
    ast = self.env.parse(content)
    required_vars = meta.find_undeclared_variables(ast)

    if variables:
        # Check for unknown variables
        unknown_vars = set(variables) - required_vars
        if unknown_vars:
            _vars = ", ".join(unknown_vars)
            msg = f"Unknown variables for prompt {identifier}: {_vars}"
            raise KeyError(msg)

    if required_vars:
        if not variables:
            req = ", ".join(required_vars)
            msg = f"Prompt {identifier} requires variables: {req}"
            raise KeyError(msg)

        # Check for missing required variables
        missing_vars = required_vars - set(variables or {})
        if missing_vars:
            _vars = ", ".join(missing_vars)
            msg = f"Missing required variables for prompt {identifier}: {_vars}"
            raise KeyError(msg)

        try:
            template = Template(content)
            content = template.render(**variables)
        except Exception as e:
            msg = f"Failed to render prompt {identifier}: {e}"
            raise ValueError(msg) from e

    return content

list_prompts async

list_prompts() -> list[str]

List available prompt identifiers.

Source code in src/llmling_agent/prompts/builtin_provider.py

async def list_prompts(self) -> list[str]:
    """List available prompt identifiers."""
    return list(self.prompts.keys())

RuntimePromptProvider

Bases: BasePromptProvider

Provider for prompts from RuntimeConfig.

Source code in src/llmling_agent/prompts/builtin_provider.py

class RuntimePromptProvider(BasePromptProvider):
    """Provider for prompts from RuntimeConfig."""

    name = "runtime"
    supports_variables = True

    def __init__(self, runtime_config: RuntimeConfig):
        self.runtime = runtime_config

    async def get_prompt(
        self,
        name: str,
        version: str | None = None,
        variables: dict[str, Any] | None = None,
    ) -> str:
        prompt = self.runtime.get_prompt(name)
        if variables:
            messages = await prompt.format(variables)
        messages = await prompt.format()
        return "\n".join(m.get_text_content() for m in messages)

    async def list_prompts(self) -> list[str]:
        return [p.name for p in self.runtime.get_prompts() if p.name]