Router

`router`

codex_ai.core.router

LLMRouter — registry for prompt builder functions.

Register builders via the @router.prompt(mode) decorator, then connect to LLMDispatcher via include_router().

Classes

`LLMRouter`

Router for grouping prompt builder functions by mode.

Example

router = LLMRouter()

@router.prompt("chat")
async def build_chat(text: str, history: list, **kw) -> PromptResult:
    return PromptResult(messages=[{"role": "user", "content": text}])

@router.prompt("summarize")
async def build_summarize(text: str, **kw) -> PromptResult:
    return PromptResult(
        messages=[{"role": "user", "content": f"Summarize: {text}"}],
        system="Be concise.",
    )

Source code in src/codex_ai/core/router.py

class LLMRouter:
    """
    Router for grouping prompt builder functions by mode.

    Example:
        ```python
        router = LLMRouter()

        @router.prompt("chat")
        async def build_chat(text: str, history: list, **kw) -> PromptResult:
            return PromptResult(messages=[{"role": "user", "content": text}])

        @router.prompt("summarize")
        async def build_summarize(text: str, **kw) -> PromptResult:
            return PromptResult(
                messages=[{"role": "user", "content": f"Summarize: {text}"}],
                system="Be concise.",
            )
        ```
    """

    def __init__(self) -> None:
        # {mode: builder_func}
        self._builders: dict[str, PromptBuilder] = {}

    def prompt(self, mode: str) -> Callable[[PromptBuilder], PromptBuilder]:
        """
        Decorator for registering a prompt builder for a given mode.

        Args:
            mode: Identifier string (e.g., ``"chat"``, ``"summarize"``).

        Returns:
            Decorator returning the original builder unchanged.
        """

        def decorator(fn: PromptBuilder) -> PromptBuilder:
            self._builders[mode] = fn
            return fn

        return decorator

    @property
    def builders(self) -> dict[str, PromptBuilder]:
        """Registered builders (read-only view)."""
        return self._builders

Attributes

`builders` `property`

Registered builders (read-only view).

Functions

`prompt(mode)`

Decorator for registering a prompt builder for a given mode.

Parameters:

Name	Type	Description	Default
`mode`	`str`	Identifier string (e.g., `"chat"`, `"summarize"`).	required

Returns:

Type	Description
`Callable[[PromptBuilder], PromptBuilder]`	Decorator returning the original builder unchanged.