tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-07-11 03:04:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(bedrock_httpx.py): compatibility fix

Browse Source
This commit is contained in:
Krrish Dholakia
2024-05-11 19:55:38 -07:00
parent f6c84f1aa6
commit 65d0be85fc
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/llms/bedrock_httpx.py
+2 -2
View File
@@ -270,7 +270,7 @@ class BedrockLLM(BaseLLM):
def process_response(
self,
model: str,
response: requests.Response | httpx.Response,
response: Union[requests.Response, httpx.Response],
model_response: ModelResponse,
stream: bool,
logging_obj: Logging,
@@ -720,7 +720,7 @@ class AWSEventStreamDecoder:
)
yield streaming_chunk
def _parse_message_from_event(self, event) -> str | None:
def _parse_message_from_event(self, event) -> Optional[str]:
response_dict = event.to_response_dict()
parsed_response = self.parser.parse(response_dict, get_response_stream_shape())
if response_dict["status_code"] != 200: