Skip to main content

Your Python AI Coder

Project description

Nemo Agent

PyPI - Version

Nemo Agent

Nemo Agent is your Python AI Coder!

https://github.com/user-attachments/assets/51cf6ad1-196c-44ab-99ba-0035365f1bbd

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, OpenAI, or Claude language models for intelligent code generation
  • Ability to import reference documents to guide the task implementation
  • Allows importing existing code projects in multiple languages to serve as a reference for the task
  • Enables the importation of csv data files to populate databases or graphs
  • 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
  • Shows the token count used for the responses

Community

Coding Ability

  • leetcode hards
  • fastapi or flask APIs
  • flask web apps
  • streamlit apps
  • tkinter apps
  • jupyter notebook
  • Note: OpenAI > Claude > mistral-nemo for most coding projects
  • Note: Not all runs will be successful with all models

Install

OpenAI or Claude Install

Requirements

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

Requirements Installation

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

OR

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

Usage

Providers

  • ollama: nemo-agent --provider ollama
  • openai: nemo-agent --provider openai
  • claude: nemo-agent --provider claude

Import Reference Documentation Into Prompt

  • Documentation files must be either: .md (Markdown) or .txt (Text) and be located in a folder
  • nemo-agent --docs example_folder

Import Existing Code Projects Into Prompt

  • Code files must be either: .py (Python), .php (PHP), .rs (Rust), .js (JavaScript), .ts (TypeScript), .toml (TOML), .json (JSON), .rb (Ruby), or .yaml (YAML) and be located in a folder
  • nemo-agent --code example_folder

Import Data Into Prompt

  • Data files must be .csv (CSV) and be located in a folder
  • nemo-agent --data example_folder

Prompting

CLI

  • nemo-agent "create a fizzbuzz script"

OR

File Prompt

  • Prompt file must be markdown (.md) or text files (.txt)
  • nemo-agent --file example.md or
  • nemo-agent --file example.txt

Run Generated Program

  • cd generated_project_folder
  • source .venv/bin/activate
  • python main.py

Default Models

  • ollama is mistral-nemo (default model)
  • openai is gpt-4o
  • claude is claude-3-5-sonnet-20241022

Select Models

  • nemo-agent "my_prompt" --provider ollama --model nemotron

OpenAI o1 Support

  • Supports o1-mini and o1-preview
  • nemo-agent "my prompt" --provider openai --model o1-mini

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.

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-2.6.0.tar.gz (14.1 kB view details)

Uploaded Source

Built Distribution

nemo_agent-2.6.0-py3-none-any.whl (13.4 kB view details)

Uploaded Python 3

File details

Details for the file nemo_agent-2.6.0.tar.gz.

File metadata

  • Download URL: nemo_agent-2.6.0.tar.gz
  • Upload date:
  • Size: 14.1 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/5.1.1 CPython/3.9.20

File hashes

Hashes for nemo_agent-2.6.0.tar.gz
Algorithm Hash digest
SHA256 e8fb558f53784b69fb9c55397d3ae0a23a9d36dcbc9c5cd0c2732bc2be55c11b
MD5 56fff719e722d2d153f34e958930696d
BLAKE2b-256 a908714bdda53c335903be0d13e3f7b825ccd19ce6029dde05718a563d55ed27

See more details on using hashes here.

File details

Details for the file nemo_agent-2.6.0-py3-none-any.whl.

File metadata

  • Download URL: nemo_agent-2.6.0-py3-none-any.whl
  • Upload date:
  • Size: 13.4 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/5.1.1 CPython/3.9.20

File hashes

Hashes for nemo_agent-2.6.0-py3-none-any.whl
Algorithm Hash digest
SHA256 9fda1a0f05f108af27cf6289a415609ebf9c7d85f690468da6e2834fce8685a9
MD5 a65ec05f48637f7d6d71de940b0f6eb3
BLAKE2b-256 a2110db16ec2d926383cf41c8cc6d42b071eb32505e4cd642409eeeae9601086

See more details on using hashes here.

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