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

feat: add aphrodite support #1153

Draft
wants to merge 1 commit into
base: main
Choose a base branch
from
Draft

feat: add aphrodite support #1153

wants to merge 1 commit into from

Conversation

AlpinDale
Copy link

This PR adds support for Aphrodite Engine. WIP as this is currently untested.

Copy link

vercel bot commented Dec 15, 2023

The latest updates on your projects. Learn more about Vercel for Git ↗︎

Name Status Preview Comments Updated (UTC)
litellm ❌ Failed (Inspect) Dec 15, 2023 3:42pm

@krrishdholakia
Copy link
Contributor

@AlpinDale isn't it already openai-compatible? meaning you could just call it like this - https://docs.litellm.ai/docs/providers/openai_compatible
Screenshot 2023-12-15 at 9 41 18 AM

@krrishdholakia
Copy link
Contributor

@AlpinDale bump on this?

@krrishdholakia krrishdholakia self-requested a review December 16, 2023 20:25
@krrishdholakia krrishdholakia self-assigned this Dec 16, 2023
@AlpinDale
Copy link
Author

@krrishdholakia hi sorry for the late reply.

I'd assume the LiteLLM OpenAI endpoint doesn't support any samplers beyond what OpenAI itself provides. Is that true? If not, I suppose we can use it as-is.

@krrishdholakia
Copy link
Contributor

Yea - we send across any unmapped kwargs straight to the provider - https://docs.litellm.ai/docs/completion/input#provider-specific-params.

You can test this out by doing

import os 
from litellm import completion

os.environ["OPENAI_API_KEY"] = "your-api-key"

# openai call
response = completion(
    model = "openai/<your-model-name>", 
    messages=[{ "content": "Hello, how are you?","role": "user"}],
    api_base = "your-api-base"
)

Let me know if this solves your problem!

@krrishdholakia
Copy link
Contributor

I don't mind adding something more specific for aphrodite, but if you have no hosted endpoint - i won't be able to add this to our ci/cd pipeline for testing.

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