integ tests

Author: dbschmigelski

tests_integ/models/test_model_bedrock.py

@@ -267,3 +267,19 @@ def test_redacted_content_handling():
     assert "reasoningContent" in result.message["content"][0]
     assert "redactedContent" in result.message["content"][0]["reasoningContent"]
     assert isinstance(result.message["content"][0]["reasoningContent"]["redactedContent"], bytes)
+
+
+def test_multi_prompt_system_content():
+    """Test multi-prompt system content blocks."""
+    system_prompt_content = [
+        {"text": "You are a helpful assistant."},
+        {"text": "Always be concise."},
+        {"text": "End responses with 'Done.'"}
+    ]
+    
+    agent = Agent(
+        system_prompt=system_prompt_content,
+        load_tools_from_directory=False
+    )
+    result = agent("Hello!")
+    # just verifying there is no failure

tests_integ/models/test_model_openai.py

@@ -221,3 +221,16 @@ def test_rate_limit_throttling_integration_no_retries(model):
         # Verify it's a rate limit error
         error_message = str(exc_info.value).lower()
         assert "rate limit" in error_message or "tokens per min" in error_message
+
+
+def test_content_blocks_handling(model):
+    """Test that content blocks are handled properly without failures."""
+    content = [
+        {"text": "What is 2+2?"},
+        {"text": "Please be brief."}
+    ]
+    
+    agent = Agent(model=model, load_tools_from_directory=False)
+    result = agent(content)
+    
+    assert "4" in result.message["content"][0]["text"]

tests_integ/test_bedrock_cache_point.py

@@ -29,3 +29,30 @@ def cache_point_callback_handler(**kwargs):
     agent = Agent(messages=messages, callback_handler=cache_point_callback_handler, load_tools_from_directory=False)
     agent("What is favorite color?")
     assert cache_point_usage > 0
+
+
+def test_bedrock_multi_prompt_cache_point():
+    """Test multi-prompt system with cache point."""
+    system_prompt_content = [
+        {"text": "You are a helpful assistant." * 500},  # Long text for cache
+        {"cachePoint": {"type": "default"}},
+        {"text": "Always respond with enthusiasm!"}
+    ]
+    
+    cache_point_usage = 0
+
+    def cache_point_callback_handler(**kwargs):
+        nonlocal cache_point_usage
+        if "event" in kwargs and kwargs["event"] and "metadata" in kwargs["event"] and kwargs["event"]["metadata"]:
+            metadata = kwargs["event"]["metadata"]
+            if "usage" in metadata and metadata["usage"]:
+                if "cacheReadInputTokens" in metadata["usage"] or "cacheWriteInputTokens" in metadata["usage"]:
+                    cache_point_usage += 1
+
+    agent = Agent(
+        system_prompt=system_prompt_content,
+        callback_handler=cache_point_callback_handler,
+        load_tools_from_directory=False
+    )
+    agent("Hello!")
+    assert cache_point_usage > 0