a simple & light 100 line package to call OpenAI, Azure, Cohere, Anthropic API Endpoints
litellm manages:
- translating inputs to completion and embedding endpoints
- guarantees consistent output, text responses will always be available at
['choices'][0]['message']['content']
from litellm import completion
## set ENV variables
os.environ["OPENAI_API_KEY"] = "openai key"
os.environ["COHERE_API_KEY"] = "cohere key"
messages = [{ "content": "Hello, how are you?","role": "user"}]
# openai call
response = completion(model="gpt-3.5-turbo", messages=messages)
# cohere call
response = completion("command-nightly", messages)
# azure openai call
response = completion("chatgpt-test", messages, azure=True)pip install litellm
- Need for simplicity: My code started to get extremely complicated managing & translating calls between Azure, OpenAI, Cohere