Python: [BREAKING] Simplify API: ChatAgent -> Agent, ChatMessage -> Message (#3747)

* [BREAKING] Rename ChatAgent -> Agent, ChatMessage -> Message, ChatClientProtocol -> SupportsChatGetResponse

Simplify the public API by removing redundant 'Chat' prefix from core types:
- ChatAgent -> Agent
- RawChatAgent -> RawAgent
- ChatMessage -> Message
- ChatClientProtocol -> SupportsChatGetResponse

Also renamed internal WorkflowMessage (was Message in _runner_context) to avoid collision.

No backward compatibility aliases - this is a clean breaking change.

* [BREAKING] Rename Agent chat_client parameter to client

* Fix rebase issues: WorkflowMessage references and broken markdown links

* Fix formatting and lint issues from code quality checks

* Fix import ordering in workflow sample files

* fixed rebase

* Fix test failures: use WorkflowMessage and A2AMessage after ChatMessage→Message rename

- Replace Message(data=..., source_id=...) with WorkflowMessage(...) in workflow tests
- Fix isinstance check in A2A agent to use A2AMessage instead of Message
- Fix import in test_workflow_observability.py (Message→WorkflowMessage)

* Fix lint, fmt, and sample errors after ChatMessage→Message rename

- Auto-fix 70+ ruff lint issues across samples (ChatMessage→Message refs)
- Fix HostedVectorStoreContent→Content.from_hosted_vector_store in file search sample
- Fix _normalize_messages→normalize_messages in custom agent sample
- Fix context.terminate→raise MiddlewareTermination in middleware samples
- Fix with_update_hook→with_transform_hook in override middleware sample
- Add TOptions_co import back to custom_chat_client sample
- Add noqa for FastAPI File() default in chatkit sample
- Fix B023 loop variable capture in weather agent sample

* fix: update Agent constructor calls from chat_client to client in declaration-only tool tests

* fix: add register_cleanup to devui lazy-loading proxy and type stub

* fixed tests and updated new pieces

* fix agui typevar

* fix merge errors

* fix merge conflicts

* fiux merge

* Remove unused links

---------

Co-authored-by: Evan Mattson <evan.mattson@microsoft.com>

python/packages/mem0/agent_framework_mem0/_context_provider.py

@@ -13,7 +13,7 @@ import sys
 from contextlib import AbstractAsyncContextManager
 from typing import TYPE_CHECKING, Any
 
-from agent_framework import ChatMessage
+from agent_framework import Message
 from agent_framework._sessions import AgentSession, BaseContextProvider, SessionContext
 from agent_framework.exceptions import ServiceInitializationError
 from mem0 import AsyncMemory, AsyncMemoryClient
@@ -131,7 +131,7 @@ class _Mem0ContextProvider(BaseContextProvider):
         if line_separated_memories:
             context.extend_messages(
                 self.source_id,
-                [ChatMessage(role="user", text=f"{self.context_prompt}\n{line_separated_memories}")],
+                [Message(role="user", text=f"{self.context_prompt}\n{line_separated_memories}")],
             )
 
     async def after_run(
@@ -145,7 +145,7 @@ class _Mem0ContextProvider(BaseContextProvider):
         """Store request/response messages to Mem0 for future retrieval."""
         self._validate_filters()
 
-        messages_to_store: list[ChatMessage] = list(context.input_messages)
+        messages_to_store: list[Message] = list(context.input_messages)
         if context.response and context.response.messages:
             messages_to_store.extend(context.response.messages)
 

python/packages/mem0/agent_framework_mem0/_provider.py

@@ -7,7 +7,7 @@ from collections.abc import MutableSequence, Sequence
 from contextlib import AbstractAsyncContextManager
 from typing import Any
 
-from agent_framework import ChatMessage, Context, ContextProvider
+from agent_framework import Context, ContextProvider, Message
 from agent_framework.exceptions import ServiceInitializationError
 from mem0 import AsyncMemory, AsyncMemoryClient
 
@@ -103,19 +103,17 @@ class Mem0Provider(ContextProvider):
     @override
     async def invoked(
         self,
-        request_messages: ChatMessage | Sequence[ChatMessage],
-        response_messages: ChatMessage | Sequence[ChatMessage] | None = None,
+        request_messages: Message | Sequence[Message],
+        response_messages: Message | Sequence[Message] | None = None,
         invoke_exception: Exception | None = None,
         **kwargs: Any,
     ) -> None:
         self._validate_filters()
 
-        request_messages_list = (
-            [request_messages] if isinstance(request_messages, ChatMessage) else list(request_messages)
-        )
+        request_messages_list = [request_messages] if isinstance(request_messages, Message) else list(request_messages)
         response_messages_list = (
             [response_messages]
-            if isinstance(response_messages, ChatMessage)
+            if isinstance(response_messages, Message)
             else list(response_messages)
             if response_messages
             else []
@@ -142,7 +140,7 @@ class Mem0Provider(ContextProvider):
             )
 
     @override
-    async def invoking(self, messages: ChatMessage | MutableSequence[ChatMessage], **kwargs: Any) -> Context:
+    async def invoking(self, messages: Message | MutableSequence[Message], **kwargs: Any) -> Context:
         """Called before invoking the AI model to provide context.
 
         Args:
@@ -155,7 +153,7 @@ class Mem0Provider(ContextProvider):
             Context: Context object containing instructions with memories.
         """
         self._validate_filters()
-        messages_list = [messages] if isinstance(messages, ChatMessage) else list(messages)
+        messages_list = [messages] if isinstance(messages, Message) else list(messages)
         input_text = "\n".join(msg.text for msg in messages_list if msg and msg.text and msg.text.strip())
 
         # Validate input text is not empty before searching (possible for function approval responses)
@@ -182,7 +180,7 @@ class Mem0Provider(ContextProvider):
         line_separated_memories = "\n".join(memory.get("memory", "") for memory in memories)
 
         return Context(
-            messages=[ChatMessage(role="user", text=f"{self.context_prompt}\n{line_separated_memories}")]
+            messages=[Message(role="user", text=f"{self.context_prompt}\n{line_separated_memories}")]
             if line_separated_memories
             else None
         )