update

Author: truemagic-coder

README.md

@@ -119,9 +119,9 @@ https://github.com/user-attachments/assets/51cf6ad1-196c-44ab-99ba-0035365f1bbd
 ## Select Models
 * `nemo-agent "my_prompt" --provider ollama --model nemotron`
 
-## OpenAI o1 Support
-* Supports `o1-mini`, `o1-preview`, and `o1`
-* `nemo-agent "my prompt" --provider openai --model o1-mini`
+## OpenAI o-series Support
+* Supports `o3-mini`, `o1-mini`, `o1-preview`, and `o1`
+* `nemo-agent "my prompt" --provider openai --model o3-mini`
 
 ## Contributing
 Contributions to Nemo Agent are welcome! Please feel free to submit a Pull Request.

nemo_agent/main.py

@@ -79,7 +79,7 @@ def __init__(self, model):
         self.token_count = 0
         self.max_tokens = 128000
         self.max_output_tokens = 16384
-        self.special_models = ["o1-preview", "o1-mini"]
+        self.special_models = ["o1-preview", "o1-mini", "o3-mini"]
 
     def count_tokens(self, text):
         return len(tiktoken.encoding_for_model("gpt-4o").encode(text))
@@ -96,6 +96,7 @@ def generate(self, prompt):
             # Use the predefined max output tokens, or adjust if prompt is very long
             max_completion_tokens = min(self.max_output_tokens, self.max_tokens - prompt_tokens)
 
+            # o1 doesn't support streaming
             if self.model == "o1":
                 response = self.openai.chat.completions.create(
                     model=self.model,

poetry.lock

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "nemo-agent"
-version = "3.1.0"
+version = "3.2.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.9"
-anthropic = "^0.43.1"
+openai = "^1.61.0"
+anthropic = "^0.45.2"
 tiktoken = "^0.8.0"