braintrust 0.3.8__tar.gz → 0.3.9__tar.gz

{braintrust-0.3.8 → braintrust-0.3.9}/PKG-INFO

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

{braintrust-0.3.8 → braintrust-0.3.9}/src/braintrust/_generated_types.py

@@ -212,6 +212,17 @@ CallEvent = Union[
 ]
 
 
+class ChatCompletionContentPartFileFile(TypedDict):
+    file_data: NotRequired[Optional[str]]
+    filename: NotRequired[Optional[str]]
+    file_id: NotRequired[Optional[str]]
+
+
+class ChatCompletionContentPartFileWithTitle(TypedDict):
+    file: ChatCompletionContentPartFileFile
+    type: Literal['file']
+
+
 class ChatCompletionContentPartImageWithTitleImageUrl(TypedDict):
     url: str
     detail: NotRequired[Optional[Union[Literal['auto'], Literal['low'], Literal['high']]]]
@@ -342,7 +353,7 @@ class ChatCompletionTool(TypedDict):
 
 
 class CodeBundleRuntimeContext(TypedDict):
-    runtime: Literal['node', 'python']
+    runtime: Literal['node', 'python', 'browser']
     version: str
 
 
@@ -570,7 +581,7 @@ class Data(CodeBundle):
 
 
 class FunctionDataFunctionData1DataRuntimeContext(TypedDict):
-    runtime: Literal['node', 'python']
+    runtime: Literal['node', 'python', 'browser']
     version: str
 
 
@@ -649,7 +660,7 @@ class FunctionIdFunctionId3(TypedDict):
 
 
 class FunctionIdFunctionId4InlineContext(TypedDict):
-    runtime: Literal['node', 'python']
+    runtime: Literal['node', 'python', 'browser']
     version: str
 
 
@@ -668,16 +679,16 @@ class FunctionIdFunctionId4(TypedDict):
 FunctionIdRef = Mapping[str, Any]
 
 
-FunctionObjectType = Literal['prompt', 'tool', 'scorer', 'task', 'agent']
+FunctionObjectType = Literal['prompt', 'tool', 'scorer', 'task', 'agent', 'custom_view']
 
 
 FunctionOutputType = Literal['completion', 'score', 'any']
 
 
-FunctionTypeEnum = Literal['llm', 'scorer', 'task', 'tool']
+FunctionTypeEnum = Literal['llm', 'scorer', 'task', 'tool', 'custom_view']
 
 
-FunctionTypeEnumNullish = Literal['llm', 'scorer', 'task', 'tool']
+FunctionTypeEnumNullish = Literal['llm', 'scorer', 'task', 'tool', 'custom_view']
 
 
 class GitMetadataSettings(TypedDict):
@@ -1854,7 +1865,7 @@ class AnyModelParams(TypedDict):
     function_call: NotRequired[Optional[Union[Literal['auto'], Literal['none'], AnyModelParamsFunctionCall]]]
     n: NotRequired[Optional[float]]
     stop: NotRequired[Optional[Sequence[str]]]
-    reasoning_effort: NotRequired[Optional[Literal['minimal', 'low', 'medium', 'high']]]
+    reasoning_effort: NotRequired[Optional[Literal['none', 'minimal', 'low', 'medium', 'high']]]
     verbosity: NotRequired[Optional[Literal['low', 'medium', 'high']]]
     top_k: NotRequired[Optional[float]]
     stop_sequences: NotRequired[Optional[Sequence[str]]]
@@ -1894,7 +1905,11 @@ class AttachmentStatus(TypedDict):
     """
 
 
-ChatCompletionContentPart = Union[ChatCompletionContentPartTextWithTitle, ChatCompletionContentPartImageWithTitle]
+ChatCompletionContentPart = Union[
+    ChatCompletionContentPartTextWithTitle,
+    ChatCompletionContentPartImageWithTitle,
+    ChatCompletionContentPartFileWithTitle,
+]
 
 
 class ChatCompletionMessageParamChatCompletionMessageParam1(TypedDict):
@@ -1993,6 +2008,14 @@ class DatasetEvent(TypedDict):
     Whether this span is a root span
     """
     origin: NotRequired[Optional[ObjectReferenceNullish]]
