[Bug]: unable to pass `input_type` parameter to Azuree AI Cohere Embed v4 embedding model

[HtutLynn]

What happened?

What happened?
According to the documentation here:

Any non-openai params, will be treated as provider-specific params, and sent in the request body as kwargs to the provider.

I expect the following to work.

response = embedding(
                model="azure_ai/embed-v-4-0",
                input=["Hello World"],
                input_type="search_document",
                dimensions=1536,  # Specify dimensions explicitly
            )

Instead I see the following error:

litellm.APIError: APIError: Azure_aiException - Embeddings.create() got an unexpected keyword argument 'input_type'

Relevant log output

litellm.APIError: APIError: Azure_aiException - Embeddings.create() got an unexpected keyword argument 'input_type'

Are you a ML Ops Team?

Yes

What LiteLLM version are you on ?

v1.71.0

Twitter / LinkedIn details

@htutlynnaung