Nemo Agent

Nemo Agent is your Python AI Coder!

nemo-demo.mp4

Features

• Runs blazing fast
• Generates Python project structures automatically using uv
• Writes Python code based on task descriptions
• Executes development tasks using AI-generated commands
• Utilizes the mistral-nemo or OpenAI language models for intelligent code generation
• Implements best practices in Python development automatically
• Writes and runs passing tests using pytest up to 80%+ test coverage
• Automatically fixes and styles code using pylint up to 7+/10
• Calculates and improves the complexity score using complexipy to be under 15
• Auto-formats the code with autopep8

Community

• Join our community - Nemo Agent Telegram Group

Coding Ability

• leetcode hards (app works - tests pass)
• fastapi or flask APIs (app works - tests pass)
• flask web apps (app works - tests pass)
• streamlit apps (app works - tests fail)
• tkinter apps (app works - tests fail)
• Note: OpenAI succeeds more often the mistral-nemo in their runs
• Note: Not all runs will be successful

Install

OpenAI Local Install

Requirements

• Python 3.9 or higher
• OpenAI API KEY
• Mac or Linux

Requirements Installation

• Install OpenAI API KEY for zsh shell
  • echo 'export OPENAI_API_KEY="YOUR_API_KEY"' >> ~/.zshrc
• pip install nemo-agent
• You are ready to use nemo-agent

OR

Mistral-Nemo Local Install

Requirements

• Python 3.9 or higher
• Ollama running mistral-nemo
• Linux with minimum specs of Ubuntu 24.04 with RTX 4070

Requirements Installation

• Ollama install instructions:
  • curl -fsSL https://ollama.com/install.sh | sh
  • ollama pull mistral-nemo
• pip install nemo-agent
• You are ready to use nemo-agent

OR

Mistral-Nemo 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

Prompt

• mistral-nemo: nemo-agent "create a fizzbuzz script"
• openai: nemo-agent "create a fizzbuzz script" --provider openai

Markdown File

• mistral-nemo: nemo-agent --file example.md
• openai: nemo-agent --file example.md --provider openai

Model overrides

• You can pass the --model flag to override the default model for the provider.
• The default model for openai is gpt-4o-2024-08-06
• The default model for ollama is mistral-nemo

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 an LLM. Every run is different as the LLM generated code is different. While it strives for accuracy and best practices, the generated code should be reviewed and tested before being used in a production environment.