Skip to content

feat: Infinispan vector-io provider#4839

Open
rigazilla wants to merge 1 commit intollamastack:mainfrom
rigazilla:infinispan-vector-io
Open

feat: Infinispan vector-io provider#4839
rigazilla wants to merge 1 commit intollamastack:mainfrom
rigazilla:infinispan-vector-io

Conversation

@rigazilla
Copy link

What does this PR do?

Add Infinispan as vector-io provider. Infinispan Home

Closes #2961

Test Plan

Ran vector io test suite, and all test passed

$ pytest -sv     --stack-config="files=inline::localfs,inference=inline::sentence-transformers,vector_io=remote::infinispan"     tests/integration/vector_io/test_vector_io.py 
================================================================ test session starts =================================================================
platform linux -- Python 3.14.2, pytest-9.0.2, pluggy-1.6.0 -- /home/rigazilla/llama/llama-stack/.venv/bin/python3
cachedir: .pytest_cache
rootdir: /home/rigazilla/llama/llama-stack
configfile: pyproject.toml
plugins: anyio-4.12.1, asyncio-1.3.0
asyncio: mode=Mode.AUTO, debug=False, asyncio_default_fixture_loop_scope=None, asyncio_default_test_loop_scope=function
collected 12 items                                                                                                                                   

tests/integration/vector_io/test_vector_io.py::test_vector_store_retrieve[vector_io=infinispan-emb=inline::sentence-transformers/nomic-ai/nomic-embed-text-v1.5:dim=768] 
instantiating llama_stack_client
llama_stack_client instantiated in 1.286s
PASSED
tests/integration/vector_io/test_vector_io.py::test_vector_store_register[vector_io=infinispan-emb=inline::sentence-transformers/nomic-ai/nomic-embed-text-v1.5:dim=768] PASSED
tests/integration/vector_io/test_vector_io.py::test_insert_chunks[vector_io=infinispan-emb=inline::sentence-transformers/nomic-ai/nomic-embed-text-v1.5:dim=768-test_case0] Warning: You are sending unauthenticated requests to the HF Hub. Please set a HF_TOKEN to enable higher rate limits and faster downloads.
WARNING  2026-02-05 17:18:41,852 huggingface_hub.utils._http:779 uncategorized: Warning: You are sending unauthenticated
         requests to the HF Hub. Please set a HF_TOKEN to enable higher rate limits and faster downloads.               
WARNING  2026-02-05 17:18:44,152                                                                                        
         transformers_modules.nomic_hyphen_ai.nomic_hyphen_bert_hyphen_2048.7710840340a098cfb869c4f65e87cf2b1b70caca.mod
         eling_hf_nomic_bert:466 uncategorized: <All keys matched successfully>                                         
PASSED
tests/integration/vector_io/test_vector_io.py::test_insert_chunks_with_precomputed_embeddings[vector_io=infinispan-emb=inline::sentence-transformers/nomic-ai/nomic-embed-text-v1.5:dim=768] PASSED
tests/integration/vector_io/test_vector_io.py::test_query_returns_valid_object_when_identical_to_embedding_in_vdb[vector_io=infinispan-emb=inline::sentence-transformers/nomic-ai/nomic-embed-text-v1.5:dim=768] PASSED
tests/integration/vector_io/test_vector_io.py::test_auto_extract_embedding_dimension[vector_io=infinispan-emb=inline::sentence-transformers/nomic-ai/nomic-embed-text-v1.5:dim=768] PASSED
tests/integration/vector_io/test_vector_io.py::test_provider_auto_selection_single_provider[vector_io=infinispan-emb=inline::sentence-transformers/nomic-ai/nomic-embed-text-v1.5:dim=768] PASSED
tests/integration/vector_io/test_vector_io.py::test_provider_id_override[vector_io=infinispan-emb=inline::sentence-transformers/nomic-ai/nomic-embed-text-v1.5:dim=768] PASSED
tests/integration/vector_io/test_vector_io.py::test_insert_chunks[vector_io=infinispan-emb=inline::sentence-transformers/nomic-ai/nomic-embed-text-v1.5:dim=768-test_case1] PASSED
tests/integration/vector_io/test_vector_io.py::test_insert_chunks[vector_io=infinispan-emb=inline::sentence-transformers/nomic-ai/nomic-embed-text-v1.5:dim=768-test_case2] PASSED
tests/integration/vector_io/test_vector_io.py::test_insert_chunks[vector_io=infinispan-emb=inline::sentence-transformers/nomic-ai/nomic-embed-text-v1.5:dim=768-test_case3] PASSED
tests/integration/vector_io/test_vector_io.py::test_insert_chunks[vector_io=infinispan-emb=inline::sentence-transformers/nomic-ai/nomic-embed-text-v1.5:dim=768-test_case4] PASSED

Even CI is happy, see: https://github.com/rigazilla/llama-stack/actions/runs/21718857111

@meta-cla
Copy link

meta-cla bot commented Feb 5, 2026

Hi @rigazilla!

Thank you for your pull request and welcome to our community.

Action Required

In order to merge any pull request (code, docs, etc.), we require contributors to sign our Contributor License Agreement, and we don't seem to have one on file for you.

Process

In order for us to review and merge your suggested changes, please sign at https://code.facebook.com/cla. If you are contributing on behalf of someone else (eg your employer), the individual CLA may not be sufficient and your employer may need to sign the corporate CLA.

Once the CLA is signed, our tooling will perform checks and validations. Afterwards, the pull request will be tagged with CLA signed. The tagging process may take up to 1 hour after signing. Please give it that time before contacting us about it.

If you have received this in error or have any questions, please contact us at cla@meta.com. Thanks!

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.

Add Infinispan as Vector provider

1 participant