support o3 and 2.5 pro

Author: truemagic-coder

README.md

@@ -55,8 +55,8 @@
 * `gemini`: `uvx nemo-agent --provider gemini`
 
 ### Supported Models 
-* `openai` is `gpt-4.1-mini` (default) and `o4-mini`
-* `gemini` is `gemini-2.5-flash-preview-04-17` (default)
+* `openai` is `gpt-4.1-mini` (default), `o4-mini`, and `o3` (requires organization verification)
+* `gemini` is `gemini-2.5-pro-preview-03-25`(default) and `gemini-2.5-flash-preview-04-17`
 
 ### Calling Non-Default Models
 * `uvx nemo-agent --model o4-mini`

nemo_agent/main.py

@@ -26,11 +26,11 @@ def __init__(self, model):
         if model == "gpt-4.1-mini":
             self.max_tokens = 32768
             self.max_output_tokens = 32768
-        elif model == "o4-mini":
+        elif model == "o4-mini" or model == "o3":
             self.max_tokens = 100000
             self.max_output_tokens = 100000
 
-        self.special_models = ["o4-mini"]
+        self.special_models = ["o4-mini", "o3"]
 
     def count_tokens(self, text):
         return len(tiktoken.encoding_for_model("gpt-4o").encode(text))
@@ -99,7 +99,7 @@ def generate(self, prompt):
 class GeminiAPI:
     def __init__(self, model):
         if model == "gpt-4.1-mini":
-            model="gemini-2.5-flash-preview-04-17"
+            model="gemini-2.5-pro-preview-03-25"
         self.model = model
         self.api_key = os.getenv("GEMINI_API_KEY")
         self.base_url="https://generativelanguage.googleapis.com/v1beta/openai/"

poetry.lock

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "nemo-agent"
-version = "4.0.1"
+version = "4.1.0"
 description = "Your Python AI Coder"
 authors = ["Bevan Hunt <bevan@bevanhunt.com>"]
 license = "MIT"
@@ -19,7 +19,7 @@ packages = [{include = "nemo_agent"}]
 [tool.poetry.dependencies]
 python = ">=3.9,<4.0"
 click = "^8.1.8"
-openai = "^1.75.0"
+openai = "^1.76.0"
 tiktoken = "^0.9.0"