Sketch

Anyone who writes pandas code in a notebook and is tired of AI suggestions that guess at column names will find Sketch aimed squarely at that problem.

Sketch is an AI code assistant for pandas users that reads the context of your data before it writes code, so the suggestions fit the table you are actually working with.

The method is simple.

Sketch uses data sketches, a family of efficient approximation algorithms, to summarize your data quickly.

It then summarizes each column and passes those summary statistics to a language model as extra context in the code-writing prompt.

The model sees what the data looks like instead of only your question, which is why the suggestions tend to be more relevant.

Setup is light.

You can use Sketch within seconds, and there is no plugin to install in an IDE.

That makes it a handy pick for quick exploration and analysis work where you want a code generator that knows your dataframe without a lot of configuration.

The project lives on GitHub.