Skip to main content

A CLI coding super-framework using AI

Project description

Dravid (DRD) - AI-Powered CLI Coding Framework

PyPI Build Status Coverage Status License

Dravid (DRD) is an advanced, AI-powered CLI coding framework (in alpha) designed to follow user instructions until the job is done, even if it means fixing errors, including installation issues. It can generate code and fix errors autonomously until the intended result is achieved.

Security and Sandbox (important note)

  • Always try in a new directory for a fresh project.
  • For existing projects, create a separate git branch or a sandbox environment. Monitor the generated commands. Git add or commit when you get results.
  • Your file content will be sent to the CLAUDE API LLM for response. Do not include sensitive files in the project.
  • Don't use hardcoded API_KEYS. Use .env and ensure it's part of .gitignore so the tool can skip reading it.

Quick preview:

  1. As shown in the video, when initializing a project where system dependencies don't exist, Dravid will attempt to fix them one by one, even if those fixes result in their own errors

https://github.com/user-attachments/assets/07784a9e-8de6-4161-9e83-8cad1fa04ae6

  1. If you have a dev server with import or reference errors, requiring dependency installation or fixes, Dravid will monitor your dev or test server and autofix. This is particularly useful for existing projects where you want to fix tests or refactor the entire project.

https://github.com/user-attachments/assets/14350e4d-6cec-4922-997f-f34e9f716189

You can also initialize Dravid in your existing project. See the Usage section for more details.

Features

  • AI-powered CLI for efficient coding and project management
  • Support for multiple AI providers: Claude (default), OpenAI, Azure OpenAI, Llama 3, Mixtral (through Anyscale, Groq, or NVIDIA), and local LLM models through Ollama
  • Image query handling capabilities
  • Robust file operations and metadata management
  • Integration with external APIs (Dravid API)
  • Built-in development server with file monitoring
  • Comprehensive error handling and reporting
  • Extensible architecture for easy feature additions

Installation

Prerequisites

  • Python 3.7+
  • pip (Python package installer)
  • API key for your chosen AI provider (environment variable should be set)

To install Dravid, run the following command:

pip install dravid

To upgrade for latest fixes:

pip install --upgrade dravid

NOTE:

Always create a fresh directory before trying to create a new project.

Usage

After installation, you can use the drd command directly from your terminal. Here are some common usage examples:

NOTE: for better results, go step by step and communicate clearly. You can also define project_guidelines.txt which will be referenced in the main query, you can use this to instruct on how the code should be generated etc.

Also, any png or jpg files that will be generated and needs to be replaced will have placeholder prefix, so you know that it has to be replaced.

Turbo dev mode

You can run the development server with automatic error fixing.

This command will start your dev server or test server and then continually fix any errors and then restart, you can sitback and sip coffee :) You can give instructions into this mode, it will generate code, fix any errors, restart. You can also provide image for reference, or any number of files for references.

drd "rails server"
drd "npm run test:watch"

The bootstrap mode

  • When you want to create a fresh project
drd --do "create a nextjs project"
  • When you to quickly fix or change something in a project without starting a dev or test server.
drd --do "make the walls permeable in the snake_game file"

With larger text (heredoc)

When you have larger string or if you want to copy paste a error stack with double quotes etc, please use this.

drd --do <<EOF
Fix this error:
....
EOF

Ask Questions or Generate Content

Ask questions or generate content:

drd --ask "how is the weather"

Generate a file directly:

drd --ask "create a MIT LICENSE file, just the file, don't respond with anything else" >> LICENSE

--ask is much faster than the execute command because it doesn't load project context or project guidelines (you can create your own project_guidelines.txt)

Image-based Queries

Use image references in your queries:

drd --do "make the home image similar to the image" --image "~/Downloads/reference.png"

AI Models and Providers

Dravid supports multiple AI providers and models to suit your needs. Here's an overview of the available options:

Default: Claude (Anthropic)

By default, Dravid uses Claude 3.5 Sonnet from Anthropic. Claude is known for its strong performance across a wide range of tasks, including coding, analysis, and creative writing.

To use Claude, set the following environment variable:

CLAUDE_API_KEY=your_claude_api_key_here

OpenAI

Dravid also supports OpenAI's models. When using OpenAI, the default model is gpt-4o.

To use OpenAI, set the following environment variables:

DRAVID_LLM=openai
OPENAI_API_KEY=your_openai_api_key_here

You can control the specific OpenAI model used by changing the OPENAI_MODEL environment variable. For example:

OPENAI_MODEL=gpt-4 # for standard GPT-4
OPENAI_MODEL=gpt-3.5-turbo # for GPT-3.5 Turbo

Other Providers

Depending on your chosen AI provider, you'll need to set different environment variables:

