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

Add gemini models for litellm/__init__.py #3473

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

Conversation

nobu007
Copy link
Contributor

@nobu007 nobu007 commented May 6, 2024

Add gemini models for litellm/init.py

Add gemini models as same as watsonx_models.

Relevant issues

None

Type

🆕 New Feature

Changes

  • gemini_models added
  • models_by_provider is also added
  • here condition changes to false by this commit.
    So used in here used, and gemini model works fine.
  • apply formatter

Testing

Works fine with this condition.

  • MODEL_NAME="gemini/gemini-1.5-pro-latest"
  • GEMINI_API_KEY=XXXX
  • LITELLM_LOCAL_MODEL_COST_MAP="True"

Notes

Needs additional fix #3493 for handling Exception from gemini.

Pre-Submission Checklist (optional but appreciated):

  • I have included relevant documentation updates (stored in /docs/my-website)

OS Tests (optional but appreciated):

  • Tested on Windows
  • Tested on MacOS
  • Tested on Linux

Copy link

vercel bot commented May 6, 2024

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 May 7, 2024 2:02pm

@nobu007 nobu007 changed the title Add gemini models Add gemini models for litellm/__init__.py May 6, 2024
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.

None yet

1 participant