tzamuncode 0.1.0__py3-none-any.whl

tzamuncode/models/vllm_client.py

@@ -0,0 +1,121 @@
+"""
+vLLM Client for TzamunCode CLI
+Provides faster inference using vLLM OpenAI-compatible API
+"""
+
+import requests
+from typing import Iterator, List, Dict, Optional
+
+
+class VLLMClient:
+    """Client for vLLM server using OpenAI-compatible API"""
+    
+    def __init__(self, base_url: str = "http://localhost:8000", model: str = "deepseek-coder-7b"):
+        self.base_url = base_url.rstrip('/')
+        self.model = model
+        self.api_url = f"{self.base_url}/v1"
+    
+    def chat(self, messages: List[Dict[str, str]], stream: bool = True) -> Iterator[str]:
+        """
+        Send chat request to vLLM server
+        
+        Args:
+            messages: List of message dicts with 'role' and 'content'
+            stream: Whether to stream the response
+            
+        Yields:
+            Response chunks as strings
+        """
+        url = f"{self.api_url}/chat/completions"
+        
+        payload = {
+            "model": self.model,
+            "messages": messages,
+            "stream": stream,
+            "temperature": 0.7,
+            "max_tokens": 2000,  # Reduced to fit within 4096 context with system prompt
+        }
+        
+        try:
+            response = requests.post(
+                url,
+                json=payload,
+                stream=stream,
+                timeout=60
+            )
+            
+            if response.status_code != 200:
+                error_detail = response.text
+                yield f"\n❌ vLLM Error ({response.status_code}): {error_detail}\n"
+                return
+            
+            response.raise_for_status()
+            
+            if stream:
+                for line in response.iter_lines():
+                    if line:
+                        line = line.decode('utf-8')
+                        if line.startswith('data: '):
+                            data = line[6:]  # Remove 'data: ' prefix
+                            if data.strip() == '[DONE]':
+                                break
+                            
+                            try:
+                                import json
+                                chunk = json.loads(data)
+                                if 'choices' in chunk and len(chunk['choices']) > 0:
+                                    delta = chunk['choices'][0].get('delta', {})
+                                    content = delta.get('content', '')
+                                    if content:
+                                        yield content
+                            except json.JSONDecodeError:
+                                continue
+            else:
+                result = response.json()
+                if 'choices' in result and len(result['choices']) > 0:
+                    content = result['choices'][0]['message']['content']
+                    yield content
+                    
+        except requests.exceptions.RequestException as e:
+            yield f"\n❌ Error connecting to vLLM server: {e}\n"
+            yield "Make sure vLLM server is running on http://localhost:8000\n"
+    
+    def list_models(self) -> List[str]:
+        """List available models from vLLM server"""
+        try:
+            url = f"{self.api_url}/models"
+            response = requests.get(url, timeout=5)
+            response.raise_for_status()
+            
+            data = response.json()
+            if 'data' in data:
+                return [model['id'] for model in data['data']]
+            return [self.model]
+        except:
+            return [self.model]
+    
+    def chat_stream(self, messages: List[Dict[str, str]]) -> Iterator[str]:
+        """
+        Stream chat responses (alias for chat with stream=True)
+        
+        Args:
+            messages: List of message dicts with 'role' and 'content'
+            
+        Yields:
+            Response chunks as strings
+        """
+        yield from self.chat(messages, stream=True)
+    
+    def generate(self, prompt: str, stream: bool = True) -> Iterator[str]:
+        """
+        Generate completion from prompt
+        
+        Args:
+            prompt: Input prompt
+            stream: Whether to stream the response
+            
+        Yields:
+            Response chunks
+        """
+        messages = [{"role": "user", "content": prompt}]
+        yield from self.chat(messages, stream=stream)

tzamuncode/utils/__init__.py

@@ -0,0 +1 @@
+"""Utilities module"""

tzamuncode/utils/file_ops.py

@@ -0,0 +1,59 @@
+"""
+File operations utilities for TzamunCode
+"""
+
+from pathlib import Path
+from typing import Optional
+import difflib
+
+class FileManager:
+    """Manage file operations"""
+    
+    def read_file(self, path: str) -> str:
+        """Read file content"""
+        file_path = Path(path)
+        if not file_path.exists():
+            raise FileNotFoundError(f"File not found: {path}")
+        return file_path.read_text()
+    
+    def write_file(self, path: str, content: str) -> None:
+        """Write content to file"""
+        file_path = Path(path)
+        
+        # Create parent directories if they don't exist
+        file_path.parent.mkdir(parents=True, exist_ok=True)
+        
+        file_path.write_text(content)
+    
+    def show_diff(self, path: str, new_content: str) -> str:
+        """Show diff between current and new content"""
+        try:
+            old_content = self.read_file(path)
+        except FileNotFoundError:
+            # New file
+            return f"+++ {path} (new file)\n" + "\n".join(f"+{line}" for line in new_content.splitlines())
+        
+        old_lines = old_content.splitlines(keepends=True)
+        new_lines = new_content.splitlines(keepends=True)
+        
+        diff = difflib.unified_diff(
+            old_lines,
+            new_lines,
+            fromfile=f"a/{path}",
+            tofile=f"b/{path}",
+            lineterm=""
+        )
+        
+        return "".join(diff)
+    
+    def file_exists(self, path: str) -> bool:
+        """Check if file exists"""
+        return Path(path).exists()
+    
+    def list_files(self, directory: str, pattern: str = "*") -> list:
+        """List files in directory matching pattern"""
+        dir_path = Path(directory)
+        if not dir_path.is_dir():
+            raise NotADirectoryError(f"Not a directory: {directory}")
+        
+        return [str(p) for p in dir_path.rglob(pattern) if p.is_file()]

tzamuncode/utils/project_scanner.py

