braintrust 0.3.13__py3-none-any.whl → 0.3.14__py3-none-any.whl

braintrust/gitutil.py

@@ -88,8 +88,12 @@ def get_past_n_ancestors(n=1000, remote=None):
         if ancestor_output is None:
             return
         ancestor = repo.commit(ancestor_output)
+        count = 0
         for _ in range(n):
+            if count >= n:
+                break
             yield ancestor.hexsha
+            count += 1
             try:
                 if ancestor.parents:
                     ancestor = ancestor.parents[0]

braintrust/logger.py

@@ -1104,7 +1104,7 @@ class _HTTPBackgroundLogger:
             _HTTPBackgroundLogger._write_payload_to_dir(payload_dir=self.all_publish_payloads_dir, payload=dataStr)
         for i in range(self.num_tries):
             start_time = time.time()
-            resp = conn.post("/logs3", data=dataStr)
+            resp = conn.post("/logs3", data=dataStr.encode("utf-8"))
             if resp.ok:
                 return
             resp_errmsg = f"{resp.status_code}: {resp.text}"

braintrust/score.py

@@ -34,6 +34,7 @@ class Score(SerializableDataClass):
 
     def as_dict(self):
         return {
+            "name": self.name,
             "score": self.score,
             "metadata": self.metadata,
         }

braintrust/test_score.py

