microsoft-agents-activity 1.2.0.dev4__tar.gz → 1.2.0.dev7__tar.gz

{microsoft_agents_activity-1.2.0.dev4 → microsoft_agents_activity-1.2.0.dev7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: microsoft-agents-activity
-Version: 1.2.0.dev4
+Version: 1.2.0.dev7
 Summary: A protocol library for Microsoft Agents
 Author: Microsoft Corporation
 License-Expression: MIT

microsoft_agents_activity-1.2.0.dev7/VERSION.txt

@@ -0,0 +1 @@
+1.2.0.dev7

{microsoft_agents_activity-1.2.0.dev4 → microsoft_agents_activity-1.2.0.dev7}/microsoft_agents/activity/activity.py

@@ -6,7 +6,8 @@ from __future__ import annotations
 import logging
 from copy import copy
 from datetime import datetime, timezone
-from typing import Optional, Any
+from typing import Optional, Any, cast, Annotated, TypeVar
+from typing_extensions import Self
 
 from pydantic import (
     Field,
@@ -48,6 +49,8 @@ from microsoft_agents.activity.errors import activity_errors
 
 logger = logging.getLogger(__name__)
 
+_EntityT = TypeVar("_EntityT", bound=Entity)
+
 
 # TODO: A2A Agent 2 is responding with None as id, had to mark it as optional (investigate)
 class Activity(AgentsModel, _ChannelIdFieldMixin):
@@ -157,7 +160,7 @@ class Activity(AgentsModel, _ChannelIdFieldMixin):
     local_timestamp: datetime = None
     local_timezone: NonEmptyString = None
     service_url: NonEmptyString = None
-    from_property: ChannelAccount = Field(None, alias="from")
+    from_property: Annotated[ChannelAccount, Field(alias="from")] = None
     conversation: ConversationAccount = None
     recipient: ChannelAccount = None
     text_format: NonEmptyString = None
@@ -509,33 +512,36 @@ class Activity(AgentsModel, _ChannelIdFieldMixin):
         .. remarks::
             The new activity sets up routing information based on this activity.
         """
-        return pick_model(
-            Activity,
-            type=ActivityTypes.message,
-            timestamp=datetime.now(timezone.utc),
-            from_property=SkipNone(
-                ChannelAccount.pick_properties(self.recipient, ["id", "name"])
-            ),
-            recipient=SkipNone(
-                ChannelAccount.pick_properties(self.from_property, ["id", "name"])
-            ),
-            reply_to_id=(
-                SkipNone(self.id)
-                if type != ActivityTypes.conversation_update
-                or self.channel_id not in ["directline", "webchat"]
-                else None
+        return cast(
+            Self,
+            pick_model(
+                self.__class__,
+                type=ActivityTypes.message,
+                timestamp=datetime.now(timezone.utc),
+                from_property=SkipNone(
+                    ChannelAccount.pick_properties(self.recipient, ["id", "name"])
+                ),
+                recipient=SkipNone(
+                    ChannelAccount.pick_properties(self.from_property, ["id", "name"])
+                ),
+                reply_to_id=(
+                    SkipNone(self.id)
+                    if self.type != ActivityTypes.conversation_update
+                    or self.channel_id not in ["directline", "webchat"]
+                    else None
+                ),
+                service_url=self.service_url,
+                channel_id=self.channel_id,
+                conversation=SkipNone(
+                    ConversationAccount.pick_properties(
+                        self.conversation, ["is_group", "id", "name"]
+                    )
+                ),
+                text=text if text else "",
+                locale=locale if locale else SkipNone(self.locale),
+                attachments=[],
+                entities=[],
             ),
-            service_url=self.service_url,
-            channel_id=self.channel_id,
-            conversation=SkipNone(
-                ConversationAccount.pick_properties(
-                    self.conversation, ["is_group", "id", "name"]
-                )
-            ),
-            text=text if text else "",
-            locale=locale if locale else SkipNone(self.locale),
-            attachments=[],
-            entities=[],
         )
 
     def create_trace(
@@ -558,33 +564,36 @@ class Activity(AgentsModel, _ChannelIdFieldMixin):
         if not value_type and value:
             value_type = type(value).__name__
 
-        return pick_model(
-            Activity,
-            type=ActivityTypes.trace,
-            timestamp=datetime.now(timezone.utc),
-            from_property=SkipNone(
-                ChannelAccount.pick_properties(self.recipient, ["id", "name"])
-            ),
-            recipient=SkipNone(
-                ChannelAccount.pick_properties(self.from_property, ["id", "name"])
+        return cast(
+            Self,
+            pick_model(
+                self.__class__,
+                type=ActivityTypes.trace,
+                timestamp=datetime.now(timezone.utc),
+                from_property=SkipNone(
+                    ChannelAccount.pick_properties(self.recipient, ["id", "name"])
+                ),
+                recipient=SkipNone(
+                    ChannelAccount.pick_properties(self.from_property, ["id", "name"])
+                ),
+                reply_to_id=(
+                    SkipNone(self.id)  # preserve unset
+                    if self.type != ActivityTypes.conversation_update
+                    or self.channel_id not in ["directline", "webchat"]
+                    else None
+                ),
+                service_url=self.service_url,
+                channel_id=self.channel_id,
+                conversation=SkipNone(
+                    ConversationAccount.pick_properties(
+                        self.conversation, ["is_group", "id", "name"]
+                    )
+                ),
+                name=SkipNone(name),
+                label=SkipNone(label),
+                value_type=SkipNone(value_type),
+                value=SkipNone(value),
             ),
-            reply_to_id=(
-                SkipNone(self.id)  # preserve unset
-                if type != ActivityTypes.conversation_update
-                or self.channel_id not in ["directline", "webchat"]
-                else None
-            ),
-            service_url=self.service_url,
-            channel_id=self.channel_id,
-            conversation=SkipNone(
-                ConversationAccount.pick_properties(
-                    self.conversation, ["is_group", "id", "name"]
-                )
-            ),
-            name=SkipNone(name),
-            label=SkipNone(label),
-            value_type=SkipNone(value_type),
-            value=SkipNone(value),
         ).as_trace_activity()
 
     @staticmethod
@@ -593,7 +602,7 @@ class Activity(AgentsModel, _ChannelIdFieldMixin):
         value: object = None,
         value_type: str | None = None,
         label: str | None = None,
-    ):
+    ) -> Activity:
         """
         Creates an instance of the :class:`microsoft_agents.activity.Activity` class as a TraceActivity object.
 
@@ -607,13 +616,16 @@ class Activity(AgentsModel, _ChannelIdFieldMixin):
         if not value_type and value:
             value_type = type(value).__name__
 
-        return pick_model(
+        return cast(
             Activity,
-            type=ActivityTypes.trace,
-            name=name,
-            label=SkipNone(label),
-            value_type=SkipNone(value_type),
-            value=SkipNone(value),
+            pick_model(
+                Activity,
+                type=ActivityTypes.trace,
+                name=name,
+                label=SkipNone(label),
+                value_type=SkipNone(value_type),
+                value=SkipNone(value),
+            ),
         )
 
     @staticmethod
@@ -636,33 +648,71 @@ class Activity(AgentsModel, _ChannelIdFieldMixin):
             Composite values are split only on the first ``:``.
         :returns: A conversation reference for the conversation that contains this activity.
         """
-        return pick_model(
+        return cast(
             ConversationReference,
-            activity_id=(
-                SkipNone(self.id)
-                if self.type != ActivityTypes.conversation_update
-                or self.channel_id not in ["directline", "webchat"]
-                else None
-            ),
-            user=copy(self.from_property),
-            agent=copy(self.recipient),
-            conversation=copy(self.conversation),
-            channel_id=(
-                self.channel_id.split(":", 1)[0]
-                if force_base_channel and self.channel_id is not None
-                else self.channel_id
+            pick_model(
+                ConversationReference,
+                activity_id=(
+                    SkipNone(self.id)
+                    if self.type != ActivityTypes.conversation_update
+                    or self.channel_id not in ["directline", "webchat"]
+                    else None
+                ),
+                user=copy(self.from_property),
+                agent=copy(self.recipient),
+                conversation=copy(self.conversation),
+                channel_id=(
+                    self.channel_id.split(":", 1)[0]
+                    if force_base_channel and self.channel_id is not None
+                    else self.channel_id
+                ),
+                locale=self.locale,
+                service_url=self.service_url,
             ),
-            locale=self.locale,
-            service_url=self.service_url,
         )
 
+    @staticmethod
+    def _convert_entity(raw_entity: Entity, entity_cls: type[_EntityT]) -> _EntityT:
+        """
+        Converts an entity to a specific entity type.
+
+        :param raw_entity: The entity to convert.
+        :param entity_cls: The class of the entity type to convert to.
+        :return: The converted entity of the specified type.
+        """
+        if isinstance(raw_entity, entity_cls):
+            return raw_entity
+        return entity_cls.model_validate(raw_entity.model_dump())
+
+    @staticmethod
+    def _convert_entity_list(
+        raw_entities: list[Entity], entity_cls: type[_EntityT]
+    ) -> list[_EntityT]:
+        """
+        Converts a list of entities to a list of a specific entity type.
+
+        :param raw_entities: The list of entities to convert.
+        :param entity_cls: The class of the entity type to convert to.
+        :return: The list of converted entities of the specified type.
+        """
+
+        entities: list[_EntityT] = []
+        for e in raw_entities:
+            entities.append(Activity._convert_entity(e, entity_cls))
+        return entities
+
     def get_product_info_entity(self) -> Optional[ProductInfo]:
         if not self.entities:
             return None
         target = EntityTypes.PRODUCT_INFO.lower()
         # validated entities can be Entity, and that prevents us from
         # making assumptions about the casing of the 'type' attribute
-        return next(filter(lambda e: e.type.lower() == target, self.entities), None)
+        raw_product_info = next(
+            filter(lambda e: e.type.lower() == target, self.entities), None
+        )
+        if raw_product_info is None:
+            return None
+        return Activity._convert_entity(raw_product_info, ProductInfo)
 
     def get_mentions(self) -> list[Mention]:
         """
@@ -676,7 +726,11 @@ class Activity(AgentsModel, _ChannelIdFieldMixin):
         """
         if not self.entities:
             return []
-        return [x for x in self.entities if x.type.lower() == EntityTypes.MENTION]
+        raw_mentions = [
+            x for x in self.entities if x.type.lower() == EntityTypes.MENTION
+        ]
+
+        return Activity._convert_entity_list(raw_mentions, Mention)
 
     def get_reply_conversation_reference(
         self, reply: ResourceResponse

{microsoft_agents_activity-1.2.0.dev4 → microsoft_agents_activity-1.2.0.dev7}/microsoft_agents/activity/config/_configure_logging.py

@@ -20,6 +20,44 @@ _NAME_TO_LEVEL = {
 }
 
 
+class ColorFormatter(logging.Formatter):
+    """Logging formatter that can add ANSI color based on severity."""
+
+    MAGENTA = "\033[1;35m"
+    RED = "\033[1;31m"
+    YELLOW = "\033[1;33m"
+    GREEN = "\033[1;32m"
+    BLUE = "\033[1;34m"
+    RESET = "\033[0m"
+
+    FMT = (
+        "%(asctime)s - %(name)s - %(levelname)s - %(message)s (%(filename)s:%(lineno)d)"
+    )
+
+    def __init__(
+        self,
+        use_colors: bool = True,
+    ) -> None:
+        super().__init__(fmt=self.FMT)
+        self._plain_formatter = logging.Formatter(fmt=self.FMT)
+        if not use_colors:
+            self._formatters: dict[int, logging.Formatter] = {}
+            return
+        self._formatters = {
+            logging.CRITICAL: logging.Formatter(
+                f"{self.MAGENTA}{self.FMT}{self.RESET}"
+            ),
+            logging.ERROR: logging.Formatter(f"{self.RED}{self.FMT}{self.RESET}"),
+            logging.WARNING: logging.Formatter(f"{self.YELLOW}{self.FMT}{self.RESET}"),
+            logging.INFO: logging.Formatter(f"{self.GREEN}{self.FMT}{self.RESET}"),
+            logging.DEBUG: logging.Formatter(f"{self.BLUE}{self.FMT}{self.RESET}"),
+        }
+
+    def format(self, record: logging.LogRecord) -> str:
+        formatter = self._formatters.get(record.levelno)
+        return (formatter or self._plain_formatter).format(record)
+
+
 def _configure_logging(logging_config: dict):
     """Configures logging based on the provided logging configuration dictionary.
 
@@ -31,9 +69,7 @@ def _configure_logging(logging_config: dict):
 
     console_handler = logging.StreamHandler()
     console_handler.setFormatter(
-        logging.Formatter(
-            "%(asctime)s - %(name)s - %(levelname)s - %(message)s (%(filename)s:%(lineno)d)"
-        )
+        ColorFormatter(use_colors=console_handler.stream.isatty())
     )
 
     for key in log_levels:
@@ -50,6 +86,7 @@ def _configure_logging(logging_config: dict):
         else:
             logger = logging.getLogger(namespace)
 
+        logger.propagate = False  # Prevent log messages from being propagated
         logger.handlers.clear()  # Remove existing handlers to prevent duplicates
         logger.addHandler(console_handler)
         logger.setLevel(level)

{microsoft_agents_activity-1.2.0.dev4 → microsoft_agents_activity-1.2.0.dev7}/microsoft_agents_activity.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: microsoft-agents-activity
-Version: 1.2.0.dev4
+Version: 1.2.0.dev7
 Summary: A protocol library for Microsoft Agents
 Author: Microsoft Corporation
 License-Expression: MIT

microsoft_agents_activity-1.2.0.dev4/VERSION.txt

@@ -1 +0,0 @@
-1.2.0.dev4