add new openai models

Author: truemagic-coder

README.md

@@ -1,6 +1,10 @@
 # Nemo Agent
 
 [![PyPI - Version](https://img.shields.io/pypi/v/nemo-agent)](https://pypi.org/project/nemo-agent/)
+[![Python 3.12+](https://img.shields.io/badge/python-3.12+-blue.svg)](https://www.python.org/downloads/)
+[![PyPI - Downloads](https://img.shields.io/pypi/dm/nemo-agent)](https://pypi.org/project/nemo-agent/)
+[![License: MIT](https://img.shields.io/badge/License-MIT-green.svg)](https://opensource.org/licenses/MIT)
+[![Libraries.io dependency status for GitHub repo](https://img.shields.io/librariesio/github/truemagic-coder/nemo-agent)](https://libraries.io/pypi/nemo-agent)
 
 [![Nemo Agent](https://cdn.cometheart.com/nemo-agent-2.png)](https://cdn.cometheart.com/nemo-agent.mp4)
 
@@ -135,7 +139,8 @@ You many want to skip tests especially if you are generating a UI application.
 * Supports any 128k input token models
 
 #### OpenAI
-* Supports `gpt-4.1`, `gpt-4.1-mini`, `gpt-4.1-nano`, `o3-mini`, `o1-mini`, `o1`, `gpt-4o`, and `gpt-4o-mini`
+* Supports standard models: `gpt-4.1`, `gpt-4.1-mini`, `gpt-4.1-nano`, `gpt-4o`, and `gpt-4o-mini`
+* Supports reasoning models: `o4-mini`, `o3-mini`, `o1-mini`, and `o1`
 
 #### Claude
 * Supports `claude-3-7-sonnet-20250219` and `claude-3-5-sonnet-20241022`

nemo_agent/main.py

@@ -88,11 +88,11 @@ def __init__(self, model):
             self.max_tokens = 16384
             self.max_output_tokens = 16384
         else:
-            # Default to o3-mini or o1
+            # Default to other o-models
             self.max_tokens = 100000
             self.max_output_tokens = 100000
 
-        self.special_models = ["o1", "o1-preview", "o1-mini", "o3-mini"]
+        self.special_models = ["o1", "o1-mini", "o3-mini", "o4-mini"]
 
     def count_tokens(self, text):
         return len(tiktoken.encoding_for_model("gpt-4o").encode(text))

poetry.lock

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "nemo-agent"
-version = "3.8.0"
+version = "3.9.0"
 description = "Your Python AI Coder"
 authors = ["Bevan Hunt <bevan@bevanhunt.com>"]
 license = "MIT"
@@ -20,7 +20,7 @@ packages = [{include = "nemo_agent"}]
 python = ">=3.9,<4.0"
 click = "^8.1.8"
 requests = "^2.32.3"
-openai = "^1.74.0"
+openai = "^1.75.0"
 anthropic = "^0.49.0"
 tiktoken = "^0.9.0"