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

fix: Do not throw when listing vector stores (#2460)

Browse source 
When trying to `list` vector_stores , if we cannot retrieve one, log an
error and return all the ones that are valid.

### Test Plan 
```
pytest -sv --stack-config=http://localhost:8321 tests/integration/vector_io/test_openai_vector_stores.py  --embedding-model all-MiniLM-L6-v2
```
Also tested for `--stack-config fireworks`
This commit is contained in:
Hardik Shah 2025-06-17 11:19:43 -07:00 • committed by GitHub
parent 53ac8532e4
commit 822307e6d5
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 8 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

12
llama_stack/distribution/routers/vector_io.py
View file

@ -163,10 +163,14 @@ class VectorIORouter(VectorIO):
vector_dbs = await self.routing_table.get_all_with_type("vector_db") vector_dbs = await self.routing_table.get_all_with_type("vector_db")
all_stores = [] all_stores = []
for vector_db in vector_dbs: for vector_db in vector_dbs:
vector_store = await self.routing_table.get_provider_impl( try:
vector_db.identifier vector_store = await self.routing_table.get_provider_impl(
).openai_retrieve_vector_store(vector_db.identifier) vector_db.identifier
all_stores.append(vector_store) ).openai_retrieve_vector_store(vector_db.identifier)
all_stores.append(vector_store)
except Exception as e:
logger.error(f"Error retrieving vector store {vector_db.identifier}: {e}")
continue
# Sort by created_at # Sort by created_at
reverse_order = order == "desc" reverse_order = order == "desc"