impl pytest add type

pyrefly/lib/lsp/non_wasm/server.rs

@@ -3107,6 +3107,11 @@ impl Server {
                 }));
             }
         };
+        if let Some(pytest_actions) =
+            transaction.pytest_fixture_type_annotation_code_actions(&handle, range, import_format)
+        {
+            push_refactor_actions(pytest_actions);
+        }
         if let Some(refactors) = transaction.extract_field_code_actions(&handle, range) {
             push_refactor_actions(refactors);
         }

pyrefly/lib/state/lsp.rs

@@ -1950,6 +1950,20 @@ impl<'a> Transaction<'a> {
         )
     }
 
+    pub fn pytest_fixture_type_annotation_code_actions(
+        &self,
+        handle: &Handle,
+        selection: TextRange,
+        import_format: ImportFormat,
+    ) -> Option<Vec<LocalRefactorCodeAction>> {
+        quick_fixes::pytest_fixture::pytest_fixture_type_annotation_code_actions(
+            self,
+            handle,
+            selection,
+            import_format,
+        )
+    }
+
     pub fn extract_function_code_actions(
         &self,
         handle: &Handle,

pyrefly/lib/state/lsp/quick_fixes/mod.rs

@@ -15,4 +15,5 @@ pub(crate) mod inline_variable;
 pub(crate) mod introduce_parameter;
 pub(crate) mod move_members;
 pub(crate) mod move_module;
+pub(crate) mod pytest_fixture;
 pub(crate) mod types;