@@ -0,0 +1,157 @@
+import json
+import unittest
+
+from .score import Score
+
+
+class TestScore(unittest.TestCase):
+    def test_as_dict_includes_all_required_fields(self):
+        """Test that as_dict() includes name, score, and metadata fields."""
+        score = Score(name="test_scorer", score=0.85, metadata={"key": "value"})
+        result = score.as_dict()
+
+        self.assertIn("name", result)
+        self.assertIn("score", result)
+        self.assertIn("metadata", result)
+
+        self.assertEqual(result["name"], "test_scorer")
+        self.assertEqual(result["score"], 0.85)
+        self.assertEqual(result["metadata"], {"key": "value"})
+
+    def test_as_dict_with_null_score(self):
+        """Test that as_dict() works correctly with null score."""
+        score = Score(name="null_scorer", score=None, metadata={})
+        result = score.as_dict()
+
+        self.assertEqual(result["name"], "null_scorer")
+        self.assertIsNone(result["score"])
+        self.assertEqual(result["metadata"], {})
+
+    def test_as_dict_with_empty_metadata(self):
+        """Test that as_dict() works correctly with empty metadata."""
+        score = Score(name="empty_metadata_scorer", score=1.0)
+        result = score.as_dict()
+
+        self.assertEqual(result["name"], "empty_metadata_scorer")
+        self.assertEqual(result["score"], 1.0)
+        self.assertEqual(result["metadata"], {})
+
+    def test_as_dict_with_complex_metadata(self):
+        """Test that as_dict() works correctly with complex nested metadata."""
+        complex_metadata = {
+            "reason": "Test reason",
+            "details": {"nested": {"deeply": "value"}},
+            "list": [1, 2, 3],
+            "bool": True,
+        }
+        score = Score(name="complex_scorer", score=0.5, metadata=complex_metadata)
+        result = score.as_dict()
+
+        self.assertEqual(result["name"], "complex_scorer")
+        self.assertEqual(result["score"], 0.5)
+        self.assertEqual(result["metadata"], complex_metadata)
+
+    def test_as_json_serialization(self):
+        """Test that as_json() produces valid JSON string."""
+        score = Score(name="json_scorer", score=0.75, metadata={"test": "data"})
+        json_str = score.as_json()
+
+        # Should be valid JSON
+        parsed = json.loads(json_str)
+
+        self.assertEqual(parsed["name"], "json_scorer")
+        self.assertEqual(parsed["score"], 0.75)
+        self.assertEqual(parsed["metadata"], {"test": "data"})
+
+    def test_from_dict_round_trip(self):
+        """Test that Score can be serialized to dict and deserialized back."""
+        original = Score(
+            name="round_trip_scorer", score=0.95, metadata={"info": "test"}
+        )
+
+        # Serialize to dict
+        as_dict = original.as_dict()
+
+        # Deserialize from dict
+        restored = Score.from_dict(as_dict)
+
+        self.assertEqual(restored.name, original.name)
+        self.assertEqual(restored.score, original.score)
+        self.assertEqual(restored.metadata, original.metadata)
+
+    def test_array_of_scores_serialization(self):
+        """Test that arrays of Score objects can be serialized correctly."""
+        scores = [
+            Score(name="score_1", score=0.8, metadata={"index": 1}),
+            Score(name="score_2", score=0.6, metadata={"index": 2}),
+            Score(name="score_3", score=None, metadata={}),
+        ]
+
+        # Serialize each score
+        serialized = [s.as_dict() for s in scores]
+
+        # Check that all scores have required fields
+        for i, s_dict in enumerate(serialized):
+            self.assertIn("name", s_dict)
+            self.assertIn("score", s_dict)
+            self.assertIn("metadata", s_dict)
+            self.assertEqual(s_dict["name"], f"score_{i + 1}")
+
+        # Check specific values
+        self.assertEqual(serialized[0]["score"], 0.8)
+        self.assertEqual(serialized[1]["score"], 0.6)
+        self.assertIsNone(serialized[2]["score"])
+
+    def test_array_of_scores_json_serialization(self):
+        """Test that arrays of Score objects can be JSON serialized."""
+        scores = [
+            Score(name="json_score_1", score=0.9),
+            Score(name="json_score_2", score=0.7),
+        ]
+
+        # Serialize to JSON
+        serialized = [s.as_dict() for s in scores]
+        json_str = json.dumps(serialized)
+
+        # Parse back
+        parsed = json.loads(json_str)
+
+        self.assertEqual(len(parsed), 2)
+        self.assertEqual(parsed[0]["name"], "json_score_1")
+        self.assertEqual(parsed[0]["score"], 0.9)
+        self.assertEqual(parsed[1]["name"], "json_score_2")
+        self.assertEqual(parsed[1]["score"], 0.7)
+
+    def test_score_validation_enforces_bounds(self):
+        """Test that Score validates score values are between 0 and 1."""
+        # Valid scores
+        Score(name="valid_0", score=0.0)
+        Score(name="valid_1", score=1.0)
+        Score(name="valid_mid", score=0.5)
+        Score(name="valid_null", score=None)
+
+        # Invalid scores
+        with self.assertRaises(ValueError):
+            Score(name="invalid_negative", score=-0.1)
+
+        with self.assertRaises(ValueError):
+            Score(name="invalid_over_one", score=1.1)
+
+    def test_score_does_not_include_deprecated_error_field(self):
+        """Test that as_dict() does not include the deprecated error field."""
+        score = Score(name="test_scorer", score=0.5)
+        result = score.as_dict()
+
+        # The error field should not be in the serialized output
+        self.assertNotIn("error", result)
+
+        # Even if error was set (though deprecated), it shouldn't be in as_dict
+        score_with_error = Score(name="error_scorer", score=0.5)
+        score_with_error.error = Exception("test")  # Set after construction
+        result_with_error = score_with_error.as_dict()
+
+        self.assertNotIn("error", result_with_error)
+
+
+if __name__ == "__main__":
+    unittest.main()

braintrust/version.py

@@ -1,4 +1,4 @@
-VERSION = "0.3.13"
+VERSION = "0.3.14"
 
 # this will be templated during the build
-GIT_COMMIT = "cef88a007fa60f4cd873f1d891a54ce5e173f3aa"
+GIT_COMMIT = "dbbc1894ef31143816e5913676301261bc44aa4c"