+    comments: NotRequired[Optional[Sequence[Any]]]
+    """
+    Optional list of comments attached to this event
+    """
+    audit_data: NotRequired[Optional[Sequence[Any]]]
+    """
+    Optional list of audit entries attached to this event
+    """
 
 
 class Experiment(TypedDict):
@@ -2075,7 +2098,7 @@ class ModelParamsModelParams(TypedDict):
     function_call: NotRequired[Optional[Union[Literal['auto'], Literal['none'], ModelParamsModelParamsFunctionCall]]]
     n: NotRequired[Optional[float]]
     stop: NotRequired[Optional[Sequence[str]]]
-    reasoning_effort: NotRequired[Optional[Literal['minimal', 'low', 'medium', 'high']]]
+    reasoning_effort: NotRequired[Optional[Literal['none', 'minimal', 'low', 'medium', 'high']]]
     verbosity: NotRequired[Optional[Literal['low', 'medium', 'high']]]
 
 
@@ -2327,6 +2350,14 @@ class ExperimentEvent(TypedDict):
     Whether this span is a root span
     """
     origin: NotRequired[Optional[ObjectReferenceNullish]]
+    comments: NotRequired[Optional[Sequence[Any]]]
+    """
+    Optional list of comments attached to this event
+    """
+    audit_data: NotRequired[Optional[Sequence[Any]]]
+    """
+    Optional list of audit entries attached to this event
+    """
 
 
 class GraphNodeGraphNode7(TypedDict):
@@ -2437,6 +2468,18 @@ class ProjectLogsEvent(TypedDict):
     """
     span_attributes: NotRequired[Optional[SpanAttributes]]
     origin: NotRequired[Optional[ObjectReferenceNullish]]
+    comments: NotRequired[Optional[Sequence[Any]]]
+    """
+    Optional list of comments attached to this event
+    """
+    audit_data: NotRequired[Optional[Sequence[Any]]]
+    """
+    Optional list of audit entries attached to this event
+    """
+    _async_scoring_state: NotRequired[Optional[Any]]
+    """
+    The async scoring state for this event
+    """
 
 
 class ProjectScore(TypedDict):

{braintrust-0.3.8 → braintrust-0.3.9}/src/braintrust/framework2.py

@@ -3,7 +3,6 @@ import json
 from typing import Any, Callable, Dict, List, Optional, Union, overload
 
 import slugify
-
 from braintrust.logger import api_conn, app_conn, login
 
 from .framework import _is_lazy_load, bcolors  # type: ignore

{braintrust-0.3.8 → braintrust-0.3.9}/src/braintrust/generated_types.py

