Skip to content

Context In Function Tools

Source code in OpAgentsOlympus/practice/context_in_function_tools.py
OpAgentsOlympus/practice/context_in_function_tools.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
45
46
47
48
49
50
import asyncio
from agents import (
    Agent,
    Runner,
    function_tool,
    GuardrailFunctionOutput,
    RunContextWrapper,
)
from config import config
from pydantic import BaseModel


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


@function_tool
async def get_user_data(
    context: RunContextWrapper[UserData],
) -> GuardrailFunctionOutput:  # Context is not accessible without RunContextWrapper
    """This tool can be used to get the user data so that the llm can understand about the user"""
    print("Getting user data...")
    return context.context


async def main():
    agent = Agent(
        name="Super Assistant",
        instructions=(
            "You are Super Assistant, an expert and proactive AI assistant. "
            "Use available tools to retrieve information you do not know. "
            "You DON'T need to ask the user for calling tools. "
        ),
        tools=[get_user_data],
    )

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

    result = await Runner.run(
        agent,
        input="What is the name of the user?",
        context=userdata,
        run_config=config,
    )
    print(result.final_output)


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