Skip to main content

High performance code search for large codebases

Project description

Context extraction at scale

Tools like Aider and Cursor are great at editing code for you once you give them the right context. But finding that context automatically is largely an unsolved problem, especially in large codebases.

LLMap is a CLI code search tool designed to solve that problem by asking
DeepSeek-V3 and DeepSeek-R1 to evaluate the relevance of each source file in your codebase to your problem.

Until recently, this would be prohibitively expensive and slow. But DeepSeek-V3 is cheap, smart, fast, and most importantly, it allows multiple concurrent requests. LLMap performs its analysis (by default) 500 files at a time, so it's reasonably fast even for large codebases.

LLMap also structures its request to take advantage of DeepSeek's caching. This means that repeated searches against the same files will be faster and less expensive.

Finally, LLMap optimizes the problem by using a multi-stage analysis to avoid spending more time than necessary analyzing obviously irrelevant files. LLMap performs 3 stages of analysis:

  1. Coarse analysis using code skeletons [DeepSeek-V3]
  2. Full source analysis of potentially relevant files from (1) [DeepSeek-V3]
  3. Refine the output of (2) to only the most relevant snippets [DeepSeek-R1]

Limitations

Currently only Java and Python files are supported by the skeletonization pass.
LLMap will process other source files, but it will perform full source analysis on all of them, which will be slower.

Extending the parsing to other languages is straightforward; contributions are welcome.

Installation

pip install llmap-ai

Get a DeepSeek API key from platform.deepseek.com.

Usage

export DEEPSEEK_API_KEY=YYY

find src/ -name "*.java" | llmap "Where is the database connection configured?"

LLMs APIs are not super reliable, so LLMap caches LLM responses in .llmap_cache by question and by processing stage, so that you don't have to start over from scratch if you get rate limited or run into another hiccup.

The cache is automatically cleaned out when execution completes successfully.

Output

LLMap prints the most relevant context found to stdout. You can save this to a file and send it to Aider or attach it to a conversation with your favorite AI chat tool.

Errors are logged to stderr.

Didn't find what you were looking for?

First, try passing --no-refine. While the refine step is usually helpful in filtering out the noise (thus taking up less of your context window), sometimes it's too aggressive.

Second, try rephrasing your question with more clues for the LLM to latch onto. Like any information retrieval tool, sometimes the way you ask can make a big difference.

  • Worse: "How can I add a WITH clause to the CQL SELECT statement?"
  • Better: "How can I add a WITH clause to the CQL SELECT statement? It will be used for adding query planning hints like which index to use."

Options

  --sample SAMPLE       Number of random files to sample
  --save-cache          Keep cache directory after completion
  --llm-concurrency LLM_CONCURRENCY
                        Maximum number of concurrent LLM requests
  --no-refine           Skip refinement and combination of analyses

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

llmap_ai-0.9.3.tar.gz (1.9 MB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

llmap_ai-0.9.3-py3-none-any.whl (2.0 MB view details)

Uploaded Python 3

File details

Details for the file llmap_ai-0.9.3.tar.gz.

File metadata

  • Download URL: llmap_ai-0.9.3.tar.gz
  • Upload date:
  • Size: 1.9 MB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.0.1 CPython/3.10.15

File hashes

Hashes for llmap_ai-0.9.3.tar.gz
Algorithm Hash digest
SHA256 fa5ee71ae406eece850a1f09c602ba9ec9f6f92cf30cb28fb095297f90788a11
MD5 1fd0fa5da293f34c35818c4044534725
BLAKE2b-256 e6ab20065e99f3ff3e3113f2ead56671c8fbf1b2fa5a768152a6c540ddbfe4bd

See more details on using hashes here.

File details

Details for the file llmap_ai-0.9.3-py3-none-any.whl.

File metadata

  • Download URL: llmap_ai-0.9.3-py3-none-any.whl
  • Upload date:
  • Size: 2.0 MB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.0.1 CPython/3.10.15

File hashes

Hashes for llmap_ai-0.9.3-py3-none-any.whl
Algorithm Hash digest
SHA256 56d3ab0e4d2a140ab0e07278e4f5811dfb7bb3a6b6a6942ce0b3f041ff265447
MD5 6f4e2cd9dbdc7d1c981aacf9f62256e5
BLAKE2b-256 c8f78bf47ae431a3b163d8cc9a6aba6a745dbd6102d2855add3036cb47041c8d

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page