From 64246658b0109fe1bf1f5daa6ea6ae17009ff096 Mon Sep 17 00:00:00 2001 From: Kenneth Reitz Date: Sat, 2 Nov 2024 16:50:09 -0400 Subject: [PATCH] Add streaming support to Amazon provider --- simplemind/providers/amazon.py | 1 + 1 file changed, 1 insertion(+) diff --git a/simplemind/providers/amazon.py b/simplemind/providers/amazon.py index c7c7363..afb6888 100644 --- a/simplemind/providers/amazon.py +++ b/simplemind/providers/amazon.py @@ -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