Minimal MCP Server for tool calling codeflash with --file --function

[aseembits93]

PR Type

Enhancement

---

Description

• Add asyncio client for optimize_code tool

• Implement MCP server with optimize_code endpoint

• Configure TestConfig and CoverageExpectation

• Enable end-to-end run_codeflash_command

---

Diagram Walkthrough

flowchart LR
  client["MCP Client"] -- "call optimize_code" --> server["MCP Server"]
  server -- "run_codeflash_command" --> codeflash["CodeFlash Command"]
  codeflash -- "return result" --> client

Loading

File Walkthrough

Relevant files

Enhancement

myclient.py Add asyncio-based MCP client                                                          myclient.py Import and instantiate Client from fastmcp Define call_tool async function Call optimize_code tool and print result Use asyncio.run for execution +11/-0    myserver.py Implement MCP server tool optimize_code                                    myserver.py Import and instantiate FastMCP server Decorate optimize_code tool implementation Setup TestConfig and coverage expectations Invoke run_codeflash_command in tool +29/-0

---

[github-actions]

PR Reviewer Guide 🔍

Here are some key observations to aid the review process:

⏱️ Estimated effort to review: 2 🔵🔵⚪⚪⚪

🧪 No relevant tests

🔒 No security concerns identified

⚡ Recommended focus areas for review Hardcoded Path The TestConfig uses an absolute file path which will not work on other environments; consider using the file parameter to set file_path dynamically or a relative path. file_path=pathlib.Path("/Users/codeflash/Downloads/codeflash-dev/codeflash/code_to_optimize/bubble_sort.py"), Undefined Variable The inline print statement uses arr which isn't defined in this context, leading to a runtime error; ensure the correct variable is passed or captured. cwd, config, 100, ['print("codeflash stdout: Sorting list")', 'print(f"result: {arr}")'] Ignored Parameters The optimize_code tool accepts file and function but currently ignores them in favor of hardcoded values; wire these parameters into the TestConfig instead. def optimize_code(file: str, function: str) -> bool: #todo add file and function name as arguments config = TestConfig( file_path=pathlib.Path("/Users/codeflash/Downloads/codeflash-dev/codeflash/code_to_optimize/bubble_sort.py"), function_name="sorter", test_framework="pytest", min_improvement_x=1.0,

[github-actions]

PR Code Suggestions ✨

Explore these optional code suggestions:

Category	Suggestion	Impact
Possible issue	Make TestConfig use parameters Use the file and function parameters to dynamically set file_path and function_name in TestConfig, and update coverage_expectations to reference the passed-in function. This ensures the tool operates on the user-specified inputs rather than hardcoded values. myserver.py [12-22] config = TestConfig( - file_path=pathlib.Path("/Users/codeflash/Downloads/codeflash-dev/codeflash/code_to_optimize/bubble_sort.py"), - function_name="sorter", + file_path=pathlib.Path(file), + function_name=function, test_framework="pytest", min_improvement_x=1.0, coverage_expectations=[ CoverageExpectation( - function_name="sorter", expected_coverage=100.0, expected_lines=[2, 3, 4, 5, 6, 7, 8, 9, 10] + function_name=function, expected_coverage=100.0, expected_lines=[2, 3, 4, 5, 6, 7, 8, 9, 10] ) ], ) Suggestion importance[1-10]: 9 __ Why: The suggestion correctly replaces hardcoded file_path and function_name with the passed file and function parameters, ensuring dynamic behavior and preventing misuse of fixed values.	High
	Fix undefined variable in prints Remove or correct the print(f"result: {arr}") statement since arr is undefined, which will cause a runtime error. Replace it with a valid variable or drop the statement. myserver.py [24-26] return run_codeflash_command( - cwd, config, 100, ['print("codeflash stdout: Sorting list")', 'print(f"result: {arr}")'] + cwd, config, 100, ['print("codeflash stdout: Sorting list")'] ) Suggestion importance[1-10]: 7 __ Why: Removing the undefined arr in the print list prevents a runtime NameError, addressing a clear bug in the call to run_codeflash_command.	Medium