microsoft-agents-a365-observability-core 0.1.0.dev30__tar.gz → 0.2.1.dev0__tar.gz

{microsoft_agents_a365_observability_core-0.1.0.dev30 → microsoft_agents_a365_observability_core-0.2.1.dev0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: microsoft-agents-a365-observability-core
-Version: 0.1.0.dev30
+Version: 0.2.1.dev0
 Summary: Telemetry, tracing, and monitoring components for AI agents
 Author-email: Microsoft <support@microsoft.com>
 License: MIT
@@ -20,6 +20,7 @@ Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Classifier: Topic :: System :: Monitoring
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown
+License-File: LICENSE
 Requires-Dist: opentelemetry-api>=1.36.0
 Requires-Dist: opentelemetry-sdk>=1.36.0
 Requires-Dist: opentelemetry-exporter-otlp>=1.36.0
@@ -41,6 +42,7 @@ Requires-Dist: mypy>=1.0.0; extra == "dev"
 Provides-Extra: test
 Requires-Dist: pytest>=7.0.0; extra == "test"
 Requires-Dist: pytest-asyncio>=0.21.0; extra == "test"
+Dynamic: license-file
 
 # microsoft-agents-a365-observability-core
 
@@ -64,7 +66,7 @@ For usage examples and detailed documentation, see the [Observability documentat
 For issues, questions, or feedback:
 
 - File issues in the [GitHub Issues](https://github.com/microsoft/Agent365-python/issues) section
-- See the [main documentation](../../../README.md) for more information
+- See the [main documentation](../../README.md) for more information
  
 ## Trademarks
  
@@ -74,5 +76,5 @@ For issues, questions, or feedback:
 
 Copyright (c) Microsoft Corporation. All rights reserved.
 
-Licensed under the MIT License - see the [LICENSE](../../../LICENSE.md) file for details.
+Licensed under the MIT License - see the [LICENSE](../../LICENSE.md) file for details.
 

{microsoft_agents_a365_observability_core-0.1.0.dev30 → microsoft_agents_a365_observability_core-0.2.1.dev0}/README.md

@@ -20,7 +20,7 @@ For usage examples and detailed documentation, see the [Observability documentat
 For issues, questions, or feedback:
 
 - File issues in the [GitHub Issues](https://github.com/microsoft/Agent365-python/issues) section
-- See the [main documentation](../../../README.md) for more information
+- See the [main documentation](../../README.md) for more information
  
 ## Trademarks
  
@@ -30,5 +30,5 @@ For issues, questions, or feedback:
 
 Copyright (c) Microsoft Corporation. All rights reserved.
 
-Licensed under the MIT License - see the [LICENSE](../../../LICENSE.md) file for details.
+Licensed under the MIT License - see the [LICENSE](../../LICENSE.md) file for details.
 

{microsoft_agents_a365_observability_core-0.1.0.dev30 → microsoft_agents_a365_observability_core-0.2.1.dev0}/microsoft_agents_a365/observability/core/__init__.py

@@ -1,4 +1,5 @@
-# Copyright (c) Microsoft. All rights reserved.
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
 
 # Microsoft Agent 365 Python SDK for OpenTelemetry tracing.
 

{microsoft_agents_a365_observability_core-0.1.0.dev30 → microsoft_agents_a365_observability_core-0.2.1.dev0}/microsoft_agents_a365/observability/core/agent_details.py

@@ -21,22 +21,25 @@ class AgentDetails:
     """A description of the AI agent's purpose or capabilities."""
 
     agent_auid: Optional[str] = None
-    """Optional Agent User ID for the agent."""
+    """Agentic User ID for the agent."""
 
     agent_upn: Optional[str] = None
-    """Optional User Principal Name (UPN) for the agent."""
+    """User Principal Name (UPN) for the agentic user."""
 
     agent_blueprint_id: Optional[str] = None
-    """Optional Blueprint/Application ID for the agent."""
+    """Blueprint/Application ID for the agent."""
 
     agent_type: Optional[AgentType] = None
     """The agent type."""
 
     tenant_id: Optional[str] = None
-    """Optional Tenant ID for the agent."""
+    """Tenant ID for the agent."""
 
     conversation_id: Optional[str] = None
     """Optional conversation ID for compatibility."""
 
     icon_uri: Optional[str] = None
     """Optional icon URI for the agent."""
+
+    agent_client_ip: Optional[str] = None
+    """Client IP address of the agent user."""

{microsoft_agents_a365_observability_core-0.1.0.dev30 → microsoft_agents_a365_observability_core-0.2.1.dev0}/microsoft_agents_a365/observability/core/config.py

@@ -1,4 +1,5 @@
-# Copyright (c) Microsoft. All rights reserved.
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
 
 import logging
 import threading
@@ -10,7 +11,8 @@ from opentelemetry.sdk.resources import SERVICE_NAME, SERVICE_NAMESPACE, Resourc
 from opentelemetry.sdk.trace import TracerProvider
 from opentelemetry.sdk.trace.export import BatchSpanProcessor, ConsoleSpanExporter
 
-from .exporters.agent365_exporter import Agent365Exporter
+from .exporters.agent365_exporter import _Agent365Exporter
+from .exporters.agent365_exporter_options import Agent365ExporterOptions
 from .exporters.utils import is_agent365_exporter_enabled
 from .trace_processor.span_processor import SpanProcessor
 
@@ -51,6 +53,8 @@ class TelemetryManager:
         logger_name: str = DEFAULT_LOGGER_NAME,
         token_resolver: Callable[[str, str], str | None] | None = None,
         cluster_category: str = "prod",
+        exporter_options: Optional[Agent365ExporterOptions] = None,
+        suppress_invoke_agent_input: bool = False,
         **kwargs: Any,
     ) -> bool:
         """
@@ -59,8 +63,13 @@ class TelemetryManager:
         :param service_name: The name of the service.
         :param service_namespace: The namespace of the service.
         :param logger_name: The name of the logger to collect telemetry from.
-        :param token_resolver: Callable that returns an auth token for a given agent + tenant.
-        :param cluster_category: Environment / cluster category (e.g., "preprod", "prod").
+        :param token_resolver: (Deprecated) Callable that returns an auth token for a given agent + tenant.
+            Use exporter_options instead.
+        :param cluster_category: (Deprecated) Environment / cluster category (e.g. "prod").
+            Use exporter_options instead.
+        :param exporter_options: Agent365ExporterOptions instance for configuring the exporter.
+            If provided, exporter_options takes precedence. If exporter_options is None, the token_resolver and cluster_category parameters are used as fallback/legacy support to construct a default Agent365ExporterOptions instance.
+        :param suppress_invoke_agent_input: If True, suppress input messages for spans that are children of InvokeAgent spans.
         :return: True if configuration succeeded, False otherwise.
         """
         try:
@@ -71,6 +80,8 @@ class TelemetryManager:
                     logger_name,
                     token_resolver,
                     cluster_category,
+                    exporter_options,
+                    suppress_invoke_agent_input,
                     **kwargs,
                 )
         except Exception as e:
@@ -84,10 +95,19 @@ class TelemetryManager:
         logger_name: str,
         token_resolver: Callable[[str, str], str | None] | None = None,
         cluster_category: str = "prod",
+        exporter_options: Optional[Agent365ExporterOptions] = None,
+        suppress_invoke_agent_input: bool = False,
         **kwargs: Any,
     ) -> bool:
         """Internal configuration method - not thread-safe, must be called with lock."""
 
+        # Check if a365 observability is already configured
+        if self._tracer_provider is not None:
+            self._logger.warning(
+                "a365 observability already configured. Ignoring repeated configure() call."
+            )
+            return True
+
         # Create resource with service information
         resource = Resource.create(
             {
@@ -96,30 +116,47 @@ class TelemetryManager:
             }
         )
 
-        # Get existing tracer provider or create new one
-        try:
-            tracer_provider = trace.get_tracer_provider()
-            # Check if it's already configured
-            if hasattr(tracer_provider, "resource") and tracer_provider.resource:
-                # Already configured, just add our span processor
-                agent_processor = SpanProcessor()
-                tracer_provider.add_span_processor(agent_processor)
-                self._tracer_provider = tracer_provider
-                self._span_processors["agent"] = agent_processor
-                return True
-        except Exception:
-            pass
-
-        # Configure tracer provider
-        tracer_provider = TracerProvider(resource=resource)
-        trace.set_tracer_provider(tracer_provider)
+        # Check if there's an existing TracerProvider (from app's OTEL setup)
+        tracer_provider = trace.get_tracer_provider()
+
+        # Determine if we should use existing provider or create new one
+        # Check if it's a real TracerProvider with a resource (not a proxy/no-op)
+        if getattr(tracer_provider, "resource", None):
+            # Use existing provider from application's OTEL setup
+            self._logger.info(
+                "Detected existing TracerProvider with resource. "
+                "Adding a365 observability processors to it."
+            )
+        else:
+            # Create new TracerProvider with our resource
+            self._logger.info("Creating new TracerProvider for a365 observability.")
+            tracer_provider = TracerProvider(resource=resource)
+            trace.set_tracer_provider(tracer_provider)
+
+        # Store reference
         self._tracer_provider = tracer_provider
 
-        if is_agent365_exporter_enabled() and token_resolver is not None:
-            exporter = Agent365Exporter(
-                token_resolver=token_resolver,
+        # Use exporter_options if provided, otherwise create default options with legacy parameters
+        if exporter_options is None:
+            exporter_options = Agent365ExporterOptions(
                 cluster_category=cluster_category,
-                **kwargs,
+                token_resolver=token_resolver,
+            )
+
+        # Extract configuration for BatchSpanProcessor
+        batch_processor_kwargs = {
+            "max_queue_size": exporter_options.max_queue_size,
+            "schedule_delay_millis": exporter_options.scheduled_delay_ms,
+            "export_timeout_millis": exporter_options.exporter_timeout_ms,
+            "max_export_batch_size": exporter_options.max_export_batch_size,
+        }
+
+        if is_agent365_exporter_enabled() and exporter_options.token_resolver is not None:
+            exporter = _Agent365Exporter(
+                token_resolver=exporter_options.token_resolver,
+                cluster_category=exporter_options.cluster_category,
+                use_s2s_endpoint=exporter_options.use_s2s_endpoint,
+                suppress_invoke_agent_input=suppress_invoke_agent_input,
             )
         else:
             exporter = ConsoleSpanExporter()
@@ -130,7 +167,7 @@ class TelemetryManager:
         # Add span processors
 
         # Create BatchSpanProcessor with optimized settings
-        batch_processor = BatchSpanProcessor(exporter)
+        batch_processor = BatchSpanProcessor(exporter, **batch_processor_kwargs)
         agent_processor = SpanProcessor()
 
         tracer_provider.add_span_processor(batch_processor)
@@ -197,6 +234,7 @@ def configure(
     logger_name: str = DEFAULT_LOGGER_NAME,
     token_resolver: Callable[[str, str], str | None] | None = None,
     cluster_category: str = "prod",
+    exporter_options: Optional[Agent365ExporterOptions] = None,
     **kwargs: Any,
 ) -> bool:
     """
@@ -205,6 +243,12 @@ def configure(
     :param service_name: The name of the service.
     :param service_namespace: The namespace of the service.
     :param logger_name: The name of the logger to collect telemetry from.
+    :param token_resolver: (Deprecated) Callable that returns an auth token for a given agent + tenant.
+        Use exporter_options instead.
+    :param cluster_category: (Deprecated) Environment / cluster category (e.g. "prod").
+        Use exporter_options instead.
+    :param exporter_options: Agent365ExporterOptions instance for configuring the exporter.
+        If provided, exporter_options takes precedence. If exporter_options is None, the token_resolver and cluster_category parameters are used as fallback/legacy support to construct a default Agent365ExporterOptions instance.
     :return: True if configuration succeeded, False otherwise.
     """
     return _telemetry_manager.configure(
@@ -213,6 +257,7 @@ def configure(
         logger_name,
         token_resolver,
         cluster_category,
+        exporter_options,
         **kwargs,
     )
 

{microsoft_agents_a365_observability_core-0.1.0.dev30 → microsoft_agents_a365_observability_core-0.2.1.dev0}/microsoft_agents_a365/observability/core/constants.py

@@ -1,4 +1,5 @@
-# Copyright (c) Microsoft. All rights reserved.
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
 
 # Constants for SDK OpenTelemetry implementation.
 
@@ -68,6 +69,7 @@ GEN_AI_CALLER_TENANT_ID_KEY = "gen_ai.caller.tenantid"
 GEN_AI_CALLER_ID_KEY = "gen_ai.caller.id"
 GEN_AI_CALLER_NAME_KEY = "gen_ai.caller.name"
 GEN_AI_CALLER_UPN_KEY = "gen_ai.caller.upn"
+GEN_AI_CALLER_CLIENT_IP_KEY = "gen_ai.caller.client.ip"
 
 # Agent to Agent caller agent dimensions
 GEN_AI_CALLER_AGENT_USER_ID_KEY = "gen_ai.caller.agent.userid"
@@ -77,6 +79,7 @@ GEN_AI_CALLER_AGENT_NAME_KEY = "gen_ai.caller.agent.name"
 GEN_AI_CALLER_AGENT_ID_KEY = "gen_ai.caller.agent.id"
 GEN_AI_CALLER_AGENT_APPLICATION_ID_KEY = "gen_ai.caller.agent.applicationid"
 GEN_AI_CALLER_AGENT_TYPE_KEY = "gen_ai.caller.agent.type"
+GEN_AI_CALLER_AGENT_USER_CLIENT_IP = "gen_ai.caller.agent.user.client.ip"
 
 # Agent-specific dimensions
 AGENT_ID_KEY = "gen_ai.agent.id"
@@ -92,13 +95,13 @@ GEN_AI_AGENT_UPN_KEY = "gen_ai.agent.upn"
 GEN_AI_AGENT_BLUEPRINT_ID_KEY = "gen_ai.agent.applicationid"
 CORRELATION_ID_KEY = "correlation.id"
 HIRING_MANAGER_ID_KEY = "hiring.manager.id"
+SESSION_DESCRIPTION_KEY = "session.description"
 
 # Execution context dimensions
 GEN_AI_EXECUTION_TYPE_KEY = "gen_ai.execution.type"
 GEN_AI_EXECUTION_PAYLOAD_KEY = "gen_ai.execution.payload"
 
 # Source metadata dimensions
-GEN_AI_EXECUTION_SOURCE_ID_KEY = "gen_ai.execution.sourceMetadata.id"
 GEN_AI_EXECUTION_SOURCE_NAME_KEY = "gen_ai.channel.name"
 GEN_AI_EXECUTION_SOURCE_DESCRIPTION_KEY = "gen_ai.channel.link"
 

{microsoft_agents_a365_observability_core-0.1.0.dev30 → microsoft_agents_a365_observability_core-0.2.1.dev0}/microsoft_agents_a365/observability/core/execute_tool_scope.py

@@ -5,6 +5,8 @@ from .agent_details import AgentDetails
 from .constants import (
     EXECUTE_TOOL_OPERATION_NAME,
     GEN_AI_EVENT_CONTENT,
+    GEN_AI_EXECUTION_SOURCE_DESCRIPTION_KEY,
+    GEN_AI_EXECUTION_SOURCE_NAME_KEY,
     GEN_AI_TOOL_ARGS_KEY,
     GEN_AI_TOOL_CALL_ID_KEY,
     GEN_AI_TOOL_DESCRIPTION_KEY,
@@ -14,6 +16,7 @@ from .constants import (
     SERVER_PORT_KEY,
 )
 from .opentelemetry_scope import OpenTelemetryScope
+from .request import Request
 from .tenant_details import TenantDetails
 from .tool_call_details import ToolCallDetails
 
@@ -26,6 +29,7 @@ class ExecuteToolScope(OpenTelemetryScope):
         details: ToolCallDetails,
         agent_details: AgentDetails,
         tenant_details: TenantDetails,
+        request: Request | None = None,
     ) -> "ExecuteToolScope":
         """Creates and starts a new scope for tool execution tracing.
 
@@ -33,17 +37,19 @@ class ExecuteToolScope(OpenTelemetryScope):
             details: The details of the tool call
             agent_details: The details of the agent making the call
             tenant_details: The details of the tenant
+            request: Optional request details for additional context
 
         Returns:
             A new ExecuteToolScope instance
         """
-        return ExecuteToolScope(details, agent_details, tenant_details)
+        return ExecuteToolScope(details, agent_details, tenant_details, request)
 
     def __init__(
         self,
         details: ToolCallDetails,
         agent_details: AgentDetails,
         tenant_details: TenantDetails,
+        request: Request | None = None,
     ):
         """Initialize the tool execution scope.
 
@@ -51,6 +57,7 @@ class ExecuteToolScope(OpenTelemetryScope):
             details: The details of the tool call
             agent_details: The details of the agent making the call
             tenant_details: The details of the tenant
+            request: Optional request details for additional context
         """
         super().__init__(
             kind="Internal",
@@ -79,6 +86,13 @@ class ExecuteToolScope(OpenTelemetryScope):
             if endpoint.port and endpoint.port != 443:
                 self.set_tag_maybe(SERVER_PORT_KEY, endpoint.port)
 
+        # Set request metadata if provided
+        if request and request.source_metadata:
+            self.set_tag_maybe(GEN_AI_EXECUTION_SOURCE_NAME_KEY, request.source_metadata.name)
+            self.set_tag_maybe(
+                GEN_AI_EXECUTION_SOURCE_DESCRIPTION_KEY, request.source_metadata.description
+            )
+
     def record_response(self, response: str) -> None:
         """Records response information for telemetry tracking.
 

{microsoft_agents_a365_observability_core-0.1.0.dev30 → microsoft_agents_a365_observability_core-0.2.1.dev0}/microsoft_agents_a365/observability/core/execution_type.py

@@ -1,4 +1,5 @@
-# Copyright (c) Microsoft. All rights reserved.
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
 
 # Execution type enum.
 

microsoft_agents_a365_observability_core-0.2.1.dev0/microsoft_agents_a365/observability/core/exporters/__init__.py

@@ -0,0 +1,8 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
+
+from .agent365_exporter_options import Agent365ExporterOptions
+
+# Agent365Exporter is not exported intentionally.
+# It should only be used internally by the observability core module.
+__all__ = ["Agent365ExporterOptions"]

{microsoft_agents_a365_observability_core-0.1.0.dev30 → microsoft_agents_a365_observability_core-0.2.1.dev0}/microsoft_agents_a365/observability/core/exporters/agent365_exporter.py

@@ -1,4 +1,5 @@
-# Copyright (c) Microsoft. All rights reserved.
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
 
 # pip install opentelemetry-sdk opentelemetry-api requests
 
@@ -9,7 +10,8 @@ import logging
 import threading
 import time
 from collections.abc import Callable, Sequence
-from typing import Any
+from typing import Any, final
+from urllib.parse import urlparse
 
 import requests
 from microsoft_agents_a365.runtime.power_platform_api_discovery import PowerPlatformApiDiscovery
@@ -17,12 +19,19 @@ from opentelemetry.sdk.trace import ReadableSpan
 from opentelemetry.sdk.trace.export import SpanExporter, SpanExportResult
 from opentelemetry.trace import StatusCode
 
+from ..constants import (
+    GEN_AI_INPUT_MESSAGES_KEY,
+    GEN_AI_OPERATION_NAME_KEY,
+    INVOKE_AGENT_OPERATION_NAME,
+)
 from .utils import (
+    get_validated_domain_override,
     hex_span_id,
     hex_trace_id,
     kind_name,
     partition_by_identity,
     status_name,
+    truncate_span,
 )
 
 # ---- Exporter ---------------------------------------------------------------
@@ -35,7 +44,8 @@ DEFAULT_MAX_RETRIES = 3
 logger = logging.getLogger(__name__)
 
 
-class Agent365Exporter(SpanExporter):
+@final
+class _Agent365Exporter(SpanExporter):
     """
     Agent 365 span exporter for Agent 365:
       * Partitions spans by (tenantId, agentId)
@@ -49,6 +59,7 @@ class Agent365Exporter(SpanExporter):
         token_resolver: Callable[[str, str], str | None],
         cluster_category: str = "prod",
         use_s2s_endpoint: bool = False,
+        suppress_invoke_agent_input: bool = False,
     ):
         if token_resolver is None:
             raise ValueError("token_resolver must be provided.")
@@ -58,6 +69,9 @@ class Agent365Exporter(SpanExporter):
         self._token_resolver = token_resolver
         self._cluster_category = cluster_category
         self._use_s2s_endpoint = use_s2s_endpoint
+        self._suppress_invoke_agent_input = suppress_invoke_agent_input
+        # Read domain override once at initialization
+        self._domain_override = get_validated_domain_override()
 
     # ------------- SpanExporter API -----------------
 
@@ -84,14 +98,29 @@ class Agent365Exporter(SpanExporter):
                 body = json.dumps(payload, separators=(",", ":"), ensure_ascii=False)
 
                 # Resolve endpoint + token
-                discovery = PowerPlatformApiDiscovery(self._cluster_category)
-                endpoint = discovery.get_tenant_island_cluster_endpoint(tenant_id)
+                if self._domain_override:
+                    endpoint = self._domain_override
+                else:
+                    discovery = PowerPlatformApiDiscovery(self._cluster_category)
+                    endpoint = discovery.get_tenant_island_cluster_endpoint(tenant_id)
+
                 endpoint_path = (
                     f"/maven/agent365/service/agents/{agent_id}/traces"
                     if self._use_s2s_endpoint
                     else f"/maven/agent365/agents/{agent_id}/traces"
                 )
-                url = f"https://{endpoint}{endpoint_path}?api-version=1"
+
+                # Construct URL - if endpoint has a scheme (http:// or https://), use it as-is
+                # Otherwise, prepend https://
+                # Note: Check for "://" to distinguish between real protocols and domain:port format
+                # (urlparse treats "example.com:8080" as having scheme="example.com")
+                parsed = urlparse(endpoint)
+                if parsed.scheme and "://" in endpoint:
+                    # Endpoint is a full URL, append path
+                    url = f"{endpoint}{endpoint_path}?api-version=1"
+                else:
+                    # Endpoint is just a domain (possibly with port), prepend https://
+                    url = f"https://{endpoint}{endpoint_path}?api-version=1"
 
                 # Debug: Log endpoint being used
                 logger.info(
@@ -140,6 +169,8 @@ class Agent365Exporter(SpanExporter):
     def force_flush(self, timeout_millis: int = 30000) -> bool:
         return True
 
+    # ------------- Helper methods -------------------
+
     # ------------- HTTP helper ----------------------
 
     @staticmethod
@@ -256,6 +287,20 @@ class Agent365Exporter(SpanExporter):
 
         # attributes
         attrs = dict(sp.attributes or {})
+
+        # Suppress input messages if configured and current span is an InvokeAgent span
+        if self._suppress_invoke_agent_input:
+            # Check if current span is an InvokeAgent span by:
+            # 1. Span name starts with "invoke_agent"
+            # 2. Has attribute gen_ai.operation.name set to INVOKE_AGENT_OPERATION_NAME
+            operation_name = attrs.get(GEN_AI_OPERATION_NAME_KEY)
+            if (
+                sp.name.startswith(INVOKE_AGENT_OPERATION_NAME)
+                and operation_name == INVOKE_AGENT_OPERATION_NAME
+            ):
+                # Remove input messages attribute
+                attrs.pop(GEN_AI_INPUT_MESSAGES_KEY, None)
+
         # events
         events = []
         for ev in sp.events:
@@ -295,7 +340,7 @@ class Agent365Exporter(SpanExporter):
         start_ns = sp.start_time
         end_ns = sp.end_time
 
-        return {
+        span_dict = {
             "traceId": hex_trace_id(ctx.trace_id),
             "spanId": hex_span_id(ctx.span_id),
             "parentSpanId": parent_span_id,
@@ -308,3 +353,6 @@ class Agent365Exporter(SpanExporter):
             "links": links,
             "status": status,
         }
+
+        # Apply truncation if needed
+        return truncate_span(span_dict)

microsoft_agents_a365_observability_core-0.2.1.dev0/microsoft_agents_a365/observability/core/exporters/agent365_exporter_options.py

@@ -0,0 +1,39 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
+
+from typing import Awaitable, Callable, Optional
+
+
+class Agent365ExporterOptions:
+    """
+    Configuration for Agent365Exporter.
+    Only cluster_category and token_resolver are required for core operation.
+    """
+
+    def __init__(
+        self,
+        cluster_category: str = "prod",
+        token_resolver: Optional[Callable[[str, str], Awaitable[Optional[str]]]] = None,
+        use_s2s_endpoint: bool = False,
+        max_queue_size: int = 2048,
+        scheduled_delay_ms: int = 5000,
+        exporter_timeout_ms: int = 30000,
+        max_export_batch_size: int = 512,
+    ):
+        """
+        Args:
+            cluster_category: Cluster region argument. Defaults to 'prod'.
+            token_resolver: Async callable that resolves the auth token (REQUIRED).
+            use_s2s_endpoint: Use the S2S endpoint instead of standard endpoint.
+            max_queue_size: Maximum queue size for the batch processor. Default is 2048.
+            scheduled_delay_ms: Delay between export batches (ms). Default is 5000.
+            exporter_timeout_ms: Timeout for the export operation (ms). Default is 30000.
+            max_export_batch_size: Maximum batch size for export operations. Default is 512.
+        """
+        self.cluster_category = cluster_category
+        self.token_resolver = token_resolver
+        self.use_s2s_endpoint = use_s2s_endpoint
+        self.max_queue_size = max_queue_size
+        self.scheduled_delay_ms = scheduled_delay_ms
+        self.exporter_timeout_ms = exporter_timeout_ms
+        self.max_export_batch_size = max_export_batch_size