Bug: BedrockLLMAgent callbacks on_llm_start and on_llm_end does not work

[rjesh-git]

Expected Behaviour

All the three callback events on_llm_start, on_llm_end and on_llm_new_token fires for BedrockLLMAgent while streaming response.

Current Behaviour

Only on_llm_new_token fires for BedrockLLMAgent while streaming response.

Code snippet

Refer to https://github.com/awslabs/multi-agent-orchestrator/blob/main/examples/fast-api-streaming/main.py#L41

Possible Solution

No response

Steps to Reproduce

https://github.com/awslabs/multi-agent-orchestrator/blob/main/examples/fast-api-streaming/main.py#L41

[brnaba-aws]

Hi @rjesh-git
Thanks for reaching out.
This example is wrong. We actually don't have any definition apart from on_llm_new_token

Here is where the callbacks are defined:

multi-agent-orchestrator/python/src/multi_agent_orchestrator/agents/agent.py

Line 25 in 58e37b5

class AgentCallbacks:

We had plan to add more callbacks.