Beyond More Context: How Granularity and Order Drive Code Completion Quality
By: Uswat Yusuf, Genevieve Caumartin, Diego Elias Costa
Potential Business Impact:
Helps computers write better code by finding good examples.
Context plays an important role in the quality of code completion, as Large Language Models (LLMs) require sufficient and relevant information to assist developers in code generation tasks. However, composing a relevant context for code completion poses challenges in large repositories: First, the limited context length of LLMs makes it impractical to include all repository files. Second, the quality of generated code is highly sensitive to noisy or irrelevant context. In this paper, we present our approach for the ASE 2025 Context Collection Challenge. The challenge entails outperforming JetBrains baselines by designing effective retrieval and context collection strategies. We develop and evaluate a series of experiments that involve retrieval strategies at both the file and chunk levels. We focus our initial experiments on examining the impact of context size and file ordering on LLM performance. Our results show that the amount and order of context can significantly influence the performance of the models. We introduce chunk-based retrieval using static analysis, achieving a 6% improvement over our best file-retrieval strategy and a 16% improvement over the no-context baseline for Python in the initial phase of the competition. Our results highlight the importance of retrieval granularity, ordering and hybrid strategies in developing effective context collection pipelines for real-world development scenarios.
Similar Papers
Relative Positioning Based Code Chunking Method For Rich Context Retrieval In Repository Level Code Completion Task With Code Language Model
Software Engineering
Helps computers finish writing code faster.
Challenge on Optimization of Context Collection for Code Completion
Software Engineering
Helps AI write code faster by reading more of it.
Deep Learning-based Code Completion: On the Impact on Performance of Contextual Information
Software Engineering
Helps computers write code faster and better.