andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-04-30 23:02:50 +00:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

improve assert

Browse Source
This commit is contained in:
lorenzejay
2026-02-10 09:40:37 -08:00
parent 2ee96ec8c7
commit 5828758ac0
1 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
lib/crewai/tests/utilities/test_summarize_integration.py
View File

@@ -114,10 +114,12 @@ class TestSummarizeDirectOpenAI:
assert messages[0]["role"] == "system"
assert messages[0]["content"] == original_system_content
# Summary should be a user message
# Summary should be a user message with <summary> block
summary_msg = messages[-1]
assert summary_msg["role"] == "user"
assert len(summary_msg["content"]) > 0
assert "<summary>" in summary_msg["content"]
assert "</summary>" in summary_msg["content"]
class TestSummarizeDirectAnthropic:
@@ -142,6 +144,8 @@ class TestSummarizeDirectAnthropic:
summary_msg = messages[-1]
assert summary_msg["role"] == "user"
assert len(summary_msg["content"]) > 0
assert "<summary>" in summary_msg["content"]
assert "</summary>" in summary_msg["content"]
class TestSummarizeDirectGemini:
@@ -166,6 +170,8 @@ class TestSummarizeDirectGemini:
summary_msg = messages[-1]
assert summary_msg["role"] == "user"
assert len(summary_msg["content"]) > 0
assert "<summary>" in summary_msg["content"]
assert "</summary>" in summary_msg["content"]
class TestCrewKickoffCompaction:
@@ -251,7 +257,9 @@ class TestSummarizePreservesFiles:
# System message preserved
assert messages[0]["role"] == "system"
# Files should be on the summary message
# Files should be on the summary message with <summary> block
summary_msg = messages[-1]
assert "<summary>" in summary_msg["content"]
assert "</summary>" in summary_msg["content"]
assert "files" in summary_msg
assert "reference.pdf" in summary_msg["files"]