Update model.py

explosion · Oct 24, 2023 · 63c7183 · 63c7183
1 parent 8363b22
commit 63c7183
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/spacy_llm/models/bedrock/model.py b/spacy_llm/models/bedrock/model.py
@@ -2,8 +2,6 @@
 import json
 import warnings
 from enum import Enum
-import requests
-from requests import HTTPError
 from typing import Any, Dict, Iterable, Optional, Type, List, Sized, Tuple
 
 from confection import SimpleFrozenDict
@@ -81,7 +79,7 @@ def __call__(self, prompts: Iterable[str]) -> Iterable[str]:
         prompts = list(prompts)
         api_config = Config(retries = dict(max_attempts = self._max_retries))
 
-        def _request(json_data: Dict[str, Any]) -> str:
+        def _request(json_data: str) -> str:
             session = self.get_session()
             print("Session:", session)
             bedrock = session.client(service_name="bedrock-runtime", config=api_config)