Python: [Breaking] Simplified Content types to a single class with classmethod constructors. (#3252)

* ported Content to a new model

* fixed linting

* fixes

* fixed data format handling

* fix for 3.10 mypy

* fix

* fix int test

python/packages/core/tests/azure/test_azure_assistants_client.py

@@ -18,7 +18,6 @@ from agent_framework import (
     ChatResponse,
     ChatResponseUpdate,
     HostedCodeInterpreterTool,
-    TextContent,
 )
 from agent_framework.azure import AzureOpenAIAssistantsClient
 from agent_framework.exceptions import ServiceInitializationError
@@ -332,7 +331,7 @@ async def test_azure_assistants_client_streaming() -> None:
             assert chunk is not None
             assert isinstance(chunk, ChatResponseUpdate)
             for content in chunk.contents:
-                if isinstance(content, TextContent) and content.text:
+                if content.type == "text" and content.text:
                     full_message += content.text
 
         assert any(word in full_message.lower() for word in ["sunny", "25", "weather", "seattle"])
@@ -358,7 +357,7 @@ async def test_azure_assistants_client_streaming_tools() -> None:
             assert chunk is not None
             assert isinstance(chunk, ChatResponseUpdate)
             for content in chunk.contents:
-                if isinstance(content, TextContent) and content.text:
+                if content.type == "text" and content.text:
                     full_message += content.text
 
         assert any(word in full_message.lower() for word in ["sunny", "25", "weather"])

python/packages/core/tests/azure/test_azure_chat_client.py

@@ -25,7 +25,6 @@ from agent_framework import (
     ChatMessage,
     ChatResponse,
     ChatResponseUpdate,
-    TextContent,
     ai_function,
 )
 from agent_framework._telemetry import USER_AGENT_KEY
@@ -304,9 +303,9 @@ async def test_azure_on_your_data(
     )
     assert len(content.messages) == 1
     assert len(content.messages[0].contents) == 1
-    assert isinstance(content.messages[0].contents[0], TextContent)
+    assert content.messages[0].contents[0].type == "text"
     assert len(content.messages[0].contents[0].annotations) == 1
-    assert content.messages[0].contents[0].annotations[0].title == "test title"
+    assert content.messages[0].contents[0].annotations[0]["title"] == "test title"
     assert content.messages[0].contents[0].text == "test"
 
     mock_create.assert_awaited_once_with(
@@ -374,9 +373,9 @@ async def test_azure_on_your_data_string(
     )
     assert len(content.messages) == 1
     assert len(content.messages[0].contents) == 1
-    assert isinstance(content.messages[0].contents[0], TextContent)
+    assert content.messages[0].contents[0].type == "text"
     assert len(content.messages[0].contents[0].annotations) == 1
-    assert content.messages[0].contents[0].annotations[0].title == "test title"
+    assert content.messages[0].contents[0].annotations[0]["title"] == "test title"
     assert content.messages[0].contents[0].text == "test"
 
     mock_create.assert_awaited_once_with(
@@ -433,7 +432,7 @@ async def test_azure_on_your_data_fail(
     )
     assert len(content.messages) == 1
     assert len(content.messages[0].contents) == 1
-    assert isinstance(content.messages[0].contents[0], TextContent)
+    assert content.messages[0].contents[0].type == "text"
     assert content.messages[0].contents[0].text == "test"
 
     mock_create.assert_awaited_once_with(
@@ -628,9 +627,7 @@ async def test_streaming_with_none_delta(
         results.append(msg)
 
     assert len(results) > 0
-    assert any(
-        isinstance(content, TextContent) and content.text == "test" for msg in results for content in msg.contents
-    )
+    assert any(content.type == "text" and content.text == "test" for msg in results for content in msg.contents)
     assert any(msg.contents for msg in results)
 
 
@@ -731,7 +728,7 @@ async def test_azure_openai_chat_client_streaming() -> None:
         assert chunk.message_id is not None
         assert chunk.response_id is not None
         for content in chunk.contents:
-            if isinstance(content, TextContent) and content.text:
+            if content.type == "text" and content.text:
                 full_message += content.text
 
     assert "Emily" in full_message or "David" in full_message
@@ -757,7 +754,7 @@ async def test_azure_openai_chat_client_streaming_tools() -> None:
         assert chunk is not None
         assert isinstance(chunk, ChatResponseUpdate)
         for content in chunk.contents:
-            if isinstance(content, TextContent) and content.text:
+            if content.type == "text" and content.text:
                 full_message += content.text
 
     assert "Emily" in full_message or "David" in full_message

python/packages/core/tests/azure/test_azure_responses_client.py

@@ -15,10 +15,10 @@ from agent_framework import (
     ChatClientProtocol,
     ChatMessage,
     ChatResponse,
+    Content,
     HostedCodeInterpreterTool,
     HostedFileSearchTool,
     HostedMCPTool,
-    HostedVectorStoreContent,
     HostedWebSearchTool,
     ai_function,
 )
@@ -48,7 +48,7 @@ async def get_weather(location: Annotated[str, "The location as a city name"]) -
     return f"The weather in {location} is sunny and 72°F."
 
 
-async def create_vector_store(client: AzureOpenAIResponsesClient) -> tuple[str, HostedVectorStoreContent]:
+async def create_vector_store(client: AzureOpenAIResponsesClient) -> tuple[str, Content]:
     """Create a vector store with sample documents for testing."""
     file = await client.client.files.create(
         file=("todays_weather.txt", b"The weather today is sunny with a high of 75F."), purpose="assistants"
@@ -61,7 +61,7 @@ async def create_vector_store(client: AzureOpenAIResponsesClient) -> tuple[str,
     if result.last_error is not None:
         raise Exception(f"Vector store file processing failed with status: {result.last_error.message}")
 
-    return file.id, HostedVectorStoreContent(vector_store_id=vector_store.id)
+    return file.id, Content.from_hosted_vector_store(vector_store_id=vector_store.id)
 
 
 async def delete_vector_store(client: AzureOpenAIResponsesClient, file_id: str, vector_store_id: str) -> None: