- clean up

tailcallhq · Sep 5, 2024 · 567bba5 · 567bba5
1 parent 763cb5d
commit 567bba5
Showing 1 changed file with 7 additions and 10 deletions.
diff --git a/src/cli/llm/infer_type_name.rs b/src/cli/llm/infer_type_name.rs
@@ -154,16 +154,13 @@ impl AsyncTransform for InferTypeName {
 
     async fn transform(&self, value: Self::Value) -> Valid<Self::Value, Self::Error> {
         match self.generate(&value).await {
-            Ok(suggested_names) => {
-                match RenameTypes::new(suggested_names.iter())
-                    .transform(value)
-                    .to_result()
-                {
-                    Ok(v) => Valid::succeed(v),
-                    Err(e) => Valid::fail(Error::Err(e.to_string())),
-                }
-            }
-            Err(err) => Valid::fail(err),
+            Ok(suggested_names) => RenameTypes::new(suggested_names.iter())
+                .transform(value)
+                .to_result()
+                .map(Valid::succeed)
+                .map_err(|e| Error::Err(e.to_string()))
+                .unwrap_or_else(Valid::fail),
+            Err(e) => Valid::fail(e),
         }
     }
 }