llama_index
llama_index copied to clipboard
Added support for Bedrock Titan Embeddings v2
Description
- Added support for Bedrock Titan Embeddings v2. This version introduces new parameters to the request body,
dimensionsandnormalize, so added new field,titan_body_kwargs, to support. - Renamed
modelfield tomodel_nameto adhere to that used in parent class,BaseEmbedding. Bumped to0.2.0consequently. - Fixed bugs with provider mappings.
- Improved unit test quality.
Version Bump?
Did I bump the version in the pyproject.toml file of the package I am updating? (Except for the llama-index-core package)
- [x] Yes
- [ ] No
Type of Change
- [x] New feature (non-breaking change which adds functionality)
How Has This Been Tested?
Please describe the tests that you ran to verify your changes. Provide instructions so we can reproduce. Please also list any relevant details for your test configuration
- [x] Added new unit/integration tests
- [x] I stared at the code and made sure it makes sense
Suggested Checklist:
- [x] I have performed a self-review of my own code
- [x] I have commented my code, particularly in hard-to-understand areas
- [x] My changes generate no new warnings
- [x] I have added tests that prove my fix is effective or that my feature works
- [x] New and existing unit tests pass locally with my changes
- [x] I ran
make format; make lintto appease the lint gods
@nerdai Just requesting re-review since the review should be stale with my two additional commits.
FWIW, I will be on vacation starting tomorrow so feel free to edit the PR directly. Added you as a collaborator to my fork.
Hello @logan-markewich and @nerdai, any idea on when this will get merged I was waiting for this feature for my project instead of monkey patching.