diff --git a/generated_types.json b/generated_types.json index f741eba9..3715ee3b 100644 --- a/generated_types.json +++ b/generated_types.json @@ -7733,6 +7733,9 @@ "maximum": 1, "description": "The sampling rate for topic automation" }, + "facet_model": { + "$ref": "#/components/schemas/TopicAutomationFacetModel" + }, "facet_functions": { "type": "array", "items": { @@ -7878,6 +7881,18 @@ ], "description": "Optional data scope for topic automation." }, + "TopicAutomationFacetModel": { + "type": [ + "string", + "null" + ], + "enum": [ + "brain-facet-latest", + "brain-facet-1", + "brain-facet-2" + ], + "description": "Optional facet model override for topic automation" + }, "TopicMapData": { "type": "object", "properties": { diff --git a/py/src/braintrust/_generated_types.py b/py/src/braintrust/_generated_types.py index c4a15701..bb516d3a 100644 --- a/py/src/braintrust/_generated_types.py +++ b/py/src/braintrust/_generated_types.py @@ -2419,6 +2419,14 @@ class TopicAutomationDataScopeTopicAutomationDataScope2(TypedDict): """ +TopicAutomationFacetModel: TypeAlias = Literal[ + 'brain-facet-latest', 'brain-facet-1', 'brain-facet-2' +] +""" +Optional facet model override for topic automation +""" + + class Function1Function1(TypedDict): type: Literal['function'] id: str @@ -3333,6 +3341,7 @@ class TopicAutomationConfig(TypedDict): """ The sampling rate for topic automation """ + facet_model: NotRequired[TopicAutomationFacetModel | None] facet_functions: Sequence[SavedFunctionId] """ Facet functions used by the topic automation diff --git a/py/src/braintrust/generated_types.py b/py/src/braintrust/generated_types.py index 7a385c38..c3ddc5d6 100644 --- a/py/src/braintrust/generated_types.py +++ b/py/src/braintrust/generated_types.py @@ -1,4 +1,4 @@ -"""Auto-generated file (content hash 2046371fcb54d28e) -- do not modify""" +"""Auto-generated file (content hash 56952ef4659f447d) -- do not modify""" from ._generated_types import ( Acl, @@ -101,6 +101,7 @@ ToolFunctionDefinition, TopicAutomationConfig, TopicAutomationDataScope, + TopicAutomationFacetModel, TopicMapData, TopicMapFunctionAutomation, TopicMapGenerationSettings, @@ -216,6 +217,7 @@ "ToolFunctionDefinition", "TopicAutomationConfig", "TopicAutomationDataScope", + "TopicAutomationFacetModel", "TopicMapData", "TopicMapFunctionAutomation", "TopicMapGenerationSettings",