ai-agents-vertex-ai-gemini-rou-when-using-vertex-ai-gemini-models-e-g-gemini-2-5--fa6dbc00

THEN Modify the image generation handler to detect Gemini models. Convert the request to a chat completion with modalities: ['text', 'image'] and use the :generateContent endpoint internally. After receiving the response, convert it back to the standard image generation format.

if model.startswith('vertex_ai/gemini'):
    # Convert to chat completion request
    chat_request = {
        'model': model,
        'messages': [{'role': 'user', 'content': prompt}],
        'modalities': ['text', 'image']
    }
    # Route to :generateContent endpoint
    response = await client.chat.completions.create(**chat_request)
    # Convert response to standard image generation format
    image_data = extract_image_from_response(response)
    return {'data': [{'url': image_data}]}