Generative AI for DataWeave: Generate Code from Unit Tests

The DataWeave Codegen project by the MuleSoft ML/AI team is a generative AI tool aimed at simplifying the use of MuleSoft’s powerful data transformation programming language, DataWeave, and making it more accessible for low-code/no-code users. 

The DataWeave language can have a steep learning curve, but DataWeave Codegen simplifies the process by letting users generate DataWeave scripts by simply providing a sample input and output data corresponding to the desired DataWeave code. By streamlining the data transformation process, the DataWeave Codegen project enables more users to build Mule applications while accelerating development and time-to-value. 

We will describe our exploration of techniques and models that are used to generate DataWeave code given the sample input and output. These results also serve as an analysis of the generalizability of current state-of-the-art models and techniques to a low-resource programming language like DataWeave and the ability of large language models (LLMs) to learn about