zinccat / ChatGPT-streamlit

Easy to use UI built with Streamlit for using ChatGPT, Claude, Stable Diffusion and beyond
GNU General Public License v3.0
17 stars 7 forks source link

Update Anthropic Client #2

Open krrishdholakia opened 1 year ago

krrishdholakia commented 1 year ago

Anthropic changed their python sdk - making this code line outdated.

https://github.com/zinccat/ChatGPT-streamlit/blob/d997ee3618ee4996d51d217faf8600d413575449/scripts/models.py#L51

Would love to know if this might help - https://github.com/BerriAI/litellm

~Simple I/O library, that standardizes all the llm api calls to the OpenAI call

from litellm import completion

## set ENV variables
# ENV variables can be set in .env file, too. Example in .env.example
os.environ["OPENAI_API_KEY"] = "openai key"
os.environ["ANTHROPIC_API_KEY"] = "anthropic key"

messages = [{ "content": "Hello, how are you?","role": "user"}]

# openai call
response = completion(model="gpt-3.5-turbo", messages=messages)

# anthropic call
response = completion("claude-v-2", messages)
krrishdholakia commented 1 year ago

cc: @zinccat