Update model.py

explosion · Oct 25, 2023 · 0a8d66e · 0a8d66e
1 parent cd342cc
commit 0a8d66e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/spacy_llm/models/bedrock/model.py b/spacy_llm/models/bedrock/model.py
@@ -69,7 +69,7 @@ def __call__(self, prompts: Iterable[str]) -> Iterable[str]:
         api_responses: List[str] = []
         prompts = list(prompts)
 
-        def _request(json_data: str) -> Dict[str, Any]:
+        def _request(json_data: str) -> str:
             try:
                 import boto3
                 import botocore