braintrust 0.3.15__py3-none-any.whl → 0.4.1__py3-none-any.whl

braintrust/wrappers/test_agno.py

@@ -5,7 +5,6 @@
 # pyright: reportUnknownVariableType=false
 # pyright: reportUnknownArgumentType=false
 import pytest
-
 from braintrust import logger
 from braintrust.test_helpers import init_test_logger
 from braintrust.wrappers.agno import setup_agno

braintrust/wrappers/test_dspy.py

@@ -4,7 +4,6 @@ Tests for DSPy integration with Braintrust.
 
 import dspy
 import pytest
-
 from braintrust import logger
 from braintrust.test_helpers import init_test_logger
 from braintrust.wrappers.dspy import BraintrustDSpyCallback

braintrust/wrappers/test_google_genai.py

@@ -2,12 +2,11 @@ import time
 from pathlib import Path
 
 import pytest
-from google.genai import types
-from google.genai.client import Client
-
 from braintrust import logger
 from braintrust.test_helpers import init_test_logger
 from braintrust.wrappers.google_genai import setup_genai
+from google.genai import types
+from google.genai.client import Client
 
 PROJECT_NAME = "test-genai-app"
 MODEL = "gemini-2.0-flash-001"
@@ -548,7 +547,6 @@ def test_error_handling(memory_logger):
     assert log["error"]
 
 
-# Test 9: Stop Sequences
 @pytest.mark.vcr
 def test_stop_sequences(memory_logger):
     """Test stop sequences parameter."""
@@ -572,3 +570,65 @@ def test_stop_sequences(memory_logger):
     assert len(spans) == 1
     span = spans[0]
     assert span["metadata"]["model"] == MODEL
+
+
+def test_attachment_in_config(memory_logger):
+    """Test that attachments in config are preserved through serialization."""
+    from braintrust.bt_json import bt_safe_deep_copy
+    from braintrust.logger import Attachment
+
+    attachment = Attachment(data=b"config data", filename="config.txt", content_type="text/plain")
+
+    # Simulate config with attachment
+    config = {"temperature": 0.5, "context_file": attachment, "max_output_tokens": 100}
+
+    # Test bt_safe_deep_copy preserves attachment
+    copied = bt_safe_deep_copy(config)
+    assert copied["context_file"] is attachment
+    assert copied["temperature"] == 0.5
+
+
+def test_nested_attachments_in_contents(memory_logger):
+    """Test that nested attachments in contents are preserved."""
+    from braintrust.bt_json import bt_safe_deep_copy
+    from braintrust.logger import Attachment, ExternalAttachment
+
+    attachment1 = Attachment(data=b"file1", filename="file1.txt", content_type="text/plain")
+    attachment2 = ExternalAttachment(url="s3://bucket/file2.pdf", filename="file2.pdf", content_type="application/pdf")
+
+    # Simulate contents with nested attachments
+    contents = [
+        {"role": "user", "parts": [{"text": "Check these files"}, {"file": attachment1}]},
+        {"role": "model", "parts": [{"text": "Analyzed"}, {"result_file": attachment2}]},
+    ]
+
+    copied = bt_safe_deep_copy(contents)
+
+    # Verify attachments preserved
+    assert copied[0]["parts"][1]["file"] is attachment1
+    assert copied[1]["parts"][1]["result_file"] is attachment2
+
+
+def test_attachment_with_pydantic_model(memory_logger):
+    """Test that attachments work alongside Pydantic model serialization."""
+    from braintrust.bt_json import bt_safe_deep_copy
+    from braintrust.logger import Attachment
+    from pydantic import BaseModel
+
+    class TestModel(BaseModel):
+        name: str
+        value: int
+
+    attachment = Attachment(data=b"model data", filename="model.txt", content_type="text/plain")
+
+    # Structure with both Pydantic model and attachment
+    data = {"model_config": TestModel(name="test", value=42), "context_file": attachment}
+
+    copied = bt_safe_deep_copy(data)
+
+    # Pydantic model should be converted to dict
+    assert isinstance(copied["model_config"], dict)
+    assert copied["model_config"]["name"] == "test"
+
+    # Attachment should be preserved
+    assert copied["context_file"] is attachment

braintrust/wrappers/test_litellm.py

@@ -3,7 +3,6 @@ import time
 
 import litellm
 import pytest
-
 from braintrust import logger
 from braintrust.test_helpers import assert_dict_matches, init_test_logger
 from braintrust.wrappers.litellm import wrap_litellm