diff --git a/developer-knowledge/conftest.py b/developer-knowledge/conftest.py index 53122cb064..8f96841067 100644 --- a/developer-knowledge/conftest.py +++ b/developer-knowledge/conftest.py @@ -97,6 +97,9 @@ class SearchDocumentChunksResponse: def __init__(self, results=None): self.results = results or [DocumentChunk()] + def __iter__(self): + return iter(self.results) + class BatchGetDocumentsResponse: def __init__(self, documents=None): self.documents = documents or []