argilla-io / argilla-plugins

🔌 Open-source plugins for with practical features for Argilla using listeners.
Apache License 2.0
6 stars 2 forks source link

inference for updating record embeddings #18

Open davidberenstein1957 opened 1 year ago

davidberenstein1957 commented 1 year ago

Automatically add predictions to records as they are logged into Argilla. This can be used for making it really easy to pre-annotated a dataset with an existing model or service.

MVP

from argilla_plugins.inference import embedder

embedder(name="dataset", model="model_name", vector_name="vector")
embedder.start()

Stretch filtering variables like query could be added to limit the sync.