janito 0.3.0__py3-none-any.whl → 0.5.0__py3-none-any.whl

janito/qa.py

@@ -1,6 +1,12 @@
 from rich.console import Console
 from rich.markdown import Markdown
-from janito.claude import ClaudeAPIAgent
+from rich.panel import Panel
+from rich.syntax import Syntax
+from rich.table import Table
+from rich.rule import Rule
+from janito.agents import AIAgent
+from janito.common import progress_send_message
+from janito.scan import show_content_stats
 
 QA_PROMPT = """Please provide a clear and concise answer to the following question about the codebase:
 
@@ -12,21 +18,48 @@ Current files:
 </files>
 
 Focus on providing factual information and explanations. Do not suggest code changes.
+Format your response using markdown with appropriate headers and code blocks.
 """
 
-def ask_question(question: str, files_content: str, claude: ClaudeAPIAgent) -> str:
+def ask_question(question: str, files_content: str) -> str:
     """Process a question about the codebase and return the answer"""
+    # Show content stats before processing
+    show_content_stats(files_content)
+    
     prompt = QA_PROMPT.format(
         question=question,
         files_content=files_content
     )
-    return claude.send_message(prompt)
+    return progress_send_message(prompt)
+
 
 def display_answer(answer: str, raw: bool = False) -> None:
-    """Display the answer in markdown or raw format"""
+    """Display the answer as markdown with consistent colors"""
     console = Console()
+    
+    # Define consistent colors
+    COLORS = {
+        'primary': '#729FCF',    # Soft blue for primary elements
+        'secondary': '#8AE234',  # Bright green for actions/success
+        'accent': '#AD7FA8',     # Purple for accents
+        'muted': '#7F9F7F',      # Muted green for less important text
+    }
+    
     if raw:
         console.print(answer)
-    else:
-        md = Markdown(answer)
-        console.print(md)
+        return
+    
+    # Display markdown answer in a panel with consistent styling
+    answer_panel = Panel(
+        Markdown(answer),
+        title="[bold]Answer[/bold]",
+        title_align="center",
+        border_style=COLORS['primary'],
+        padding=(1, 2)
+    )
+    
+    console.print("\n")
+    console.print(Rule(style=COLORS['accent']))
+    console.print(answer_panel)
+    console.print(Rule(style=COLORS['accent']))
+    console.print("\n")

janito/review.py

@@ -0,0 +1,13 @@
+from rich.console import Console
+from rich.markdown import Markdown
+from janito.common import progress_send_message
+from janito.agents import AIAgent
+
+def review_text(text: str, raw: bool = False) -> None:
+    """Review the provided text using Claude"""
+    console = Console()
+    response = progress_send_message(f"Please review this text and provide feedback:\n\n{text}")
+    if raw:
+        console.print(response)
+    else:
+        console.print(Markdown(response))

janito/scan.py

@@ -1,10 +1,13 @@
 from pathlib import Path
-from typing import List, Tuple
+from typing import List, Tuple, Set
 from rich.console import Console
 from rich.columns import Columns
+from rich.panel import Panel
 from janito.config import config
 from pathspec import PathSpec
 from pathspec.patterns import GitWildMatchPattern
+from collections import defaultdict
+
 
 
 SPECIAL_FILES = ["README.md", "__init__.py", "__main__.py"]
