gustavz / DataChad

Ask questions about any data source by leveraging langchains
https://datachad.streamlit.app/
Apache License 2.0
314 stars 80 forks source link

add support for claude, azure, cohere, replicate #7

Open krrishdholakia opened 1 year ago

krrishdholakia commented 1 year ago

Hi @gustavz ,

Noticed you're calling OpenAI. I'm working on litellm (simple library to standardize LLM API Calls - https://github.com/BerriAI/litellm) and was wondering if we could be helpful.

Added support for Claude, Cohere, Azure and Llama2 (via Replicate) by adding an elif and setting client in the langchain ChatOpenAI class. The code is pretty similar to the OpenAI class - as litellm follows the same pattern as the openai-python sdk.

Would love to know if this helps.

Happy to add additional tests / update documentation, if the initial PR looks good to you.