microsoft-agents-a365-tooling-extensions-semantickernel 0.2.1.dev7__py3-none-any.whl → 0.2.1.dev10__py3-none-any.whl

microsoft_agents_a365/tooling/extensions/semantickernel/services/mcp_tool_registration_service.py

@@ -13,15 +13,23 @@ servers to agents.
 import logging
 import os
 import re
-from typing import Any, Optional
+import uuid
+from datetime import datetime, timezone
+from typing import Any, List, Optional, Sequence
+
+# Third-party imports
 from semantic_kernel import kernel as sk
 from semantic_kernel.connectors.mcp import MCPStreamableHttpPlugin
+from semantic_kernel.contents import AuthorRole, ChatHistory, ChatMessageContent
+
+# Local imports
 from microsoft_agents.hosting.core import Authorization, TurnContext
+from microsoft_agents_a365.runtime import OperationError, OperationResult
 from microsoft_agents_a365.runtime.utility import Utility
+from microsoft_agents_a365.tooling.models import ChatHistoryMessage, ToolOptions
 from microsoft_agents_a365.tooling.services.mcp_tool_server_configuration_service import (
     McpToolServerConfigurationService,
 )
-from microsoft_agents_a365.tooling.models import ToolOptions
 from microsoft_agents_a365.tooling.utils.constants import Constants
 from microsoft_agents_a365.tooling.utils.utility import (
     get_mcp_platform_authentication_scope,
@@ -119,7 +127,9 @@ class McpToolRegistrationService:
         for server in servers:
             try:
                 headers = {
-                    Constants.Headers.AUTHORIZATION: f"{Constants.Headers.BEARER_PREFIX} {auth_token}",
+                    Constants.Headers.AUTHORIZATION: (
+                        f"{Constants.Headers.BEARER_PREFIX} {auth_token}"
+                    ),
                 }
 
                 headers[Constants.Headers.USER_AGENT] = Utility.get_user_agent_header(
@@ -129,7 +139,8 @@ class McpToolRegistrationService:
                 # Use the URL from server (always populated by the configuration service)
                 server_url = server.url
 
-                # Use mcp_server_name if available (not None or empty), otherwise fall back to mcp_server_unique_name
+                # Use mcp_server_name if available (not None or empty),
+                # otherwise fall back to mcp_server_unique_name
                 server_name = server.mcp_server_name or server.mcp_server_unique_name
 
                 plugin = MCPStreamableHttpPlugin(
@@ -144,10 +155,11 @@ class McpToolRegistrationService:
                 # Add plugin to kernel
                 kernel.add_plugin(plugin, server_name)
 
-                # Store reference to keep plugin alive throughout application lifecycle
-                # By storing plugin references in _connected_plugins, we prevent Python's garbage collector from cleaning up the plugin objects
-                # The connections remain active throughout the application lifecycle
-                # Tools can be successfully invoked because their underlying connections are still alive
+                # Store reference to keep plugin alive throughout application lifecycle.
+                # By storing plugin references in _connected_plugins, we prevent
+                # Python's garbage collector from cleaning up the plugin objects.
+                # The connections remain active throughout the application lifecycle.
+                # Tools can be invoked because their underlying connections stay alive.
                 self._connected_plugins.append(plugin)
 
                 self._logger.info(
@@ -181,6 +193,354 @@ class McpToolRegistrationService:
         clean_name = re.sub(r"[^a-zA-Z0-9_]", "_", server_name)
         return f"{clean_name}Tools"
 
+    # ============================================================================
+    # SEND CHAT HISTORY - Semantic Kernel-specific implementations
+    # ============================================================================
+
+    async def send_chat_history(
+        self,
+        turn_context: TurnContext,
+        chat_history: ChatHistory,
+        limit: Optional[int] = None,
+        options: Optional[ToolOptions] = None,
+    ) -> OperationResult:
+        """
+        Send Semantic Kernel chat history to the MCP platform.
+
+        This method extracts messages from a Semantic Kernel ChatHistory object,
+        converts them to ChatHistoryMessage format, and sends them to the MCP
+        platform for real-time threat protection.
+
+        Args:
+            turn_context: TurnContext from the Agents SDK containing conversation info.
+            chat_history: Semantic Kernel ChatHistory object containing messages.
+            limit: Optional maximum number of messages to send. If specified,
+                   sends the most recent N messages. If None, sends all messages.
+            options: Optional configuration for the request.
+
+        Returns:
+            OperationResult indicating success or failure.
+
+        Raises:
+            ValueError: If turn_context or chat_history is None.
+
+        Example:
+            >>> from semantic_kernel.contents import ChatHistory
+            >>> from microsoft_agents_a365.tooling.extensions.semantickernel import (
+            ...     McpToolRegistrationService
+            ... )
+            >>>
+            >>> service = McpToolRegistrationService()
+            >>> chat_history = ChatHistory()
+            >>> chat_history.add_user_message("Hello!")
+            >>> chat_history.add_assistant_message("Hi there!")
+            >>>
+            >>> result = await service.send_chat_history(
+            ...     turn_context, chat_history, limit=50
+            ... )
+            >>> if result.succeeded:
+            ...     print("Chat history sent successfully")
+        """
+        # Validate inputs
+        if turn_context is None:
+            raise ValueError("turn_context cannot be None")
+        if chat_history is None:
+            raise ValueError("chat_history cannot be None")
+
+        try:
+            # Extract messages from ChatHistory
+            messages = list(chat_history.messages)
+            self._logger.debug(f"Extracted {len(messages)} messages from ChatHistory")
+
+            # Apply limit if specified
+            if limit is not None and limit > 0 and len(messages) > limit:
+                self._logger.info(f"Applying limit of {limit} to {len(messages)} messages")
+                messages = messages[-limit:]  # Take the most recent N messages
+
+            # Delegate to the list-based method
+            return await self.send_chat_history_messages(
+                turn_context=turn_context,
+                messages=messages,
+                options=options,
+            )
+        except ValueError:
+            # Re-raise validation errors
+            raise
+        except Exception as ex:
+            self._logger.error(f"Failed to send chat history: {ex}")
+            return OperationResult.failed(OperationError(ex))
+
+    async def send_chat_history_messages(
+        self,
+        turn_context: TurnContext,
+        messages: Sequence[ChatMessageContent],
+        options: Optional[ToolOptions] = None,
+    ) -> OperationResult:
+        """
+        Send Semantic Kernel chat history messages to the MCP platform.
+
+        This method accepts a sequence of Semantic Kernel ChatMessageContent objects,
+        converts them to ChatHistoryMessage format, and sends them to the MCP
+        platform for real-time threat protection.
+
+        Args:
+            turn_context: TurnContext from the Agents SDK containing conversation info.
+            messages: Sequence of Semantic Kernel ChatMessageContent objects to send.
+            options: Optional configuration for the request.
+
+        Returns:
+            OperationResult indicating success or failure.
+
+        Raises:
+            ValueError: If turn_context or messages is None.
+
+        Example:
+            >>> from semantic_kernel.contents import ChatMessageContent, AuthorRole
+            >>> from microsoft_agents_a365.tooling.extensions.semantickernel import (
+            ...     McpToolRegistrationService
+            ... )
+            >>>
+            >>> service = McpToolRegistrationService()
+            >>> messages = [
+            ...     ChatMessageContent(role=AuthorRole.USER, content="Hello!"),
+            ...     ChatMessageContent(role=AuthorRole.ASSISTANT, content="Hi there!"),
+            ... ]
+            >>>
+            >>> result = await service.send_chat_history_messages(
+            ...     turn_context, messages
+            ... )
+            >>> if result.succeeded:
+            ...     print("Chat history sent successfully")
+        """
+        # Validate inputs
+        if turn_context is None:
+            raise ValueError("turn_context cannot be None")
+        if messages is None:
+            raise ValueError("messages cannot be None")
+
+        self._logger.info(f"Sending {len(messages)} Semantic Kernel messages as chat history")
+
+        # Set default options
+        if options is None:
+            options = ToolOptions(orchestrator_name=self._orchestrator_name)
+        elif options.orchestrator_name is None:
+            options.orchestrator_name = self._orchestrator_name
+
+        try:
+            # Convert Semantic Kernel messages to ChatHistoryMessage format
+            chat_history_messages = self._convert_sk_messages_to_chat_history(messages)
+
+            # Call core service even with empty chat_history_messages
+            if len(chat_history_messages) == 0:
+                self._logger.info(
+                    "Empty chat history messages (either no input or all filtered), "
+                    "still sending to register user message"
+                )
+
+            self._logger.debug(
+                f"Converted {len(chat_history_messages)} messages to ChatHistoryMessage format"
+            )
+
+            # Delegate to core service
+            return await self._mcp_server_configuration_service.send_chat_history(
+                turn_context=turn_context,
+                chat_history_messages=chat_history_messages,
+                options=options,
+            )
+        except ValueError:
+            # Re-raise validation errors from the core service
+            raise
+        except Exception as ex:
+            self._logger.error(f"Failed to send chat history messages: {ex}")
+            return OperationResult.failed(OperationError(ex))
+
+    # ============================================================================
+    # PRIVATE HELPER METHODS - Message Conversion
+    # ============================================================================
+
+    def _convert_sk_messages_to_chat_history(
+        self,
+        messages: Sequence[ChatMessageContent],
+    ) -> List[ChatHistoryMessage]:
+        """
+        Convert Semantic Kernel ChatMessageContent objects to ChatHistoryMessage format.
+
+        Args:
+            messages: Sequence of Semantic Kernel ChatMessageContent objects.
+
+        Returns:
+            List of ChatHistoryMessage objects. Messages that cannot be converted
+            are filtered out with a warning log.
+        """
+        chat_history_messages: List[ChatHistoryMessage] = []
+
+        for idx, message in enumerate(messages):
+            converted = self._convert_single_sk_message(message, idx)
+            if converted is not None:
+                chat_history_messages.append(converted)
+
+        self._logger.info(
+            f"Converted {len(chat_history_messages)} of {len(messages)} messages "
+            "to ChatHistoryMessage format"
+        )
+        return chat_history_messages
+
+    def _convert_single_sk_message(
+        self,
+        message: ChatMessageContent,
+        index: int = 0,
+    ) -> Optional[ChatHistoryMessage]:
+        """
+        Convert a single Semantic Kernel message to ChatHistoryMessage format.
+
+        Args:
+            message: Single Semantic Kernel ChatMessageContent message.
+            index: Index of the message in the list (for logging).
+
+        Returns:
+            ChatHistoryMessage object or None if conversion fails.
+        """
+        try:
+            # Skip None messages
+            if message is None:
+                self._logger.warning(f"Skipping null message at index {index}")
+                return None
+
+            # Map role to string
+            role = self._map_author_role(message.role)
+
+            # Extract content
+            content = self._extract_content(message)
+            if not content or not content.strip():
+                self._logger.warning(f"Skipping message at index {index} with empty content")
+                return None
+
+            # Extract or generate ID
+            msg_id = self._extract_or_generate_id(message, index)
+
+            # Extract or generate timestamp
+            timestamp = self._extract_or_generate_timestamp(message, index)
+
+            self._logger.debug(
+                f"Converting message {index}: role={role}, "
+                f"id={msg_id}, has_timestamp={timestamp is not None}"
+            )
+
+            return ChatHistoryMessage(
+                id=msg_id,
+                role=role,
+                content=content,
+                timestamp=timestamp,
+            )
+        except Exception as ex:
+            self._logger.error(f"Failed to convert message at index {index}: {ex}")
+            return None
+
+    def _map_author_role(self, role: AuthorRole) -> str:
+        """
+        Map Semantic Kernel AuthorRole enum to lowercase string.
+
+        Args:
+            role: AuthorRole enum value.
+
+        Returns:
+            Lowercase string representation of the role.
+        """
+        return role.name.lower()
+
+    def _extract_content(self, message: ChatMessageContent) -> str:
+        """
+        Extract text content from a ChatMessageContent.
+
+        Args:
+            message: Semantic Kernel ChatMessageContent object.
+
+        Returns:
+            Content string (may be empty). Returns empty string for unexpected
+            types to avoid unintentionally exposing sensitive data.
+        """
+        content = message.content
+
+        if content is None:
+            return ""
+
+        # If content is already a string, return it directly
+        if isinstance(content, str):
+            return content
+
+        # For unexpected types, log a warning and return empty string to avoid
+        # unintentionally stringifying objects that might contain sensitive data
+        content_type = type(content).__name__
+        self._logger.warning(
+            f"Unexpected content type '{content_type}' encountered. "
+            "Returning empty string to avoid potential data exposure."
+        )
+        return ""
+
+    def _extract_or_generate_id(
+        self,
+        message: ChatMessageContent,
+        index: int,
+    ) -> str:
+        """
+        Extract message ID from metadata or generate a new UUID.
+
+        Args:
+            message: Semantic Kernel ChatMessageContent object.
+            index: Message index for logging.
+
+        Returns:
+            Message ID string.
+        """
+        # Try to get existing ID from metadata
+        if message.metadata and "id" in message.metadata:
+            existing_id = message.metadata["id"]
+            if existing_id:
+                return str(existing_id)
+
+        # Generate new UUID
+        generated_id = str(uuid.uuid4())
+        self._logger.debug(f"Generated UUID {generated_id} for message at index {index}")
+        return generated_id
+
+    def _extract_or_generate_timestamp(
+        self,
+        message: ChatMessageContent,
+        index: int,
+    ) -> datetime:
+        """
+        Extract timestamp from metadata or generate current UTC time.
+
+        Args:
+            message: Semantic Kernel ChatMessageContent object.
+            index: Message index for logging.
+
+        Returns:
+            Timestamp as datetime object.
+        """
+        # Try to get existing timestamp from metadata
+        if message.metadata:
+            existing_timestamp = message.metadata.get("timestamp") or message.metadata.get(
+                "created_at"
+            )
+            if existing_timestamp:
+                if isinstance(existing_timestamp, datetime):
+                    return existing_timestamp
+                elif isinstance(existing_timestamp, (int, float)):
+                    # Unix timestamp
+                    return datetime.fromtimestamp(existing_timestamp, tz=timezone.utc)
+                elif isinstance(existing_timestamp, str):
+                    try:
+                        return datetime.fromisoformat(existing_timestamp.replace("Z", "+00:00"))
+                    except (ValueError, TypeError) as ex:
+                        self._logger.debug(
+                            f"Failed to parse timestamp '{existing_timestamp}' at index {index}: {ex}"
+                        )
+
+        # Use current UTC time
+        self._logger.debug(f"Using current UTC time for message at index {index}")
+        return datetime.now(timezone.utc)
+
     # ============================================================================
     # Cleanup Methods
     # ============================================================================

{microsoft_agents_a365_tooling_extensions_semantickernel-0.2.1.dev7.dist-info → microsoft_agents_a365_tooling_extensions_semantickernel-0.2.1.dev10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: microsoft-agents-a365-tooling-extensions-semantickernel
-Version: 0.2.1.dev7
+Version: 0.2.1.dev10
 Summary: Semantic Kernel integration for Agent 365 Tooling SDK
 Author-email: Microsoft <support@microsoft.com>
 License: MIT

{microsoft_agents_a365_tooling_extensions_semantickernel-0.2.1.dev7.dist-info → microsoft_agents_a365_tooling_extensions_semantickernel-0.2.1.dev10.dist-info}/RECORD

@@ -1,7 +1,7 @@
 microsoft_agents_a365/tooling/extensions/semantickernel/__init__.py,sha256=IvWvvW3Yajk0vX0Il0zisWbN0F-2EIWBmSpAkTeEETw,398
 microsoft_agents_a365/tooling/extensions/semantickernel/services/__init__.py,sha256=gS6eaKMSKI0PCvXzjUt5gx8xvwMcy2Ym0XLj8cJOqvQ,373
-microsoft_agents_a365/tooling/extensions/semantickernel/services/mcp_tool_registration_service.py,sha256=kwl95y7yUfeWMXzy3eLWrIf8c58nVKNQ81gQd1AZcjY,8449
-microsoft_agents_a365_tooling_extensions_semantickernel-0.2.1.dev7.dist-info/METADATA,sha256=eRNAnGPM7EcsvLOrEUWyn6ugZiG7gnMmKer_0wJAl4Y,3364
-microsoft_agents_a365_tooling_extensions_semantickernel-0.2.1.dev7.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-microsoft_agents_a365_tooling_extensions_semantickernel-0.2.1.dev7.dist-info/top_level.txt,sha256=G3c2_4sy5_EM_BWO67SbK2tKj4G8XFn-QXRbh8g9Lgk,22
-microsoft_agents_a365_tooling_extensions_semantickernel-0.2.1.dev7.dist-info/RECORD,,
+microsoft_agents_a365/tooling/extensions/semantickernel/services/mcp_tool_registration_service.py,sha256=BYQdXnwqYgpnC1lddH8ibJqNG0Rc69oidPqxprMwIsc,22052
+microsoft_agents_a365_tooling_extensions_semantickernel-0.2.1.dev10.dist-info/METADATA,sha256=jDx-obviiNxGWsxNZw838bKGosOf0WN8Xmyn-B1wwHw,3365
+microsoft_agents_a365_tooling_extensions_semantickernel-0.2.1.dev10.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+microsoft_agents_a365_tooling_extensions_semantickernel-0.2.1.dev10.dist-info/top_level.txt,sha256=G3c2_4sy5_EM_BWO67SbK2tKj4G8XFn-QXRbh8g9Lgk,22
+microsoft_agents_a365_tooling_extensions_semantickernel-0.2.1.dev10.dist-info/RECORD,,