diff --git a/apps/utils/pr_agent/algo/__init__.py b/apps/utils/pr_agent/algo/__init__.py
index e4e7e4c..7449158 100644
--- a/apps/utils/pr_agent/algo/__init__.py
+++ b/apps/utils/pr_agent/algo/__init__.py
@@ -26,8 +26,11 @@ MAX_TOKENS = {
     'o1-preview-2024-09-12': 128000,  # 128K, but may be limited by config.max_model_tokens
     'o1-2024-12-17': 204800,  # 200K, but may be limited by config.max_model_tokens
     'o1': 204800,  # 200K, but may be limited by config.max_model_tokens
+    'o3': 204800,  # 200K, but may be limited by config.max_model_tokens
     'o3-mini': 204800,  # 200K, but may be limited by config.max_model_tokens
     'o3-mini-2025-01-31': 204800,  # 200K, but may be limited by config.max_model_tokens
+    'o4-mini': 204800,  # 200K, but may be limited by config.max_model_tokens
+    'o4-mini-high': 204800,  # 200K, but may be limited by config.max_model_tokens
     'claude-instant-1': 100000,
     'claude-2': 100000,
     'command-nightly': 4096,