Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

adding support for Palm, Cohere, HF Inference Endpoints, Replicate #349

Open
wants to merge 5 commits into
base: main
Choose a base branch
from

Conversation

krrishdholakia
Copy link

@krrishdholakia krrishdholakia commented Aug 15, 2023

Hi @pycui @Shaunwei @ZongZiWang

Noticed you're calling Anthropic and 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.

I added support for palm, cohere, replicate, and huggingface inference endpoints by replacing the ChatOpenAI langchain integration with ChatLiteLLM. 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.

@pycui
Copy link
Collaborator

pycui commented Aug 15, 2023

Can you showcase how would we actually use Palm, Cohere, HF Inference Endpoints? Thanks

@krrishdholakia
Copy link
Author

Yes! will do

@krrishdholakia
Copy link
Author

Updated ReadME
Screenshot 2023-09-02 at 2 41 31 PM

Also tested this on local with AI21 - https://www.ai21.com/studio/pricing

Screenshot 2023-09-02 at 2 34 07 PM Screenshot 2023-09-02 at 2 34 01 PM

@krrishdholakia
Copy link
Author

krrishdholakia commented Sep 2, 2023

In addition, we also enable easy A/B testing of new LLM models in production:

https://docs.litellm.ai/docs/tutorials/ab_test_llms

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

@pycui @Shaunwei

@krrishdholakia
Copy link
Author

Bump on this @pycui @Shaunwei @ZongZiWang

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants