feat(api): support entity spans format (#264)

reinfer · Apr 3, 2024 · 028337a · 028337a
1 parent 0c22d50
commit 028337a
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -2,6 +2,7 @@
 
 - Add `get emails`
 - Added support for `--auto-increase-up-to` when creating quotas.
+- Support spans format for entities 
 
 # v0.22.2
 

diff --git a/api/src/resources/comment.rs b/api/src/resources/comment.rs
@@ -775,12 +775,26 @@ fn should_skip_serializing_entities(maybe_entities: &Option<Entities>) -> bool {
 }
 
 #[derive(Debug, Clone, PartialEq, Deserialize, Serialize, Eq)]
-pub struct NewEntity {
+#[serde(untagged)]
+pub enum NewEntity {
+    WithSpan(NewEntityWithSpan),
+    WithSpans(NewEntityWithSpans),
+}
+
+#[derive(Debug, Clone, PartialEq, Deserialize, Serialize, Eq)]
+pub struct NewEntityWithSpan {
     pub name: EntityName,
     pub formatted_value: String,
     pub span: NewEntitySpan,
 }
 
+#[derive(Debug, Clone, PartialEq, Deserialize, Serialize, Eq)]
+pub struct NewEntityWithSpans {
+    pub name: EntityName,
+    pub formatted_value: String,
+    pub spans: Vec<NewEntitySpan>,
+}
+
 #[derive(Debug, PartialEq, Clone, Deserialize, Serialize, Eq)]
 pub struct NewEntitySpan {
     content_part: String,
@@ -793,7 +807,7 @@ pub struct NewEntitySpan {
 pub struct Entity {
     pub name: EntityName,
     pub formatted_value: String,
-    pub span: EntitySpan,
+    pub spans: Vec<EntitySpan>,
 }
 
 #[derive(Debug, PartialEq, Clone, Deserialize, Serialize, Eq)]