Abstract: A method comprises generating a map of relationships between functions, statements and data structures in identified code by traversing a syntax tree for the identified code, the syntax tree for the identified code includes a hierarchy of nodes, each node comprising at least one identifying name; modifying nodes of the syntax tree of the identified node by truncating code unrelated to the code generation task to generate a modified syntax tree for the identified code; adding to the modified syntax tree, for each modified node, a new node indicating that the modified node is truncated, and traversing the at least one modified syntax tree to generate a second code that is shorter than at least one identified code. The method also includes providing the generated second code, the at least one modified syntax tree, and the code generation task as input to a language model to generate output code.

Abstract: A method comprises generating a map of relationships between functions, statements and data structures in identified code by traversing a syntax tree for the identified code, the syntax tree for the identified code includes a hierarchy of nodes, each node comprising at least one identifying name; modifying nodes of the syntax tree of the identified node by truncating code unrelated to the code generation task to generate a modified syntax tree for the identified code; adding to the modified syntax tree, for each modified node, a new node indicating that the modified node is truncated, and traversing the at least one modified syntax tree to generate a second code that is shorter than at least one identified code. The method also includes providing the generated second code, the at least one modified syntax tree, and the code generation task as input to a language model to generate output code.