Create README.md
Browse files
README.md
CHANGED
|
@@ -0,0 +1,746 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# π€ AI Codebase Assistant β Complete Documentation
|
| 2 |
+
|
| 3 |
+
> An AI-powered developer tool that ingests any local code repository, answers questions about it using RAG (Retrieval-Augmented Generation), detects bugs, measures complexity, generates docs, proposes new files, and integrates with the GitHub MCP API β all from an interactive CLI.
|
| 4 |
+
|
| 5 |
+
---
|
| 6 |
+
|
| 7 |
+
## Table of Contents
|
| 8 |
+
|
| 9 |
+
1. [Project Overview](#1-project-overview)
|
| 10 |
+
2. [Repository Structure](#2-repository-structure)
|
| 11 |
+
3. [Architecture & Flow](#3-architecture--flow)
|
| 12 |
+
4. [Component Deep-Dive](#4-component-deep-dive)
|
| 13 |
+
5. [Setup & Installation](#5-setup--installation)
|
| 14 |
+
6. [Configuration (.env)](#6-configuration-env)
|
| 15 |
+
7. [Running the CLI](#7-running-the-cli)
|
| 16 |
+
8. [CLI Features β All 9 Options](#8-cli-features--all-9-options)
|
| 17 |
+
9. [REST API](#9-rest-api)
|
| 18 |
+
10. [Key Design Decisions & Bug Fixes](#10-key-design-decisions--bug-fixes)
|
| 19 |
+
11. [Extending the Project](#11-extending-the-project)
|
| 20 |
+
|
| 21 |
+
---
|
| 22 |
+
|
| 23 |
+
## 1. Project Overview
|
| 24 |
+
|
| 25 |
+
The **AI Codebase Assistant** is a local developer tool that:
|
| 26 |
+
|
| 27 |
+
- π **Ingests** any code repository (Python, JS, TS, Java, Go, Markdown)
|
| 28 |
+
- π **Answers questions** about the code using RAG + LLM
|
| 29 |
+
- π **Detects bugs** via LLM-powered code review
|
| 30 |
+
- π **Measures cyclomatic complexity** using Radon
|
| 31 |
+
- π **Explains functions** in plain English
|
| 32 |
+
- π **Generates module docs and READMEs** automatically
|
| 33 |
+
- π οΈ **Proposes & creates new files** using AI, saved to your chosen directory
|
| 34 |
+
- π **Lists 44 GitHub MCP tools** via the GitHub Copilot MCP API
|
| 35 |
+
|
| 36 |
+
**LLM Providers supported:**
|
| 37 |
+
| Provider | Model | Notes |
|
| 38 |
+
|---|---|---|
|
| 39 |
+
| Groq | `llama-3.3-70b-versatile` | Free tier, recommended |
|
| 40 |
+
| Gemini | Configurable | Paid, stricter quota |
|
| 41 |
+
|
| 42 |
+
**Embedding:** Google Gemini Embedding API (`models/gemini-embedding-001`)
|
| 43 |
+
**Vector Store:** ChromaDB (local persistent)
|
| 44 |
+
|
| 45 |
+
---
|
| 46 |
+
|
| 47 |
+
## 2. Repository Structure
|
| 48 |
+
|
| 49 |
+
```
|
| 50 |
+
Codebase Assistant/
|
| 51 |
+
β
|
| 52 |
+
βββ cli.py # β Main entry point (interactive CLI)
|
| 53 |
+
βββ app.py # β FastAPI REST server (optional)
|
| 54 |
+
βββ config.py # β All settings via pydantic-settings + .env
|
| 55 |
+
βββ llm.py # β LLM abstraction (Gemini / Groq) + prompt templates
|
| 56 |
+
βββ mcp_client.py # β GitHub MCP async context-manager client
|
| 57 |
+
βββ requirements.txt # β All Python dependencies
|
| 58 |
+
βββ .env # β API keys and configuration (not committed)
|
| 59 |
+
β
|
| 60 |
+
βββ rag/ # ββ RAG Pipeline ββββββββββββββββββββββββββββββ
|
| 61 |
+
β βββ repository_loader.py # Walk directory, read files β CodeDocument
|
| 62 |
+
β βββ splitter.py # Split CodeDocuments into chunks with metadata
|
| 63 |
+
β βββ embedding.py # Embed chunks/queries via Gemini Embeddings
|
| 64 |
+
β βββ retriever.py # ChromaDB vector store: build / load / query
|
| 65 |
+
β βββ rag_chain.py # Orchestrate retrieve β build context β LLM
|
| 66 |
+
β
|
| 67 |
+
βββ services/ # ββ Feature Services ββββββββββββββββββββββββββ
|
| 68 |
+
β βββ code_analysis.py # Bug detection, complexity, function explainer
|
| 69 |
+
β βββ documentation.py # Module docs, README generator
|
| 70 |
+
β βββ file_creator.py # AI file proposal + local disk write
|
| 71 |
+
β
|
| 72 |
+
βββ api/ # ββ REST API (FastAPI) ββββββββββββββββββββββββ
|
| 73 |
+
β βββ routes.py # All HTTP endpoints (mirrors CLI features)
|
| 74 |
+
β
|
| 75 |
+
βββ vector_db/ # ββ ChromaDB Storage (auto-created) βββββββββββ
|
| 76 |
+
βββ chroma.sqlite3 # Persisted vector embeddings
|
| 77 |
+
```
|
| 78 |
+
|
| 79 |
+
---
|
| 80 |
+
|
| 81 |
+
## 3. Architecture & Flow
|
| 82 |
+
|
| 83 |
+
### 3.1 Full System Architecture
|
| 84 |
+
|
| 85 |
+
```mermaid
|
| 86 |
+
graph TB
|
| 87 |
+
subgraph USER["User Interface"]
|
| 88 |
+
CLI["cli.py\n(Interactive CLI)"]
|
| 89 |
+
API["app.py\n(FastAPI REST API)"]
|
| 90 |
+
end
|
| 91 |
+
|
| 92 |
+
subgraph RAG["RAG Pipeline"]
|
| 93 |
+
RL["repository_loader.py\nWalk & read files"]
|
| 94 |
+
SP["splitter.py\nChunk by language"]
|
| 95 |
+
EM["embedding.py\nGemini Embeddings"]
|
| 96 |
+
RT["retriever.py\nChromaDB"]
|
| 97 |
+
RC["rag_chain.py\nOrchestrator"]
|
| 98 |
+
end
|
| 99 |
+
|
| 100 |
+
subgraph SERVICES["Feature Services"]
|
| 101 |
+
CA["code_analysis.py\nBugs Β· Complexity Β· Explain"]
|
| 102 |
+
DOC["documentation.py\nModule Docs Β· README"]
|
| 103 |
+
FC["file_creator.py\nAI File Creator"]
|
| 104 |
+
end
|
| 105 |
+
|
| 106 |
+
subgraph EXTERNAL["External APIs"]
|
| 107 |
+
GROQ["Groq API\nllama-3.3-70b"]
|
| 108 |
+
GEM["Gemini API\nEmbeddings"]
|
| 109 |
+
MCP["GitHub MCP\n44 tools"]
|
| 110 |
+
end
|
| 111 |
+
|
| 112 |
+
LLM["llm.py\nLLM Abstraction Layer"]
|
| 113 |
+
CFG["config.py\n.env Settings"]
|
| 114 |
+
DB[("vector_db/\nChromaDB")]
|
| 115 |
+
|
| 116 |
+
CLI --> RAG
|
| 117 |
+
CLI --> SERVICES
|
| 118 |
+
API --> RAG
|
| 119 |
+
API --> SERVICES
|
| 120 |
+
|
| 121 |
+
RL --> SP --> EM --> RT
|
| 122 |
+
RT --> DB
|
| 123 |
+
RC --> RT
|
| 124 |
+
RC --> LLM
|
| 125 |
+
|
| 126 |
+
CA --> LLM
|
| 127 |
+
DOC --> LLM
|
| 128 |
+
FC --> LLM
|
| 129 |
+
|
| 130 |
+
LLM --> GROQ
|
| 131 |
+
LLM --> GEM
|
| 132 |
+
EM --> GEM
|
| 133 |
+
CLI --> MCP
|
| 134 |
+
|
| 135 |
+
CFG -.->|settings| RAG
|
| 136 |
+
CFG -.->|settings| LLM
|
| 137 |
+
CFG -.->|settings| SERVICES
|
| 138 |
+
```
|
| 139 |
+
|
| 140 |
+
### 3.2 Ingest Flow (Step-by-step)
|
| 141 |
+
|
| 142 |
+
```mermaid
|
| 143 |
+
flowchart LR
|
| 144 |
+
A["User provides\nrepo path"] --> B["repository_loader.py\nwalk directories\nskip: .git venv __pycache__"]
|
| 145 |
+
B --> C["Filter by\nallowed extensions\n.py .js .ts .java .go .md"]
|
| 146 |
+
C --> D["Read each file\nβ CodeDocument\n(content, path, language, size)"]
|
| 147 |
+
D --> E["splitter.py\nLanguage-aware chunking\nRecursiveCharacterTextSplitter"]
|
| 148 |
+
E --> F["Each chunk gets\nmetadata: file_path\nlanguage Β· start_line Β· end_line"]
|
| 149 |
+
F --> G["embedding.py\nGemini embed_documents\nβ float vectors"]
|
| 150 |
+
G --> H["retriever.py\nDelete old collection\nCreate fresh ChromaDB\ncollection.add(...)"]
|
| 151 |
+
H --> I["β Vector store ready"]
|
| 152 |
+
```
|
| 153 |
+
|
| 154 |
+
### 3.3 RAG Query Flow
|
| 155 |
+
|
| 156 |
+
```mermaid
|
| 157 |
+
flowchart LR
|
| 158 |
+
Q["User question"] --> E["embed_query\n(Gemini)"]
|
| 159 |
+
E --> S["ChromaDB\ncollection.query\ntop-k chunks"]
|
| 160 |
+
S --> C["build_context\nformat chunks\nwith file/line headers"]
|
| 161 |
+
C --> P["build_prompt\nqa template\ncontext + question"]
|
| 162 |
+
P --> L["LLM\n(Groq / Gemini)"]
|
| 163 |
+
L --> A["Answer + Sources\n(file_path, line range)"]
|
| 164 |
+
```
|
| 165 |
+
|
| 166 |
+
### 3.4 CLI Menu Flow
|
| 167 |
+
|
| 168 |
+
```mermaid
|
| 169 |
+
flowchart TD
|
| 170 |
+
START([Start cli.py]) --> REPO["βΆ Enter repo path"]
|
| 171 |
+
REPO --> INGEST["Ingest Repository\nload β split β embed β store"]
|
| 172 |
+
INGEST --> MENU["Show Menu\nOptions 1β9"]
|
| 173 |
+
|
| 174 |
+
MENU --> O1["1 Ask a question\nβ RAG Query"]
|
| 175 |
+
MENU --> O2["2 Detect bugs\nβ LLM code review"]
|
| 176 |
+
MENU --> O3["3 Cyclomatic complexity\nβ Radon"]
|
| 177 |
+
MENU --> O4["4 Explain function\nβ AST + LLM"]
|
| 178 |
+
MENU --> O5["5 Module docs\nβ LLM"]
|
| 179 |
+
MENU --> O6["6 Generate README\nβ LLM"]
|
| 180 |
+
MENU --> O7["7 Propose & create file\nβ LLM + local write"]
|
| 181 |
+
MENU --> O8["8 List GitHub MCP tools\nβ GitHub Copilot MCP"]
|
| 182 |
+
MENU --> O9["9 Re-ingest repo\nβ new repo path"]
|
| 183 |
+
MENU --> O0["0 Exit"]
|
| 184 |
+
|
| 185 |
+
O1 & O2 & O3 & O4 & O5 & O6 & O7 & O8 & O9 --> MENU
|
| 186 |
+
O0 --> END([Goodbye!])
|
| 187 |
+
```
|
| 188 |
+
|
| 189 |
+
---
|
| 190 |
+
|
| 191 |
+
## 4. Component Deep-Dive
|
| 192 |
+
|
| 193 |
+
### 4.1 `config.py` β Settings
|
| 194 |
+
|
| 195 |
+
All configuration lives in one `pydantic-settings` class loaded from `.env`:
|
| 196 |
+
|
| 197 |
+
| Setting | Default | Description |
|
| 198 |
+
|---|---|---|
|
| 199 |
+
| `llm_provider` | `"groq"` | `"groq"` or `"gemini"` |
|
| 200 |
+
| `groq_api_key` | `""` | From `.env` |
|
| 201 |
+
| `groq_model` | `"llama-3.3-70b-versatile"` | Free Groq model |
|
| 202 |
+
| `gemini_api_key` | `""` | From `.env` |
|
| 203 |
+
| `embedding_model` | `"models/gemini-embedding-001"` | Always Gemini for embeddings |
|
| 204 |
+
| `vector_db_path` | `"./vector_db"` | ChromaDB storage directory |
|
| 205 |
+
| `chunk_size` | `1200` | Characters per chunk |
|
| 206 |
+
| `chunk_overlap` | `100` | Overlap between chunks |
|
| 207 |
+
| `allowed_extensions` | `[.py .js .ts .go .java .md]` | File types to load |
|
| 208 |
+
| `max_file_size_kb` | `1042` | Skip files larger than this |
|
| 209 |
+
| `github_mcp_url` | GitHub Copilot MCP endpoint | For option 8 |
|
| 210 |
+
| `github_mcp_token` | `""` | GitHub PAT from `.env` |
|
| 211 |
+
|
| 212 |
+
### 4.2 `rag/repository_loader.py` β File Ingestion
|
| 213 |
+
|
| 214 |
+
```
|
| 215 |
+
load_repository(root_path)
|
| 216 |
+
βββ os.walk(root_path)
|
| 217 |
+
βββ Skip: .git, node_modules, __pycache__, venv, .venv, dist, build
|
| 218 |
+
βββ should_include(fpath) β checks extension + file size
|
| 219 |
+
βββ read_file_with_metadata(fpath) β CodeDocument(content, file_path, language, size_bytes)
|
| 220 |
+
```
|
| 221 |
+
|
| 222 |
+
**Supported languages detected by extension:**
|
| 223 |
+
|
| 224 |
+
| Extension | Language |
|
| 225 |
+
|---|---|
|
| 226 |
+
| `.py` | python |
|
| 227 |
+
| `.js` | javascript |
|
| 228 |
+
| `.ts` | typescript |
|
| 229 |
+
| `.java` | java |
|
| 230 |
+
| `.go` | go |
|
| 231 |
+
| `.md` | markdown |
|
| 232 |
+
|
| 233 |
+
### 4.3 `rag/splitter.py` β Chunking
|
| 234 |
+
|
| 235 |
+
Uses **LangChain's `RecursiveCharacterTextSplitter`** with language-aware splitting:
|
| 236 |
+
- For Python/JS/TS/Java/Go β uses syntax-aware boundaries (functions, classes)
|
| 237 |
+
- For Markdown/unknown β falls back to generic character splitting
|
| 238 |
+
- Each chunk carries: `content`, `file_path`, `language`, `chunk_index`, `start_line`, `end_line`
|
| 239 |
+
|
| 240 |
+
### 4.4 `rag/embedding.py` β Embeddings
|
| 241 |
+
|
| 242 |
+
- Uses `GoogleGenerativeAIEmbeddings` (`models/gemini-embedding-001`)
|
| 243 |
+
- `embed_document(chunks)` β batch embeds all chunks, mutates dicts in-place
|
| 244 |
+
- `embed_query(query)` β single query vector for similarity search
|
| 245 |
+
|
| 246 |
+
### 4.5 `rag/retriever.py` β Vector Store
|
| 247 |
+
|
| 248 |
+
> **Critical fix applied:** collection is deleted and recreated on every ingest to prevent stale data from previous repos bleeding through.
|
| 249 |
+
|
| 250 |
+
```python
|
| 251 |
+
build_vector_store(chunks) # delete β create β add (fresh each ingest)
|
| 252 |
+
load_vector_store() # get_or_create for reading
|
| 253 |
+
retrieve_relevant_chunks(q, k) # embed query β cosine similarity β top-k
|
| 254 |
+
```
|
| 255 |
+
|
| 256 |
+
### 4.6 `rag/rag_chain.py` β Orchestration
|
| 257 |
+
|
| 258 |
+
```python
|
| 259 |
+
run_rag_query(query, k=5)
|
| 260 |
+
1. retrieve_relevant_chunks(query, k)
|
| 261 |
+
2. build_context(chunks) # format with file/line headers
|
| 262 |
+
3. build_prompt(query, context) # fill qa template
|
| 263 |
+
4. llm.generate(prompt)
|
| 264 |
+
5. return { answer, sources }
|
| 265 |
+
```
|
| 266 |
+
|
| 267 |
+
### 4.7 `llm.py` β LLM Abstraction
|
| 268 |
+
|
| 269 |
+
Abstract `BaseLLM` with two concrete providers:
|
| 270 |
+
|
| 271 |
+
| Class | Provider | API |
|
| 272 |
+
|---|---|---|
|
| 273 |
+
| `GeminiLLM` | Google Gemini | `google-genai` SDK |
|
| 274 |
+
| `GroqLLM` | Groq | `groq` SDK, chat completions |
|
| 275 |
+
|
| 276 |
+
**Prompt templates (`build_prompt`):**
|
| 277 |
+
|
| 278 |
+
| `task_type` | Used by |
|
| 279 |
+
|---|---|
|
| 280 |
+
| `"qa"` | RAG query, function explain, module docs, README |
|
| 281 |
+
| `"bug_finding"` | Bug detection (returns JSON) |
|
| 282 |
+
| `"docstring"` | Docstring generation |
|
| 283 |
+
| `"file_creation"` | AI file proposal |
|
| 284 |
+
|
| 285 |
+
### 4.8 `mcp_client.py` β GitHub MCP Client
|
| 286 |
+
|
| 287 |
+
Async context-manager pattern using `AsyncExitStack` to keep all `anyio` cancel scopes in the **same task**:
|
| 288 |
+
|
| 289 |
+
```python
|
| 290 |
+
async with get_github_mcp_client() as client:
|
| 291 |
+
tools = await client.list_tools()
|
| 292 |
+
result = await client.call_tool("create_branch", {...})
|
| 293 |
+
```
|
| 294 |
+
|
| 295 |
+
Available GitHub MCP tools (44 total) include: `search_code`, `list_issues`, `create_pull_request`, `get_file_contents`, `push_files`, `create_branch`, `fork_repository`, `search_repositories`, and many more.
|
| 296 |
+
|
| 297 |
+
### 4.9 `services/code_analysis.py`
|
| 298 |
+
|
| 299 |
+
| Function | How it works |
|
| 300 |
+
|---|---|
|
| 301 |
+
| `explain_function(file, name)` | Python `ast` extracts the function source β RAG context β LLM |
|
| 302 |
+
| `detect_bugs(file)` | Read file β `bug_finding` prompt β LLM returns JSON list |
|
| 303 |
+
| `analyze_complexity(file)` | `radon.cc_visit` β cyclomatic complexity + rank AβF |
|
| 304 |
+
|
| 305 |
+
### 4.10 `services/file_creator.py`
|
| 306 |
+
|
| 307 |
+
```
|
| 308 |
+
propose_new_file(description, context_query)
|
| 309 |
+
βββ RAG retrieve relevant context
|
| 310 |
+
βββ build_prompt(description, context, "file_creation")
|
| 311 |
+
βββ LLM generates complete file content
|
| 312 |
+
βββ infer_file_path(description) β LLM suggests relative path
|
| 313 |
+
|
| 314 |
+
apply_approved_file(proposal, confirmed, base_dir)
|
| 315 |
+
βββ Strip leading / or \ from LLM path
|
| 316 |
+
βββ os.path.join(base_dir, relative_path)
|
| 317 |
+
βββ os.makedirs(parent_dirs, exist_ok=True)
|
| 318 |
+
βββ open(abs_path, "w").write(content)
|
| 319 |
+
```
|
| 320 |
+
|
| 321 |
+
---
|
| 322 |
+
|
| 323 |
+
## 5. Setup & Installation
|
| 324 |
+
|
| 325 |
+
### Prerequisites
|
| 326 |
+
|
| 327 |
+
| Requirement | Version |
|
| 328 |
+
|---|---|
|
| 329 |
+
| Python | 3.11+ |
|
| 330 |
+
| pip | Latest |
|
| 331 |
+
| Internet | For API calls |
|
| 332 |
+
|
| 333 |
+
### Step 1 β Clone / Download the project
|
| 334 |
+
|
| 335 |
+
```bash
|
| 336 |
+
git clone <your-repo-url>
|
| 337 |
+
cd "Codebase Assistant"
|
| 338 |
+
```
|
| 339 |
+
|
| 340 |
+
### Step 2 β Create a virtual environment
|
| 341 |
+
|
| 342 |
+
```bash
|
| 343 |
+
# Windows (PowerShell)
|
| 344 |
+
python -m venv .venv
|
| 345 |
+
.venv\Scripts\Activate.ps1
|
| 346 |
+
|
| 347 |
+
# macOS / Linux
|
| 348 |
+
python3 -m venv .venv
|
| 349 |
+
source .venv/bin/activate
|
| 350 |
+
```
|
| 351 |
+
|
| 352 |
+
### Step 3 β Install dependencies
|
| 353 |
+
|
| 354 |
+
```bash
|
| 355 |
+
pip install -r requirements.txt
|
| 356 |
+
```
|
| 357 |
+
|
| 358 |
+
> [!TIP]
|
| 359 |
+
> If you hit permission issues on Windows, use:
|
| 360 |
+
> `pip install -r requirements.txt --user`
|
| 361 |
+
|
| 362 |
+
### Step 4 β Create your `.env` file
|
| 363 |
+
|
| 364 |
+
Create a file named `.env` in the project root:
|
| 365 |
+
|
| 366 |
+
```env
|
| 367 |
+
# Choose your LLM provider: "groq" (free) or "gemini"
|
| 368 |
+
LLM_PROVIDER=groq
|
| 369 |
+
|
| 370 |
+
# Groq β free at https://console.groq.com
|
| 371 |
+
GROQ_API_KEY=gsk_your_key_here
|
| 372 |
+
|
| 373 |
+
# Gemini β get at https://aistudio.google.com
|
| 374 |
+
GEMINI_API_KEY=your_gemini_key_here
|
| 375 |
+
|
| 376 |
+
# GitHub PAT for MCP tools β create at https://github.com/settings/tokens
|
| 377 |
+
# Required scopes: repo, read:org
|
| 378 |
+
GITHUB_MCP_TOKEN=github_pat_your_token_here
|
| 379 |
+
```
|
| 380 |
+
|
| 381 |
+
> [!IMPORTANT]
|
| 382 |
+
> `GEMINI_API_KEY` is **always required** regardless of LLM provider, because embeddings always use Gemini.
|
| 383 |
+
|
| 384 |
+
### Step 5 β Run the CLI
|
| 385 |
+
|
| 386 |
+
```bash
|
| 387 |
+
.venv\Scripts\python.exe cli.py # Windows
|
| 388 |
+
python cli.py # macOS / Linux
|
| 389 |
+
```
|
| 390 |
+
|
| 391 |
+
---
|
| 392 |
+
|
| 393 |
+
## 6. Configuration (.env)
|
| 394 |
+
|
| 395 |
+
```env
|
| 396 |
+
# βββ LLM Provider ββββββββββββββββββββββββββββββββββββββββ
|
| 397 |
+
LLM_PROVIDER=groq # "groq" | "gemini"
|
| 398 |
+
|
| 399 |
+
# βββ Groq (recommended β free tier) βββββββββββββββββββββ
|
| 400 |
+
GROQ_API_KEY=gsk_...
|
| 401 |
+
GROQ_MODEL=llama-3.3-70b-versatile
|
| 402 |
+
|
| 403 |
+
# βββ Gemini ββββββββββββββββββββββββββββββββββββββββββββββ
|
| 404 |
+
GEMINI_API_KEY=...
|
| 405 |
+
LLM_MODEL=gemini-1.5-flash # only used if LLM_PROVIDER=gemini
|
| 406 |
+
EMBEDDING_MODEL=models/gemini-embedding-001
|
| 407 |
+
|
| 408 |
+
# βββ RAG / Vector DB βββββββββββββββββββββββββββββββββββββ
|
| 409 |
+
VECTOR_DB_PATH=./vector_db
|
| 410 |
+
CHUNK_SIZE=1200
|
| 411 |
+
CHUNK_OVERLAP=100
|
| 412 |
+
|
| 413 |
+
# βββ File loader βββββββββββββββββββββββββββββββββββββββββ
|
| 414 |
+
# comma-separated extensions
|
| 415 |
+
ALLOWED_EXTENSIONS=[".py",".js",".ts",".go",".java",".md"]
|
| 416 |
+
MAX_FILE_SIZE_KB=1042
|
| 417 |
+
|
| 418 |
+
# βββ GitHub MCP ββββββββββββββββββββββββββββββββββββββββββ
|
| 419 |
+
GITHUB_MCP_URL=https://api.githubcopilot.com/mcp/
|
| 420 |
+
GITHUB_MCP_TOKEN=github_pat_...
|
| 421 |
+
```
|
| 422 |
+
|
| 423 |
+
---
|
| 424 |
+
|
| 425 |
+
## 7. Running the CLI
|
| 426 |
+
|
| 427 |
+
```
|
| 428 |
+
ββββββββββββββββββββββββββββββββββββββββββββ
|
| 429 |
+
β AI Codebase Assistant CLI β
|
| 430 |
+
ββββββββββββββββββββββββββββββββββββββββββββ
|
| 431 |
+
|
| 432 |
+
βΆ Enter the path to the repository you want to analyse: D:\MyProject
|
| 433 |
+
|
| 434 |
+
ββ Ingesting Repository ββββββββββββββββββββββ
|
| 435 |
+
Loading files from: D:\MyProject
|
| 436 |
+
β Loaded 12 files
|
| 437 |
+
β Split into 47 chunks
|
| 438 |
+
Embedding chunks (this may take a moment)...
|
| 439 |
+
β Embedded 47 chunks
|
| 440 |
+
β Vector store ready (ChromaDB)
|
| 441 |
+
|
| 442 |
+
ββ What would you like to do? ββββββββββββββββ
|
| 443 |
+
1 Ask a question about the codebase
|
| 444 |
+
2 Detect bugs in a file
|
| 445 |
+
3 Cyclomatic complexity analysis
|
| 446 |
+
4 Explain a function
|
| 447 |
+
5 Generate module documentation
|
| 448 |
+
6 Generate README for a repository
|
| 449 |
+
7 Propose & create a new file (AI)
|
| 450 |
+
8 List GitHub MCP tools
|
| 451 |
+
9 Re-ingest a repository
|
| 452 |
+
0 Exit
|
| 453 |
+
|
| 454 |
+
βΆ Choose an option [0β9]:
|
| 455 |
+
```
|
| 456 |
+
|
| 457 |
+
---
|
| 458 |
+
|
| 459 |
+
## 8. CLI Features β All 9 Options
|
| 460 |
+
|
| 461 |
+
### Option 1 β Ask a Question (RAG Query)
|
| 462 |
+
|
| 463 |
+
```
|
| 464 |
+
βΆ Your question: How does the authentication work?
|
| 465 |
+
βΆ Number of source chunks to retrieve? [default: 5] 3
|
| 466 |
+
|
| 467 |
+
ββ Answer ββββββββββββββββββββββββββββββββββββ
|
| 468 |
+
The authentication uses JWT tokens ...
|
| 469 |
+
|
| 470 |
+
ββ Sources βββββββββββββββββββββββββββββββββββ
|
| 471 |
+
β’ src/auth/middleware.py lines 12β45
|
| 472 |
+
β’ src/auth/tokens.py lines 1β30
|
| 473 |
+
```
|
| 474 |
+
|
| 475 |
+
**Pipeline:** Embed question β ChromaDB similarity search β Build context β LLM β Answer + cited sources
|
| 476 |
+
|
| 477 |
+
---
|
| 478 |
+
|
| 479 |
+
### Option 2 β Detect Bugs
|
| 480 |
+
|
| 481 |
+
```
|
| 482 |
+
βΆ File path to analyse: src/payment.py
|
| 483 |
+
|
| 484 |
+
ββ Found 2 issue(s) ββββββββββββββββββββββββββ
|
| 485 |
+
[HIGH] Line 34: SQL query uses string concatenation
|
| 486 |
+
β Use parameterized queries to prevent SQL injection
|
| 487 |
+
|
| 488 |
+
[MEDIUM] Line 67: Exception swallowed silently
|
| 489 |
+
β Log or re-raise the exception
|
| 490 |
+
```
|
| 491 |
+
|
| 492 |
+
**Pipeline:** Read file β `bug_finding` prompt β LLM returns JSON β parsed and displayed
|
| 493 |
+
|
| 494 |
+
---
|
| 495 |
+
|
| 496 |
+
### Option 3 β Cyclomatic Complexity
|
| 497 |
+
|
| 498 |
+
```
|
| 499 |
+
βΆ File path to analyse: src/processor.py
|
| 500 |
+
|
| 501 |
+
ββ 4 function(s) βββββββββββββββββββββββββββββ
|
| 502 |
+
[A] process_order complexity=2 ββ
|
| 503 |
+
[B] validate_cart complexity=5 βββββ
|
| 504 |
+
[C] apply_discounts complexity=8 ββββββββ
|
| 505 |
+
[F] handle_edge_cases complexity=18 ββββββββββββββββββ
|
| 506 |
+
```
|
| 507 |
+
|
| 508 |
+
**Rank scale:** A (1β5, simple) β F (26+, untestable)
|
| 509 |
+
|
| 510 |
+
---
|
| 511 |
+
|
| 512 |
+
### Option 4 β Explain a Function
|
| 513 |
+
|
| 514 |
+
```
|
| 515 |
+
βΆ File path: src/utils.py
|
| 516 |
+
βΆ Function name: parse_date_range
|
| 517 |
+
|
| 518 |
+
ββ Explanation βββββββββββββββββββββββββββββββ
|
| 519 |
+
parse_date_range takes a string like "2024-01-01:2024-12-31"
|
| 520 |
+
and returns a tuple of (start_date, end_date) as datetime objects ...
|
| 521 |
+
```
|
| 522 |
+
|
| 523 |
+
**Pipeline:** Python `ast` module extracts exact function source β RAG retrieves usages β LLM explains
|
| 524 |
+
|
| 525 |
+
---
|
| 526 |
+
|
| 527 |
+
### Option 5 β Generate Module Documentation
|
| 528 |
+
|
| 529 |
+
```
|
| 530 |
+
βΆ File path: src/database.py
|
| 531 |
+
|
| 532 |
+
ββ Documentation βββββββββββββββββββββββββββββ
|
| 533 |
+
## database.py
|
| 534 |
+
|
| 535 |
+
### Overview
|
| 536 |
+
This module provides the database connection layer ...
|
| 537 |
+
|
| 538 |
+
### Functions
|
| 539 |
+
- `connect(url)` β Establishes a connection ...
|
| 540 |
+
- `execute(query, params)` β Runs a parameterized query ...
|
| 541 |
+
```
|
| 542 |
+
|
| 543 |
+
---
|
| 544 |
+
|
| 545 |
+
### Option 6 β Generate README
|
| 546 |
+
|
| 547 |
+
```
|
| 548 |
+
βΆ Repository root path: D:\MyProject
|
| 549 |
+
|
| 550 |
+
ββ README Preview ββββββββββββββββββββββββββββ
|
| 551 |
+
# MyProject
|
| 552 |
+
|
| 553 |
+
## Overview
|
| 554 |
+
A FastAPI application that ...
|
| 555 |
+
|
| 556 |
+
βΆ Save to README.md in that directory? [y/N] y
|
| 557 |
+
β Saved to D:\MyProject\README.md
|
| 558 |
+
```
|
| 559 |
+
|
| 560 |
+
---
|
| 561 |
+
|
| 562 |
+
### Option 7 β Propose & Create a New File
|
| 563 |
+
|
| 564 |
+
```
|
| 565 |
+
βΆ Describe the file you want to create: Rectangle area calculator in JavaScript
|
| 566 |
+
βΆ Optional context query (or press Enter to skip):
|
| 567 |
+
|
| 568 |
+
ββ Proposed File βββββββββββββββββββββββββββββ
|
| 569 |
+
Path: /src/geometry/rectangle.js
|
| 570 |
+
|
| 571 |
+
class Rectangle { ...full generated code... }
|
| 572 |
+
|
| 573 |
+
βΆ Write this file to disk? [y/N] y
|
| 574 |
+
βΆ Save under which directory? [default: D:\MyProject]
|
| 575 |
+
|
| 576 |
+
β File written: D:\MyProject\src\geometry\rectangle.js
|
| 577 |
+
```
|
| 578 |
+
|
| 579 |
+
**Pipeline:** RAG context (optional) β `file_creation` prompt β LLM generates code β user confirms β write to `base_dir/relative_path`
|
| 580 |
+
|
| 581 |
+
---
|
| 582 |
+
|
| 583 |
+
### Option 8 β List GitHub MCP Tools
|
| 584 |
+
|
| 585 |
+
```
|
| 586 |
+
ββ List GitHub MCP Tools βββββββββββββββββββββ
|
| 587 |
+
44 tools available:
|
| 588 |
+
β’ create_branch Create a new branch in a GitHub repository
|
| 589 |
+
β’ create_pull_request Create a new pull request ...
|
| 590 |
+
β’ search_code Fast and precise code search ...
|
| 591 |
+
β’ list_issues List issues in a GitHub repository ...
|
| 592 |
+
...
|
| 593 |
+
```
|
| 594 |
+
|
| 595 |
+
**Connection:** Uses `streamable_http_client` β `ClientSession` β GitHub Copilot MCP endpoint
|
| 596 |
+
|
| 597 |
+
---
|
| 598 |
+
|
| 599 |
+
### Option 9 β Re-ingest a Repository
|
| 600 |
+
|
| 601 |
+
```
|
| 602 |
+
βΆ New repository path to ingest: D:\AnotherProject
|
| 603 |
+
ββ Ingesting Repository ββββββββββββββββββββββ
|
| 604 |
+
β Loaded 8 files ...
|
| 605 |
+
```
|
| 606 |
+
|
| 607 |
+
Wipes the ChromaDB collection and ingests the new repo fresh. All subsequent queries answer from the new repo only.
|
| 608 |
+
|
| 609 |
+
---
|
| 610 |
+
|
| 611 |
+
## 9. REST API
|
| 612 |
+
|
| 613 |
+
Start the FastAPI server:
|
| 614 |
+
|
| 615 |
+
```bash
|
| 616 |
+
.venv\Scripts\python.exe -m uvicorn app:app --reload --port 8000
|
| 617 |
+
```
|
| 618 |
+
|
| 619 |
+
Open docs at: `http://localhost:8000/docs`
|
| 620 |
+
|
| 621 |
+
### Endpoints
|
| 622 |
+
|
| 623 |
+
| Method | Path | Description |
|
| 624 |
+
|---|---|---|
|
| 625 |
+
| `GET` | `/health` | Health check |
|
| 626 |
+
| `POST` | `/api/query` | RAG question answering |
|
| 627 |
+
| `POST` | `/api/analyze/bugs?file_path=...` | Bug detection |
|
| 628 |
+
| `POST` | `/api/analyze/complexity?file_path=...` | Cyclomatic complexity |
|
| 629 |
+
| `POST` | `/api/analyze/explain` | Explain a function |
|
| 630 |
+
| `POST` | `/api/docs/module?file_path=...` | Module documentation |
|
| 631 |
+
| `POST` | `/api/docs/readme?root_path=...` | README generation |
|
| 632 |
+
| `POST` | `/api/files/propose` | Propose a new file |
|
| 633 |
+
| `POST` | `/api/files/approve` | Write approved file to disk |
|
| 634 |
+
|
| 635 |
+
### Example β Query
|
| 636 |
+
|
| 637 |
+
```bash
|
| 638 |
+
curl -X POST http://localhost:8000/api/query \
|
| 639 |
+
-H "Content-Type: application/json" \
|
| 640 |
+
-d '{"query": "How does authentication work?", "k": 5}'
|
| 641 |
+
```
|
| 642 |
+
|
| 643 |
+
```json
|
| 644 |
+
{
|
| 645 |
+
"answer": "Authentication is handled by ...",
|
| 646 |
+
"sources": [
|
| 647 |
+
{"file_path": "src/auth.py", "start_line": 10, "end_line": 45}
|
| 648 |
+
]
|
| 649 |
+
}
|
| 650 |
+
```
|
| 651 |
+
|
| 652 |
+
---
|
| 653 |
+
|
| 654 |
+
## 10. Key Design Decisions & Bug Fixes
|
| 655 |
+
|
| 656 |
+
### Bug Fix 1 β Collection Isolation (retriever.py)
|
| 657 |
+
|
| 658 |
+
**Problem:** ChromaDB used `upsert` on a shared `"codebase"` collection β stale chunks from previously ingested repos leaked into new queries.
|
| 659 |
+
|
| 660 |
+
**Fix:** On every ingest, `delete_collection` + `create_collection` ensures a clean slate:
|
| 661 |
+
|
| 662 |
+
```python
|
| 663 |
+
# Before (bug)
|
| 664 |
+
collection = client.get_or_create_collection("codebase")
|
| 665 |
+
collection.upsert(...)
|
| 666 |
+
|
| 667 |
+
# After (fix)
|
| 668 |
+
client.delete_collection("codebase") # wipe old repo
|
| 669 |
+
collection = client.create_collection("codebase")
|
| 670 |
+
collection.add(...)
|
| 671 |
+
```
|
| 672 |
+
|
| 673 |
+
---
|
| 674 |
+
|
| 675 |
+
### Bug Fix 2 β GitHub MCP Cancel Scope (mcp_client.py)
|
| 676 |
+
|
| 677 |
+
**Problem:** Manually calling `__aenter__`/`__aexit__` on `anyio`-backed context managers across tasks causes `RuntimeError: Attempted to exit cancel scope in a different task`.
|
| 678 |
+
|
| 679 |
+
**Fix:** Use `AsyncExitStack` to nest all context managers inside a single `async with`:
|
| 680 |
+
|
| 681 |
+
```python
|
| 682 |
+
async with get_github_mcp_client() as client:
|
| 683 |
+
tools = await client.list_tools()
|
| 684 |
+
```
|
| 685 |
+
|
| 686 |
+
---
|
| 687 |
+
|
| 688 |
+
### Bug Fix 3 β Wrong Unpack Count (mcp_client.py)
|
| 689 |
+
|
| 690 |
+
**Problem:** `streamable_http_client` yields 2 values, not 3. Unpacking 3 caused `ValueError: not enough values to unpack`.
|
| 691 |
+
|
| 692 |
+
```python
|
| 693 |
+
# Before (bug)
|
| 694 |
+
read, write, _ = await ctx.__aenter__()
|
| 695 |
+
|
| 696 |
+
# After (fix)
|
| 697 |
+
read, write = await stack.enter_async_context(streamable_http_client(...))
|
| 698 |
+
```
|
| 699 |
+
|
| 700 |
+
---
|
| 701 |
+
|
| 702 |
+
### Bug Fix 4 β File Save Path (file_creator.py)
|
| 703 |
+
|
| 704 |
+
**Problem:** Generated files saved to the current working directory regardless of user input.
|
| 705 |
+
|
| 706 |
+
**Fix:**
|
| 707 |
+
- Strip leading `/\` from LLM path to make it always relative
|
| 708 |
+
- Accept `base_dir` defaulting to `_current_repo` (the ingested repo path)
|
| 709 |
+
- Create all parent directories automatically
|
| 710 |
+
|
| 711 |
+
---
|
| 712 |
+
|
| 713 |
+
## 11. Extending the Project
|
| 714 |
+
|
| 715 |
+
### Add a new LLM provider
|
| 716 |
+
|
| 717 |
+
1. Add a new class in `llm.py` extending `BaseLLM`
|
| 718 |
+
2. Add the provider name to `get_llm_client()` factory
|
| 719 |
+
3. Add matching settings in `config.py`
|
| 720 |
+
|
| 721 |
+
### Add support for new file types
|
| 722 |
+
|
| 723 |
+
1. Add extension β language in `LANGUAGE_BY_EXT` in `repository_loader.py`
|
| 724 |
+
2. Add extension to `allowed_extensions` in `config.py`
|
| 725 |
+
3. If LangChain has a `Language` enum for it, add to `LANGUAGE_MAP` in `splitter.py`
|
| 726 |
+
|
| 727 |
+
### Use a GitHub MCP tool in a feature
|
| 728 |
+
|
| 729 |
+
```python
|
| 730 |
+
async with get_github_mcp_client() as client:
|
| 731 |
+
result = await client.call_tool("create_branch", {
|
| 732 |
+
"owner": "myuser",
|
| 733 |
+
"repo": "myrepo",
|
| 734 |
+
"branch": "feature/new-branch",
|
| 735 |
+
"from_branch": "main"
|
| 736 |
+
})
|
| 737 |
+
```
|
| 738 |
+
|
| 739 |
+
### Add a new CLI option
|
| 740 |
+
|
| 741 |
+
1. Write a `feature_xxx()` function in `cli.py`
|
| 742 |
+
2. Add `("Label", feature_xxx)` to the `MENU` list
|
| 743 |
+
3. Add a matching FastAPI endpoint in `api/routes.py`
|
| 744 |
+
|
| 745 |
+
---
|
| 746 |
+
|