repgen-ai 0.1.0__tar.gz

repgen_ai-0.1.0/MANIFEST.in

@@ -0,0 +1,3 @@
+include requirements.txt
+include README.md
+recursive-include repgen *

repgen_ai-0.1.0/PKG-INFO

@@ -0,0 +1,199 @@
+Metadata-Version: 2.4
+Name: repgen-ai
+Version: 0.1.0
+Summary: Automated reproduction generation for bug reports using LLMs
+Home-page: https://github.com/mehilshah/RepGen
+Author: Mehil B. Shah
+Author-email: Mehil Shah <shahmehil@dal.ca>
+Project-URL: Homepage, https://github.com/mehilshah/RepGen
+Project-URL: Bug Tracker, https://github.com/mehilshah/RepGen/issues
+Classifier: Programming Language :: Python :: 3
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+Requires-Dist: annoy
+Requires-Dist: numpy
+Requires-Dist: pylint
+Requires-Dist: rank_bm25
+Requires-Dist: requests
+Requires-Dist: scikit_learn
+Requires-Dist: sentence_transformers
+Requires-Dist: torch
+Requires-Dist: transformers
+Requires-Dist: pandas
+Requires-Dist: openai
+Requires-Dist: rich
+Requires-Dist: mkdocs
+Requires-Dist: mkdocs-material
+Requires-Dist: watchfiles
+Requires-Dist: black
+Requires-Dist: isort
+Requires-Dist: mypy
+Requires-Dist: flake8
+Requires-Dist: pre-commit
+Requires-Dist: rich-argparse
+Dynamic: author
+Dynamic: home-page
+Dynamic: requires-python
+
+# RepGen - Automated Bug Reproduction
+
+<div align="center">
+
+<p align="center">
+  <img src="https://img.shields.io/badge/Status-Active%20Research-4CAF50?style=flat-square" />
+  <img src="https://img.shields.io/badge/Python-3.12-3776AB?style=flat-square&logo=python&logoColor=white" />
+ <img src="https://img.shields.io/github/actions/workflow/status/mehilshah/RepGen/ci.yml?branch=main&label=Build&style=flat-square&logo=github" />
+  <img src="https://img.shields.io/badge/Execution-Dockerized-2496ED?style=flat-square&logo=docker&logoColor=white" />
+  <img src="https://img.shields.io/badge/License-MIT-lightgrey?style=flat-square" />
+</p>
+
+[Features](#features) • [Quick Start](#quick-start) • [VS Code](#vs-code-extension) • [Documentation](#documentation) • [Paper](https://arxiv.org/abs/2512.14990)
+
+</div>
+
+
+## Overview
+
+**RepGen** is a production-grade tool that leverages state-of-the-art Large Language Models (LLMs) to automatically reproduce bugs in software libraries. By analyzing bug reports and repository context, RepGen plans a reproduction strategy and generates executable Python scripts to replicate the issue.
+
+Ideally suited for Deep Learning libraries, RepGen automates the tedious first step of debugging: creating a Minimum Reproducible Example (MRE).
+
+## Features
+
+- **Smart Retrieval**: Uses hybrid search (BM25 + Semantic) to find relevant code snippets and training loops for context.
+- **Multi-Backend Support**: Seamlessly switch between LLM providers:
+  - **Ollama**: Run locally with models like `qwen2.5-coder`, `llama3`, or `mistral`.
+  - **OpenAI**: Utilize `gpt-4o` and `gpt-3.5-turbo` for high-precision generation.
+  - **Gemini** & **Claude**: leverage multimodal and reasoning capabilities.
+- **Remote Input Handling**:
+  - Direct support for **GitHub Issue URLs** (fetches content via API).
+  - Direct support for **Git Repository URLs** (clones automatically to temp workspace).
+- **Professional VS Code Extension**: A fully integrated sidebar to run reproductions directly in your editor.
+- **Docker Ready**: Full containerization for easy deployment.
+
+---
+
+## Quick Start
+
+The easiest way to run RepGen is using **Docker**. This ensures a consistent environment with all dependencies pre-installed.
+
+```bash
+# 1. Build the image
+cd RepGen
+docker build -t repgen .
+
+# 2. Run the server (mounting the volume for development if needed)
+docker run -p 8000:8000 repgen
+```
+
+The API will be available at `http://localhost:8000`.
+
+---
+
+## Developer Setup
+
+If you prefer to run locally or contribute to the core logic:
+
+```bash
+# 1. Create a virtual environment
+python3 -m venv venv
+source venv/bin/activate  # On Windows: .\venv\Scripts\activate
+
+# 2. Install RepGen in editable mode
+pip install -e .
+
+# 3. Run the CLI
+repgen
+```
+
+---
+
+## VS Code Extension
+
+RepGen includes a premium VS Code extension for a seamless workflow.
+
+### Features
+- **Persistent Sidebar**: Stays active while you work.
+- **Direct Code Display**: Generated reproduction scripts appear directly in the sidebar.
+- **"Open in Editor"**: One-click to open generated code in a new editor tab for review.
+- **Status Tracking**:  Real-time progress indicators.
+
+### Installation
+1. Navigate to `extensions/vscode`.
+2. Run `npm install` && `npm run compile`.
+3. Open the folder in VS Code and press `F5` to launch the Extension Development Host.
+
+---
+
+## Ecosystem
+
+### REST API
+The core engine runs as a FastAPI service.
+- **Start**: `uvicorn repgen.server:app --reload --reload-dir repgen`
+- **Docs**: `http://localhost:8000/docs`
+
+### Web Dashboard
+A modern, React-based UI to manage reproduction tasks visually.
+- **Location**: `ui/`
+- **Start**: `cd ui && npm install && npm run dev`
+
+### Browser Extension
+Injects a "Reproduce" button directly into GitHub Issues.
+- **Location**: `extensions/chrome/`
+
+---
+
+## Usage Examples
+
+### CLI Automation
+```bash
+repgen \
+  --bug-report https://github.com/owner/repo/issues/123 \
+  --repo-path https://github.com/owner/repo.git \
+  --backend openai \
+  --model gpt-4o
+```
+
+### Configuration
+Set API keys via environment variables:
+```bash
+export OPENAI_API_KEY="sk-..."
+export GEMINI_API_KEY="AIza..."
+```
+
+---
+
+## Contributing
+
+Contributions are welcome! Please check out the issues tab or submit a PR.
+
+## Issues
+
+If you encounter any issues, please open a GitHub issue or contact Mehil Shah at [shahmehil@dal.ca](mailto:shahmehil@dal.ca).
+
+## License
+
+MIT © [Mehil Shah](https://github.com/mehilshah)
+
+
+## Future Work and Research Directions
+
+1. **Scalable Execution via Cluster Schedulers**
+   Integrate RepGen with workload managers such as **SLURM** to offload reproduction tasks to HPC or GPU clusters. This would enable asynchronous execution, queue-based scheduling, and automated user notifications once reproduction artifacts are ready.
+
+2. **Automated Verification in Isolated Sandboxes**
+   Extend RepGen with sandboxed execution environments that automatically validate whether a generated reproduction script successfully triggers the reported bug. This would close the loop between generation and confirmation.
+
+3. **Bug-Type–Aware Verification Strategies**
+   Develop specialized verification mechanisms tailored to different bug classes (e.g., crashes, numerical instability, performance regressions, nondeterministic failures). Each class may require distinct success criteria and instrumentation.
+
+4. **Fine-Grained Bug Localization and Understanding**
+   Move beyond reproduction toward **bug comprehension**, including identifying the most likely fault-inducing components, APIs, or configuration parameters involved in the failure.
+
+5. **Understanding Practitioner Adoption Barriers**
+   Read papers about why practitioners underutilize automated debugging tools, and insights from these papers can guide usability improvements and feature prioritization.
+
+6. **CI/CD and GitHub Actions Integration**
+   Integrate RepGen directly into **GitHub Actions** and other CI pipelines, enabling automated bug reproduction as part of issue triage, regression testing, or pull request validation workflows.

repgen_ai-0.1.0/README.md

@@ -0,0 +1,160 @@
+# RepGen - Automated Bug Reproduction
+
+<div align="center">
+
+<p align="center">
+  <img src="https://img.shields.io/badge/Status-Active%20Research-4CAF50?style=flat-square" />
+  <img src="https://img.shields.io/badge/Python-3.12-3776AB?style=flat-square&logo=python&logoColor=white" />
+ <img src="https://img.shields.io/github/actions/workflow/status/mehilshah/RepGen/ci.yml?branch=main&label=Build&style=flat-square&logo=github" />
+  <img src="https://img.shields.io/badge/Execution-Dockerized-2496ED?style=flat-square&logo=docker&logoColor=white" />
+  <img src="https://img.shields.io/badge/License-MIT-lightgrey?style=flat-square" />
+</p>
+
+[Features](#features) • [Quick Start](#quick-start) • [VS Code](#vs-code-extension) • [Documentation](#documentation) • [Paper](https://arxiv.org/abs/2512.14990)
+
+</div>
+
+
+## Overview
+
+**RepGen** is a production-grade tool that leverages state-of-the-art Large Language Models (LLMs) to automatically reproduce bugs in software libraries. By analyzing bug reports and repository context, RepGen plans a reproduction strategy and generates executable Python scripts to replicate the issue.
+
+Ideally suited for Deep Learning libraries, RepGen automates the tedious first step of debugging: creating a Minimum Reproducible Example (MRE).
+
+## Features
+
+- **Smart Retrieval**: Uses hybrid search (BM25 + Semantic) to find relevant code snippets and training loops for context.
+- **Multi-Backend Support**: Seamlessly switch between LLM providers:
+  - **Ollama**: Run locally with models like `qwen2.5-coder`, `llama3`, or `mistral`.
+  - **OpenAI**: Utilize `gpt-4o` and `gpt-3.5-turbo` for high-precision generation.
+  - **Gemini** & **Claude**: leverage multimodal and reasoning capabilities.
+- **Remote Input Handling**:
+  - Direct support for **GitHub Issue URLs** (fetches content via API).
+  - Direct support for **Git Repository URLs** (clones automatically to temp workspace).
+- **Professional VS Code Extension**: A fully integrated sidebar to run reproductions directly in your editor.
+- **Docker Ready**: Full containerization for easy deployment.
+
+---
+
+## Quick Start
+
+The easiest way to run RepGen is using **Docker**. This ensures a consistent environment with all dependencies pre-installed.
+
+```bash
+# 1. Build the image
+cd RepGen
+docker build -t repgen .
+
+# 2. Run the server (mounting the volume for development if needed)
+docker run -p 8000:8000 repgen
+```
+
+The API will be available at `http://localhost:8000`.
+
+---
+
+## Developer Setup
+
+If you prefer to run locally or contribute to the core logic:
+
+```bash
+# 1. Create a virtual environment
+python3 -m venv venv
+source venv/bin/activate  # On Windows: .\venv\Scripts\activate
+
+# 2. Install RepGen in editable mode
+pip install -e .
+
+# 3. Run the CLI
+repgen
+```
+
+---
+
+## VS Code Extension
+
+RepGen includes a premium VS Code extension for a seamless workflow.
+
+### Features
+- **Persistent Sidebar**: Stays active while you work.
+- **Direct Code Display**: Generated reproduction scripts appear directly in the sidebar.
+- **"Open in Editor"**: One-click to open generated code in a new editor tab for review.
+- **Status Tracking**:  Real-time progress indicators.
+
+### Installation
+1. Navigate to `extensions/vscode`.
+2. Run `npm install` && `npm run compile`.
+3. Open the folder in VS Code and press `F5` to launch the Extension Development Host.
+
+---
+
+## Ecosystem
+
+### REST API
+The core engine runs as a FastAPI service.
+- **Start**: `uvicorn repgen.server:app --reload --reload-dir repgen`
+- **Docs**: `http://localhost:8000/docs`
+
+### Web Dashboard
+A modern, React-based UI to manage reproduction tasks visually.
+- **Location**: `ui/`
+- **Start**: `cd ui && npm install && npm run dev`
+
+### Browser Extension
+Injects a "Reproduce" button directly into GitHub Issues.
+- **Location**: `extensions/chrome/`
+
+---
+
+## Usage Examples
+
+### CLI Automation
+```bash
+repgen \
+  --bug-report https://github.com/owner/repo/issues/123 \
+  --repo-path https://github.com/owner/repo.git \
+  --backend openai \
+  --model gpt-4o
+```
+
+### Configuration
+Set API keys via environment variables:
+```bash
+export OPENAI_API_KEY="sk-..."
+export GEMINI_API_KEY="AIza..."
+```
+
+---
+
+## Contributing
+
+Contributions are welcome! Please check out the issues tab or submit a PR.
+
+## Issues
+
+If you encounter any issues, please open a GitHub issue or contact Mehil Shah at [shahmehil@dal.ca](mailto:shahmehil@dal.ca).
+
+## License
+
+MIT © [Mehil Shah](https://github.com/mehilshah)
+
+
+## Future Work and Research Directions
+
+1. **Scalable Execution via Cluster Schedulers**
+   Integrate RepGen with workload managers such as **SLURM** to offload reproduction tasks to HPC or GPU clusters. This would enable asynchronous execution, queue-based scheduling, and automated user notifications once reproduction artifacts are ready.
+
+2. **Automated Verification in Isolated Sandboxes**
+   Extend RepGen with sandboxed execution environments that automatically validate whether a generated reproduction script successfully triggers the reported bug. This would close the loop between generation and confirmation.
+
+3. **Bug-Type–Aware Verification Strategies**
+   Develop specialized verification mechanisms tailored to different bug classes (e.g., crashes, numerical instability, performance regressions, nondeterministic failures). Each class may require distinct success criteria and instrumentation.
+
+4. **Fine-Grained Bug Localization and Understanding**
+   Move beyond reproduction toward **bug comprehension**, including identifying the most likely fault-inducing components, APIs, or configuration parameters involved in the failure.
+
+5. **Understanding Practitioner Adoption Barriers**
+   Read papers about why practitioners underutilize automated debugging tools, and insights from these papers can guide usability improvements and feature prioritization.
+
+6. **CI/CD and GitHub Actions Integration**
+   Integrate RepGen directly into **GitHub Actions** and other CI pipelines, enabling automated bug reproduction as part of issue triage, regression testing, or pull request validation workflows.

repgen_ai-0.1.0/pyproject.toml

@@ -0,0 +1,45 @@
+[build-system]
+requires = ["setuptools>=61.0"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "repgen-ai"
+version = "0.1.0"
+description = "Automated reproduction generation for bug reports using LLMs"
+readme = "README.md"
+authors = [
+  { name = "Mehil Shah", email = "shahmehil@dal.ca" },
+]
+classifiers = [
+    "Programming Language :: Python :: 3",
+    "License :: OSI Approved :: MIT License",
+    "Operating System :: OS Independent",
+]
+requires-python = ">=3.8"
+dependencies = [
+    "annoy",
+    "numpy",
+    "pylint",
+    "rank_bm25",
+    "requests",
+    "scikit_learn",
+    "sentence_transformers",
+    "torch",
+    "transformers",
+    "pandas",
+    "openai",
+    "rich",
+    "mkdocs",
+    "mkdocs-material",
+    "watchfiles",
+    "black",
+    "isort",
+    "mypy",
+    "flake8",
+    "pre-commit",
+    "rich-argparse",
+]
+
+[project.urls]
+"Homepage" = "https://github.com/mehilshah/RepGen"
+"Bug Tracker" = "https://github.com/mehilshah/RepGen/issues"

repgen_ai-0.1.0/repgen/__init__.py

@@ -0,0 +1,51 @@
+from typing import Optional
+
+from .core import RepGenService
+
+
+def reproduce(
+    bug_report_source: str,
+    repo_source: str,
+    backend: str = "openai",
+    model: str = "gpt-4o",
+    api_key: Optional[str] = None,
+    commit: Optional[str] = None,
+    output_dir: str = "./repgen_results",
+) -> str:
+    """
+    Generates a reproduction script for a given bug report and repository.
+
+    Args:
+        bug_report_source: URL or path to the bug report.
+        repo_source: URL or path to the repository.
+        backend: The LLM backend to use (default: "openai").
+        model: The model name to use (default: "gpt-4o").
+        api_key: API key for the backend (optional).
+        commit: Specific commit hash to checkout (optional).
+        output_dir: Directory to store intermediate artifacts (default: "./repgen_results").
+
+    Returns:
+        The generated reproduction script as a string.
+
+    Raises:
+        RuntimeError: If reproduction fails.
+    """
+    service = RepGenService(output_dir=output_dir)
+    result = service.run_reproduction(
+        bug_report_source=bug_report_source,
+        repo_source=repo_source,
+        backend=backend,
+        model=model,
+        commit=commit,
+        api_key=api_key,
+    )
+
+    if result["success"] and result["files"]:
+        # Return the content of the first generated file (usually the reproduction script)
+        return result["files"][0]["content"]
+    else:
+        error_msg = result.get("error", "Unknown error during reproduction")
+        raise RuntimeError(f"Reproduction failed: {error_msg}")
+
+
+__all__ = ["reproduce", "RepGenService"]