Skip to content

Bugfix/usage for openrouter (#11627) #11785

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 2 commits into
base: main
Choose a base branch
from

Conversation

krrishdholakia
Copy link
Contributor

  • "Enhance usage tracking by adding cost and is_byok fields, refactor Usage handling and calculation logic for better modularity and accuracy."

  • Remove unused model_config attribute in utils.py

  • Refactor usage handling in streaming responses to ensure propagation of cost and token details, enhance chunk processing, and fix PromptTokensDetails to PromptTokensDetailsWrapper.

  • Remove redundant logging and usage handling logic for cleaner streaming response processing.

  • Add handling and mapping of usage data to model response in streaming handler

  • Refactor streaming_handler token parsing for readability; update test to ignore new usage metadata fields.

  • Fix linter issues

  • Refactor: Remove redundant comment in streaming_chunk_builder_utils.py

Title

Relevant issues

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

Type

🆕 New Feature
🐛 Bug Fix
🧹 Refactoring
📖 Documentation
🚄 Infrastructure
✅ Test

Changes

* "Enhance usage tracking by adding cost and is_byok fields, refactor Usage handling and calculation logic for better modularity and accuracy."

* Remove unused model_config attribute in utils.py

* Refactor usage handling in streaming responses to ensure propagation of cost and token details, enhance chunk processing, and fix PromptTokensDetails to PromptTokensDetailsWrapper.

* Remove redundant logging and usage handling logic for cleaner streaming response processing.

* Add handling and mapping of `usage` data to model response in streaming handler

* Refactor streaming_handler token parsing for readability; update test to ignore new usage metadata fields.

* Fix linter issues

* Refactor: Remove redundant comment in streaming_chunk_builder_utils.py
Copy link

vercel bot commented Jun 17, 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 Jun 18, 2025 1:52am

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