more broad with o1

This commit is contained in:
Brandon Hancock
2025-02-06 14:16:09 -05:00
parent e2b460374a
commit f4310aea3b

View File

@@ -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"