send the file paths with the functions in the current diff

Author: mohammedahmed18

codeflash/lsp/beta.py

@@ -45,7 +45,7 @@ class ProvideApiKeyParams:
 @server.feature("getOptimizableFunctionsInCurrentDiff")
 def get_functions_in_current_git_diff(
     server: CodeflashLanguageServer, _params: OptimizableFunctionsParams
-) -> dict[str, str | list[str]]:
+) -> 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,
@@ -55,8 +55,10 @@ def get_functions_in_current_git_diff(
         module_root=server.optimizer.args.module_root,
         previous_checkpoint_functions={},
     )
-    qualified_names: list[str] = [func.qualified_name for funcs in file_to_funcs_to_optimize.values() for func in funcs]
-    return {"functions": qualified_names, "status": "success"}
+    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()
+    }
+    return {"functions": file_to_qualified_names, "status": "success"}
 
 
 @server.feature("getOptimizableFunctions")
@@ -143,7 +145,8 @@ def validate_project(server: CodeflashLanguageServer, _params: FunctionOptimizat
         server.show_message_log("pyproject.toml is not valid", "Error")
         return {
             "status": "error",
-            "message": "pyproject.toml is not valid",  # keep the error message the same, the extension is matching "pyproject.toml" in the error message to show the codeflash init instructions
+            # keep the error message the same, the extension is matching "pyproject.toml" in the error message to show the codeflash init instructions
+            "message": "pyproject.toml is not valid",
         }
 
     args = process_args(server)