# For lists like web_search_results, take the last (most complete) one
combined_provider_fields[key] = value

---

for key, value in fields.items():
    if key not in combined_provider_fields:
        combined_provider_fields[key] = value
    elif key == 'server_side_tool_invocations' and isinstance(value, list):
        # Merge by id, first-wins per key (matches _extract_server_side_tool_invocations behavior)
        existing = {e.get('id'): e for e in combined_provider_fields[key]}
        for inv in value:
            inv_id = inv.get('id', '')
            if inv_id in existing:
                for k, v in inv.items():
                    if v is not None and k not in existing[inv_id]:
                        existing[inv_id][k] = v
            else:
                existing[inv_id] = dict(inv)
        combined_provider_fields[key] = list(existing.values())
    elif key == 'thought_signatures' and isinstance(value, list):
        # Accumulate all thought signatures across chunks
        combined_provider_fields[key] = combined_provider_fields[key] + value
    elif isinstance(value, list) and isinstance(
        combined_provider_fields[key], list
    ):
        combined_provider_fields[key] = value
    else:
        combined_provider_fields[key] = value

---

import litellm
from litellm import acompletion

model = "gemini/gemini-3-flash-preview"

def simple_add(a: int, b: int) -> int:
    "Add two numbers together"
    return a + b

tool_schema = {'type': 'function', 'function': {
    'name': 'simple_add', 'description': 'Add two numbers',
    'parameters': {'type': 'object', 'properties': {
        'a': {'type': 'integer'}, 'b': {'type': 'integer'}
    }, 'required': ['a', 'b']}
}}

msgs = [{'role': 'user', 'content': "Search the web for Brisbane's population and also use simple_add to add 1+1."}]

# First turn - streaming with search + tools
chunks = []
r = await acompletion(model=model, messages=msgs, stream=True,
    tools=[tool_schema], web_search_options={"search_context_size": "low"},
    include_server_side_tool_invocations=True, stream_options={"include_usage": True})
async for chunk in r: chunks.append(chunk)

# Rebuild response from chunks
result = litellm.stream_chunk_builder(chunks)
assistant_msg = result.choices[0].message

# Add tool result
msgs.append(assistant_msg)
tc = [t for t in assistant_msg.tool_calls if not t.id.startswith('srvtoolu_')][0]
msgs.append({'role': 'tool', 'tool_call_id': tc.id, 'name': 'simple_add', 'content': '2'})

# Second turn fails
r2 = await acompletion(model=model, messages=msgs, tools=[tool_schema],
    web_search_options={"search_context_size": "low"},
    include_server_side_tool_invocations=True)
# => BadRequestError: "contents[1].parts[1]: Corrupted tool call context."

---