from loom import (
    Agent,
    KnowledgeDocument,
    KnowledgeQuery,
    KnowledgeSource,
    Model,
    RunContext,
    Runtime,
)

agent = Agent(
    model=Model.anthropic("claude-sonnet-4"),
    instructions="Answer internal documentation questions from the supplied evidence.",
    knowledge=[
        KnowledgeSource.inline(
            "handbook",
            [
                KnowledgeDocument(
                    title="On-call Policy",
                    content="Primary on-call engineers must acknowledge incidents within 10 minutes.",
                ),
                KnowledgeDocument(
                    title="Deploy Policy",
                    content="Production deploys require approval from the release owner.",
                ),
            ],
        )
    ],
    runtime=Runtime.sdk(),
)

knowledge = agent.resolve_knowledge(
    KnowledgeQuery(
        text="Who can approve a production deploy?",
        goal="Answer deploy approval question",
        top_k=3,
    )
)

result = await agent.run(
    "Answer the user question using the retrieved documentation",
    context=RunContext(
        inputs={"user_question": "Who can approve a production deploy?"},
        knowledge=knowledge,
    ),
)