Skip to main content

Blazing fast local AI Agent for Python coding

Project description

Nemo Agent

PyPI - Version

Nemo Agent

Nemo Agent is a blazing fast local AI Agent for Python coding

Features

  • Runs blazing fast locally on an RTX 4070 or greater
  • Generates Python project structures automatically using poetry
  • Writes Python code based on task descriptions
  • Executes development tasks using AI-generated commands
  • Utilizes the mistral-nemo language model for intelligent code generation
  • Implements best practices in Python development automatically
  • Writes and runs passing tests up to 80%+ test coverage using pytest
  • Automatically fixes and styles code using pylint
  • Utilizes git to commit and track changes

Ability

  • Can solve some leetcode hards on some runs in about 30 seconds on an RTX 4090
  • Can apply time complexity constraints in task requirements

Demo Video

Watch on Youtube

Installation

You can install Nemo Agent using pip:

pip install nemo-agent

Usage

After installation, you can use NemoAgent from the command line:

nemo-agent "Your task description here" If you run nemo-agent without any arguments, it will prompt you to enter a task.

Examples

Create a simple calculator:

nemo-agent "Create a simple calculator"

Generate a fizzbuzz program:

nemo-agent "Create a fizzbuzz script"

Cloud-Ready

  • RunPod is the recommended hosting provider for RTX 4090 compute - setup link

Limitations

  • LLM codegen issues mean that web APIs like Flask and FastAPI are not supported.
  • Due to the LLM getting confused with paths and files - writing 1 code and 1 test file is only supported.

How It Works

Nemo Agent leverages mistral-nemo, Ollama, phidata, poetry, pytest, pylint, and other libraries to build a complete working solution for your task.

Requirements

  • Python 3.9 or higher
  • Git
  • Ollama running mistral-nemo
  • Linux with a minimum of an RTX 4070

Contributing

Contributions to Nemo Agent are welcome! Please feel free to submit a Pull Request.

License

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

Disclaimer

Nemo Agent generates code using AI. While it strives for accuracy and best practices, the generated code should be reviewed and tested before being used in a production environment.

Project details


Release history Release notifications | RSS feed

Download files

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

Source Distribution

nemo_agent-0.5.3.tar.gz (17.1 kB view hashes)

Uploaded Source

Built Distribution

nemo_agent-0.5.3-py3-none-any.whl (17.0 kB view hashes)

Uploaded Python 3

Supported by

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