This commit is contained in:
Mason Daugherty 2025-08-04 13:14:04 -04:00
parent 719c9dfaaa
commit d4ac8ff5f7
No known key found for this signature in database
2 changed files with 4 additions and 7 deletions

View File

@ -90,7 +90,7 @@ def _generate_response_from_error(error: BaseException) -> list[AIMessageV1]:
if hasattr(response, "status_code"): if hasattr(response, "status_code"):
metadata["status_code"] = response.status_code metadata["status_code"] = response.status_code
if hasattr(error, "request_id"): if hasattr(error, "request_id"):
metadata["request_id"] = error.request_id # type: ignore[arg-type] metadata["request_id"] = error.request_id
# Permit response_metadata without model_name, model_provider fields # Permit response_metadata without model_name, model_provider fields
generations = [AIMessageV1(content=[], response_metadata=metadata)] # type: ignore[arg-type] generations = [AIMessageV1(content=[], response_metadata=metadata)] # type: ignore[arg-type]
else: else:

View File

@ -385,15 +385,12 @@ def _init_tool_calls(content: list[types.ContentBlock]) -> list[types.ContentBlo
new_content.append(block) new_content.append(block)
continue continue
try: try:
args_ = ( args_str = block.get("args")
parse_partial_json(cast("str", block.get("args") or "")) args_ = parse_partial_json(args_str) if args_str else {}
if block.get("args")
else {}
)
if isinstance(args_, dict): if isinstance(args_, dict):
new_content.append( new_content.append(
create_tool_call( create_tool_call(
name=cast("str", block.get("name") or ""), name=cast("str", block.get("name")),
args=args_, args=args_,
id=cast("str", block.get("id", "")), id=cast("str", block.get("id", "")),
) )