trending/harness-sdk
1
0
Fork 0
mirror of https://github.com/strands-agents/harness-sdk.git synced 2026-10-02 02:44:48 +08:00
Code Issues Packages Projects Releases Wiki Activity

test(bedrock): stabilize the cache-point-after-document integ test (#4119)

Browse Source
This commit is contained in:
opieter-aws
2026-09-02 16:48:09 -04:00
committed by GitHub
parent 2c18d8f757
commit e13bfcd4c7
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
strands-py/tests_integ/models/test_model_bedrock.py
+3 -1
View File
@@ -860,6 +860,8 @@ def test_a_caller_ttl_does_not_conflict_with_the_tools_ttl(quiet_strands_logging
assert result.metrics.latest_agent_invocation.usage.get("cacheWriteInputTokens", 0) > 0
# A ~100 KB PDF plus the SDK's throttling backoff can outlast the global 90s timeout.
@pytest.mark.timeout(300)
@pytest.mark.parametrize("document_format", ["csv", "pdf"])
def test_a_cache_point_after_a_document_is_accepted(quiet_strands_logging, letter_pdf, document_format):
"""Bedrock refuses a cache point directly after a non-PDF document but allows one after a PDF.
@@ -878,6 +880,6 @@ def test_a_cache_point_after_a_document_is_accepted(quiet_strands_logging, lette
model = BedrockModel(model_id=_CACHE_TTL_MODEL_ID, cache_config=CacheConfig(strategy="auto"))
agent = Agent(model=model, load_tools_from_directory=False, callback_handler=None)
result = agent([{"text": _durable_prefix()}, document, {"cachePoint": {"type": "default"}}, {"text": "Summarize."}])
result = agent([{"text": _durable_prefix()}, document, {"cachePoint": {"type": "default"}}, {"text": "Reply OK."}])
assert result.stop_reason == "end_turn"