chore(provider): add document capability

SigmaNight · Jan 26, 2025 · 95c3a50 · 95c3a50
1 parent 2066b96
commit 95c3a50
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/basilisk/provider_capability.py b/basilisk/provider_capability.py
@@ -2,6 +2,7 @@
 
 
 class ProviderCapability(Enum):
+	DOCUMENT = "document"
 	IMAGE = "image"
 	TEXT = "text"
 	STT = "stt"

diff --git a/basilisk/provider_engine/anthropic_engine.py b/basilisk/provider_engine/anthropic_engine.py
@@ -33,6 +33,7 @@ class AnthropicEngine(BaseEngine):
 	capabilities: set[ProviderCapability] = {
 		ProviderCapability.TEXT,
 		ProviderCapability.IMAGE,
+		ProviderCapability.DOCUMENT,
 	}
 	supported_attachment_formats: set[str] = {
 		"image/gif",