handle response

tembo-io · Nov 20, 2023 · e1fbf5a · e1fbf5a
1 parent 2018ad4
commit e1fbf5a
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/transformers/types.rs b/src/transformers/types.rs
@@ -2,13 +2,14 @@ use serde::{Deserialize, Serialize};
 
 #[derive(Deserialize, Debug)]
 pub struct EmbeddingResponse {
-    pub data: Vec<DataObject>,
+    pub data: Vec<EmbeddingObject>,
+    pub model: String,
 }
 
 #[derive(serde::Deserialize, Debug)]
-pub struct DataObject {
+pub struct EmbeddingObject {
     // object: String,
-    // index: usize,
+    pub index: usize,
     pub embedding: Vec<f64>,
 }