Skip to content

Commit fdaf3af

Browse files
committed
Removing chunk from _classify_blocks since it'll be backend chunk as oppose to an AI generated chunk
1 parent 214cabe commit fdaf3af

2 files changed

Lines changed: 0 additions & 7 deletions

File tree

backend/tenantfirstaid/chat.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,8 +31,6 @@ def _classify_blocks(
3131
yield TextChunk(content=content_block["text"])
3232
case "letter":
3333
yield LetterChunk(content=content_block["content"])
34-
case "done":
35-
yield DoneChunk()
3634
case _:
3735
current_app.logger.warning(
3836
f"Unhandled block type: {content_block['type']}"

backend/tests/test_chat.py

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -45,11 +45,6 @@ def test_unknown_block_type_is_skipped(self, app):
4545
result = chunks([{"type": "image", "image": "..."}])
4646
assert result == []
4747

48-
def test_done_passthrough(self):
49-
result = chunks([{"type": "done"}])
50-
assert len(result) == 1
51-
assert result[0].type == "done"
52-
5348
def test_done_chunk_serializes_correctly(self):
5449
assert json.loads(DoneChunk().model_dump_json()) == {"type": "done"}
5550

0 commit comments

Comments
 (0)