Skip to content

Add llamafile as a provider #10203

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

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

Conversation

peteski22
Copy link

@peteski22 peteski22 commented Apr 22, 2025

Title

Implement llamafile as a supported provider.

Relevant issues

Closes: #3225

Pre-Submission checklist

Please complete all items before asking a LiteLLM maintainer to review your PR

  • I have Added testing in the tests/litellm/ directory, Adding at least 1 test is a hard requirement - see details
  • I have added a screenshot of my new test passing locally
  • My PR passes all unit tests on (make test-unit)[https://docs.litellm.ai/docs/extras/contributing_code]
  • My PR's scope is as isolated as possible, it only solves 1 specific problem

Type

🆕 New Feature
📖 Documentation
✅ Test

Changes

  • The provider can be set to llamafile, including via model (e.g. model="llamafile/mistralai/Mistral-7B-Instruct-v0.2")
  • When provider is set to llamafile, API key defaults to fake-api-key if it isn't configured by the user or available in the secret manager (under LLAMAFILE_API_KEY)
  • When provider is set to llamafile, base API URL will fallback to http://127.0.0.1:8080/v1 if it isn't configured by the user or available in the secret manager (under LLAMAFILE_API_BASE)

Vercel URLs:

Receipts

image
image
image
image

Copy link

vercel bot commented Apr 22, 2025

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

Name Status Preview Comments Updated (UTC)
litellm ✅ Ready (Inspect) Visit Preview 💬 Add feedback Apr 23, 2025 7:29pm

@CLAassistant
Copy link

CLAassistant commented Apr 22, 2025

CLA assistant check
All committers have signed the CLA.

Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

this is missing a test to confirm it would

  1. work with .completion(model="llamafile/<my-custom-model",..)
  2. work with get_optional_params() -> used by completion for translating params

Copy link
Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

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.

[Feature]: Add support for Llamafile provider
3 participants