chore: add missing modes

phil65 · Nov 21, 2024 · eb7a001 · eb7a001
1 parent 77c1c77
commit eb7a001
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/llmling/core/capabilities.py b/src/llmling/core/capabilities.py
@@ -39,7 +39,13 @@
     "user",
 ]
 
-Mode = Literal["chat", "completion"]
+Mode = Literal[
+    "completion",
+    "embedding",
+    "image_generation",
+    "chat",
+    "audio_transcription",
+]
 
 Provider = Literal[
     "openai",