Skip to content

Add azure_ai cohere rerank v3.5 #12283

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

Conversation

dcieslak19973
Copy link
Contributor

Title

Add azure_ai/cohere-rerank-v3.5

Relevant issues

Fixes #10847

Pre-Submission checklist

  • 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
  • My PR's scope is as isolated as possible, it only solves 1 specific problem

Type

🆕 New Feature

Changes

@Copilot Copilot AI review requested due to automatic review settings July 3, 2025 14:28
Copy link

vercel bot commented Jul 3, 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 Jul 3, 2025 4:44pm

@CLAassistant
Copy link

CLAassistant commented Jul 3, 2025

CLA assistant check
All committers have signed the CLA.

Copy link
Contributor

@Copilot Copilot AI left a comment

Choose a reason for hiding this comment

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

Pull Request Overview

Adds support for the Azure Cohere Rerank v3.5 model by defining its token limits and cost parameters in both the main and backup pricing files.

  • Introduces a new JSON entry for azure_ai/cohere-rerank-v3.5 in the primary pricing definition
  • Mirrors the same addition in the backup JSON

Reviewed Changes

Copilot reviewed 2 out of 2 changed files in this pull request and generated no comments.

File Description
model_prices_and_context_window.json Added new model entry for azure_ai/cohere-rerank-v3.5
litellm/model_prices_and_context_window_backup.json Synchronized backup to include the same azure_ai/cohere-rerank-v3.5 entry
Comments suppressed due to low confidence (1)

model_prices_and_context_window.json:4071

  • Add a unit test in tests/litellm to verify that the new model azure_ai/cohere-rerank-v3.5 is correctly loaded with the expected token limits and cost values.
    "azure_ai/cohere-rerank-v3.5": {

Copy link
Contributor

@ishaan-jaff ishaan-jaff left a comment

Choose a reason for hiding this comment

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

LGTM

@ishaan-jaff ishaan-jaff merged commit d480cea into BerriAI:main Jul 3, 2025
5 of 6 checks passed
@dcieslak19973 dcieslak19973 deleted the litellm-10847-add-azure_ai-cohere-35-pricing-20250703 branch July 3, 2025 17:06
jugaldb pushed a commit that referenced this pull request Jul 3, 2025
* Add azure_ai cohere rerank v3.5

* Fix CI error
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.

Add "azure_ai/cohere-rerank-v3.5" in "model_prices_and_context_window.json"
3 participants