From a1445b2f03464c1abc0a65f2ab62678a0577351f Mon Sep 17 00:00:00 2001
From: Himadri Bhattacharjee <107522312+lavafroth@users.noreply.github.com>
Date: Mon, 14 Jul 2025 17:36:29 +0530
Subject: [PATCH] fmt: cargo fmt

---
 src/lsp.rs   | 16 ++++++++++------
 src/state.rs | 22 ++++++++++++++--------
 2 files changed, 24 insertions(+), 14 deletions(-)

diff --git a/src/lsp.rs b/src/lsp.rs
index 021323d..499a770 100644
--- a/src/lsp.rs
+++ b/src/lsp.rs
@@ -77,7 +77,10 @@ impl LanguageServer for Backend {
         let uri = params.text_document.uri;
         let Some(lang) = url_extension(&uri) else {
             self.client
-                .log_message(MessageType::ERROR, "unable to determine filetype, file has no extension")
+                .log_message(
+                    MessageType::ERROR,
+                    "unable to determine filetype, file has no extension",
+                )
                 .await;
             return Ok(None);
         };
@@ -93,14 +96,15 @@ impl LanguageServer for Backend {
         let action_response = match comment.action {
             Action::Generate => {
                 range.start = range.end;
-                self.appstate.generate(&lang, comment.description, 1)
+                self.appstate
+                    .generate(&lang, comment.description, 1)
                     .map(|v| v.into_iter().map(|s| format!("{s}\n")).collect())
                     .map_err(|e| e.to_string())
             }
-            Action::Refactor => {
-                self.appstate.refactor(&lang, comment.description, selected_text, 1)
-                    .map_err(|e| e.to_string())
-            }
+            Action::Refactor => self
+                .appstate
+                .refactor(&lang, comment.description, selected_text, 1)
+                .map_err(|e| e.to_string()),
         };
 
         let closest_matches = match action_response {
diff --git a/src/state.rs b/src/state.rs
index e03b2ac..6aac8cc 100644
--- a/src/state.rs
+++ b/src/state.rs
@@ -1,10 +1,10 @@
+use crate::mutation;
 use derive_more::Display;
 use derive_more::Error;
-use tree_sitter::Parser;
-use std::collections::HashMap;
-use hora::index::hnsw_idx::HNSWIndex;
 use hora::core::ann_index::ANNIndex;
-use crate::mutation;
+use hora::index::hnsw_idx::HNSWIndex;
+use std::collections::HashMap;
+use tree_sitter::Parser;
 
 #[derive(Debug, Display, Error)]
 pub enum Error {
@@ -68,7 +68,8 @@ impl Refactor {
                     Err(e) => {
                         tracing::error!(
                             collection_index = index,
-                            "failed to apply mutations from collection {}", e
+                            "failed to apply mutations from collection {}",
+                            e
                         );
                         None
                     }
@@ -79,7 +80,7 @@ impl Refactor {
     }
 }
 pub struct Generate {
-    pub dict: HashMap<String, HNSWIndex<f32, String>>
+    pub dict: HashMap<String, HNSWIndex<f32, String>>,
 }
 
 impl Generate {
@@ -99,7 +100,6 @@ pub struct State {
 }
 
 impl State {
-    
     pub fn generate(&self, lang: &str, prompt: &str, top_k: usize) -> Result<Vec<String>, Error> {
         let Ok(target) = self.embed.embed(prompt) else {
             return Err(Error::EmbedFailed);
@@ -108,7 +108,13 @@ impl State {
         self.generate.search(lang, &target, top_k)
     }
 
-    pub fn refactor(&self, lang: &str, prompt: &str, body: &str, top_k: usize) -> Result<Vec<String>, Error> {
+    pub fn refactor(
+        &self,
+        lang: &str,
+        prompt: &str,
+        body: &str,
+        top_k: usize,
+    ) -> Result<Vec<String>, Error> {
         let Ok(target) = self.embed.embed(prompt) else {
             return Err(Error::EmbedFailed);
         };