[Bug]: Gpt-images and dall-e-3 problem with cost

[devbysb]

What happened?

Hello, I'm trying to track the cost for the azure/gpt-image-1 and azure/dall-e-3 models, but nothing is coming back.

I tried modifying my configuration like this, but it didn't improve. In the logs, the request remains at 0.

 - model_name: azure-ai-openai-dall-e
    litellm_params:
        model: azure/dall-e-3
        api_version: 2024-02-01-preview
        api_base: https://myendpoint.openai.azure.com
        api_key: mykey
        base_model: dall-e-3
    model_info:
        mode: image_generation
        input_cost_per_pixel: 0.00000007629
        output_cost_per_token: 0.0

  - model_name: azure-ai-openai-gpt-image-1 
    litellm_params:
      model: azure/gpt-image-1
        api_base: https://myendpoint.openai.azure.com
        api_key: mykey
    model_info:
      input_cost_per_pixel: 0.000000040054321
      output_cost_per_pixel: 0.0

Thank you for your help

Relevant log output

Are you a ML Ops Team?

No

What LiteLLM version are you on ?

1.74.3

Twitter / LinkedIn details

No response

[devbysb]

HI, I have seen in model_prices_and_context_window.json

    "azure/gpt-image-1": {
        "mode": "image_generation",
        "input_cost_per_pixel": 4.0054321e-08,
        "output_cost_per_pixel": 0.0,
        "litellm_provider": "azure",
        "supported_endpoints": [
            "/v1/images/generations"
        ]
    },

It is input_cost_per_pixel or output_cost_per_pixel to use image generation ?