Skip to main content

An AI-powered tool to automatically document, refactor, and format Python code.

Project description

AutoDoc AI

PyPI version License: MIT

AutoDoc AI is a smart command-line developer tool that automates the tedious parts of maintaining high-quality Python code. It leverages Large Language Models (LLMs) to automatically generate docstrings, suggest intelligent refactorings, and integrate seamlessly into a modern development workflow.

This project was built on a foundation of solid Object-Oriented Programming principles and design patterns, including the Visitor, Strategy, Factory, and Adapter patterns.


Core Features

  • AI-Powered Docstring Generation: Automatically create high-quality, context-aware docstrings for any function or class that's missing them.
  • Intelligent Docstring Correction: Uses AI to evaluate existing docstrings and replaces poor-quality or placeholder documentation with new, improved versions.
  • AI-Powered Refactoring:
    • Safe Variable Renaming: Automatically renames poorly named local variables within function scope.
    • Intelligent Naming Suggestions: Acts as an AI-powered linter, suggesting better names for classes and functions.
  • Seamless Git Integration: Use the --diff flag to only process files that have been changed in your current Git branch.
  • Codebase-Wide Scanning: Run autodoc on a single file or an entire directory, with automatic .gitignore respect.
  • Automatic Formatting: Integrates with the black code formatter to ensure all generated and refactored code is perfectly styled.
  • Highly Configurable: Set project-wide defaults for style, AI provider, and behavior in your pyproject.toml file.

Installation

You can install AutoDoc AI directly from PyPI:

pip install autodoc-ai-paudelnirajan

First-Time Setup

Before you can use the AI features, you need to configure your API key. The easiest way is to use the built-in init command in your project's root directory.

autodoc init

This will guide you through creating or updating a .env file with your Groq API key and preferred model.

Configuration

AutoDoc uses a .env file for secrets and a pyproject.toml file for project-wide settings.

1. API Credentials (.env)

The autodoc init command will create this for you. It should look like this:

GROQ_API_KEY="gsk_YourActualGroqApiKeyHere"
GROQ_MODEL_NAME="llama3-8b-8192"

2. Project Defaults (pyproject.toml)

You can set default behaviors for your project in pyproject.toml. These settings will be used unless they are overridden by a command-line flag.

# In your pyproject.toml

[tool.autodoc]
# Settings for your AutoDoc tool
strategy = "groq"
style = "google"           # 'google', 'numpy', or 'rst'
overwrite_existing = true  # Regenerate poor-quality docstrings
refactor = true            # Enable AI-powered refactoring

Usage

AutoDoc is a flexible command-line tool with two main commands: init and run.

Get a list of commands:

autodoc --help

Get detailed help for the run command and all its flags:

autodoc run --help

Run on a specific file (dry run, prints to console):

autodoc run path/to/your/file.py

Run on an entire directory and save changes in-place:

autodoc run src/ --in-place

Run in "Git mode" to only process changed files (most common use case):

autodoc run . --diff --in-place

Enable all AI features to perform a full quality pass:

autodoc run . --in-place --overwrite-existing --refactor

How It Works

AutoDoc uses a "Surgical Hybrid" architecture that combines the strengths of traditional and modern tooling:

  1. Fast Local Analysis: It uses Python's ast (Abstract Syntax Tree) module to rapidly parse code and identify potential issues.
  2. Targeted AI Intelligence: Only when a potential issue is found does it send the small, relevant code snippet to an LLM for intelligent analysis.
  3. Precise Code Modification: The AI's response is used to perform a safe and precise modification of the AST, which is then unparsed back into valid Python code.
  4. Toolchain Integration: Finally, it uses black to ensure the final code is perfectly formatted, integrating seamlessly into the existing Python ecosystem.

License

This project is licensed under the MIT License.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

autodoc_ai_paudelnirajan-0.1.3.tar.gz (14.5 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

autodoc_ai_paudelnirajan-0.1.3-py3-none-any.whl (13.6 kB view details)

Uploaded Python 3

File details

Details for the file autodoc_ai_paudelnirajan-0.1.3.tar.gz.

File metadata

File hashes

Hashes for autodoc_ai_paudelnirajan-0.1.3.tar.gz
Algorithm Hash digest
SHA256 bce22bd4b7aeaf2a712ff0e1377a96ad76febd8df95962dd5a4107e21043c2ad
MD5 48adeb15514426f57d4d1f785fff41ee
BLAKE2b-256 3fbac38e178e1bd89a0f499397f591f9b99c35ee7fe50a40893eebb178b4963c

See more details on using hashes here.

File details

Details for the file autodoc_ai_paudelnirajan-0.1.3-py3-none-any.whl.

File metadata

File hashes

Hashes for autodoc_ai_paudelnirajan-0.1.3-py3-none-any.whl
Algorithm Hash digest
SHA256 80796dc37f662dae6cc8a11b8fa59a36d1f3636eeb89199754bfc9f9a56f5d0e
MD5 19f87295ebb7abd4cefef29d34d11e66
BLAKE2b-256 30aa1c2d89c6d4f93693c498327a8acda5d4204ff369c81301132c7d950e891c

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page