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

[Backport 8.14] VectorStore for GenAI integrations #2541

Merged
merged 1 commit into from
May 2, 2024

Conversation

github-actions[bot]
Copy link

@github-actions github-actions bot commented May 2, 2024

Backport c2b0ca3 from #2528.

* ElasticsearchStore

* Update elasticsearch/store/_utilities.py

Co-authored-by: Quentin Pradet <[email protected]>

* rename; depend on client; async only

* generate _sync files

* add cleanup step for _sync generation

* fix formatting

* more linting fixes

* batch embedding call; infer num_dimensions

* revert accidental changes

* keep field names only in store; apply metadata mappings in store

* fix typos in file names

* use `elasticsearch_url` fixture; create conftest.py

* export relevant classes

* remove Semantic strategy

wait for `semantic_text` to land

* es_query is sync

* async strategies

* cleanup old file

* add docker-compose service with model deployment

* optional dependencies for MMR

* only test sync parts

* cleanup unasync script

* nox: install optional deps

* fix tests with requests remembering Transport

* fix numpy typing

* add user agent default argument

* move to `elasticsearch.helpers.vectorstore`

* use Protocol over ABC

* revert Protocol change because Python 3.7

* address PR feedback:

- Strategy suffix
- Sphinx docstrings
- add user agent to EmbeddingService
- raise ConflictError
- various cleanup

* improve docstring

* fix metadata mappings issue

* address PR feedback

* add error tests for strategies

* canonical names, keyword args only

* fix sparse vector strategy bug (duplicate `size`)

* all wildcard deletes in compose ES

---------

Co-authored-by: Quentin Pradet <[email protected]>
(cherry picked from commit c2b0ca3)
@pquentin pquentin closed this May 2, 2024
@pquentin pquentin reopened this May 2, 2024
@pquentin pquentin merged commit 590ee66 into 8.14 May 2, 2024
12 checks passed
@pquentin pquentin deleted the backport-2528-to-8.14 branch May 2, 2024 17:09
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants