How to type stream response of createReactAgent?

[riywo]

If I invoke stream() method directly on a model, the returned type is typed with AIMessageChunk.

However, if I invoke it on the agent created by createReactAgent(), it's any.

Is there any way to specify the type of the return message on createReactAgent() call? Of course I can just type cast the stream object, but I'd like to know whether it's the only option.

const model = new ChatOpenAI();
const stream = await model.stream(); // <= IterableReadableStream<AIMessageChunk>

const agent = createReacAgent();
const stream = await agent.stream(); // <= IterableReadableStream<any>