@@ -1,4 +1,4 @@
-"""Auto-generated file (internal git SHA 93e76a7bcdf0f874a1827af017d26ac37995a47b) -- do not modify"""
+"""Auto-generated file (internal git SHA 8e9c0a96b3cf291360978c17580f72f6817bd6c8) -- do not modify"""
 
 from ._generated_types import (
     Acl,
@@ -14,6 +14,8 @@ from ._generated_types import (
     BraintrustModelParams,
     CallEvent,
     ChatCompletionContentPart,
+    ChatCompletionContentPartFileFile,
+    ChatCompletionContentPartFileWithTitle,
     ChatCompletionContentPartImageWithTitle,
     ChatCompletionContentPartText,
     ChatCompletionContentPartTextWithTitle,
@@ -111,6 +113,8 @@ __all__ = [
     "BraintrustModelParams",
     "CallEvent",
     "ChatCompletionContentPart",
+    "ChatCompletionContentPartFileFile",
+    "ChatCompletionContentPartFileWithTitle",
     "ChatCompletionContentPartImageWithTitle",
     "ChatCompletionContentPartText",
     "ChatCompletionContentPartTextWithTitle",

{braintrust-0.3.8 → braintrust-0.3.9}/src/braintrust/logger.py

@@ -4488,6 +4488,18 @@ def render_message(render: Callable[[str], str], message: PromptMessage):
                             "image_url": {**c["image_url"], "url": render(c["image_url"]["url"])},
                         }
                     )
+                elif c["type"] == "file":
+                    rendered_content.append(
+                        {
+                            **c,
+                            "file": {
+                                **c["file"],
+                                "file_data": render(c["file"]["file_data"]),
+                                **({} if "file_id" not in c["file"] else {"file_id": render(c["file"]["file_id"])}),
+                                **({} if "filename" not in c["file"] else {"filename": render(c["file"]["filename"])}),
+                            },
+                        }
+                    )
                 else:
                     raise ValueError(f"Unknown content type: {c['type']}")
 

{braintrust-0.3.8 → braintrust-0.3.9}/src/braintrust/test_logger.py

@@ -20,7 +20,7 @@ from braintrust import (
     logger,
 )
 from braintrust.id_gen import OTELIDGenerator, get_id_generator
-from braintrust.logger import _deep_copy_event, _extract_attachments, parent_context, render_mustache
+from braintrust.logger import _deep_copy_event, _extract_attachments, parent_context, render_message, render_mustache
 from braintrust.prompt import PromptChatBlock, PromptData, PromptMessage, PromptSchema
 from braintrust.test_helpers import (
     assert_dict_matches,
@@ -485,6 +485,46 @@ class TestLogger(TestCase):
         with self.assertRaises(ValueError):
             prompt.build(items=["only_one"], strict=True)
 
+    def test_render_message_with_file_content_parts(self):
+        """Test render_message with mixed text, image, and file content parts including all file fields."""
+        message = PromptMessage(
+            role="user",
+            content=[
+                {"type": "text", "text": "Here is a {{item}}:"},
+                {"type": "image_url", "image_url": {"url": "{{image_url}}"}},
+                {
+                    "type": "file",
+                    "file": {
+                        "file_data": "{{file_data}}",
+                        "file_id": "{{file_id}}",
+                        "filename": "{{filename}}",
+                    },
+                },
+            ],
+        )
+
+        rendered = render_message(
+            lambda template: template.replace("{{item}}", "document")
+            .replace("{{image_url}}", "https://example.com/image.png")
+            .replace("{{file_data}}", "base64data")
+            .replace("{{file_id}}", "file-456")
+            .replace("{{filename}}", "report.pdf"),
+            message,
+        )
+
+        assert rendered["content"] == [
+            {"type": "text", "text": "Here is a document:"},
+            {"type": "image_url", "image_url": {"url": "https://example.com/image.png"}},
+            {
+                "type": "file",
+                "file": {
+                    "file_data": "base64data",
+                    "file_id": "file-456",
+                    "filename": "report.pdf",
+                },
+            },
+        ]
+
 
 def test_noop_permalink_issue_1837():
     # fixes issue #BRA-1837

braintrust-0.3.9/src/braintrust/version.py

@@ -0,0 +1,4 @@
+VERSION = "0.3.9"
+
+# this will be templated during the build
+GIT_COMMIT = "238ae256a3a1ab93e682d603c4b53652fa0059f1"

{braintrust-0.3.8 → braintrust-0.3.9}/src/braintrust.egg-info/PKG-INFO

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

braintrust-0.3.8/src/braintrust/version.py

@@ -1,4 +0,0 @@
-VERSION = "0.3.8"
-
-# this will be templated during the build
-GIT_COMMIT = "133d19e1e5c4b0b87017942d4ca37f731af5a91e"