Skip to content

RepoQA v0.1.0
Compare
Choose a tag to compare
@ganler ganler released this 26 Apr 09:58
· 32 commits to main since this release
v0.1.0
b873b72

RepoQA for Long-Context Code Understanding

Introduction

RepoQA is a benchmark that aims to exercise LLM's long-context code understanding ability.

  • Multi-Lingual: RepoQA now supports repositories from 5 programming languages:
    • Python
    • C++
    • TypeScript
    • Rust
    • Java
  • Application-driven: RepoQA aims to evaluate LLMs on long-context tasks that can reflect real-life uses. Before RepoQA, long-context evaluators mainly focus on using synthetic tasks to examine the vulnerable parts of the LLM's long context, such as "Needle in the Code" by CodeQwen and "Needle in a Haystack".
  • The first RepoQA task we propose is 🔍 Searching Needle Function:
    • 500 sub-tasks = 5 PLs x 10 repos x 10 needles
    • Asks the model to search the corresponding function (we call it needle function) given a precise natural language description

RepoQA is easy to use

  • Supports following backends
    • OpenAI
    • Anthropic
    • vLLM
    • HuggingFace transformers
    • Google Generative AI API (Gemini)
  • 🚀 Evaluation can be done in one command
  • 🏆 A leaderboard: https://evalplus.github.io/repoqa.html

Quick examples

pip install repoqa
repoqa.search_needle_function --model "gpt4-turbo" --backend openai
repoqa.search_needle_function --model "claude-3-haiku-20240307" --backend anthropic
repoqa.search_needle_function --model "Qwen/CodeQwen1.5-7B-Chat" --backend vllm
repoqa.search_needle_function --model "Qwen/CodeQwen1.5-7B-Chat" --backend hf --trust-remote-code
repoqa.search_needle_function --model "gemini-1.5-pro-latest" --backend google

Resources

Assets 2
Loading