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

chore: fix flaky unit test and add proper shutdown for file batches

Browse source
This commit is contained in:
Swapna Lekkala 2025-10-07 14:09:28 -07:00
parent 1970b4aa4b
commit cf418f3394
9 changed files with 39 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_stack/providers/inline/vector_io/faiss/faiss.py
View file

@ -225,8 +225,8 @@ class FaissVectorIOAdapter(OpenAIVectorStoreMixin, VectorIO, VectorDBsProtocolPr
await self.initialize_openai_vector_stores()
async def shutdown(self) -> None:
# Cleanup if needed
pass
# Clean up mixin resources (file batch tasks)
await super().shutdown()
async def health(self) -> HealthResponse:
"""

4
llama_stack/providers/inline/vector_io/sqlite_vec/sqlite_vec.py
View file

@ -434,8 +434,8 @@ class SQLiteVecVectorIOAdapter(OpenAIVectorStoreMixin, VectorIO, VectorDBsProtoc
await self.initialize_openai_vector_stores()
async def shutdown(self) -> None:
# nothing to do since we don't maintain a persistent connection
pass
# Clean up mixin resources (file batch tasks)
await super().shutdown()
async def list_vector_dbs(self) -> list[VectorDB]:
return [v.vector_db for v in self.cache.values()]