llama-stack-mirror/llama_stack
Dinesh Yeduguru 57bafd0f8c
fix faiss serialize and serialize of index (#464)
faiss serialize index returns a np object, that we first need to save to
buffer and then write to sqllite. Since we are using json, we need to
base64 encode the data.

Same in the read path, we base64 decode and read into np array and then
call into deserialize index.

tests:
torchrun $CONDA_PREFIX/bin/pytest -v -s -m "faiss"
llama_stack/providers/tests/memory/test_memory.py

Co-authored-by: Dinesh Yeduguru <dineshyv@fb.com>
2024-11-15 18:02:48 -08:00
..
apis No more model_id warnings 2024-11-15 12:20:18 -08:00
cli Add a verify-download command to llama CLI (#457) 2024-11-14 11:47:51 -08:00
distribution No more model_id warnings 2024-11-15 12:20:18 -08:00
providers fix faiss serialize and serialize of index (#464) 2024-11-15 18:02:48 -08:00
scripts Add a test for CLI, but not fully done so disabled 2024-09-19 13:27:07 -07:00
templates Update provider types and prefix with inline:: 2024-11-12 12:54:44 -08:00
__init__.py API Updates (#73) 2024-09-17 19:51:35 -07:00