stamp: clippy & fmt

Author: kallebysantos

ext/ai/lib.rs

@@ -200,10 +200,17 @@ async fn init_gte(state: Rc<RefCell<OpState>>) -> Result<(), Error> {
         .map(|i| *i as i64)
         .collect::<Vec<_>>();
 
-      let input_ids_array = TensorRef::from_array_view(([input_ids.len(), 1], &*input_ids))?;
-      let attention_mask_array = TensorRef::from_array_view(([1, encoded_prompt.len()], &*attention_mask))?;
-      let token_type_ids_array = TensorRef::from_array_view(([1, encoded_prompt.len()], &*token_type_ids))?;
-
+      let input_ids_array =
+        TensorRef::from_array_view(([input_ids.len(), 1], &*input_ids))?;
+      let attention_mask_array = TensorRef::from_array_view((
+        [1, encoded_prompt.len()],
+        &*attention_mask,
+      ))?;
+
+      let token_type_ids_array = TensorRef::from_array_view((
+        [1, encoded_prompt.len()],
+        &*token_type_ids,
+      ))?;
 
       let Ok(mut guard) = session.lock() else {
         let err = anyhow!("failed to lock session");
@@ -223,10 +230,12 @@ async fn init_gte(state: Rc<RefCell<OpState>>) -> Result<(), Error> {
       let embeddings = embeddings.into_dimensionality::<Ix3>()?;
 
       let result = if do_mean_pooling {
-      let attention_mask_array_clone= Array1::from_iter(attention_mask.iter().cloned());
-      let attention_mask_array_clone= attention_mask_array_clone.view()
-        .insert_axis(Axis(0))
-        .insert_axis(Axis(2));
+        let attention_mask_array_clone =
+          Array1::from_iter(attention_mask.iter().cloned());
+        let attention_mask_array_clone = attention_mask_array_clone
+          .view()
+          .insert_axis(Axis(0))
+          .insert_axis(Axis(2));
 
         println!("attention_mask: {attention_mask_array_clone:?}");
         mean_pool(embeddings, attention_mask_array_clone)

ext/ai/onnxruntime/mod.rs

@@ -56,8 +56,11 @@ pub async fn op_ai_ort_init_session(
   };
 
   let mut state = state.borrow_mut();
-  let mut sessions =
-    { state.try_take::<Vec<Arc<Mutex<Session>>>>().unwrap_or_default() };
+  let mut sessions = {
+    state
+      .try_take::<Vec<Arc<Mutex<Session>>>>()
+      .unwrap_or_default()
+  };
 
   sessions.push(model.get_session());
   state.put(sessions);

ext/ai/onnxruntime/model.rs

@@ -36,20 +36,23 @@ impl Model {
   fn new(session_with_id: SessionWithId) -> Result<Self> {
     let (input_names, output_names) = {
       let Ok(session_guard) = session_with_id.session.lock() else {
-      return Err(anyhow!("Could not lock model session {}", session_with_id.id));
-    };
-
-    let input_names = session_guard
-      .inputs
-      .iter()
-      .map(|input| input.name.clone())
-      .collect::<Vec<_>>();
-
-    let output_names = session_guard
-      .outputs
-      .iter()
-      .map(|output| output.name.clone())
-      .collect::<Vec<_>>();
+        return Err(anyhow!(
+          "Could not lock model session {}",
+          session_with_id.id
+        ));
+      };
+
+      let input_names = session_guard
+        .inputs
+        .iter()
+        .map(|input| input.name.clone())
+        .collect::<Vec<_>>();
+
+      let output_names = session_guard
+        .outputs
+        .iter()
+        .map(|output| output.name.clone())
+        .collect::<Vec<_>>();
 
       (input_names, output_names)
     };
@@ -75,13 +78,11 @@ impl Model {
   pub async fn from_id(id: &str) -> Option<Self> {
     let session = {
       get_session(id)
-      .await
-      .map(|it| SessionWithId::from((id.to_string(), it)))
+        .await
+        .map(|it| SessionWithId::from((id.to_string(), it)))
     };
 
-    let Some(session) = session else {
-      return None;
-    };
+    let session = session?;
 
     Self::new(session).ok()
   }

ext/ai/onnxruntime/session.rs

@@ -76,13 +76,13 @@ pub(crate) fn get_session_builder() -> Result<SessionBuilder, AnyError> {
 }
 
 fn cpu_execution_provider() -> ExecutionProviderDispatch {
-    // NOTE(Nyannacha): See the comment above. This makes `enable_cpu_mem_arena` set to
-    // False.
-    //
-    // Backgrounds:
-    // [1]: https://docs.rs/ort/2.0.0-rc.4/src/ort/execution_providers/cpu.rs.html#9-18
-    // [2]: https://docs.rs/ort/2.0.0-rc.4/src/ort/execution_providers/cpu.rs.html#46-50
-    CPUExecutionProvider::default().build()
+  // NOTE(Nyannacha): See the comment above. This makes `enable_cpu_mem_arena` set to
+  // False.
+  //
+  // Backgrounds:
+  // [1]: https://docs.rs/ort/2.0.0-rc.4/src/ort/execution_providers/cpu.rs.html#9-18
+  // [2]: https://docs.rs/ort/2.0.0-rc.4/src/ort/execution_providers/cpu.rs.html#46-50
+  CPUExecutionProvider::default().build()
 }
 
 fn cuda_execution_provider() -> Option<ExecutionProviderDispatch> {
@@ -92,13 +92,12 @@ fn cuda_execution_provider() -> Option<ExecutionProviderDispatch> {
 
   if is_cuda_available {
     Some(cuda.build())
-  }else{
+  } else {
     None
   }
 }
 
-fn get_execution_providers(
-) -> Vec<ExecutionProviderDispatch> {
+fn get_execution_providers() -> Vec<ExecutionProviderDispatch> {
   let cpu = cpu_execution_provider();
 
   if let Some(cuda) = cuda_execution_provider() {

ext/ai/onnxruntime/tensor.rs

@@ -115,23 +115,23 @@ pub enum JsTensorType {
   /// Brain 16-bit floating point number, equivalent to [`half::bf16`] (requires the `half` feature).
   Bfloat16,
   Complex64,
-	Complex128,
-	/// 8-bit floating point number with 4 exponent bits and 3 mantissa bits, with only NaN values and no infinite
-	/// values.
-	Float8E4M3FN,
-	/// 8-bit floating point number with 4 exponent bits and 3 mantissa bits, with only NaN values, no infinite
-	/// values, and no negative zero.
-	Float8E4M3FNUZ,
-	/// 8-bit floating point number with 5 exponent bits and 2 mantissa bits.
-	Float8E5M2,
-	/// 8-bit floating point number with 5 exponent bits and 2 mantissa bits, with only NaN values, no infinite
-	/// values, and no negative zero.
-	Float8E5M2FNUZ,
-	/// 4-bit unsigned integer.
-	Uint4,
-	/// 4-bit signed integer.
-	Int4,
-	Undefined
+  Complex128,
+  /// 8-bit floating point number with 4 exponent bits and 3 mantissa bits, with only NaN values and no infinite
+  /// values.
+  Float8E4M3FN,
+  /// 8-bit floating point number with 4 exponent bits and 3 mantissa bits, with only NaN values, no infinite
+  /// values, and no negative zero.
+  Float8E4M3FNUZ,
+  /// 8-bit floating point number with 5 exponent bits and 2 mantissa bits.
+  Float8E5M2,
+  /// 8-bit floating point number with 5 exponent bits and 2 mantissa bits, with only NaN values, no infinite
+  /// values, and no negative zero.
+  Float8E5M2FNUZ,
+  /// 4-bit unsigned integer.
+  Uint4,
+  /// 4-bit signed integer.
+  Int4,
+  Undefined,
 }
 
 #[derive(Serialize, Deserialize)]
@@ -282,7 +282,7 @@ impl ToJsTensor {
       TensorElementType::String => todo!(),
       TensorElementType::Float16 => todo!(),
       TensorElementType::Bfloat16 => todo!(),
-      _ => todo!()
+      _ => todo!(),
     };
 
     Ok(Self {