From 6bf2db71d1428ba6f8d48b77fbb117b7cc019be2 Mon Sep 17 00:00:00 2001 From: Krish Dholakia Date: Sat, 12 Aug 2023 20:05:03 -0700 Subject: [PATCH] Update README.md --- README.md | 15 +++------------ 1 file changed, 3 insertions(+), 12 deletions(-) diff --git a/README.md b/README.md index 164cbb927a..be90151697 100644 --- a/README.md +++ b/README.md @@ -22,10 +22,10 @@ pip install litellm ```python from litellm import completion -import os -# set API keys -os.environ['OPENAI_API_KEY'] = "" +## 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"}] @@ -34,15 +34,6 @@ 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) - -# hugging face call -response = completion(model="stabilityai/stablecode-completion-alpha-3b-4k", messages=messages, hugging_face=True) - -# openrouter call -response = completion("google/palm-2-codechat-bison", messages) ``` Code Sample: [Getting Started Notebook](https://colab.research.google.com/drive/1gR3pY-JzDZahzpVdbGBtrNGDBmzUNJaJ?usp=sharing)