Azure OpenAI

DRAVID_LLM=azure
AZURE_OPENAI_API_KEY=your_azure_api_key_here
AZURE_OPENAI_API_VERSION=your_api_version_here
AZURE_OPENAI_ENDPOINT=your_azure_endpoint_here
AZURE_OPENAI_DEPLOYMENT_NAME=your_deployment_name_here

Custom Provider compatible with OpenAI spec(e.g., Anyscale, Togeether, Groq, NVIDIA, or your own)

DRAVID_LLM=custom
DRAVID_LLM_API_KEY=your_custom_api_key_here
DRAVID_LLM_ENDPOINT=your_custom_endpoint_here
DRAVID_LLM_MODEL=your_preferred_model_here

Ollama (for local LLM models)

DRAVID_LLM=ollama
DRAVID_LLM_MODEL=your_preferred_local_model_here

Project Structure

  • src/drd/: Main source code directory
    • cli/: Command-line interface modules
    • api/: API interaction and parsing modules
    • utils/: Utility functions and helpers
    • prompts/: AI prompt templates
    • metadata/: Project metadata management
  • tests/: Test suite for the project

Contributing

We welcome contributions to Dravid! Please see our Contributing Guide for more details on how to get started.

Development

To install Dravid, you need Python 3.7+ and Poetry. Follow these steps:

  1. Clone the repository:

    git clone https://github.com/vysakh0/dravid.git
    cd dravid
    
  2. Install dependencies using Poetry:

    poetry install
    
  3. Set up environment variables: Create a .env file in the project root and add your API keys:

    CLAUDE_API_KEY=your_claude_api_key_here
    
  4. You can use Dravid to add features or functionalities to the project. As this project uses drd.json and has used Dravid to build Dravid.

poetry run drd "refactor api_utils"

or

poetry run drd "add tests for utils/utils"
poetry run drd --ask "who are you"

Video examples

https://github.com/user-attachments/assets/2bcd2969-2746-4115-a879-18b8333a3053

https://github.com/user-attachments/assets/15112577-0d45-44be-b564-74bee548ac66

https://github.com/user-attachments/assets/25b82c1f-e357-405b-9b85-2488a2d2b771

Testing

After adding some functionalities, if you want to test how it works, I suggest creating a directory called myapp or testapp or test-app in the root of this project. These folder names are already in .gitignore.

cd myapp
poetry run drd "create a simple elixir project"

To run the test suite:

poetry run pytest

Errors or Exception

Installation

  • Make sure you are installing with python 3.8 and above, please check your python version and pip version if you have both python2 and python3 in your system.
  • If you get "SyntaxError: multiple exception types must be parenthesized" after installation and when you do drd --version it means the pygments library (dependency of openai) needs upgrade
pip install --upgrade pygments

License

This project is licensed under the MIT License - see the LICENSE file for details.

Acknowledgments

  • Special thanks to the creators of the Claude AI model, which powers many of Dravid's capabilities

Contact

For questions, suggestions, or issues, please open an issue on the GitHub repository or contact the maintainers directly.

Happy coding with Dravid!

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

dravid-0.13.5.tar.gz (40.2 kB view details)

Uploaded Source

Built Distribution

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

dravid-0.13.5-py3-none-any.whl (54.1 kB view details)

Uploaded Python 3

File details

Details for the file dravid-0.13.5.tar.gz.

File metadata

  • Download URL: dravid-0.13.5.tar.gz
  • Upload date:
  • Size: 40.2 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: poetry/1.8.3 CPython/3.12.0 Darwin/23.5.0

File hashes

Hashes for dravid-0.13.5.tar.gz
Algorithm Hash digest
SHA256 c75873091d97eb88872c3c0a85851b289929914563ed6be4d31d5609b324f604
MD5 c8dc88a3241ae690ad692b5fc6acd51f
BLAKE2b-256 a9cbac4d98a7c64deea84d15edb3cc863f9ef680816cd01d57d9ec937b69864a

See more details on using hashes here.

File details

Details for the file dravid-0.13.5-py3-none-any.whl.

File metadata

  • Download URL: dravid-0.13.5-py3-none-any.whl
  • Upload date:
  • Size: 54.1 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: poetry/1.8.3 CPython/3.12.0 Darwin/23.5.0

File hashes

Hashes for dravid-0.13.5-py3-none-any.whl
Algorithm Hash digest
SHA256 6b80b0436f42d9870355ee24d0552c689f837534ca66d777b6450524a1d655dd
MD5 428d80d22c94ac4675b6448987aecb34
BLAKE2b-256 4c290c41e31466c986fa44b2ced528dc4249e9a6a27fb9af7a0997417d403340

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