microsoft-agents-activity 1.1.0.dev8__tar.gz → 1.2.0__tar.gz

{microsoft_agents_activity-1.1.0.dev8 → microsoft_agents_activity-1.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: microsoft-agents-activity
-Version: 1.1.0.dev8
+Version: 1.2.0
 Summary: A protocol library for Microsoft Agents
 Author: Microsoft Corporation
 License-Expression: MIT
@@ -35,6 +35,24 @@ This library is part of the **Microsoft 365 Agents SDK for Python** - a comprehe
     <th style="width:20%">Date</th>
     <th style="width:60%">Release Notes</th>
   </tr>
+  <tr>
+    <td>1.2.0</td>
+    <td>2026-07-17</td>
+    <td>
+      <a href="https://github.com/microsoft/Agents-for-python/blob/main/changelog.md#microsoft-365-agents-sdk-for-python---release-notes-v120">
+        1.2.0 Release Notes
+      </a>
+    </td>
+  </tr>
+  <tr>
+    <td>1.1.0</td>
+    <td>2026-06-19</td>
+    <td>
+      <a href="https://github.com/microsoft/Agents-for-python/blob/main/changelog.md#microsoft-365-agents-sdk-for-python---release-notes-v110">
+        1.1.0 Release Notes
+      </a>
+    </td>
+  </tr>
   <tr>
     <td>1.0.0</td>
     <td>2026-05-22</td>
@@ -44,6 +62,15 @@ This library is part of the **Microsoft 365 Agents SDK for Python** - a comprehe
       </a>
     </td>
   </tr>
+  <tr>
+    <td>0.9.1</td>
+    <td>2026-05-04</td>
+    <td>
+      <a href="https://github.com/microsoft/Agents-for-python/blob/main/changelog.md#microsoft-365-agents-sdk-for-python---release-notes-v091">
+        0.9.1 Release Notes
+      </a>
+    </td>
+  </tr>
   <tr>
     <td>0.9.0</td>
     <td>2026-04-15</td>
@@ -110,6 +137,7 @@ We offer the following PyPI packages to create conversational experiences based
 | `microsoft-agents-hosting-core` | [![PyPI](https://img.shields.io/pypi/v/microsoft-agents-hosting-core)](https://pypi.org/project/microsoft-agents-hosting-core/) | Core library for Microsoft Agents hosting. |
 | `microsoft-agents-hosting-aiohttp` | [![PyPI](https://img.shields.io/pypi/v/microsoft-agents-hosting-aiohttp)](https://pypi.org/project/microsoft-agents-hosting-aiohttp/) | Configures aiohttp to run the Agent. |
 | `microsoft-agents-hosting-teams` | [![PyPI](https://img.shields.io/pypi/v/microsoft-agents-hosting-teams)](https://pypi.org/project/microsoft-agents-hosting-teams/) | Provides classes to host an Agent for Teams. |
+| `microsoft-agents-hosting-dialogs` | [![PyPI](https://img.shields.io/pypi/v/microsoft-agents-hosting-dialogs)](https://pypi.org/project/microsoft-agents-hosting-dialogs/) | Dialog system with waterfall dialogs, prompts, and multi-turn conversation management. |
 | `microsoft-agents-storage-blob` | [![PyPI](https://img.shields.io/pypi/v/microsoft-agents-storage-blob)](https://pypi.org/project/microsoft-agents-storage-blob/) | Extension to use Azure Blob as storage. |
 | `microsoft-agents-storage-cosmos` | [![PyPI](https://img.shields.io/pypi/v/microsoft-agents-storage-cosmos)](https://pypi.org/project/microsoft-agents-storage-cosmos/) | Extension to use CosmosDB as storage. |
 | `microsoft-agents-authentication-msal` | [![PyPI](https://img.shields.io/pypi/v/microsoft-agents-authentication-msal)](https://pypi.org/project/microsoft-agents-authentication-msal/) | MSAL-based authentication for Microsoft Agents. |

microsoft_agents_activity-1.2.0/VERSION.txt

@@ -0,0 +1 @@
+1.2.0

{microsoft_agents_activity-1.1.0.dev8 → microsoft_agents_activity-1.2.0}/microsoft_agents/activity/__init__.py

@@ -45,6 +45,8 @@ from .entity import (
     ProductInfo,
     Thing,
     StreamInfo,
+    ActivityTreatment,
+    ActivityTreatmentTypes,
 )
 from .error import Error
 from .error_response import ErrorResponse
@@ -199,5 +201,7 @@ __all__ = [
     "load_configuration_from_env",
     "ChannelAdapterProtocol",
     "TurnContextProtocol",
+    "ActivityTreatment",
+    "ActivityTreatmentTypes",
     "TokenOrSignInResourceResponse",
 ]

{microsoft_agents_activity-1.1.0.dev8 → microsoft_agents_activity-1.2.0}/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
@@ -497,7 +500,7 @@ class Activity(AgentsModel, _ChannelIdFieldMixin):
         """
         return Activity(type=ActivityTypes.message)
 
-    def create_reply(self, text: str = None, locale: str = None):
+    def create_reply(self, text: str | None = None, locale: str | None = None):
         """
         Creates a new message activity as a response to this activity.
 
@@ -509,37 +512,44 @@ 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"])
+        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=[],
             ),
-            reply_to_id=(
-                SkipNone(self.id)
-                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"]
-                )
-            ),
-            text=text if text else "",
-            locale=locale if locale else SkipNone(self.locale),
-            attachments=[],
-            entities=[],
         )
 
     def create_trace(
-        self, name: str, value: object = None, value_type: str = None, label: str = None
+        self,
+        name: str,
+        value: object = None,
+        value_type: str | None = None,
+        label: str | None = None,
     ):
         """
         Creates a new trace activity based on this activity.
@@ -554,39 +564,45 @@ 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
     def create_trace_activity(
-        name: str, value: object = None, value_type: str = None, label: str = None
-    ):
+        name: str,
+        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.
 
@@ -600,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
@@ -629,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]:
         """
@@ -669,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
@@ -732,21 +793,14 @@ class Activity(AgentsModel, _ChannelIdFieldMixin):
         if self.type is None:
             return False
 
-        type_attribute = f"ActivityTypes.{str(self.type)}".lower()
-        activity_type = str(activity_type).lower()
-
-        result = type_attribute.startswith(activity_type)
+        type_value = str(getattr(self.type, "value", self.type)).lower()
+        activity_type_value = str(
+            getattr(activity_type, "value", activity_type)
+        ).lower()
 
-        if result:
-            result = len(type_attribute) == len(activity_type)
-
-            if not result:
-                result = (
-                    len(type_attribute) > len(activity_type)
-                    and type_attribute[len(activity_type)] == "/"
-                )
-
-        return result
+        return type_value == activity_type_value or type_value.startswith(
+            f"{activity_type_value}/"
+        )
 
     def add_ai_metadata(
         self,

{microsoft_agents_activity-1.1.0.dev8 → microsoft_agents_activity-1.2.0}/microsoft_agents/activity/channel_adapter_protocol.py

@@ -2,7 +2,7 @@
 # Licensed under the MIT License.
 
 from abc import abstractmethod
-from typing import Protocol, List, Callable, Awaitable, Optional
+from typing import Protocol, Callable, Awaitable, Optional
 
 from .turn_context_protocol import TurnContextProtocol
 from microsoft_agents.activity import (
@@ -18,8 +18,8 @@ class ChannelAdapterProtocol(Protocol):
 
     @abstractmethod
     async def send_activities(
-        self, context: TurnContextProtocol, activities: List[Activity]
-    ) -> List[ResourceResponse]:
+        self, context: TurnContextProtocol, activities: list[Activity]
+    ) -> list[ResourceResponse]:
         pass
 
     @abstractmethod
@@ -54,7 +54,7 @@ class ChannelAdapterProtocol(Protocol):
         claims_identity: dict,
         continuation_activity: Activity,
         callback: Callable[[TurnContextProtocol], Awaitable],
-        audience: str = None,
+        audience: str | None = None,
     ):
         pass
 

microsoft_agents_activity-1.2.0/microsoft_agents/activity/config/_coercion.py

@@ -0,0 +1,99 @@
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License.
+
+"""Coercion helpers for values loaded from environment configuration.
+
+:func:`load_configuration_from_env` keeps every value as the raw string (or a
+nested ``dict``) read from the environment. Consumers that expect a bool, int or
+float must coerce explicitly: a naive ``bool("false")`` is ``True`` and would
+silently invert a flag, and ``int("5") + 1`` style math fails on values that may
+legitimately arrive as strings. These helpers centralize that coercion so every
+config consumer parses env values the same, fail-safe way.
+"""
+
+from __future__ import annotations
+
+from typing import Any
+
+_TRUTHY = ("1", "true")
+_FALSY = ("0", "false")
+
+
+def coerce_bool(value: Any, default: bool | None = None, name: str = "value") -> bool:
+    """Coerce a config value (possibly an env string) to a bool.
+
+    Environment variables always arrive as strings, so ``bool("false")`` would
+    be ``True``. Recognized truthy spellings are ``1/true`` and falsy spellings
+    are ``0/false`` (case-insensitive).
+
+    An empty/whitespace-only string or ``None`` is treated as *unset* (not as an
+    explicit ``false``). For unset and unrecognized values: if ``default`` is
+    provided it is returned, otherwise a ``ValueError`` is raised naming the
+    offending setting -- mirroring :func:`coerce_int` and :func:`coerce_float`.
+    """
+    if isinstance(value, bool):
+        return value
+    if isinstance(value, (int, float)):
+        return bool(value)
+    if value is not None:
+        text = str(value).strip().lower()
+        if text in _TRUTHY:
+            return True
+        if text in _FALSY:
+            return False
+        if text:
+            if default is not None:
+                return default
+            raise ValueError(f"Invalid boolean for {name}: {value!r}")
+    # Unset (None or empty/whitespace-only string).
+    if default is not None:
+        return default
+    raise ValueError(f"Invalid boolean for {name}: {value!r}")
+
+
+def coerce_int(value: Any, default: int, name: str = "value") -> int:
+    """Coerce a config value (possibly an env string) to an int.
+
+    Empty/``None`` values fall back to ``default``. Bools are treated as unset
+    (a stray ``True``/``False`` is not a meaningful count). Numeric strings such
+    as ``"5"`` or ``"5.0"`` are accepted; anything else raises ``ValueError``
+    naming the offending setting.
+    """
+    if value is None:
+        return default
+    if isinstance(value, bool):
+        return default
+    if isinstance(value, int):
+        return value
+    text = str(value).strip()
+    if not text:
+        return default
+    try:
+        return int(text, 10)
+    except ValueError:
+        try:
+            return int(float(text))
+        except ValueError as ex:
+            raise ValueError(f"Invalid integer for {name}: {value!r}") from ex
+
+
+def coerce_float(value: Any, default: float, name: str = "value") -> float:
+    """Coerce a config value (possibly an env string) to a float.
+
+    Empty/``None`` values fall back to ``default``. Bools are treated as unset.
+    Anything that is not parseable as a float raises ``ValueError`` naming the
+    offending setting.
+    """
+    if value is None:
+        return default
+    if isinstance(value, bool):
+        return default
+    if isinstance(value, (int, float)):
+        return float(value)
+    text = str(value).strip()
+    if not text:
+        return default
+    try:
+        return float(text)
+    except ValueError as ex:
+        raise ValueError(f"Invalid number for {name}: {value!r}") from ex

{microsoft_agents_activity-1.1.0.dev8 → microsoft_agents_activity-1.2.0}/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.1.0.dev8 → microsoft_agents_activity-1.2.0}/microsoft_agents/activity/conversation_reference.py

@@ -4,22 +4,18 @@
 from __future__ import annotations
 
 from uuid import uuid4 as uuid
-from typing import Optional
-import logging
+from typing import Optional, Annotated
 
 from pydantic import Field
 
 from .channel_account import ChannelAccount
 from ._channel_id_field_mixin import _ChannelIdFieldMixin
-from .channel_id import ChannelId
 from .conversation_account import ConversationAccount
 from .agents_model import AgentsModel
 from ._type_aliases import NonEmptyString
 from .activity_types import ActivityTypes
 from .activity_event_names import ActivityEventNames
 
-logger = logging.getLogger(__name__)
-
 
 class ConversationReference(AgentsModel, _ChannelIdFieldMixin):
     """An object relating to a particular point in a conversation.
@@ -48,7 +44,7 @@ class ConversationReference(AgentsModel, _ChannelIdFieldMixin):
     # optionals here are due to webchat
     activity_id: Optional[NonEmptyString] = None
     user: Optional[ChannelAccount] = None
-    agent: ChannelAccount = Field(None, alias="bot")
+    agent: Annotated[ChannelAccount, Field(alias="bot")] = None
     conversation: ConversationAccount
     locale: Optional[NonEmptyString] = None
     service_url: NonEmptyString = None

{microsoft_agents_activity-1.1.0.dev8 → microsoft_agents_activity-1.2.0}/microsoft_agents/activity/entity/__init__.py

@@ -1,6 +1,7 @@
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # Licensed under the MIT License.
 
+from .activity_treatment import ActivityTreatment, ActivityTreatmentTypes
 from .mention import Mention
 from .entity import Entity
 from .entity_types import EntityTypes
@@ -35,4 +36,6 @@ __all__ = [
     "Place",
     "ProductInfo",
     "Thing",
+    "ActivityTreatment",
+    "ActivityTreatmentTypes",
 ]