diff --git a/docs/my-website/docs/providers/gemini.md b/docs/my-website/docs/providers/gemini.md index df9ad16ed49..6d3cfc39c02 100644 --- a/docs/my-website/docs/providers/gemini.md +++ b/docs/my-website/docs/providers/gemini.md @@ -3,6 +3,7 @@ ## Pre-requisites * `pip install -q google-generativeai` +# Gemini-Pro ## Sample Usage ```python import litellm @@ -15,11 +16,54 @@ response = completion( ) ``` +# Gemini-Pro-Vision LiteLLM Supports the following image types passed in `url` - Images with direct links - https://storage.googleapis.com/github-repo/img/gemini/intro/landmark3.jpg - Image in local storage - ./localimage.jpeg +## Sample Usage +```python +import os +import litellm +from dotenv import load_dotenv +# Load the environment variables from .env file +load_dotenv() +os.environ["GEMINI_API_KEY"] = os.getenv('GEMINI_API_KEY') + +prompt = 'Describe the image in a few sentences.' +# Note: You can pass here the URL or Path of image directly. +image_url = 'https://storage.googleapis.com/github-repo/img/gemini/intro/landmark3.jpg' + +# Create the messages payload according to the documentation +messages = [ + { + "role": "user", + "content": [ + { + "type": "text", + "text": prompt + }, + { + "type": "image_url", + "image_url": {"url": image_url} + } + ] + } +] + +# Make the API call to Gemini model +response = litellm.completion( + model="gemini/gemini-pro-vision", + messages=messages, +) + +# Extract the response content +content = response.get('choices', [{}])[0].get('message', {}).get('content') + +# Print the result +print(content) +``` ## Chat Models | Model Name | Function Call | Required OS Variables |