update

Author: truemagic-coder

README.md

@@ -62,13 +62,14 @@ https://github.com/user-attachments/assets/51cf6ad1-196c-44ab-99ba-0035365f1bbd
 
 #### Requirements
 * Python 3.9 or higher
-* Ollama running `qwen2.5-coder:32b`
-* Linux with minimum specs of Ubuntu 24.04 with RTX 4090
+* Ollama running `qwen2.5-coder:14b`
+* Linux with minimum spec of Ubuntu 24.04 with RTX 4070 or;
+* Mac with minimum spec of Mac Mini M2 Pro with 16MB 
 
 #### Requirements Installation
 * Ollama install instructions:
     * `curl -fsSL https://ollama.com/install.sh | sh`
-    * `ollama pull qwen2.5-coder:32b`
+    * `ollama pull qwen2.5-coder:14b`
 * `pip install nemo-agent`
 * You are ready to use `nemo-agent`
 
@@ -110,7 +111,7 @@ https://github.com/user-attachments/assets/51cf6ad1-196c-44ab-99ba-0035365f1bbd
 * `python main.py`
 
 ## Default Models 
-* `ollama` is `qwen2.5-coder:32b` (default model)
+* `ollama` is `qwen2.5-coder:14b` (default model)
 * `openai` is `gpt-4o`
 * `claude` is `claude-3-5-sonnet-20241022`
 * `gemini` is `gemini-1.5-pro`

nemo_agent/main.py

@@ -69,7 +69,7 @@ def generate(self, prompt):
 
 class OpenAIAPI:
     def __init__(self, model):
-        if model == "qwen2.5-coder:32b":
+        if model == "qwen2.5-coder:14b":
             model = "gpt-4o"
         self.model = model
         self.api_key = os.getenv("OPENAI_API_KEY")
@@ -155,7 +155,7 @@ def generate(self, prompt):
 
 class GeminiAPI:
     def __init__(self, model):
-        if model == "qwen2.5-coder:32b":
+        if model == "qwen2.5-coder:14b":
             model="gemini-1.5-pro"
         self.model = model
         self.api_key = os.getenv("GEMINI_API_KEY")
@@ -223,7 +223,7 @@ def generate(self, prompt):
 
 class ClaudeAPI:
     def __init__(self, model):
-        if model == "qwen2.5-coder:32b":
+        if model == "qwen2.5-coder:14b":
             model = "claude-3-5-sonnet-20241022"
         self.model = model
         self.api_key = os.getenv("ANTHROPIC_API_KEY")
@@ -288,7 +288,7 @@ class NemoAgent:
     WRITE_RETRY_DELAY = 1  # second
 
     def __init__(
-        self, task: str, model: str = "qwen2.5-coder:32b", provider: str = "ollama"
+        self, task: str, model: str = "qwen2.5-coder:14b", provider: str = "ollama"
     ):
         self.task = task
         self.model = model
@@ -576,8 +576,7 @@ def implement_solution(self, max_attempts=3):
                 3. IMPORTANT: Do not add any code comments to the files.
                 4. IMPORTANT: Always follow PEP8 style guide, follow best practices for Python, use snake_case naming, and provide meaningful docstrings.
                 5. CRITICAL: Your response should ONLY contain the code blocks and the pip dependencies required for both the test and code files. Do not include any additional information.
-                6. CRITICAL: Create a main method to run the app in main.py and if a web app run the app on port 8080.
-
+                6. CRITICAL: Create a main method to run the app in main.py.
                 7. CRITICAL: Enclose your entire response between ^^^start^^^ and ^^^end^^^ markers.
                 8. IMPORTANT: Use the reference documentation provided to guide your implementation including the required dependencies.
                 9. IMPORTANT: Use the code content as a reference to build a working solution based on the task provided by the user in Python.
@@ -998,7 +997,7 @@ def main
     type=click.Path(exists=True),
     help="Path to a markdown file containing the task",
 )
-@click.option("--model", default="qwen2.5-coder:32b", help="The model to use for the LLM")
+@click.option("--model", default="qwen2.5-coder:14b", help="The model to use for the LLM")
 @click.option(
     "--provider",
     default="ollama",
@@ -1026,7 +1025,7 @@ def main
 def cli(
     task: str = None,
     file: str = None,
-    model: str = "qwen2.5-coder:32b",
+    model: str = "qwen2.5-coder:14b",
     provider: str = "ollama",
     zip: str = None,
     docs: str = None,

poetry.lock

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "nemo-agent"
-version = "3.0.4"
+version = "3.1.0"
 description = "Your Python AI Coder"
 authors = ["Bevan Hunt <bevan@bevanhunt.com>"]
 license = "MIT"
@@ -20,8 +20,8 @@ packages = [{include = "nemo_agent"}]
 python = "<4.0,>=3.9"
 click = "^8.1.8"
 requests = "^2.32.3"
-openai = "^1.59.6"
-anthropic = "^0.42.0"
+openai = "^1.59.9"
+anthropic = "^0.43.1"
 tiktoken = "^0.8.0"