fix(api): fix fetch & advance stream requests (#251)

reinfer · Jan 25, 2024 · f2edc08 · f2edc08
1 parent 0ac7cce
commit f2edc08
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,3 +1,7 @@
+# unreleaswed
+- Fix a bug where stream responses were not correctly parsed 
+- Fix a bug where streams were not correctly advanced 
+
 # v0.21.4
 - Add messages filters
 - Fixes `required` field error when interacting with datasets

diff --git a/api/src/lib.rs b/api/src/lib.rs
@@ -1439,6 +1439,7 @@ impl Endpoints {
                 &stream_name.dataset.0,
                 "streams",
                 &stream_name.stream.0,
+                "advance",
             ],
         )
     }

diff --git a/api/src/resources/comment.rs b/api/src/resources/comment.rs
@@ -624,7 +624,7 @@ pub struct Label {
 
 #[derive(Debug, Clone, Deserialize, Serialize, PartialEq, Eq)]
 pub struct PredictedLabel {
-    pub name: LabelName,
+    pub name: Vec<String>,
     #[serde(skip_serializing_if = "Option::is_none")]
     pub sentiment: Option<NotNan<f64>>,
     pub probability: NotNan<f64>,

diff --git a/cli/src/commands/get/comments.rs b/cli/src/commands/get/comments.rs
@@ -613,7 +613,7 @@ fn get_comments_from_uids(
                                     auto_threshold_labels
                                         .iter()
                                         .map(|auto_threshold_label| PredictedLabel {
-                                            name: LabelName(auto_threshold_label.name.join(" > ")),
+                                            name: auto_threshold_label.name.to_owned(),
                                             sentiment: None,
                                             probability: auto_threshold_label.probability,
                                             auto_thresholds: Some(