nemo-agent 1.1.0

Blazing fast local AI Agent for Python coding

Nemo Agent

Nemo Agent is an 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

Coding 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

Local Install

Requirements

• Python 3.9 or higher
• git
• curl
• Ollama running mistral-nemo
• Ubuntu (22.04 or 24.04) with a minimum of an RTX 4070

Requirements Installation

• Python, curl, git are pre-installed on Ubuntu
• Ollama install instructions:
  • curl -fsSL https://ollama.com/install.sh | sh
  • ollama pull mistral-nemo
• nemo-agent install:
  • pip install nemo-agent
• You are ready to use nemo-agent

Cloud Install

Requirements

• RunPod account setup with your SSH and billing information

RunPod Setup

• Make sure you have setup your SSH keys
• Select a 4090 pod
• Select the RunPod Pytorch 2.1.1 template
• Edit the template:
  • Set Container Disk to 60 GB
  • Set Expose HTTP Ports to 8888, 11434
  • Add environment variables with OLLAMA_HOST key and 0.0.0.0 value
• Deploy your pod
• After deploying then login via SSH
• Run on the pod: (curl -fsSL https://ollama.com/install.sh | sh && ollama serve > ollama.log 2>&1) & and then press CTRL+C to exit
• Run on the pod: ollama pull mistral-nemo
• Run on the pod: pip install nemo-agent
• You are ready to use nemo-agent

Usage

After installation, you can use Nemo Agent 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"

Limitations

• Currently Nemo Agent can't build web apps as it gets confused - this is a WIP

Roadmap

• Refactor legacy Python code projects to nemo-agent code
• Convert legacy non-Python projects to nemo-agent code
• Give nemo-agent up-to-date knowledge about PyPi libraries
• Give nemo-agent up-to-date knowledge of the latest Python language features
• Give nemo-agent up-to-date knowledge about other language libraries
• Enable nemo-agent to write Py03 extensions

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.