From f4310aea3b3c56d3dd6aec2e0b6ba538f901d294 Mon Sep 17 00:00:00 2001
From: Brandon Hancock <brandon@brandonhancock.io>
Date: Thu, 6 Feb 2025 14:16:09 -0500
Subject: [PATCH] more broad with o1

---
 src/crewai/llm.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/crewai/llm.py b/src/crewai/llm.py
index 0f8f62a8d..be6f3426b 100644
--- a/src/crewai/llm.py
+++ b/src/crewai/llm.py
@@ -218,7 +218,7 @@ class LLM:
 
         # For O1 models, system messages are not supported.
         # Convert any system messages into assistant messages.
-        if self.model.lower() in {"o1-preview", "o1-mini"}:
+        if "o1" in self.model.lower():
             for message in messages:
                 if message.get("role") == "system":
                     message["role"] = "assistant"