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

feat: DIA-1360: Add token cost KPI to the Prompt aggregate-level subset metrics #300

Merged
merged 4 commits into from
Sep 10, 2024

Conversation

robot-ci-heartex
Copy link
Collaborator

@robot-ci-heartex robot-ci-heartex commented Sep 7, 2024

Hi @niklub!

This PR was automaticaly generated via Follow Merge.
Please ensure that all linked upstream Pull Requests are merged before proceeding with this one.


🚀 This description was created by Ellipsis for commit 27462d9

Summary:

This PR adds handling for batch failed predictions and includes token cost metrics in batch predictions, updating types, methods, and tests accordingly.

Key points:

  • Added PromptsBatchFailedPredictionsRequestFailedPredictionsItem and PromptsBatchFailedPredictionsResponse types in src/label_studio_sdk/prompts/types/.
  • Introduced batch_failed_predictions method in src/label_studio_sdk/prompts/client.py for handling failed predictions.
  • Updated batch_predictions method to include token cost metrics in src/label_studio_sdk/prompts/client.py.
  • Modified PromptsBatchPredictionsRequestResultsItem to include token and cost fields.
  • Updated tests in tests/test_prompts.py to cover new methods and fields.

Generated with ❤️ by ellipsis.dev

@robot-ci-heartex
Copy link
Collaborator Author

Hi @fern-api[bot]!

Unfortunately you don't have membership in HumanSignal organization, your PR wasn't synced with HumanSignal/label-studio.

@robot-ci-heartex
Copy link
Collaborator Author

Hi @fern-api[bot]!

Unfortunately you don't have membership in HumanSignal organization, your PR wasn't synced with HumanSignal/label-studio.

@robot-ci-heartex
Copy link
Collaborator Author

Hi @fern-api[bot]!

Unfortunately you don't have membership in HumanSignal organization, your PR wasn't synced with HumanSignal/label-studio.

@robot-ci-heartex robot-ci-heartex marked this pull request as ready for review September 10, 2024 12:58
@niklub niklub merged commit b71c6d0 into master Sep 10, 2024
7 of 9 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants