feat(cli): round trip entity def flags (#309)

* feat(cli): round trip entity def flags
reinfer · Aug 29, 2024 · e9daf5c · e9daf5c
1 parent 4776788
commit e9daf5c
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,6 +1,7 @@
 # Unreleased
 - Add dataset flags to `create-dataset`
 - Add `parse aic-classification-csv`
+- Round trip `_entity_def_flags`
 
 # v0.31.0
 - Add `get keyed sync states`

diff --git a/api/src/resources/entity_def.rs b/api/src/resources/entity_def.rs
@@ -14,6 +14,15 @@ pub struct EntityDef {
     pub name: Name,
     pub title: String,
     pub trainable: bool,
+    #[serde(default)]
+    #[serde(rename = "_entity_def_flags")]
+    pub entity_def_flags: Vec<EntityDefFlag>,
+}
+
+#[derive(Debug, Clone, Deserialize, Serialize, PartialEq, Eq)]
+#[serde(rename_all(serialize = "snake_case", deserialize = "snake_case"))]
+pub enum EntityDefFlag {
+    AppearsVerbatim,
 }
 
 #[derive(Debug, Clone, Deserialize, Serialize, PartialEq, Eq)]
@@ -22,6 +31,9 @@ pub struct NewEntityDef {
     pub name: Name,
     pub title: String,
     pub trainable: bool,
+    #[serde(rename = "_entity_def_flags")]
+    #[serde(default)]
+    pub entity_def_flags: Vec<EntityDefFlag>,
 }
 
 #[derive(Debug, Clone, Deserialize, Serialize, PartialEq, Eq)]