Can we have a similar implementation for AzureOpenAI ? https://github.com/themanojdesai/python-a2a/blob/main/python_a2a/client/llm/openai.py