Use Enum for ResponseFormatType

GooeyAI · Sep 13, 2024 · bee12c4 · bee12c4
1 parent 83cacc6
commit bee12c4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/daras_ai_v2/language_model.py b/daras_ai_v2/language_model.py
@@ -459,7 +459,9 @@ def get_entry_text(entry: ConversationEntry) -> str:
     )
 
 
-ResponseFormatType = typing.Literal["text", "json_object"]
+class ResponseFormatType(str, GooeyEnum):
+    text = "text"
+    json_object = "json_object"
 
 
 def run_language_model(