[TLM] support o1 preview

cleanlab · Oct 10, 2024 · 76582ec · 76582ec
1 parent 5c6543c
commit 76582ec
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cleanlab_studio/studio/trustworthy_language_model.py b/cleanlab_studio/studio/trustworthy_language_model.py
@@ -753,7 +753,7 @@ class TLMOptions(TypedDict):
 
     Args:
         model (str, default = "gpt-4o-mini"): underlying LLM to use (better models will yield better results).
-        Models currently supported include "gpt-4o-mini", "gpt-3.5-turbo-16k", "gpt-4", "gpt-4o", "claude-3-haiku".
+        Models currently supported include "gpt-4o-mini", "gpt-3.5-turbo-16k", "gpt-4", "gpt-4o", "o1-preview", "claude-3-haiku".
 
         max_tokens (int, default = 512): the maximum number of tokens to generate in the TLM response.
         This number will impact the maximum number of tokens you will see in the output response, and also the number of tokens