blitzcoder 1.0.0__py3-none-any.whl

blitzcoder/__init__.py

@@ -0,0 +1,24 @@
+"""
+BlitzCoder - AI-Powered Development Assistant
+
+A comprehensive AI-powered development assistant that helps with code generation,
+refactoring, project scaffolding, and development tasks.
+"""
+
+__version__ = "1.0.0"
+__author__ = "BlitzCoder Team"
+__author_email__ = "raghunandanerukulla@gmail.com"
+__description__ = "AI-Powered Development Assistant"
+__url__ = "https://github.com/Raghu6798/Blitz_Coder"
+__license__ = "MIT"
+__keywords__ = ["ai", "code-generation", "development", "assistant", "cli"]
+
+# Import main components
+from . import cli
+
+__all__ = [
+    "cli",
+    "__version__",
+    "__author__",
+    "__description__",
+] 

blitzcoder/cli/__init__.py

@@ -0,0 +1,17 @@
+"""
+BlitzCoder CLI - AI-Powered Development Assistant
+
+A command-line interface for AI-powered code generation, refactoring, and project management.
+"""
+
+from .cli_coder import cli, run_agent_with_memory, search_memories_cli
+
+__version__ = "1.0.0"
+__author__ = "BlitzCoder Team"
+__description__ = "AI-Powered Development Assistant CLI"
+
+__all__ = [
+    "cli",
+    "run_agent_with_memory", 
+    "search_memories_cli"
+]

blitzcoder/cli/cli_coder.py

