refactor

Author: mohammedahmed18

codeflash/lsp/beta.py

@@ -53,23 +53,12 @@ class OptimizableFunctionsInCommitParams:
 server = CodeflashLanguageServer("codeflash-language-server", "v1.0", protocol_cls=CodeflashLanguageServerProtocol)
 
 
-# TODO: use _group_functions_by_file once https://github.com/codeflash-ai/codeflash/pull/688 is merged
 @server.feature("getOptimizableFunctionsInCurrentDiff")
 def get_functions_in_current_git_diff(
     server: CodeflashLanguageServer, _params: OptimizableFunctionsParams
 ) -> dict[str, str | dict[str, list[str]]]:
     functions = get_functions_within_git_diff(uncommitted_changes=True)
-    file_to_funcs_to_optimize, _ = filter_functions(
-        modified_functions=functions,
-        tests_root=server.optimizer.test_cfg.tests_root,
-        ignore_paths=[],
-        project_root=server.optimizer.args.project_root,
-        module_root=server.optimizer.args.module_root,
-        previous_checkpoint_functions={},
-    )
-    file_to_qualified_names: dict[str, list[str]] = {
-        str(path): [f.qualified_name for f in funcs] for path, funcs in file_to_funcs_to_optimize.items()
-    }
+    file_to_qualified_names = _group_functions_by_file(server, functions)
     return {"functions": file_to_qualified_names, "status": "success"}