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

fix: last_id when listing files in vector store

Browse source 
# What does this PR do?


## Test Plan
This commit is contained in:
Eric Huang 2025-11-05 11:17:21 -08:00
parent 628e38b3d5
commit 2514822e6a
2 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/llama_stack/providers/utils/memory/openai_vector_store_mixin.py
View file

@ -886,8 +886,8 @@ class OpenAIVectorStoreMixin(ABC):
# Determine pagination info
has_more = len(file_objects) > limit
first_id = file_objects[0].id if file_objects else None
last_id = file_objects[-1].id if file_objects else None
first_id = limited_files[0].id if file_objects else None
last_id = limited_files[-1].id if file_objects else None
return VectorStoreListFilesResponse(
data=limited_files,

2
tests/integration/vector_io/test_openai_vector_stores.py
View file

@ -825,7 +825,7 @@ def test_openai_vector_store_list_files(
assert first_page.has_more
assert len(first_page.data) == 2
assert first_page.first_id == first_page.data[0].id
assert first_page.last_id != first_page.data[-1].id
assert first_page.last_id == first_page.data[-1].id
next_page = compat_client.vector_stores.files.list(
vector_store_id=vector_store.id, limit=2, after=first_page.data[-1].id