phoenix-oss/llama-stack
1
0
Fork 0
forked from phoenix-oss/llama-stack-mirror
Code Pull requests Releases Packages 2 Activity Actions
llama-stack/llama_stack/providers/tests/vector_io
History
Exact
Yuan Tang a79a083e39
Fix broken pgvector provider and memory leaks (#947) 
This PR fixes the broken pgvector provider as well as wraps all cursor
object creations with context manager to ensure that they get properly
closed to avoid potential memory leaks.

```
> pytest llama_stack/providers/tests/vector_io/test_vector_io.py   -m "pgvector" --env EMBEDDING_DIMENSION=384 --env PGVECTOR_PORT=7432 --env PGVECTOR_DB=db --env PGVECTOR_USER=user --env PGVECTOR_PASSWORD=pass   -v -s --tb=short --disable-warnings

llama_stack/providers/tests/vector_io/test_vector_io.py::TestVectorIO::test_banks_list[-pgvector] PASSED
llama_stack/providers/tests/vector_io/test_vector_io.py::TestVectorIO::test_banks_register[-pgvector] PASSED
llama_stack/providers/tests/vector_io/test_vector_io.py::TestVectorIO::test_query_documents[-pgvector] The scores are: [0.8168284974053789, 0.8080469278964486, 0.8050996198466661]
PASSED
```

---------

Signed-off-by: Yuan Tang <terrytangyuan@gmail.com>
2025-02-05 09:32:05 -08:00
..
fixtures [memory refactor][2/n] Update faiss and make it pass tests (#830) 2025-01-22 10:02:15 -08:00
__init__.py [memory refactor][2/n] Update faiss and make it pass tests (#830) 2025-01-22 10:02:15 -08:00
conftest.py Fix broken pgvector provider and memory leaks (#947) 2025-02-05 09:32:05 -08:00
fixtures.py [memory refactor][2/n] Update faiss and make it pass tests (#830) 2025-01-22 10:02:15 -08:00
test_vector_io.py Fix broken pgvector provider and memory leaks (#947) 2025-02-05 09:32:05 -08:00
test_vector_store.py [memory refactor][6/n] Update naming and routes (#839) 2025-01-22 10:39:13 -08:00