pith. sign in

arxiv: 2305.02783 · v4 · pith:JWLHN3ZNnew · submitted 2023-05-02 · 💻 cs.SE · cs.AI· cs.CL· cs.PL

Automated Code generation for Information Technology Tasks in YAML through Large Language Models

classification 💻 cs.SE cs.AIcs.CLcs.PL
keywords ansiblegenerationcodelanguageansible-yamlautomationmodelmodels
0
0 comments X
read the original abstract

The recent improvement in code generation capabilities due to the use of large language models has mainly benefited general purpose programming languages. Domain specific languages, such as the ones used for IT Automation, have received far less attention, despite involving many active developers and being an essential component of modern cloud platforms. This work focuses on the generation of Ansible-YAML, a widely used markup language for IT Automation. We present Ansible Wisdom, a natural-language to Ansible-YAML code generation tool, aimed at improving IT automation productivity. Ansible Wisdom is a transformer-based model, extended by training with a new dataset containing Ansible-YAML. We also develop two novel performance metrics for YAML and Ansible to capture the specific characteristics of this domain. Results show that Ansible Wisdom can accurately generate Ansible script from natural language prompts with performance comparable or better than existing state of the art code generation models. In few-shot settings we asses the impact of training with Ansible, YAML data and compare with different baselines including Codex-Davinci-002. We also show that after finetuning, our Ansible specific model (BLEU: 66.67) can outperform a much larger Codex-Davinci-002 (BLEU: 50.4) model, which was evaluated in few shot settings.

This paper has not been read by Pith yet.

discussion (0)

Sign in with ORCID, Apple, or X to comment. Anyone can read and Pith papers without signing in.

Forward citations

Cited by 1 Pith paper

Reviewed papers in the Pith corpus that reference this work. Sorted by Pith novelty score.

  1. Text2DSL: LLM-Based Code Generation for Domain-Specific Languages

    cs.AI 2026-06 unverdicted novelty 6.0

    Formalizes Text2DSL, introduces PolkitBench dataset with 4,204 pairs, and shows structured prompt context boosts syntactic validity to 98.6-99.4%, structural validity by up to 35.5 pp, and CodeBLEU by 60-95% on two Mo...