(draft)feat: add cost estimation per provider for Goose CLI #3312
Conversation
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Co-authored-by: Chaitanya Rahalkar <[email protected]>
Co-authored-by: Angie Jones <[email protected]>
Co-authored-by: Copilot <[email protected]>
…ck#3260) Co-authored-by: dianed-square <[email protected]> Co-authored-by: Angie Jones <[email protected]>
Co-authored-by: Rizel Scarlett <[email protected]>
GaryZhous
force-pushed
the
GaryZ/Refine-CLI
branch
from
c17fe3b to
55869eb
Compare
GaryZhous
changed the title
Contributor
Author
|
this is just a draft |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
13 participants
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Built on top of #2941, this pull request enhances the
goose-clisession module by introducing functionality to estimate and display cost information for token usage based on provider pricing data. It includes new methods for normalizing model names, fetching pricing data, and calculating costs, as well as configuration updates to enable cost display. Instead of hard-coding the cost calculation, I make use of Jack'sPricing_Cachewhich he created in #2992.Pricing Integration:
initialize_pricing_cacheto session startup: Ensures pricing cache is initialized, with error handling to log warnings if initialization fails.estimate_cost_usdfunction: Calculates token usage cost in USD using provider pricing data and normalized model names.display_cost_usagefunction: Displays cost information in USD if pricing data is available, including token breakdown.Model Name Normalization:
normalize_model_namefunction: Cleans up model names by removing suffixes (e.g., "-latest", date-like patterns) and converting version numbers for compatibility with pricing data. Otherwise we are not able to access the pricing info from the cache since Openrouter does the model naming differently. (e.g.claude-3-5-haiku-latestisclaude-3.5-haikuon Openrouter)Configuration Updates:
GOOSE_CLI_SHOW_COSTto toggle cost display andGOOSE_PROVIDERto specify the provider name.These changes collectively improve the
goose-clitool by providing users with detailed cost insights based on their session's token usage. For models whose pricing is unknown, nothing will show up.How to enable
After