Fleming
A set of Databricks scripts that build a corpus from GitHub repository content, train and serve a semantic search model on CPU, and summarize repository code with the Azure OpenAI API.
Fleming is the open-source "brain" behind an internal AI Discovery Tool. It provides scripts to build a text corpus from a dataframe, and to train, register, serve, and query a semantic search model on Databricks, running on CPU rather than GPU to reduce cost.
A separate class ingests files from a list of GitHub repositories in an organization and tokenizes their contents into a dataframe, and another class sends that content to the Azure OpenAI API with a documentation prompt to generate a code summary.
Databricks handles model training and serving, combined with GitHub data ingestion, to provide an end-to-end pipeline for discovering and analyzing repository content.