litellm/docs/my-website/docs/providers/palm.md
2024-01-20 19:13:33 +09:00

921 B

PaLM API - Google

Pre-requisites

  • pip install -q google-generativeai

Sample Usage

from litellm import completion
import os

os.environ['PALM_API_KEY'] = ""
response = completion(
    model="palm/chat-bison", 
    messages=[{"role": "user", "content": "write code for saying hi from LiteLLM"}]
)

Sample Usage - Streaming

from litellm import completion
import os

os.environ['PALM_API_KEY'] = ""
response = completion(
    model="palm/chat-bison", 
    messages=[{"role": "user", "content": "write code for saying hi from LiteLLM"}],
    stream=True
)

for chunk in response:
    print(chunk)

Chat Models

Model Name Function Call Required OS Variables
chat-bison completion('palm/chat-bison', messages) os.environ['PALM_API_KEY']