forked from phoenix/litellm-mirror
39 lines
1.2 KiB
Markdown
39 lines
1.2 KiB
Markdown
# litellm
|
|
[](https://discord.gg/wuPM9dRgDw)
|
|
|
|
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']`
|
|
# usage
|
|
* Code Sample: [Getting Started Notebook](https://colab.research.google.com/drive/1gR3pY-JzDZahzpVdbGBtrNGDBmzUNJaJ?usp=sharing)
|
|
```python
|
|
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)
|
|
```
|
|
|
|
# installation
|
|
```
|
|
pip install litellm
|
|
```
|
|
|
|
# why did I build this
|
|
- **Need for simplicity**: My code started to get extremely complicated managing & translating calls between Azure, OpenAI, Cohere
|
|
|
|
# Support
|
|
Contact me at ishaan@berri.ai
|