@@ -0,0 +1,268 @@
+import os
+import uuid
+import click
+from rich.console import Console
+from rich.panel import Panel
+from rich.prompt import Prompt
+from rich.markdown import Markdown
+from rich.progress import Progress, SpinnerColumn, BarColumn, TimeElapsedColumn
+
+from langgraph.graph import StateGraph, START, END, MessagesState
+from langgraph.store.memory import InMemoryStore
+from langgraph.checkpoint.memory import InMemorySaver
+from langgraph.prebuilt import ToolNode, tools_condition
+from langgraph.store.base import BaseStore
+
+from langchain_google_genai import ChatGoogleGenerativeAI
+from langchain_core.runnables import RunnableConfig
+from langchain_core.messages import AIMessage, HumanMessage, SystemMessage
+
+# --- Console ---
+console = Console()
+
+# --- Your tools ---
+from main.graphapi import (
+    print_welcome_banner, show_info, run_agent_with_memory, search_memories,
+    run_uvicorn_and_capture_logs, current_directory, change_directory,
+    navigate_entire_codebase_given_path, extract_content_within_a_file,
+    refactoring_code, explain_code, execute_python_code, error_detection,
+    agent_refactor_code, run_shell_commands, generate_project_structure,
+    generate_architecture_plan, generate_folder_creation_script,
+    generate_file_content, scaffold_and_generate_files,
+    look_for_file_or_directory, create_or_delete_file, write_code_to_file,
+    inspect_a_file, look_for_directory, windows_powershell_command,
+    semantic_memory_store, SYSTEM_PROMPT
+)
+
+class AgentState(MessagesState):
+    documents: list[str]
+
+tools = [
+    run_uvicorn_and_capture_logs,
+    current_directory,
+    change_directory,
+    navigate_entire_codebase_given_path,
+    extract_content_within_a_file,
+    refactoring_code,
+    explain_code,
+    execute_python_code,
+    error_detection,
+    agent_refactor_code,
+    run_shell_commands,
+    generate_project_structure,
+    generate_architecture_plan,
+    generate_folder_creation_script,
+    generate_file_content,
+    scaffold_and_generate_files,
+    look_for_file_or_directory,
+    create_or_delete_file,
+    write_code_to_file,
+    inspect_a_file,
+    look_for_directory,
+    windows_powershell_command
+]
+
+def validate_google_api_key(api_key: str) -> bool:
+    try:
+        model = ChatGoogleGenerativeAI(model="gemini-2.0-flash", api_key=api_key)
+        response = model.invoke(["Hello"])  # Simple call
+        return True
+    except Exception as e:
+        print(f"[bold red]API key validation failed:[/bold red] {e}")
+        return False
+
+def print_agent_response(text: str, title: str = "BlitzCoder"):
+    """Display agent output inside a Rich panel box with orange color"""
+    console.print(Panel.fit(Markdown(text), title=f"[bold orange1]{title}[/bold orange1]", border_style="orange1"))
+
+def update_memory(state: AgentState, config: RunnableConfig, *, store: BaseStore):
+    user_id = config["configurable"].get("user_id", "default")
+    namespace = (user_id, "memories")
+    messages = state["messages"]
+
+    if len(messages) >= 2:
+        user_msg = messages[-2] if isinstance(messages[-2], HumanMessage) else None
+        ai_msg = messages[-1] if isinstance(messages[-1], AIMessage) else None
+
+        if user_msg and ai_msg:
+            memory_id = str(uuid.uuid4())
+            memory_content = {
+                "memory": f"User asked: {user_msg.content} | Assistant responded: {ai_msg.content[:200]}...",
+                "context": "conversation",
+                "user_query": user_msg.content,
+                "ai_response": ai_msg.content,
+                "timestamp": str(uuid.uuid4()),  # Replace with actual timestamp if needed
+            }
+
+            store.put(
+                namespace,
+                memory_id,
+                memory_content,
+                index=["memory", "context", "user_query"],
+            )
+
+    return state
+
+
+
+
+def retrieve_and_enhance_context(state: AgentState, config: RunnableConfig, *, store: BaseStore):
+    user_id = config["configurable"].get("user_id", "default")
+    namespace = (user_id, "memories")
+    latest_message = state["messages"][-1]
+
+    if isinstance(latest_message, HumanMessage):
+        query = latest_message.content
+        memories = store.search(namespace, query=query, limit=5)
+        if memories:
+            memory_context = [
+                f"Previous context: {memory.value.get('memory', '')}" for memory in memories
+            ]
+            context_info = "\n".join(memory_context)
+            enhanced_query = f"""Based on our previous conversations:
+{context_info}
+
+Current question: {query}
+
+Please respond considering our conversation history and any relevant context from previous interactions."""
+
+            enhanced_messages = state["messages"][:-1] + [HumanMessage(content=enhanced_query)]
+            return {"messages": enhanced_messages}
+
+    return state
+
+def enhanced_tool_calling_llm(state: AgentState, config: RunnableConfig, *, store: BaseStore):
+    api_key = os.environ.get("GOOGLE_API_KEY")
+    if not api_key:
+        raise RuntimeError("GOOGLE_API_KEY is not set. Please provide it.")
+
+    gemini_model = ChatGoogleGenerativeAI(
+        model="gemini-2.0-flash",
+        api_key=api_key,
+        max_tokens=100000
+    )
+
+    enhanced_state = retrieve_and_enhance_context(state, config, store=store)
+    messages = [
+        msg for msg in enhanced_state["messages"]
+        if not (hasattr(msg, "role") and getattr(msg, "role", None) == "system")
+    ]
+    messages = [SystemMessage(content=SYSTEM_PROMPT)] + messages
+    llm_with_tool = gemini_model.bind_tools(tools)
+    response = llm_with_tool.invoke(messages)
+    return {"messages": [response]}
+
+builder = StateGraph(AgentState)
+builder.add_node("enhanced_llm", enhanced_tool_calling_llm)
+builder.add_node("update_memory", update_memory)
+builder.add_node("tools", ToolNode(tools))
+builder.add_edge(START, "enhanced_llm")
+builder.add_conditional_edges("enhanced_llm", tools_condition, {"tools": "tools", "__end__": "update_memory"})
+builder.add_edge("tools", "enhanced_llm")
+builder.add_edge("update_memory", END)
+
+checkpointer = InMemorySaver()
+semantic_graph = builder.compile(checkpointer=checkpointer, store=semantic_memory_store)
+
+def run_agent_with_memory(query: str, user_id: str = "default", thread_id: str = None):
+    if thread_id is None:
+        thread_id = str(uuid.uuid4())
+
+    config = {
+        "configurable": {"thread_id": thread_id, "user_id": user_id},
+        "recursion_limit": 100
+    }
+
+    output_buffer = ""
+
+    with Progress(
+        SpinnerColumn(),
+        "[progress.description]{task.description}",
+        BarColumn(),
+        TimeElapsedColumn(),
+        console=console,
+        transient=True
+    ) as progress:
+        task = progress.add_task("[cyan]Initializing AI agent...", total=None)
+        progress.update(task, description="[yellow]Retrieving relevant memories...")
+
+        for chunk, metadata in semantic_graph.stream(
+            {"messages": [HumanMessage(content=query)]},
+            config=config,
+            stream_mode="messages",
+        ):
+            if hasattr(chunk, "content") and chunk.content:
+                if not output_buffer:
+                    progress.update(task, description="[green]Receiving AI response...")
+                output_buffer += chunk.content
+
+        progress.stop()
+
+    if output_buffer.strip():
+        print_agent_response(output_buffer.strip())
+
+@click.group()
+def cli():
+    """BlitzCoder CLI - AI-Powered Dev Assistant"""
+    pass
+
+@cli.command()
+@click.option('--google-api-key', help='Google API key for Gemini model')
+def chat(google_api_key):
+    """Start interactive chat with BlitzCoder AI agent."""
+    if google_api_key:
+        if validate_google_api_key(google_api_key):
+            os.environ["GOOGLE_API_KEY"] = google_api_key
+        else:
+            console.print("[bold red]❌ Invalid API key. Please try again.[/bold red]")
+            exit(1)
+    else:
+        google_api_key = Prompt.ask("🔑 [bold green]Paste your API key[/bold green]", password=True)
+        if validate_google_api_key(google_api_key):
+            os.environ["GOOGLE_API_KEY"] = google_api_key
+        else:
+            console.print("[bold red]❌ Invalid API key. Please try again.[/bold red]")
+            exit(1)
+
+    print_welcome_banner()
+    user_id = str(uuid.uuid4())
+    thread_id = str(uuid.uuid4())
+
+    while True:
+        query = Prompt.ask("[bold orange1]Enter your query[/bold orange1]", console=console)
+        if query.lower() in {"bye", "exit"}:
+            show_info("Exiting interactive agent loop.")
+            break
+        if query.startswith("search:"):
+            search_query = query[7:].strip()
+            search_memories(user_id, search_query)
+            continue
+        run_agent_with_memory(query, user_id, thread_id)
+
+@cli.command()
+@click.option('--user-id', default=None, help='User ID for memory search')
+@click.option('--query', prompt='Search query', help='Query to search in memories')
+@click.option('--google-api-key', help='Google API key for Gemini model')
+def search_memories_cli(user_id, query, google_api_key):
+    """Search your agent memories."""
+    if google_api_key:
+        os.environ["GOOGLE_API_KEY"] = google_api_key
+    else:
+        console.print(Panel.fit(
+            "[bold orange1]🔑 Google API Key Required[/bold orange1]\n\n"
+            "Please paste your API key below (input is hidden):",
+            border_style="red"
+        ))
+        google_api_key = Prompt.ask("🔑 [bold green]Paste your API key[/bold green]", password=True)
+        if validate_google_api_key(google_api_key):
+            os.environ["GOOGLE_API_KEY"] = google_api_key
+        else:
+            console.print("[bold red]❌ Invalid API key. Please try again.[/bold red]")
+            exit(1)
+
+    if not user_id:
+        user_id = str(uuid.uuid4())
+    search_memories(user_id, query)
+
+if __name__ == "__main__":
+    cli()

