Skip to content

Commit 5d65656

Browse files
committed
New unit test
1 parent c14753c commit 5d65656

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

tests/unit/models/test_requests.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -122,6 +122,16 @@ def test_get_documents(self) -> None:
122122
assert documents[1]["content"] == "kind: Pod\n metadata:\n name: private-reg"
123123
assert documents[1]["mime_type"] == "application/yaml"
124124

125+
def test_get_documents_no_attachments(self) -> None:
126+
"""Test the get_documents method."""
127+
attachments = []
128+
qr = QueryRequest(
129+
query="Tell me about Kubernetes",
130+
attachments=attachments,
131+
)
132+
documents = qr.get_documents()
133+
assert len(documents) == 0
134+
125135
def test_validate_provider_and_model(self) -> None:
126136
"""Test the validate_provider_and_model method."""
127137
qr = QueryRequest(

0 commit comments

Comments
 (0)