8eb2cf5058
## Problem
When generating JavaScript/TypeScript tests with vitest, the CLI was
adding duplicate import statements, causing parse errors:
```
[PARSE_ERROR] Error: Identifier `describe` has already been declared
```
The AI service instructs the LLM to generate:
```javascript
import { describe, test, expect, vi, beforeEach, afterEach } from 'vitest';
```
But inject_test_globals() in edit_tests.py was prepending:
```javascript
import { vi, describe, it, expect, beforeEach, afterEach, beforeAll, test } from 'vitest'
```
The substring check `if global_import.strip() not in test.source` failed
because these strings differ in:
- Identifier order (vi, describe vs describe, test)
- Different identifiers (it, beforeAll in hardcoded vs not in LLM)
- Trailing punctuation (newline vs semicolon)
## Solution
Replace strict substring matching with regex pattern detection that
identifies ANY vitest/jest/mocha import, regardless of identifier order
or exact format.
Uses existing regex patterns (_VITEST_IMPORT_RE, _JEST_GLOBALS_IMPORT_RE)
to detect framework imports before injection.
## Testing
- Added comprehensive unit tests covering vitest, jest, and mocha
- All tests pass
- Verified with prek (no linting/type errors)
## Impact
Fixes ALL JavaScript/TypeScript test generation failures.
Trace IDs:
- 023f2643-1709-4f88-b31a-74403e150f06
- 49344657-b8f0-4397-8d9c-ba3b4505899d
- 4d433c2f-5939-4e52-9717-3fbdd7a16816
- (and 5 others)
Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
|
||
|---|---|---|
| .claude | ||
| .codex | ||
| .gemini | ||
| .github | ||
| code_to_optimize | ||
| codeflash | ||
| codeflash-benchmark | ||
| codeflash-java-runtime | ||
| docs | ||
| experiments | ||
| packages/codeflash | ||
| tests | ||
| .gitignore | ||
| .mcp.json | ||
| .pre-commit-config.yaml | ||
| CLAUDE.md | ||
| codeflash.code-workspace | ||
| LICENSE | ||
| mypy_allowlist.txt | ||
| pyproject.toml | ||
| README.md | ||
| SECURITY.md | ||
| tessl.json | ||
| uv.lock | ||
Codeflash is a general purpose optimizer for Python that helps you improve the performance of your Python code while maintaining its correctness. It uses advanced LLMs to generate multiple optimization ideas for your code, tests them to be correct and benchmarks them for performance. It then creates merge-ready pull requests containing the best optimization found, which you can review and merge.
How to use Codeflash -
- Optimize an entire existing codebase by running
codeflash --all - Automate optimizing all future code you will write by installing Codeflash as a GitHub action.
- Optimize a Python workflow
python myscript.pyend-to-end by runningcodeflash optimize myscript.py
Codeflash is used by top engineering teams at Pydantic (PRs Merged), Roboflow (PRs Merged 1, PRs Merged 2), Unstructured (PRs Merged 1, PRs Merged 2), Langflow (PRs Merged) and many others to ship performant, expert level code.
Codeflash is great at optimizing AI Agents, Computer Vision algorithms, PyTorch code, numerical code, backend code or anything else you might write with Python.
Installation
To install Codeflash, run:
pip install codeflash
Add codeflash as a development time dependency if you are using package managers like uv or poetry.
Quick Start
-
To configure Codeflash for a project, at the root directory of your project where the pyproject.toml file is located, run:
codeflash init- It will ask you a few questions about your project like the location of your code and tests
- Ask you to generate an API Key to access Codeflash's LLMs
- Install a GitHub app to open Pull Requests on GitHub.
- Ask if you want to setup a GitHub actions which will optimize all your future code.
- The codeflash config is then saved in the pyproject.toml file.
-
Optimize your entire codebase:
codeflash --allThis can take a while to run for a large codebase, but it will keep opening PRs as it finds optimizations.
-
Optimize a script:
codeflash optimize myscript.py
Documentation
For detailed installation and usage instructions, visit our documentation at docs.codeflash.ai
Demo
- Optimizing the performance of new code for a Pull Request through GitHub Actions. This lets you ship code quickly while ensuring it remains performant.
https://github.com/user-attachments/assets/38f44f4e-be1c-4f84-8db9-63d5ee3e61e5
- Optiming a workflow end to end automatically with
codeflash optimize
https://github.com/user-attachments/assets/355ba295-eb5a-453a-8968-7fb35c70d16c
Support
Join our community for support and discussions. If you have any questions, feel free to reach out to us using one of the following methods:
License
Codeflash is licensed under the BSL-1.1 License. See the LICENSE file for details.