blitzcoder-1.0.0.dist-info/METADATA

@@ -0,0 +1,228 @@
+Metadata-Version: 2.4
+Name: blitzcoder
+Version: 1.0.0
+Summary: AI-powered development assistant for code generation, refactoring, and project management
+Home-page: https://github.com/Raghu6798/Blitz_Coder
+Author: BlitzCoder Team
+Author-email: BlitzCoder Team <raghunandanerukulla@gmail.com>
+Maintainer-email: Raghu Nandan Erukulla <raghunandanerukulla@gmail.com>
+License: MIT
+Project-URL: Homepage, https://github.com/Raghu6798/BlitzCoder
+Project-URL: Repository, https://github.com/Raghu6798/BlitzCoder
+Project-URL: Documentation, https://github.com/Raghu6798/BlitzCoder#readme
+Project-URL: Bug Tracker, https://github.com/Raghu6798/BlitzCoder/issues
+Keywords: ai,code-generation,development,assistant,cli
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Software Development :: Libraries :: Python Modules
+Classifier: Topic :: Software Development :: Code Generators
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: python-dotenv>=0.0.1
+Requires-Dist: langgraph>=0.4.9
+Requires-Dist: langchain-groq>=0.3.4
+Requires-Dist: langchain-sambanova>=0.1.5
+Requires-Dist: langchain>=0.3.26
+Requires-Dist: loguru>=0.7.3
+Requires-Dist: langchain-google-genai>=2.1.5
+Requires-Dist: mem0ai>=0.1.111
+Requires-Dist: click>=8.1.8
+Requires-Dist: click-help-colors>=0.9.4
+Requires-Dist: rich>=14.0.0
+Requires-Dist: langfuse>=3.0.5
+Provides-Extra: dev
+Requires-Dist: pytest>=7.0.0; extra == "dev"
+Requires-Dist: pytest-cov>=4.0.0; extra == "dev"
+Requires-Dist: black>=22.0.0; extra == "dev"
+Requires-Dist: isort>=5.0.0; extra == "dev"
+Requires-Dist: flake8>=5.0.0; extra == "dev"
+Requires-Dist: mypy>=1.0.0; extra == "dev"
+Dynamic: author
+Dynamic: home-page
+Dynamic: license-file
+Dynamic: requires-python
+
+# BlitzCoder
+
+⚡ **AI-Powered Development Assistant** - A comprehensive CLI tool for code generation, refactoring, and project management.
+
+## Features
+
+- 🤖 **AI-Powered Code Generation** - Generate code using Google's Gemini model
+- 🔧 **Code Refactoring** - Automatically refactor and improve existing code
+- 📁 **Project Scaffolding** - Create complete project structures with architecture plans
+- 🧠 **Memory System** - Remember previous conversations and context
+- 🛠️ **Development Tools** - File inspection, execution, and management tools
+- 🔍 **Code Analysis** - Explain and analyze code functionality
+
+## Installation
+
+### Option 1: Install from Source (Recommended)
+
+```bash
+# Clone the repository
+git clone https://github.com/Raghu6798/Blitz_Coder.git
+cd BlitzCoder/blitz_cli
+
+# Install in development mode
+python install.py
+```
+
+### Option 2: Manual Installation
+
+```bash
+cd blitz_cli
+pip install -e .
+```
+
+### Option 3: Direct Script Execution
+
+```bash
+# Windows
+python scripts/blitzcoder.bat
+
+# Linux/Mac
+python scripts/blitzcoder
+```
+
+## Quick Start
+
+### 1. Set up your API Keys
+
+You'll need a Google API key for the Gemini model:
+
+```bash
+# Set environment variable
+export GOOGLE_API_KEY="your-api-key-here"
+
+# Or on Windows
+set GOOGLE_API_KEY=your-api-key-here
+```
+
+### 2. Start Interactive Chat
+
+```bash
+blitzcoder chat
+```
+
+### 3. Search Memories
+
+```bash
+blitzcoder search-memories --query "your search term"
+```
+
+## Usage Examples
+
+### Interactive Chat Mode
+
+```bash
+blitzcoder chat
+```
+
+This starts an interactive session where you can:
+- Ask questions about code
+- Request code generation
+- Get help with refactoring
+- Search through previous conversations
+
+### Search Previous Conversations
+
+```bash
+blitzcoder search-memories --query "React component"
+```
+
+### Use with API Key Parameter
+
+```bash
+blitzcoder chat --google-api-key "your-api-key"
+```
+
+## Available Commands
+
+| Command | Description |
+|---------|-------------|
+| `chat` | Start interactive AI chat session |
+| `search-memories` | Search through conversation history |
+
+## Development
+
+### Project Structure
+
+```
+blitz_cli/
+├── src/
+│   └── blitzcoder/
+│       ├── cli/
+│       │   ├── __init__.py
+│       │   └── cli_coder.py
+│       └── __init__.py
+├── config/
+│   └── templates/
+├── scripts/
+│   ├── blitzcoder
+│   └── blitzcoder.bat
+├── setup.py
+├── pyproject.toml
+└── install.py
+```
+
+### Running Tests
+
+```bash
+# Install development dependencies
+pip install -e ".[dev]"
+
+# Run tests
+pytest
+```
+
+### Code Formatting
+
+```bash
+# Format code
+black src/
+isort src/
+
+# Type checking
+mypy src/
+```
+
+## Configuration
+
+The package uses environment variables for configuration:
+
+- `GOOGLE_API_KEY` - Required for Gemini model access
+- `GROQ_API_KEY` - Optional for additional models
+- `NOVITA_API_KEY` - Optional for embeddings
+
+## Contributing
+
+1. Fork the repository
+2. Create a feature branch
+3. Make your changes
+4. Add tests if applicable
+5. Submit a pull request
+
+## License
+
+This project is licensed under the MIT License - see the LICENSE file for details.
+
+## Support
+
+- 📧 Email: raghunandanerukulla@gmail.com
+- 🐛 Issues: [GitHub Issues](https://github.com/Raghu6798/BlitzCoder/issues)
+- 📖 Documentation: [GitHub README](https://github.com/Raghu6798/BlitzCoder#readme)
+
+## Acknowledgments
+
+- Built with [LangGraph](https://github.com/langchain-ai/langgraph)
+- Powered by [Google Gemini](https://ai.google.dev/)
+- Enhanced with [Rich](https://github.com/Textualize/rich) for beautiful CLI output 

blitzcoder-1.0.0.dist-info/RECORD

@@ -0,0 +1,11 @@
+blitzcoder/__init__.py,sha256=sCFGHGooTUEgyh0kJpxvYUs4H7XZV9KggOmskqWAQV4,654
+blitzcoder/cli/__init__.py,sha256=nlr4GF2Wexad3gFuJXNVbWNMMJfwimeSkDpxfNIvM7M,435
+blitzcoder/cli/cli_coder.py,sha256=YZTTBu3Vi2X_LJHntEuRl1pAwJ7Hft9bwodCyXpAHZc,9872
+blitzcoder-1.0.0.dist-info/licenses/LICENSE,sha256=1M-_Pyfb8aMu69WmyeaGvl6tfPEBTQDB3yh6rBJ9u3o,1092
+main/__init__.py,sha256=3xDRz8e-8BxSuvhWMt7PkbK_j26KoypeTmrPkq3svE4,178
+main/graphapi.py,sha256=7QDbcMblebVs51A6vIUGmHGh7L5D4nVy3NTrQkxydLo,53276
+blitzcoder-1.0.0.dist-info/METADATA,sha256=4D8QU4E_WD80wiDlundMTpeZjmnoXb1_mNaGr8hj0jM,5994
+blitzcoder-1.0.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+blitzcoder-1.0.0.dist-info/entry_points.txt,sha256=9ut47J-ccJUi6hJoRK11atEgiQh870fPwUb7MJkyF8I,60
+blitzcoder-1.0.0.dist-info/top_level.txt,sha256=vtMqG24lFn7wtg0bPLsSRpyHjVzAKw8fqzNShubqL-I,16
+blitzcoder-1.0.0.dist-info/RECORD,,

blitzcoder-1.0.0.dist-info/WHEEL

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

blitzcoder-1.0.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+blitzcoder = blitzcoder.cli.cli_coder:cli

blitzcoder-1.0.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 BlitzCoder Team
+
+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. 

blitzcoder-1.0.0.dist-info/top_level.txt

@@ -0,0 +1,2 @@
+blitzcoder
+main

main/__init__.py

@@ -0,0 +1,9 @@
+"""
+BlitzCoder Main Module
+
+This module contains the core functionality for the BlitzCoder AI development assistant.
+"""
+
+from . import graphapi
+
+__all__ = ["graphapi"]