Skip to content

Context In Guardrails

Source code in OpAgentsOlympus/practice/context_in_guardrails.py
OpAgentsOlympus/practice/context_in_guardrails.py
 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
import asyncio
from agents import (
    Agent,
    Runner,
    input_guardrail,
    GuardrailFunctionOutput,
    TResponseInputItem,
)
from config import config
from pydantic import BaseModel


class UserData(BaseModel):
    name: str
    height: float


@input_guardrail
async def input_guardrail(
    context, agent: Agent, input: str | list[TResponseInputItem]
) -> GuardrailFunctionOutput:  # Context is accessible without RunContextWrapper
    print(
        f"Input Guardrail of agent {agent.name} started with context: {context.context}"
    )
    return GuardrailFunctionOutput(output_info="Worked!", tripwire_triggered=False)


async def main():
    agent = Agent(
        name="Super Assistant",
        instructions="You are a helpful assistant.",
        input_guardrails=[input_guardrail],
    )

    userdata = UserData(name="Daniel", height=5.11)

    result = await Runner.run(
        agent, input="how are you?", context=userdata, run_config=config
    )
    print(result.final_output)


if __name__ == "__main__":
    asyncio.run(main())