phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-03 18:00:36 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Added tests + docs + CI for Elasticsearch

Browse source
This commit is contained in:
Enrico Zimuel 2025-10-31 18:18:27 +01:00
parent 6f39c8994d
commit 22b27e6275
No known key found for this signature in database
GPG key ID: 6CB203F6934A69F1
9 changed files with 215 additions and 112 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tests/integration/vector_io/test_openai_vector_stores.py
View file

@ -34,6 +34,7 @@ def skip_if_provider_doesnt_support_openai_vector_stores(client_with_models):
"remote::pgvector",
"remote::qdrant",
"remote::weaviate",
"remote::elasticsearch",
]:
return
@ -54,6 +55,7 @@ def skip_if_provider_doesnt_support_openai_vector_stores_search(client_with_mode
"remote::pgvector",
"remote::qdrant",
"remote::weaviate",
"remote::elasticsearch",
],
"keyword": [
"inline::milvus",
@ -61,6 +63,7 @@ def skip_if_provider_doesnt_support_openai_vector_stores_search(client_with_mode
"remote::milvus",
"remote::pgvector",
"remote::weaviate",
"remote::elasticsearch",
],
"hybrid": [
"inline::milvus",
@ -68,6 +71,7 @@ def skip_if_provider_doesnt_support_openai_vector_stores_search(client_with_mode
"remote::milvus",
"remote::pgvector",
"remote::weaviate",
"remote::elasticsearch",
],
}
supported_providers = search_mode_support.get(search_mode, [])

2
tests/integration/vector_io/test_vector_io.py
View file

@ -154,6 +154,7 @@ def test_insert_chunks_with_precomputed_embeddings(
"inline::milvus": {"score_threshold": -1.0},
"inline::qdrant": {"score_threshold": -1.0},
"remote::qdrant": {"score_threshold": -1.0},
"remote::elasticsearch": {"score_threshold": -1.0},
}
vector_store_name = "test_precomputed_embeddings_db"
register_response = client_with_empty_registry.vector_stores.create(
@ -203,6 +204,7 @@ def test_query_returns_valid_object_when_identical_to_embedding_in_vdb(
"inline::milvus": {"score_threshold": 0.0},
"remote::qdrant": {"score_threshold": 0.0},
"inline::qdrant": {"score_threshold": 0.0},
"remote::elasticsearch": {"score_threshold": 0.0},
}
vector_store_name = "test_precomputed_embeddings_db"
register_response = client_with_empty_registry.vector_stores.create(