@@ -14,12 +17,13 @@ def _scan_paths(paths: List[Path], workdir: Path = None) -> Tuple[List[str], Lis
     content_parts = []
     file_items = []
     skipped_files = []
+    processed_files: Set[Path] = set()  # Track processed files
     console = Console()
     
     # Load gitignore if it exists
     gitignore_path = workdir / '.gitignore' if workdir else None
     gitignore_spec = None
-    if gitignore_path and gitignore_path.exists():
+    if (gitignore_path and gitignore_path.exists()):
         with open(gitignore_path) as f:
             gitignore = f.read()
         gitignore_spec = PathSpec.from_lines(GitWildMatchPattern, gitignore.splitlines())
@@ -34,7 +38,8 @@ def _scan_paths(paths: List[Path], workdir: Path = None) -> Tuple[List[str], Lis
         """
         if level > 1:
             return
-
+        
+        path = path.resolve()
         relative_base = workdir
         if path.is_dir():
             relative_path = path.relative_to(relative_base)
@@ -43,8 +48,10 @@ def _scan_paths(paths: List[Path], workdir: Path = None) -> Tuple[List[str], Lis
             # Check for special files
             special_found = []
             for special_file in SPECIAL_FILES:
-                if (path / special_file).exists():
+                special_path = path / special_file
+                if special_path.exists() and special_path.resolve() not in processed_files:
                     special_found.append(special_file)
+                    processed_files.add(special_path.resolve())
             if special_found:
                 file_items[-1] = f"[blue]•[/blue] {relative_path}/ [cyan]({', '.join(special_found)})[/cyan]"
                 for special_file in special_found:
@@ -63,9 +70,15 @@ def _scan_paths(paths: List[Path], workdir: Path = None) -> Tuple[List[str], Lis
                     rel_path = str(item.relative_to(workdir))
                     if gitignore_spec.match_file(rel_path):
                         continue
-                scan_path(item, level+1)
+                if item.resolve() not in processed_files:  # Skip if already processed
+                    scan_path(item, level+1)
 
         else:
+            resolved_path = path.resolve()
+            if resolved_path in processed_files:  # Skip if already processed
+                return
+                
+            processed_files.add(resolved_path)
             relative_path = path.relative_to(relative_base)
             # check if file is binary
             try:
@@ -97,26 +110,67 @@ def collect_files_content(paths: List[Path], workdir: Path = None) -> str:
     if file_items and config.verbose:
         console.print("\n[bold blue]Contents being analyzed:[/bold blue]")
         console.print(Columns(file_items, padding=(0, 4), expand=True))
+        
+    if config.verbose:
+        for part in content_parts:
+            if part.startswith('<file>'):
+                # Extract filename from XML content
+                path_start = part.find('<path>') + 6
+                path_end = part.find('</path>')
+                if path_start > 5 and path_end > path_start:
+                    filepath = part[path_start:path_end]
+                    console.print(f"[dim]Adding content from:[/dim] {filepath}")
     
     return "\n".join(content_parts)
 
+
 def preview_scan(paths: List[Path], workdir: Path = None) -> None:
     """Preview what files and directories would be scanned"""
     console = Console()
     _, file_items = _scan_paths(paths, workdir)
     
-    # Change message based on whether we're scanning included paths or workdir
-    if len(paths) == 1 and paths[0] == workdir:
-        console.print(f"\n[bold blue]Scanning working directory:[/bold blue] {workdir.absolute()}")
+    # Display working directory status
+    console.print("\n[bold blue]Analysis Paths:[/bold blue]")
+    console.print(f"[cyan]Working Directory:[/cyan] {workdir.absolute()}")
+    
+    # Show if working directory is being scanned
+    is_workdir_scanned = any(p.resolve() == workdir.resolve() for p in paths)
+    if is_workdir_scanned:
+        console.print("[green]✓ Working directory will be scanned[/green]")
     else:
-        console.print(f"\n[bold blue]Working directory:[/bold blue] {workdir.absolute()}")
-        console.print("\n[bold blue]Scanning included paths:[/bold blue]")
+        console.print("[yellow]! Working directory will not be scanned[/yellow]")
+    
+    # Show included paths relative to working directory
+    if len(paths) > (1 if is_workdir_scanned else 0):
+        console.print("\n[cyan]Additional Included Paths:[/cyan]")
         for path in paths:
-            console.print(f"  • {path.absolute()}")
-            
-    console.print("\n[bold blue]Files that would be analyzed:[/bold blue]")
+            if path.resolve() != workdir.resolve():
+                try:
+                    rel_path = path.relative_to(workdir)
+                    console.print(f"  • ./{rel_path}")
+                except ValueError:
+                    # Path is outside working directory
+                    console.print(f"  • {path.absolute()}")
+    
+    console.print("\n[bold blue]Files that will be analyzed:[/bold blue]")
     console.print(Columns(file_items, padding=(0, 4), expand=True))
 
 def is_dir_empty(path: Path) -> bool:
     """Check if directory is empty, ignoring hidden files"""
-    return not any(item for item in path.iterdir() if not item.name.startswith('.'))
+    return not any(item for item in path.iterdir() if not item.name.startswith('.'))
+
+def show_content_stats(content: str) -> None:
+    if not content:
+        return
+        
+    dir_counts = defaultdict(int)
+    for line in content.split('\n'):
+        if line.startswith('<path>'):
+            path = Path(line.replace('<path>', '').replace('</path>', '').strip())
+            dir_counts[str(path.parent)] += 1
+    
+    console = Console()
+    stats = [f"{directory} ({count} files)" for directory, count in dir_counts.items()]
+    columns = Columns(stats, equal=True, expand=True)
+    panel = Panel(columns, title="Work Context")
+    console.print(panel)

janito/tests/test_fileparser.py

@@ -0,0 +1,26 @@
+import pytest
+from pathlib import Path
+from janito.fileparser import validate_file_path, validate_file_content
+
+def test_validate_file_path():
+    # Valid paths
+    assert validate_file_path(Path("test.py")) == (True, "")
+    assert validate_file_path(Path("folder/test.py")) == (True, "")
+    
+    # Invalid paths
+    assert validate_file_path(Path("/absolute/path.py"))[0] == False
+    assert validate_file_path(Path("../escape.py"))[0] == False
+    assert validate_file_path(Path("test?.py"))[0] == False
+    assert validate_file_path(Path("test*.py"))[0] == False
+
+def test_validate_file_content():
+    # Valid content
+    assert validate_file_content("print('hello')") == (True, "")
+    assert validate_file_content("# Empty file with comment\n") == (True, "")
+    
+    # Invalid content
+    assert validate_file_content("")[0] == False
+    
+    # Test large content
+    large_content = "x" * (1024 * 1024 + 1)  # Slightly over 1MB
+    assert validate_file_content(large_content)[0] == False

janito/version.py

@@ -0,0 +1,23 @@
+"""Version management module for Janito."""
+from pathlib import Path
+from typing import Optional
+import tomli
+from importlib.metadata import version as pkg_version
+
+def get_version() -> str:
+    """
+    Get Janito version from package metadata or pyproject.toml.
+    
+    Returns:
+        str: The version string
+    """
+    try:
+        return pkg_version("janito")
+    except Exception:
+        # Fallback to pyproject.toml
+        pyproject_path = Path(__file__).parent.parent / "pyproject.toml"
+        if pyproject_path.exists():
+            with open(pyproject_path, "rb") as f:
+                pyproject_data = tomli.load(f)
+                return pyproject_data.get("project", {}).get("version", "unknown")
+        return "unknown"

janito-0.5.0.dist-info/METADATA

@@ -0,0 +1,146 @@
+Metadata-Version: 2.3
+Name: janito
+Version: 0.5.0
+Summary: A CLI tool for software development tasks powered by AI
+Project-URL: Homepage, https://github.com/joaompinto/janito
+Project-URL: Repository, https://github.com/joaompinto/janito.git
+Author-email: João Pinto <lamego.pinto@gmail.com>
+License: MIT
+Classifier: Development Status :: 4 - Beta
+Classifier: Environment :: Console
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Topic :: Software Development
+Requires-Python: >=3.8
+Requires-Dist: anthropic
+Requires-Dist: pathspec
+Requires-Dist: rich
+Requires-Dist: tomli
+Requires-Dist: typer
+Description-Content-Type: text/markdown
+
+# 🤖 Janito CLI
+
+A CLI tool for software development tasks powered by AI. Janito is your friendly AI-powered software development buddy that helps with coding tasks like refactoring, documentation updates, and code optimization.
+
+## 📥 Installation
+
+1. Install using pip:
+```bash
+pip install janito
+```
+
+2. Verify installation:
+```bash
+janito --version
+```
+
+## ⚙️ Setup
+
+1. Get your Anthropic API key from [Anthropic's website](https://www.anthropic.com/)
+
+2. Set your API key:
+```bash
+# Linux/macOS
+export ANTHROPIC_API_KEY='your-api-key-here'
+
+# Windows (Command Prompt)
+set ANTHROPIC_API_KEY=your-api-key-here
+
+# Windows (PowerShell)
+$env:ANTHROPIC_API_KEY='your-api-key-here'
+```
+
+3. (Optional) Configure default test command:
+```bash
+export JANITO_TEST_CMD='pytest'  # or your preferred test command
+```
+
+## 🚀 Quick Start
+
+### Basic Usage
+
+```bash
+# Add docstrings to your code
+janito "add docstrings to this file"
+
+# Optimize a function
+janito "optimize the main function"
+
+# Get code explanations
+janito --ask "explain this code"
+```
+
+### Common Scenarios
+
+1. **Code Refactoring**
+```bash
+# Refactor with test validation
+janito "refactor this code to use list comprehension" --test "pytest"
+
+# Refactor specific directory
+janito "update imports" -i ./src
+```
+
+2. **Documentation Updates**
+```bash
+# Add or update docstrings
+janito "add type hints and docstrings"
+
+# Generate README
+janito "create a README for this project"
+```
+
+3. **Code Analysis**
+```bash
+# Get code explanations
+janito --ask "what does this function do?"
+
+# Find potential improvements
+janito --ask "suggest optimizations for this code"
+```
+
+## 🛠️ Command Reference
+
+### Syntax
+```bash
+janito [OPTIONS] [REQUEST]
+```
+
+### Key Options
+
+| Option | Description |
+|--------|-------------|
+| `REQUEST` | The AI request/instruction (in quotes) |
+| `-w, --working-dir PATH` | Working directory [default: current] |
+| `-i, --include PATH` | Include directory int the working context (can be multiple)|
+| `--ask QUESTION` | Ask questions without making changes |
+| `--test COMMAND` | Run tests before applying changes |
+| `--debug` | Enable debug logging |
+| `--verbose` | Enable verbose mode |
+| `--version` | Show version information |
+| `--help` | Show help message |
+
+## 🔑 Key Features
+
+- 🤖 AI-powered code analysis and modifications
+- 💻 Interactive console mode
+- ✅ Syntax validation for Python files
+- 👀 Change preview and confirmation
+- 🧪 Test command execution
+- 📜 Change history tracking
+
+## 📚 Additional Information
+
+- Requires Python 3.8+
+- Changes are backed up in `.janito/changes_history/`
+- Environment variables:
+  - `ANTHROPIC_API_KEY`: Required for API access
+  - `JANITO_TEST_CMD`: Default test command (optional)
+
+## 📄 License
+
+MIT License - see [LICENSE](LICENSE) file for details.

janito-0.5.0.dist-info/RECORD

@@ -0,0 +1,45 @@
+janito/__init__.py,sha256=T1hTuIOJgEUjXNHplUuMZIgG_QxC1RH2ZcBe6WJ1XBE,1099
+janito/__main__.py,sha256=ogcCBQ2YHkQhdhrXF7OV-DthLiL-Zw24XtRRYscT2Yo,2018
+janito/_contextparser.py,sha256=iDX6nlqUQr-lj7lkEI4B6jHLci_Kxl-XWOaEiAQtVxA,4531
+janito/changehistory.py,sha256=4Mu60og1pySlyfB36aqHnNKyH0yuSj-5rvBjmmvKGYw,1510
+janito/common.py,sha256=imIwx0FCDe7p71ze9TBvn_W0Qz99Am3X5U9eHYd9vws,790
+janito/config.py,sha256=ocg0lyab9ysgczKaqJTAtv0ZRa2VDMwclTJBgps7Vxw,1171
+janito/fileparser.py,sha256=rJEme9QWbtnsjs9KuO3YI7B499DnfedWtzBONXqrjfU,12925
+janito/prompts.py,sha256=M8IveFkG3fgLe87_9kTNdWAVkqmsD0T4eP6VmOqHAOQ,2607
+janito/qa.py,sha256=MSzu9yvLCwXr9sKylTXtMfTy_2fRfPl3CU2NXf0OrS4,1977
+janito/review.py,sha256=5Oc6BfxMGNmKbIeDP5_EiAKUDeQwVOD0YL7iqfgJLRE,471
+janito/scan.py,sha256=QBXu64t8CjjJhiFdO-0npwdSPys9IX1oOOOyh1cGVIE,7678
+janito/version.py,sha256=ylfPwGtdY8dEOFJ-DB9gKUQLggqRCvoLxhpnwjzCM94,739
+janito/agents/__init__.py,sha256=VPBXIh526D11NrCSnuXUerYT7AI3rynp2klCgz94tPk,648
+janito/agents/agent.py,sha256=3uGiUrvj9oCW6_oK-oMQQJ77K8jZFv7mAdXlIG1dxNY,751
+janito/agents/claudeai.py,sha256=bl1MeALicf6arX_m9GmtAj0i3UoUiTbbXytVjLam2S8,2395
+janito/agents/openai.py,sha256=tNtlzFJMWFv38Z62opR23u6xXlZ9L4xX_mf2f3wjrLU,2082
+janito/agents/test.py,sha256=xoN1q9DUSYpUbnvTP1qZsEfxYrZfocJlt9DkIuMDvvY,1552
+janito/analysis/__init__.py,sha256=QVhIoZdOc4eYhQ9ZRoZZiwowUaa-PQ0_7HV_cx4eOZU,734
+janito/analysis/display.py,sha256=714423TxelieAyBe0M5A78i99qj_VfggZUWRbk2owzU,5204
+janito/analysis/options.py,sha256=tgsLZOtNy35l4v4F6p_adGk3CrWhrDN0Ba_tw4Zzj5Q,3852
+janito/analysis/prompts.py,sha256=7NI7XyL6gUeJWTfVP_X0ohtSnYb6gpu4Thy5kxzoeD8,2454
+janito/change/__init__.py,sha256=ElyiGt6KmwufQouir-0l0ZtxNZdeLWTFRc4Vbg0RW3s,467
+janito/change/applier.py,sha256=GXs0DRfb48WNoSyAH0uETjpzzFLvmUZGkb6VydKzazk,11601
+janito/change/content.py,sha256=R8lbOFF2ANV_NF3S6zCV60-Q7hGFWiS-zmJBgu-r5SU,2475
+janito/change/indentation.py,sha256=qvsGX9ZPm5cRftXJSKSVbK9-Po_95HCAc7sG98BxhIY,1191
+janito/change/position.py,sha256=Nlkg0sGUvqYwtKA19ne1kT3ixiVYKUcy_40Cdr-gfl8,6385
+janito/changeviewer/__init__.py,sha256=5kR3fsAYSgnA0Hlty0fMC6392aSK7WifH2ZIW-INjBc,298
+janito/changeviewer/diff.py,sha256=2nW1Yzu7rzzIsuHWg-t2EJfsoVdbRaRJk3ewQBjQdxY,1117
+janito/changeviewer/panels.py,sha256=F1yWUd6Vh535rV20NnFrVh6uvpc3w5CDBSNQUf_vOC8,9426
+janito/changeviewer/styling.py,sha256=qv7DVfu57Qy93XUWIWdyqZWhgcfm-BaJVPslSuch05s,2461
+janito/changeviewer/themes.py,sha256=BWa4l82fNM0pj01UYK9q_C7t_AqTwNbR8i0-F4I7KI8,1593
+janito/cli/__init__.py,sha256=3gyMSaEAH2N4mXfZTLsHXKxXDxdhlYUeAPYQuhnOVBE,77
+janito/cli/commands.py,sha256=6vPT60oT-1jJd_qdy-Si_pcf7yMCGmwpvQ43FZNJjUs,2001
+janito/cli/functions.py,sha256=5Nn4muPv1KrInhcernByQqn_T0q2bCxPeDA1BoPw4K8,11399
+janito/cli/registry.py,sha256=R1sI45YonxjMSLhAle7Dt18X_devrMsLt0ljb-rNza4,690
+janito/console/__init__.py,sha256=0zxlJR88ESfuXtyzARnWNKcZ1rTyWLZwzXrfDQ7NHyw,77
+janito/console/commands.py,sha256=kuI2w1LfEw0kbF_vOAYVENoMQYC42fXmDFZgxVKJKo8,3422
+janito/console/core.py,sha256=DTAP_bhw_DTklUyPbGdjSzllUBMg7jXW0SklXkRNMI8,2264
+janito/console/display.py,sha256=VNyHGHA_MG-dLPTCylgOKgHAl0muwhmsvg3nXFU-OZo,5380
+janito/tests/test_fileparser.py,sha256=20CfwfnFtVm3_qU11qRAdcBZQXaLlkcHgbqWDOeazj4,997
+janito-0.5.0.dist-info/METADATA,sha256=Jq4fJLGNs-4O8IOB2iy_8JrszlWbFzOmwrZRjQxIs_Y,3659
+janito-0.5.0.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+janito-0.5.0.dist-info/entry_points.txt,sha256=wIo5zZxbmu4fC-ZMrsKD0T0vq7IqkOOLYhrqRGypkx4,48
+janito-0.5.0.dist-info/licenses/LICENSE,sha256=xLIUXRPjtsgQml2zD1Pn4LpgiyZ49raw6jZDlO_gZdo,1062
+janito-0.5.0.dist-info/RECORD,,

janito/changeviewer.py

@@ -1,64 +0,0 @@
-from pathlib import Path
-from rich.console import Console
-from rich.text import Text
-from typing import TypedDict
-import difflib
-
-class FileChange(TypedDict):
-    """Type definition for a file change"""
-    description: str
-    new_content: str
-
-def show_file_changes(console: Console, filepath: Path, original: str, new_content: str, description: str) -> None:
-    """Display side by side comparison of file changes"""
-    half_width = (console.width - 3) // 2
-    
-    # Show header
-    console.print(f"\n[bold blue]Changes for {filepath}[/bold blue]")
-    console.print(f"[dim]{description}[/dim]\n")
-    
-    # Show side by side content
-    console.print(Text("OLD".center(half_width) + "│" + "NEW".center(half_width), style="blue bold"))
-    console.print(Text("─" * half_width + "┼" + "─" * half_width, style="blue"))
-    
-    old_lines = original.splitlines()
-    new_lines = new_content.splitlines()
-    
-    for i in range(max(len(old_lines), len(new_lines))):
-        old = old_lines[i] if i < len(old_lines) else ""
-        new = new_lines[i] if i < len(new_lines) else ""
-        
-        old_text = Text(f"{old:<{half_width}}", style="red" if old != new else None)
-        new_text = Text(f"{new:<{half_width}}", style="green" if old != new else None)
-        console.print(old_text + Text("│", style="blue") + new_text)
-
-def show_diff_changes(console: Console, filepath: Path, original: str, new_content: str, description: str) -> None:
-    """Display file changes using unified diff format"""
-    # Show header
-    console.print(f"\n[bold blue]Changes for {filepath}[/bold blue]")
-    console.print(f"[dim]{description}[/dim]\n")
-    
-    # Generate diff
-    diff = difflib.unified_diff(
-        original.splitlines(keepends=True),
-        new_content.splitlines(keepends=True),
-        fromfile='old',
-        tofile='new',
-        lineterm=''
-    )
-    
-    # Print diff with colors
-    for line in diff:
-        if line.startswith('+++'):
-            console.print(Text(line.rstrip(), style="bold green"))
-        elif line.startswith('---'):
-            console.print(Text(line.rstrip(), style="bold red"))
-        elif line.startswith('+'):
-            console.print(Text(line.rstrip(), style="green"))
-        elif line.startswith('-'):
-            console.print(Text(line.rstrip(), style="red"))
-        elif line.startswith('@@'):
-            console.print(Text(line.rstrip(), style="cyan"))
-        else:
-            console.print(Text(line.rstrip()))
-

janito/claude.py

@@ -1,74 +0,0 @@
-from rich.traceback import install
-import anthropic
-import os
-from typing import Optional
-from rich.progress import Progress, SpinnerColumn, TextColumn
-from threading import Event
-
-# Install rich traceback handler
-install(show_locals=True)
-
-class ClaudeAPIAgent:
-    """Handles interaction with Claude API, including message handling"""
-    def __init__(self, api_key: Optional[str] = None, system_prompt: str = None):
-        if not system_prompt:
-            raise ValueError("system_prompt is required")
-        self.api_key = api_key or os.getenv('ANTHROPIC_API_KEY')
-        if not self.api_key:
-            raise ValueError("ANTHROPIC_API_KEY environment variable is required")
-        self.client = anthropic.Client(api_key=self.api_key)
-        self.model = "claude-3-5-sonnet-20241022"
-        self.stop_progress = Event()
-        self.system_message = system_prompt
-        self.last_prompt = None
-        self.last_full_message = None
-        self.last_response = None
-        self.messages_history = []
-        if system_prompt:
-            self.messages_history.append(("system", system_prompt))
-
-    def send_message(self, message: str, stop_event: Event = None) -> str:
-        """Send message to Claude API and return response"""
-        try:
-            self.messages_history.append(("user", message))
-            # Store the full message
-            self.last_full_message = message
-            
-            try:
-                # Check if already cancelled
-                if stop_event and stop_event.is_set():
-                    return ""
-                
-                # Start API request
-                response = self.client.messages.create(
-                    model=self.model,  # Use discovered model
-                    system=self.system_message,
-                    max_tokens=4000,
-                    messages=[
-                        {"role": "user", "content": message}
-                    ],
-                    temperature=0,
-                )
-                
-                # Handle response
-                response_text = response.content[0].text
-                
-                # Only store and process response if not cancelled
-                if not (stop_event and stop_event.is_set()):
-                    self.last_response = response_text
-                    self.messages_history.append(("assistant", response_text))
-                
-                # Always return the response, let caller handle cancellation
-                return response_text
-                
-            except KeyboardInterrupt:
-                if stop_event:
-                    stop_event.set()
-                return ""
-                
-        except Exception as e:
-            error_msg = f"Error: {str(e)}"
-            self.messages_history.append(("error", error_msg))
-            if stop_event and stop_event.is_set():
-                return ""
-            return error_msg

janito/console.py

@@ -1,60 +0,0 @@
-from prompt_toolkit import PromptSession
-from prompt_toolkit.history import FileHistory
-from pathlib import Path
-from rich.console import Console
-from janito.claude import ClaudeAPIAgent
-from janito.prompts import build_request_analisys_prompt, SYSTEM_PROMPT
-from janito.scan import collect_files_content
-from janito.__main__ import handle_option_selection
-
-def start_console_session(workdir: Path, include: list[Path] = None) -> None:
-    """Start an interactive console session using prompt_toolkit"""
-    console = Console()
-    claude = ClaudeAPIAgent(system_prompt=SYSTEM_PROMPT)
-
-    # Setup prompt session with history
-    history_file = workdir / '.janito' / 'console_history'
-    history_file.parent.mkdir(parents=True, exist_ok=True)
-    session = PromptSession(history=FileHistory(str(history_file)))
-
-    from importlib.metadata import version
-    try:
-        ver = version("janito")
-    except:
-        ver = "dev"
-
-    console.print("\n[bold blue]╔═══════════════════════════════════════════╗[/bold blue]")
-    console.print("[bold blue]║           Janito AI Assistant             ║[/bold blue]")
-    console.print("[bold blue]║                v" + ver.ljust(8) + "                  ║[/bold blue]")
-    console.print("[bold blue]╠═══════════════════════════════════════════╣[/bold blue]")
-    console.print("[bold blue]║  Your AI-powered development companion    ║[/bold blue]")
-    console.print("[bold blue]╚═══════════════════════════════════════════╝[/bold blue]")
-    console.print("\n[cyan]Type your requests or 'exit' to quit[/cyan]\n")
-
-    while True:
-        try:
-            request = session.prompt("janito> ")
-            if request.lower() in ('exit', 'quit'):
-                break
-
-            if not request.strip():
-                continue
-
-            # Get current files content
-            paths_to_scan = [workdir] if workdir else []
-            if include:
-                paths_to_scan.extend(include)
-            files_content = collect_files_content(paths_to_scan, workdir)
-
-            # Get initial analysis
-            initial_prompt = build_request_analisys_prompt(files_content, request)
-            initial_response = claude.send_message(initial_prompt)
-
-            # Show response and handle options
-            console.print(initial_response)
-            handle_option_selection(claude, initial_response, request, False, workdir, include)
-
-        except KeyboardInterrupt:
-            continue
-        except EOFError:
-            break