{braintrust-0.3.13.dist-info → braintrust-0.3.14.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: braintrust
-Version: 0.3.13
+Version: 0.3.14
 Summary: SDK for integrating Braintrust
 Home-page: https://www.braintrust.dev
 Author: Braintrust

{braintrust-0.3.13.dist-info → braintrust-0.3.14.dist-info}/RECORD

@@ -10,11 +10,11 @@ braintrust/framework.py,sha256=Uf2DogzQG4suCcmABeQzc-5SWCO5tB2ic1OeQFZo_o4,57618
 braintrust/framework2.py,sha256=DTnCql0_Be9kYZ7QkgJHEUAW4_tU-vuqni3vxdCcnyk,16784
 braintrust/generated_types.py,sha256=StMC58Mg_Imc4Z-xpLzqZql2E5K14b92FAnxj0aOrtw,4521
 braintrust/git_fields.py,sha256=oT1h1nVkidGrIIamtIQfOpbuEzd1n-nup9bx54J_u0A,1480
-braintrust/gitutil.py,sha256=bEk38AlNtT-umtdCJ9lnSXlbKXsvjBOyTTsmzUKiVtM,5586
+braintrust/gitutil.py,sha256=01hCZG291992Knj9q1gOjb42XvfzruMm5yeA68FufAw,5676
 braintrust/graph_util.py,sha256=lABIOMzxHf6E5LfDYfqa4OUR4uaW7xgUYNq5WGewD4w,5594
 braintrust/http_headers.py,sha256=9ZsDcsAKG04SGowsgchZktD6rG_oSTKWa8QyGUPA4xE,154
 braintrust/id_gen.py,sha256=PVkz-pS-9AzgmnAgpV-jgOFFo4hfl6e3IP9dVt6FouQ,1595
-braintrust/logger.py,sha256=NTbxgXa_qsu7bwl4bUoLuj9HNDuaGQiDlu0-addkAQc,210560
+braintrust/logger.py,sha256=Pesk94By7gPiVLoqA7NJjlLq4XslP7EmioRrusjVd6c,210576
 braintrust/merge_row_batch.py,sha256=NX4jRE9uuFB3Z7btrarQp_di84_NGTjvzpJhksn82W8,9882
 braintrust/oai.py,sha256=DW_i6bXXDprwZ_uwcdIE25MUxx5y5DDdqr0ucdIWP9I,33187
 braintrust/object.py,sha256=vYLyYWncsqLD00zffZUJwGTSkcJF9IIXmgIzrx3Np5c,632
@@ -23,7 +23,7 @@ braintrust/prompt.py,sha256=c7UR-UeJ8Hf-DGyFRyUxSgXEOkZFhVd9-IhvyPhy1xI,1938
 braintrust/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 braintrust/queue.py,sha256=5Mjb5FvqC9hq9f4qykCQIxrZ8MFrQlCvf7XIlKyYRGA,3738
 braintrust/resource_manager.py,sha256=Rs4yEd_ShV5YV6WC0YqxkQKXEfBtbTUZ4S4G78oF5Mk,723
-braintrust/score.py,sha256=aHs5bMDI7e8Un4VdfAejHM8T38PDqdiia4YDU303YkY,3632
+braintrust/score.py,sha256=44MyCUiMb9Bn1kyYqa3Kej33Cx2-5MnQ-_Rcl5pJfQw,3663
 braintrust/serializable_data_class.py,sha256=TjlLpUmVAJ_VW7-aiEecBVQ5d3k4bgQsVeJo_5yIAU0,2471
 braintrust/span_identifier_v1.py,sha256=_i33QR-LIalyBkV28cWWh0OUPtA7cjv-VCc894-XdD0,5209
 braintrust/span_identifier_v2.py,sha256=qLi2iXs-KmIfrQ3BjW6_yLpMYqKh-7_LrotiR4cplwc,9024
@@ -37,12 +37,13 @@ braintrust/test_id_gen.py,sha256=mgArTyEBV-Xv21ARHPSHEPBsJshZrvIiPjBLNOKsAko,249
 braintrust/test_logger.py,sha256=Z3euqSHmc4kqsyKo838tkYGK8B1rTkIdxl7QZYR1CLk,98516
 braintrust/test_otel.py,sha256=janmEtu6dyFQL8N68WjRTS-65Kr5vSNSSwIaBWqXlyw,27243
 braintrust/test_queue.py,sha256=MdH6R9uSk_4akY4Db514Cpukwiy2RJ76Lqts1nQwZJY,8432
+braintrust/test_score.py,sha256=3hgwO_g_yq9VxX6no2YXfEMpBegVeUYLPh0NX4T5rHs,5879
 braintrust/test_serializable_data_class.py,sha256=b04Ym64YtC6GJRGbKIN4J20RG1QN1FlnODwtEQh4sv0,1897
 braintrust/test_span_components.py,sha256=UnF6ZL4k41XZ-CnfbjuqLeK4MZLtHTMdID3CMh3g1os,16344
 braintrust/test_util.py,sha256=gyqe2JspRP7oXlp6ENztZe2fdRTOEMZMKpQi00y1DSc,4538
 braintrust/test_version.py,sha256=hk5JKjEFbNJ_ONc1VEkqHquflzre34RpFhCEYLTK8iA,1051
 braintrust/util.py,sha256=Ec6sRkQw5BckGrFjdA4YTyu_2BaKmHh4tWDwAi_ysOw,7227
-braintrust/version.py,sha256=YyNQY-UJ0O_hVg8XR4VJGGwb-qChpEoYLN6jfWC4rEQ,118
+braintrust/version.py,sha256=3wAzsDK5T1yqm2oCx82UysM0PDJBVh8gI90sxUVKfQg,118
 braintrust/xact_ids.py,sha256=bdyp88HjlyIkglgLSqYlCYscdSH6EWVyE14sR90Xl1s,658
 braintrust/cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 braintrust/cli/__main__.py,sha256=wCBKHGVmn3IT_yMXk5qfDwyI2SV2gf1tLr0NTxm9T8k,1519
@@ -109,8 +110,8 @@ braintrust/wrappers/claude_agent_sdk/__init__.py,sha256=CSXJWy-z2fHF7h4VJjLSnXJv
 braintrust/wrappers/claude_agent_sdk/_wrapper.py,sha256=uzElIOwwPmF_Y5fbWcKWEPC8HnSzW7byzpiuVKK0TXE,15613
 braintrust/wrappers/claude_agent_sdk/test_wrapper.py,sha256=0NmohdECudFvWtc-5PbANtTXzexkkwIJhGbujydDrT8,6826
 braintrust/wrappers/google_genai/__init__.py,sha256=PGFMuR3c4Gc3SUt24eP7z5AzdS2Dc1uF1d3QPCnLnuo,16018
-braintrust-0.3.13.dist-info/METADATA,sha256=cZM8kCyxf7v0SHVRVPHBElZcVIis1GcwcXTOea54WPo,3702
-braintrust-0.3.13.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-braintrust-0.3.13.dist-info/entry_points.txt,sha256=Zpc0_09g5xm8as5jHqqFq7fhwO0xHSNct_TrEMONS7Q,60
-braintrust-0.3.13.dist-info/top_level.txt,sha256=hw1-y-UFMf60RzAr8x_eM7SThbIuWfQsQIbVvqSF83A,11
-braintrust-0.3.13.dist-info/RECORD,,
+braintrust-0.3.14.dist-info/METADATA,sha256=9ibXkb190ZtT4_UgxlPjgU4y6eKJI_Y6h0eyrzm9oEs,3702
+braintrust-0.3.14.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+braintrust-0.3.14.dist-info/entry_points.txt,sha256=Zpc0_09g5xm8as5jHqqFq7fhwO0xHSNct_TrEMONS7Q,60
+braintrust-0.3.14.dist-info/top_level.txt,sha256=hw1-y-UFMf60RzAr8x_eM7SThbIuWfQsQIbVvqSF83A,11
+braintrust-0.3.14.dist-info/RECORD,,