mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-17 15:12:35 +00:00
[fix] Fix mock vector DB schema in Qdrant tests
This commit is contained in:
parent
26b4340de3
commit
fe1b76a94c
1 changed files with 3 additions and 1 deletions
|
|
@ -54,7 +54,9 @@ def mock_vector_db(vector_db_id) -> MagicMock:
|
||||||
mock_vector_db.identifier = vector_db_id
|
mock_vector_db.identifier = vector_db_id
|
||||||
mock_vector_db.embedding_dimension = 384
|
mock_vector_db.embedding_dimension = 384
|
||||||
mock_vector_db.model_dump_json.return_value = (
|
mock_vector_db.model_dump_json.return_value = (
|
||||||
'{"identifier": "' + vector_db_id + '", "embedding_model": "embedding_model", "embedding_dimension": 384}'
|
'{"identifier": "'
|
||||||
|
+ vector_db_id
|
||||||
|
+ '", "provider_id": "qdrant", "embedding_model": "embedding_model", "embedding_dimension": 384}'
|
||||||
)
|
)
|
||||||
return mock_vector_db
|
return mock_vector_db
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue