Function build_lint_agent

openkb/agent/linter.py:47–89

Build the semantic knowledge-lint agent. Args: wiki_root: Absolute path to the wiki directory. model: LLM model name. language: Language code for wiki content (e.g. 'en', 'fr'). Returns: Configured :class:`~agents.Agent` instance.

(wiki_root: str, model: str, language: str = "en")

def build_lint_agent(wiki_root: str, model: str, language: str = "en") -> Agent:
    """Build the semantic knowledge-lint agent.

    Args:
        wiki_root: Absolute path to the wiki directory.
        model: LLM model name.
        language: Language code for wiki content (e.g. 'en', 'fr').

    Returns:
        Configured :class:`~agents.Agent` instance.
    """
    schema_md = get_agents_md(Path(wiki_root))
    instructions = _LINTER_INSTRUCTIONS_TEMPLATE.format(schema_md=schema_md)
    instructions += f"\n\nIMPORTANT: Write the lint report in {language} language."

    @function_tool
    def list_files(directory: str) -> str:
        """List all Markdown files in a wiki subdirectory.

        Args:
            directory: Subdirectory path relative to wiki root (e.g. 'summaries').
        """
        return list_wiki_files(directory, wiki_root)

    @function_tool
    def read_file(path: str) -> str:
        """Read a Markdown file from the wiki.

        Args:
            path: File path relative to wiki root (e.g. 'summaries/paper.md').
        """
        return read_wiki_file(path, wiki_root)

    return Agent(
        name="wiki-linter",
        instructions=instructions,
        tools=[list_files, read_file],
        model=f"litellm/{model}",
        model_settings=ModelSettings(
            extra_headers=get_extra_headers() or None,
            extra_args=get_timeout_extra_args(),
        ),
    )


async def run_knowledge_lint(kb_dir: Path, model: str) -> str: