mirror of
https://github.com/BerriAI/litellm.git
synced 2026-10-02 02:11:58 +00:00
fix: address typing and streaming test expectations
This commit is contained in:
parent
d2d096f88a
commit
bc4953b0b9
2 changed files with 8 additions and 5 deletions
|
|
@ -1979,9 +1979,12 @@ class AmazonConverseConfig(BaseConfig):
|
|||
message.content = None
|
||||
returned_finish_reason = "tool_calls"
|
||||
except Exception:
|
||||
tool_call = self._text_content_tool_call_transformation(
|
||||
message.content, tools
|
||||
)
|
||||
if message.content is not None:
|
||||
tool_call = self._text_content_tool_call_transformation(
|
||||
message.content, tools
|
||||
)
|
||||
else:
|
||||
tool_call = None
|
||||
if tool_call is not None:
|
||||
message.tool_calls = [tool_call]
|
||||
message.content = None
|
||||
|
|
@ -2280,7 +2283,7 @@ class AmazonConverseConfig(BaseConfig):
|
|||
|
||||
text_tool_call_tools = optional_params.get("tools")
|
||||
if text_tool_call_tools is None:
|
||||
request_data = data
|
||||
request_data: Optional[Union[dict, str]] = data
|
||||
if isinstance(data, str):
|
||||
try:
|
||||
request_data = json.loads(data)
|
||||
|
|
|
|||
|
|
@ -1707,7 +1707,7 @@ class AWSEventStreamDecoder:
|
|||
if "trace" in chunk_data:
|
||||
trace = chunk_data.get("trace")
|
||||
model_response_provider_specific_fields["trace"] = trace
|
||||
delta_content = text
|
||||
delta_content: Optional[str] = text
|
||||
if tool_use is not None and delta_content == "":
|
||||
delta_content = None
|
||||
elif (
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue