No description
Find a file
Kevin Turcios 0abc6bf1e3
async: parallelize endpoint epilogue DB writes (#2490)
## Summary

Parallelize independent DB writes at the end of 4 endpoints using
`asyncio.TaskGroup`. With psycopg3 connection pooling (#2489), each task
gets its own connection from the pool.

### Endpoints optimized

| Endpoint | Before | After |
|----------|--------|-------|
| **Refinement** | `log_features` then `update_optimization_cost` |
`TaskGroup` (concurrent) |
| **Explanations** | `update_optimization_cost` inside inner fn | Moved
to handler, `TaskGroup` with `log_features` |
| **Optimization review** | `update_optimization_cost` inside inner fn |
Moved to handler, `TaskGroup` with `update_optimization_features_review`
|
| **Ranker** | `update_optimization_cost` inside inner fn | Moved to
handler, `TaskGroup` with `log_features` |

Each endpoint saves ~87ms (one DB round-trip) by overlapping two
independent writes.

### Comprehensive audit

All 13 endpoints were audited — no remaining async antipatterns found:
- No blocking calls in async paths
- No `await`-in-loop patterns
- LLM clients already use connection reuse
- All other endpoints have at most 1 DB write in the epilogue

## Test plan

- [x] All 538 tests passing
- [ ] Verify under load in staging

---------

Co-authored-by: claude[bot] <41898282+claude[bot]@users.noreply.github.com>
Co-authored-by: Claude Sonnet 4.6 <noreply@anthropic.com>
Co-authored-by: Kevin Turcios <KRRT7@users.noreply.github.com>
Co-authored-by: codeflash-ai[bot] <148906541+codeflash-ai[bot]@users.noreply.github.com>
2026-04-01 06:15:16 -05:00
.claude fix: rename skill files to be Windows-compatible 2026-02-17 05:01:30 +00:00
.codex fix: rename skill files to be Windows-compatible 2026-02-17 05:01:30 +00:00
.gemini fix: rename skill files to be Windows-compatible 2026-02-17 05:01:30 +00:00
.github async: parallelize endpoint epilogue DB writes (#2490) 2026-04-01 06:15:16 -05:00
.idea more cleanup 2026-01-28 22:23:54 +02:00
.tessl chore: switch tessl to managed mode (#2491) 2026-03-27 07:21:14 -05:00
.vscode Revert "CF-1041 observability v2 " need more changes and testing (#2375) 2026-02-06 01:18:17 +05:30
cli/code-to-optimize codeflash-omni-java (#2335) 2026-02-13 23:26:55 +05:30
deployment/onprem-simple local setup (#1898) 2025-11-17 12:35:09 -08:00
django async: parallelize endpoint epilogue DB writes (#2490) 2026-04-01 06:15:16 -05:00
experiments move sqlalchemy, gen_inspired_tests, and mistral 2025-12-30 15:22:39 -05:00
js make healthcheck public in cfapi 2026-03-29 09:09:48 +02:00
tiles test: add evals for all three Tessl tiles 2026-02-14 22:25:30 -05:00
.dockerignore local setup (#1898) 2025-11-17 12:35:09 -08:00
.editorconfig consistency in formatting across ide & js projs (#1499) 2025-03-04 23:52:45 +00:00
.gitattributes chore: add gh-aw duplicate code detector workflow (#2418) 2026-02-14 18:14:55 -05:00
.gitignore chore: switch tessl to managed mode (#2491) 2026-03-27 07:21:14 -05:00
.gitmodules add the new submodule at cli 2025-02-13 01:05:51 -05:00
.mcp.json feat: add Tessl tiles for codeflash-internal (rules, docs, skills) 2026-02-14 22:16:33 -05:00
.pre-commit-config.yaml add pre-k GHA 2025-12-30 01:17:05 -05:00
.prettierrc Rename some auth0 mgmt things 2023-12-10 17:40:07 -08:00
AGENTS.md feat: add Tessl tiles for codeflash-internal (rules, docs, skills) 2026-02-14 22:16:33 -05:00
CLAUDE.md feat: add Tessl tiles for codeflash-internal (rules, docs, skills) 2026-02-14 22:16:33 -05:00
lefthook.yml saga4 misc fixes (#2018) 2025-11-14 20:23:58 -08:00
mypy.ini Byesian analysis implementation 2025-01-17 17:44:24 -08:00
package-lock.json Initial js support in aiservice 2026-01-14 22:15:27 -08:00
package.json Initial js support in aiservice 2026-01-14 22:15:27 -08:00
README.md Update README.md 2024-12-27 11:55:26 -08:00
secretlint.config.js add secret scanner and monorepo hook (#1201) 2024-11-09 14:23:39 +00:00
tessl.json chore: switch tessl to managed mode (#2491) 2026-03-27 07:21:14 -05:00

CodeFlash MonoRepo

Here's the projects that are part of the CodeFlash MonoRepo:

  • CodeFlash Client - /cli/
  • CodeFlash Python Django ai service - /django/aiservice
  • CodeFlash NodeJS CF API - /js/cf-api
  • CodeFlash Webapp - /js/cf-webapp

Project Setup

Prerequisites

  • Node.js and npm: Ensure Node.js is installed and npm is set up for installation of pre-commit hook(Lefthook).
  • Python and Mamba: Ensure Python is installed and Mamba is set up.

post clone run npm install to install all the dependencies at root level.

Glossary

Optimization

  • Codeflash Optimizer - The overarching technology that solves Code optimization.
  • Function to Optimize - The target function that we want to optimize.
  • Optimization Candidate - generated code that we think might be an optimization of the code to optimize.
  • Helper function - This a function being called by, and is under the code path of the function to optimize.
  • Read-Write Context - The part of the code context provided to the LLM that it can modify. Aka - Code To Optimize
  • Read-Only Context - The part of the context that is only provided as info to the LLM. It is not expected to be modified.

Test generation

  • Verification - System to verify if the optimization candidate has the same functional behavior as the function to optimize.
  • Existing Tests - All the existing tests that are present in a repo.
  • Generated Test - The tests that we create for the user using the LLM.
  • Tracer - Our technology that collects and dumps the input arguments and other info for a Python executable.
  • Replay test - This test reruns all the inputs for a function to optimize that were collected by the tracer.
  • Inspired Regression tests - Newly generated Tests that were "inspired" by existing tests. That means these are new test cases that are generated by the llm understanding how the code works by looking at the existing test cases and function to optimize.
  • Comparator - Our function that compares any two Python objects and returns True if they are equal and False if they are not equal.

Infra and Systems

  • CF API - The javascript webservice that currently serves the GitHub App.
  • AI Service - The Python Django service that serves the AI endpoints.
  • Webapp - The react web application written in Next.js. Users can generate API Key etc here.
  • PostHog - Our events tracking and product analytics 3rd party tool.
  • Sentry - Our code crash telemetry service that helps us understand how codeflash fails.