@@ -0,0 +1,193 @@
+"""
+Project scanner for understanding codebase structure
+"""
+
+from pathlib import Path
+from typing import Dict, List, Set
+import os
+
+
+class ProjectScanner:
+    """Scan and understand project structure"""
+    
+    # Common files to ignore
+    IGNORE_DIRS = {
+        '.git', '__pycache__', 'node_modules', 'venv', 'env',
+        '.venv', 'dist', 'build', '.pytest_cache', '.mypy_cache',
+        'coverage', '.coverage', 'htmlcov'
+    }
+    
+    IGNORE_FILES = {
+        '.pyc', '.pyo', '.pyd', '.so', '.dll', '.dylib',
+        '.class', '.o', '.obj', '.exe'
+    }
+    
+    # Language detection
+    LANGUAGE_EXTENSIONS = {
+        '.py': 'Python',
+        '.js': 'JavaScript',
+        '.ts': 'TypeScript',
+        '.jsx': 'React',
+        '.tsx': 'React TypeScript',
+        '.go': 'Go',
+        '.rs': 'Rust',
+        '.java': 'Java',
+        '.cpp': 'C++',
+        '.c': 'C',
+        '.rb': 'Ruby',
+        '.php': 'PHP',
+        '.swift': 'Swift',
+        '.kt': 'Kotlin',
+        '.sh': 'Shell',
+        '.md': 'Markdown',
+        '.json': 'JSON',
+        '.yaml': 'YAML',
+        '.yml': 'YAML',
+        '.toml': 'TOML',
+        '.html': 'HTML',
+        '.css': 'CSS',
+        '.scss': 'SCSS',
+        '.sql': 'SQL',
+    }
+    
+    def __init__(self, workspace: Path):
+        self.workspace = Path(workspace)
+    
+    def scan(self) -> Dict:
+        """
+        Scan project and return structure information
+        
+        Returns:
+            Dict with project metadata
+        """
+        structure = {
+            'total_files': 0,
+            'total_dirs': 0,
+            'languages': set(),
+            'key_files': [],
+            'file_tree': [],
+        }
+        
+        for root, dirs, files in os.walk(self.workspace):
+            # Filter ignored directories
+            dirs[:] = [d for d in dirs if d not in self.IGNORE_DIRS]
+            
+            root_path = Path(root)
+            rel_root = root_path.relative_to(self.workspace)
+            
+            structure['total_dirs'] += len(dirs)
+            
+            for file in files:
+                # Skip ignored files
+                if any(file.endswith(ext) for ext in self.IGNORE_FILES):
+                    continue
+                
+                structure['total_files'] += 1
+                
+                file_path = root_path / file
+                rel_path = file_path.relative_to(self.workspace)
+                
+                # Detect language
+                ext = file_path.suffix
+                if ext in self.LANGUAGE_EXTENSIONS:
+                    structure['languages'].add(self.LANGUAGE_EXTENSIONS[ext])
+                
+                # Identify key files
+                if self._is_key_file(file):
+                    structure['key_files'].append(str(rel_path))
+                
+                # Add to file tree (limit depth)
+                if len(rel_path.parts) <= 3:
+                    structure['file_tree'].append(str(rel_path))
+        
+        structure['languages'] = sorted(list(structure['languages']))
+        
+        return structure
+    
+    def _is_key_file(self, filename: str) -> bool:
+        """Check if file is a key project file"""
+        key_files = {
+            'README.md', 'README.rst', 'README.txt',
+            'setup.py', 'setup.cfg', 'pyproject.toml',
+            'requirements.txt', 'Pipfile', 'poetry.lock',
+            'package.json', 'package-lock.json', 'yarn.lock',
+            'Cargo.toml', 'go.mod', 'pom.xml', 'build.gradle',
+            'Makefile', 'Dockerfile', 'docker-compose.yml',
+            '.gitignore', 'LICENSE', 'CONTRIBUTING.md',
+            'main.py', 'app.py', 'index.js', 'main.go',
+            '__init__.py', '__main__.py'
+        }
+        
+        return filename in key_files
+    
+    def find_files(self, pattern: str) -> List[str]:
+        """Find files matching a pattern"""
+        matches = []
+        
+        for root, dirs, files in os.walk(self.workspace):
+            dirs[:] = [d for d in dirs if d not in self.IGNORE_DIRS]
+            
+            root_path = Path(root)
+            
+            for file in files:
+                if pattern in file:
+                    file_path = root_path / file
+                    rel_path = file_path.relative_to(self.workspace)
+                    matches.append(str(rel_path))
+        
+        return matches
+    
+    def get_project_type(self) -> str:
+        """Detect project type"""
+        project_types = []
+        
+        # Check root directory
+        try:
+            root_files = set(os.listdir(self.workspace))
+        except:
+            return 'Unknown'
+        
+        # Check for multiple project types
+        if 'package.json' in root_files:
+            project_types.append('Node.js')
+        
+        if 'requirements.txt' in root_files or 'setup.py' in root_files or 'pyproject.toml' in root_files:
+            project_types.append('Python')
+        
+        if 'Cargo.toml' in root_files:
+            project_types.append('Rust')
+        
+        if 'go.mod' in root_files:
+            project_types.append('Go')
+        
+        if 'pom.xml' in root_files or 'build.gradle' in root_files:
+            project_types.append('Java')
+        
+        # Check for frontend frameworks
+        if 'package.json' in root_files:
+            try:
+                import json
+                pkg_path = self.workspace / 'package.json'
+                with open(pkg_path) as f:
+                    pkg = json.load(f)
+                    deps = {**pkg.get('dependencies', {}), **pkg.get('devDependencies', {})}
+                    
+                    if 'react' in deps:
+                        project_types.append('React')
+                    elif 'vue' in deps:
+                        project_types.append('Vue')
+                    elif 'angular' in deps or '@angular/core' in deps:
+                        project_types.append('Angular')
+                    elif 'next' in deps:
+                        project_types.append('Next.js')
+            except:
+                pass
+        
+        # Check subdirectories for additional context
+        if 'backend' in root_files and 'frontend' in root_files:
+            return 'Full-Stack (' + ', '.join(project_types) + ')' if project_types else 'Full-Stack'
+        
+        if project_types:
+            return ', '.join(project_types)
+        
+        return 'Unknown'

