fix(openai): normalize tool call name in create_stream() to match create()

python/packages/autogen-ext/src/autogen_ext/models/openai/_openai_client.py

@@ -1030,6 +1030,8 @@ async def create_stream(
         # Determine the content and thought based on what was collected
         if full_tool_calls:
             # This is a tool call response
+            for tool_call in full_tool_calls.values():
+                tool_call.name = normalize_name(tool_call.name)
             content = list(full_tool_calls.values())
             if content_deltas:
                 # Store any text alongside tool calls as thoughts

python/packages/autogen-ext/tests/models/test_openai_model_client.py

@@ -1670,6 +1670,58 @@ async def _mock_create(*args: Any, **kwargs: Any) -> ChatCompletion | AsyncGener
     assert chunks[-1].thought == "Hello Another Hello Yet Another Hello"
 
 
+@pytest.mark.asyncio
+async def test_tool_calling_with_stream_normalizes_tool_name(monkeypatch: pytest.MonkeyPatch) -> None:
+    async def _mock_create_stream(*args: Any, **kwargs: Any) -> AsyncGenerator[ChatCompletionChunk, None]:
+        model = resolve_model(kwargs.get("model", "gpt-4o"))
+        chunk = MockChunkDefinition(
+            chunk_choice=ChunkChoice(
+                finish_reason="tool_calls",
+                index=0,
+                delta=ChoiceDelta(
+                    content=None,
+                    role="assistant",
+                    tool_calls=[
+                        ChoiceDeltaToolCall(
+                            index=0,
+                            id="call_1",
+                            type="function",
+                            function=ChoiceDeltaToolCallFunction(
+                                name="functions.get_weather!",
+                                arguments=json.dumps({"city": "Seattle"}),
+                            ),
+                        )
+                    ],
+                ),
+            ),
+            usage=None,
+        )
+        yield ChatCompletionChunk(
+            id="id",
+            choices=[chunk.chunk_choice],
+            created=0,
+            model=model,
+            object="chat.completion.chunk",
+            usage=chunk.usage,
+        )
+
+    async def _mock_create(*args: Any, **kwargs: Any) -> ChatCompletion | AsyncGenerator[ChatCompletionChunk, None]:
+        return _mock_create_stream(*args, **kwargs)
+
+    monkeypatch.setattr(AsyncCompletions, "create", _mock_create)
+
+    model_client = OpenAIChatCompletionClient(model="gpt-4o", api_key="placeholder")
+    stream = model_client.create_stream(messages=[UserMessage(content="What is the weather?", source="user")])
+    chunks: List[str | CreateResult] = []
+    async for chunk in stream:
+        chunks.append(chunk)
+
+    assert isinstance(chunks[-1], CreateResult)
+    assert isinstance(chunks[-1].content, list)
+    assert len(chunks[-1].content) == 1
+    assert chunks[-1].content[0].name == "functions_get_weather_"
+
+
 @pytest.mark.asyncio
 async def test_tool_calls_assistant_message_content_field(monkeypatch: pytest.MonkeyPatch) -> None:
     """Test that AssistantMessage with tool calls includes required content field.