How to get RunContextWrapper fields from Run.run_streamed ?

[muneebarshadd]

Question

I created an agent with hooks that updates the context field:

agent= Agent[MyContext]( name="agent name", instructions="some instructions", hooks=AgentHooks(), )

These are my hooks:

class MyContext(BaseModel):
    queryConverted: bool = False
    greetings: bool = False


class AgentHooks(RunHooks):


    async def on_start(
        self, context: RunContextWrapper[MyContext], agent: Agent[MyContext]
    ) -> None:
        """Called before the agent is invoked. Called each time the current agent changes."""
        pass

    async def on_end(
        self,
        context: RunContextWrapper[MyContext],
        agent: Agent[MyContext],
        output: Any,
    ) -> None:
        """Called when the agent produces a final output."""
        context.context.queryConverted= True
        context.context.greetings= False

I am running the agent like this:

result = Runner.run_streamed(agent, input="This is my query?", context=MyContext())

I am streaming the events using this line of code:

async for event in result.stream_events()

and I am trying to access the MyContext fields from the output. Is there a way to do this? Basically I am looking for a way to get the context when I run the agent. The examples on github just print the context fields but I dont want to do that

[rm-openai]

It's not there right now. We could add it, though as a workaround you could just do this:

context = MyContext()
...
result = Runner.run_streamed(..., context)
...
async for event in result.stream_event():
  # You can access context here, in the scope from when you created it

[muneebarshadd]

I tries to access context here but it wont update the fields.

Can you show me an example on how you can access context from the result as a work around?