tzamuncode-0.1.0.dist-info/METADATA

@@ -0,0 +1,200 @@
+Metadata-Version: 2.4
+Name: tzamuncode
+Version: 0.1.0
+Summary: TzamunCode - AI Coding Assistant powered by local models
+Author-email: "Tzamun Arabia IT Co." <info@tzamun.com>
+License: MIT
+Project-URL: Homepage, https://tzamun.com
+Project-URL: Documentation, https://docs.tzamun.com/tzamuncode
+Project-URL: Repository, https://github.com/tzamun/tzamuncode-cli
+Keywords: ai,coding,assistant,ollama,cli
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: typer>=0.9.0
+Requires-Dist: rich>=13.0.0
+Requires-Dist: textual>=0.40.0
+Requires-Dist: readchar>=4.0.0
+Requires-Dist: langchain>=0.1.0
+Requires-Dist: langchain-community>=0.0.10
+Requires-Dist: openai>=1.0.0
+Requires-Dist: requests>=2.31.0
+Requires-Dist: gitpython>=3.1.40
+Requires-Dist: pathspec>=0.11.0
+Requires-Dist: tiktoken>=0.5.0
+Requires-Dist: prompt-toolkit>=3.0.0
+Provides-Extra: dev
+Requires-Dist: pytest>=7.4.0; extra == "dev"
+Requires-Dist: black>=23.0.0; extra == "dev"
+Requires-Dist: ruff>=0.1.0; extra == "dev"
+Requires-Dist: mypy>=1.5.0; extra == "dev"
+Dynamic: license-file
+
+# TzamunCode CLI 🚀
+
+> AI Coding Assistant powered by local models - Built in Saudi Arabia 🇸🇦
+
+**TzamunCode** is a privacy-first AI coding assistant that runs entirely on your local infrastructure using Ollama and vLLM. No cloud dependencies, no API costs, complete control.
+
+## ✨ Features
+
+- 🤖 **Agentic AI** - Multi-step planning and execution
+- 📝 **Code Generation** - Create files, functions, and entire projects
+- ✏️ **Multi-file Editing** - Edit multiple files in one operation
+- 🔧 **Tool Calling** - Git operations, file search, command execution
+- 🎯 **Context Aware** - Understands your project structure
+- 🔒 **Privacy First** - Everything runs locally
+- ⚡ **Fast** - Powered by vLLM for optimized inference
+- 🌍 **Multi-model** - Use any Ollama model (15+ available)
+
+## 🚀 Quick Start
+
+### Installation
+
+```bash
+# Clone the repository
+git clone https://github.com/tzamun/tzamuncode-cli.git
+cd tzamuncode-cli
+
+# Install
+pip install -e .
+
+# Or install from PyPI (when published)
+pip install tzamuncode
+```
+
+### Prerequisites
+
+- Python 3.9+
+- Ollama running locally (http://localhost:11434)
+- At least one Ollama model installed
+
+### Basic Usage
+
+```bash
+# Start interactive chat
+tzamuncode chat
+
+# Generate code
+tzamuncode generate "Create a Flask REST API with authentication"
+
+# Edit a file
+tzamuncode edit app.py "Add error handling to all routes"
+
+# Explain code
+tzamuncode explain main.py
+
+# Quick alias
+tzc chat
+```
+
+## 📖 Documentation
+
+### Commands
+
+#### `chat` - Interactive Chat
+```bash
+tzamuncode chat
+tzamuncode chat --model qwen2.5:32b
+```
+
+#### `generate` - Code Generation
+```bash
+tzamuncode generate "Create a Python web scraper"
+tzamuncode generate "Add unit tests for user.py" --output tests/
+```
+
+#### `edit` - File Editing
+```bash
+tzamuncode edit app.py "Refactor to use async/await"
+tzamuncode edit . "Add type hints to all functions"
+```
+
+#### `explain` - Code Explanation
+```bash
+tzamuncode explain complex_function.py
+tzamuncode explain --detailed auth.py
+```
+
+#### `review` - Code Review
+```bash
+tzamuncode review pull_request.diff
+tzamuncode review --strict src/
+```
+
+### Configuration
+
+Create `~/.tzamuncode/config.yaml`:
+
+```yaml
+# Default model
+model: qwen2.5:32b
+
+# Ollama settings
+ollama:
+  base_url: http://localhost:11434
+  timeout: 120
+
+# vLLM settings (optional, for faster inference)
+vllm:
+  enabled: true
+  base_url: http://localhost:8000
+  model: deepseek-coder-7b
+
+# Preferences
+preferences:
+  show_diff: true
+  auto_apply: false
+  max_context: 64000
+  temperature: 0.7
+```
+
+## 🏗️ Architecture
+
+```
+┌─────────────────────────────────────┐
+│     TzamunCode CLI                  │
+│  (Typer + Rich UI)                  │
+└─────────────────────────────────────┘
+           ↓
+┌─────────────────────────────────────┐
+│   Agentic Layer (LangChain)         │
+│  - Multi-step planning              │
+│  - Tool calling                     │
+│  - Context management               │
+└─────────────────────────────────────┘
+           ↓
+┌─────────────────────────────────────┐
+│   AI Backend                        │
+│  - Ollama (15+ models)              │
+│  - vLLM (fast inference)            │
+└─────────────────────────────────────┘
+```
+
+## 🤝 Contributing
+
+We welcome contributions! See [CONTRIBUTING.md](CONTRIBUTING.md) for guidelines.
+
+## 📄 License
+
+MIT License - see [LICENSE](LICENSE) for details.
+
+## 🌟 Built by Tzamun Arabia IT Co.
+
+**TzamunCode** is part of the Tzamun AI ecosystem:
+- **TzamunAI** - AI platform with 15+ models
+- **TzamunERP** - ERPNext + AI integration
+- **Auxly** - AI coding assistant for IDEs
+- **AccessHub** - Privileged Access Management
+
+Visit [tzamun.com](https://tzamun.com) to learn more.
+
+---
+
+Made with ❤️ in Saudi Arabia 🇸🇦

tzamuncode-0.1.0.dist-info/RECORD

@@ -0,0 +1,29 @@
+tzamuncode/__init__.py,sha256=cNcSNEPzBmnBV86T-NDk_EgBLbfV2VlWy0vgxXLRSTU,229
+tzamuncode/agents/__init__.py,sha256=ZcQU1197DpPw2Gq3dWH99E6nsgOv0F6dojKO4BbV6kQ,39
+tzamuncode/agents/coder.py,sha256=852XLqSSFZpgIpIrbLuTn4BHFSES2DJaAXQfFJjXRVY,3977
+tzamuncode/agents/tools.py,sha256=xIiQGhdUXEui3Wy6zyl0g6JgR9wUOwsrxlKRUm_Sx9Y,5253
+tzamuncode/auth/__init__.py,sha256=BLT9ylnDYj8iyIS_Mb-IZrd5Br7eS1OjiPTTOb019mM,43
+tzamuncode/auth/auth_manager.py,sha256=y-YWaVrfhjzW33IXP3dw1eqtTOW56jo9iB-Ef8pznps,4970
+tzamuncode/cli/__init__.py,sha256=2GidVBgUv1bt9mf4sS1a6kLP5p9xyxT_YMjG4PpGIAk,32
+tzamuncode/cli/agentic_commands.py,sha256=9xyDJhKZiT6__7N-FYfGyFAC_hS_xnqgWU6RQMQerKY,4390
+tzamuncode/cli/auth_commands.py,sha256=6fhnm3Ta6QggcjiDq9On8hd0NrKbDQZq5f195m9l1n8,3999
+tzamuncode/cli/commands.py,sha256=hGQvyFo1aYzExrZaXmxrbJSHJU9dC-2OzVG0lM-2Lqg,6521
+tzamuncode/cli/enhanced_chat.py,sha256=9FaGPOKRXIIGIvdZHlhdxAMwKgVA9qorM7YDmSuzmBE,11193
+tzamuncode/cli/interactive_chat.py,sha256=K-mpagYwYtVAEfvTYIOb5r_srydB7EG1k6jgATuXBsk,10649
+tzamuncode/cli/main.py,sha256=bX7WHyjLa_zeiTowBDtDB0WXbTSg79flm6iFBnjlVvM,12726
+tzamuncode/cli/realtime_chat.py,sha256=a8JCU2KwoJiQmXS89h8zDdaDPl1f-QQzSZIrasZ6WFU,40202
+tzamuncode/cli/realtime_chat_methods.py,sha256=N4iJWFV6bLIG7SUuGjZTB_RzqntIxdu5Svy54rpARLQ,7186
+tzamuncode/cli/tui_chat.py,sha256=EbAUFE8vHG2S7LN9qWVkzay3vTvQuXPFXwGZ0CRprHI,10612
+tzamuncode/config/__init__.py,sha256=702VfKiaLtKlGiTc2655Fgb1twtivhPaXZKk67wqfK0,27
+tzamuncode/models/__init__.py,sha256=obVaZ746k9ZW9OBOF4I2ZLwN293d11iyLjEyj2oZSMk,36
+tzamuncode/models/ollama.py,sha256=Szwi9XMuSl1jD6Q3ffLrWqPlrU39ra5lA62NVESsCE0,4040
+tzamuncode/models/vllm_client.py,sha256=rJzjYNOYoK09SStDEkt6YIm-Gm5XzL7UGRec2ArR5ZI,4309
+tzamuncode/utils/__init__.py,sha256=pwnlYHOEveP6R5snPZnGlTE7erq2-eNyYx1aSf_O77c,23
+tzamuncode/utils/file_ops.py,sha256=di50l5oV04o_P-iQcB8UrOcIf8TVEU6LikA8etpYp7U,1892
+tzamuncode/utils/project_scanner.py,sha256=SvfxtbSn5nBvuqJNgIRVjVGlvTymfIvG0GxVHxYcA0c,6367
+tzamuncode-0.1.0.dist-info/licenses/LICENSE,sha256=SKMpoBdfYod6hLsPmTRIIPBFkDfdDNqkSWVd357RTSc,1077
+tzamuncode-0.1.0.dist-info/METADATA,sha256=rT8-usbRGWHygFttCf8X9_Eb4NRvjle0JIRBXd1Xir8,5652
+tzamuncode-0.1.0.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+tzamuncode-0.1.0.dist-info/entry_points.txt,sha256=XUYoWwk091ZyGE9oB2K6iZqix2H4efQM7YNhOA7FSHU,85
+tzamuncode-0.1.0.dist-info/top_level.txt,sha256=0phBwTnCSCYPBHwWuJdyJ_3WmVWEoCKIhqT4IIwDyPc,11
+tzamuncode-0.1.0.dist-info/RECORD,,

tzamuncode-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (82.0.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

tzamuncode-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,3 @@
+[console_scripts]
+tzamuncode = tzamuncode.cli.main:app
+tzc = tzamuncode.cli.main:app

tzamuncode-0.1.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Tzamun Arabia IT Co.
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

tzamuncode-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+tzamuncode