Merge pull request #332 from cleanlab/tlm_support_o1_preview

[TLM] support o1 preview

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