[Bug]: Support to Google AI Studio

[Vidit-Ostwal]

What happened?

I am trying to call litellm with google studio api, but got this error.

Here is the code

import litellm

from litellm import completion
import os

# litellm._turn_on_debug()
response = completion(
  model="gemini/gemini-1.5-pro",
  messages=[{'role': 'system', 'content': 'Determine if the following feedback indicates that the user is satisfied or if further changes are needed. Respond with \'True\' if further changes are needed, or \'False\' if the user is satisfied. **Important** Do not include any additional commentary outside of your \'True\' or \'False\' response.\n\nFeedback: "Yellow"'}],
  api_key=''
)

the error I am getting

BadRequestError: litellm.BadRequestError: VertexAIException BadRequestError - {
  "error": {
    "code": 400,
    "message": "* GenerateContentRequest.contents: contents is not specified\n",
    "status": "INVALID_ARGUMENT"
  }
}

Relevant log output

Are you a ML Ops Team?

No

What LiteLLM version are you on ?

1.60.2

Twitter / LinkedIn details

No response

[keenranger]

    messages=[
        {
            "role": "system",
            "content": "Determine if the following feedback indicates that the user is satisfied or if further changes are needed. Respond with 'True' if further changes are needed, or 'False' if the user is satisfied. **Important** Do not include any additional commentary outside of your 'True' or 'False' response.\n\nFeedback: \"Yellow\"",
        },
        {
            "role": "user",
            "content": "What is the color of the sky?",
        },
    ]

Why don't you add user message.

[Vidit-Ostwal]

Thanks @keenranger, this works, this was pretty dumb from my side.