Add streaming support to Amazon provider

2026-06-05 22:50:18 +00:00 · 2024-11-02 16:50:09 -04:00
parent f0aff7814b
commit 64246658b0
1 changed files with 1 additions and 0 deletions
@@ -17,6 +17,7 @@ DEFAULT_MAX_TOKENS = 5_000
 class Amazon(BaseProvider):
    NAME = PROVIDER_NAME
    DEFAULT_MODEL = DEFAULT_MODEL
+    supports_streaming = True

    def __init__(self, profile_name: str | None = None):
        self.profile_name = profile_name or settings.AMAZON_PROFILE_NAME