-
Notifications
You must be signed in to change notification settings - Fork 131
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
Gradient.ai - updating the SDK #110
Gradient.ai - updating the SDK #110
Conversation
integrations/gradient/src/gradient_haystack/embedders/gradient_document_embedder.py
Show resolved
Hide resolved
LGTM, let's make sure that the tests are ran and we're gtg! |
integrations/gradient/src/gradient_haystack/embedders/gradient_document_embedder.py
Outdated
Show resolved
Hide resolved
All tests passing on my end. |
Thanks folks! FYI, in #103 we removed all the |
@masci Alright. |
integrations/gradient/src/gradient_haystack/embedders/gradient_document_embedder.py
Outdated
Show resolved
Hide resolved
integrations/gradient/src/gradient_haystack/embedders/gradient_document_embedder.py
Show resolved
Hide resolved
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Cool, merging!
Awesome, thanks guys! |
Our SDK
gradientai >= 1.4.0
supports nowasync
API and threaded batching in our API - the embedding model now exposes.embed
=> This PR aims to improve speed of embeddings over large numbers of documents.Dependencies:
gradientai
from1.3.0
to1.4.0
. The SDK always had a minium python version of 3.8, except for some initial experiments (e.g.1.0.0.dev1
). Therefore, I raised the miniumum version to1.4.0
tqdm
as optional. It should be pre-installed, and we are using no advanced options other thanfrom tqdm import tqdm
, no version pin required. I think this will improve developer experience, especially since the embeddings are now faster on the server side.@mhaligowski Can you give this a check before I mark as ready?