Skip to content

Commit 59c4976

Browse files
authored
feat(api): round trip attribution_method (#410)
1 parent b547fd7 commit 59c4976

File tree

2 files changed

+14
-0
lines changed

2 files changed

+14
-0
lines changed

CHANGELOG.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
# Unreleased
22
- Check status code when getting octet stream
3+
- Round trip and add default for `attribution_method`
34

45
# v0.38.8
56
- Only ask for ai unit consumpion consent when unpacking CM packages

api/src/resources/dataset.rs

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -132,6 +132,19 @@ pub struct GptIxpModelConfig {
132132
pub flags: Vec<GptIxpFlag>,
133133
#[serde(rename = "iterative_config", skip_serializing_if = "Option::is_none")]
134134
pub iterative_config: Option<IterativeConfig>,
135+
136+
#[serde(default)]
137+
pub attribution_method: AttributionMethod,
138+
}
139+
140+
#[derive(
141+
Clone, Copy, Debug, Eq, PartialEq, Ord, PartialOrd, Hash, Serialize, Deserialize, Default,
142+
)]
143+
#[serde(rename_all = "snake_case")]
144+
pub enum AttributionMethod {
145+
#[default]
146+
TableHeuristic,
147+
WordIds,
135148
}
136149

137150
#[derive(Clone, Default, Debug, PartialEq, Serialize, Deserialize, Eq)]

0 commit comments

Comments
 (0)