Your Coding Intent is Secretly in the Context and You Should Deliberately Infer It Before Completion

------------------------------------- ## Контекст ------------------------------------- Функциональность репозиториев с кодом постоянно растет, что делает важной задачу реализации функций с помощью л LLM (Large Language Models). Несмотря на то, что при наличии докстрингов (docstrings) модели показывают высокую точность, в реальных репозиториях такой информации часто недостаточно. Это приводит к снижению качества результатов. Наша модель адресует эту проблему, разбив задачу на три этапа: интент-инференс, интерактивная рефинементация, и генерация кода. Специфичен фреймворк при решении задачи, который позволяет лучше использовать контекстный код перед функцией. ------------------------------------- ## Метод ------------------------------------- Мы предлагаем трехэтапный подход. 1) **Интент-инференс**: Модель анализирует код перед функцией, ищет скрытые сигналы о приоритете. 2) **Интерактивная рефинементация**: Модель предлагает несколько вариантов интента, где пользователь может выбрать или отредактировать. 3) **Генерация кода**: По уточненному интенту LLM генерирует функцию. Мы используем 40,000 примеров с аннотированными рассуждениями и докстрингами для обучения. ------------------------------------- ## Результаты ------------------------------------- Мы проводили эксперименты на двух датасетах: DevEval и ComplexCodeEval. Наша модель показала до 20% относительного увеличения точности в сравнении с базовыми моделями, особенно при использовании интерактивного рефинемента. Также, люди, работавшие с нашим подходом, отзывались о повышении качества результатов в реальных сценариях. ------------------------------------- ## Значимость ------------------------------------- Наш подход может применяться в различных областях, где требуется точность реализации функций на основе контекста. Он улучшает качество генерации кода, особенно в ситуациях, когда документации недоступна. Кроме того, он может способствовать улучшению интеллектуальных систем разработки, например, в системах автоматизированной генерации кода. ------------------------------------- ## Выводы ------------------------------------- Мы представили многоэтапный подход, который позволяет моделям лучше использовать контекстный код. Мы получили заметные улучшения в точности генерации кода с использованием интерактивной рефинементации. Будущие исследования будут фокусироваться на расширении этого подхода для более сложных сценариев и улучшении его точности и работы в реальных условиях.