azure-ai-agentserver-core 1.0.0b2__py3-none-any.whl

azure/ai/agentserver/__init__.py

@@ -0,0 +1 @@
+__path__ = __import__("pkgutil").extend_path(__path__, __name__)  # type: ignore

azure/ai/agentserver/core/__init__.py

@@ -0,0 +1,14 @@
+# ---------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# ---------------------------------------------------------
+__path__ = __import__("pkgutil").extend_path(__path__, __name__)
+
+from ._version import VERSION
+from .logger import configure as config_logging
+from .server.base import FoundryCBAgent
+from .server.common.agent_run_context import AgentRunContext
+
+config_logging()
+
+__all__ = ["FoundryCBAgent", "AgentRunContext"]
+__version__ = VERSION

azure/ai/agentserver/core/_version.py

@@ -0,0 +1,9 @@
+# coding=utf-8
+# --------------------------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# Licensed under the MIT License. See License.txt in the project root for license information.
+# Code generated by Microsoft (R) Python Code Generator.
+# Changes may cause incorrect behavior and will be lost if the code is regenerated.
+# --------------------------------------------------------------------------
+
+VERSION = "1.0.0b2"

azure/ai/agentserver/core/constants.py

@@ -0,0 +1,13 @@
+# ---------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# ---------------------------------------------------------
+class Constants:
+    # well-known environment variables
+    AZURE_AI_PROJECT_ENDPOINT = "AZURE_AI_PROJECT_ENDPOINT"
+    AGENT_ID = "AGENT_ID"
+    AGENT_NAME = "AGENT_NAME"
+    AGENT_PROJECT_RESOURCE_ID = "AGENT_PROJECT_NAME"
+    OTEL_EXPORTER_ENDPOINT = "OTEL_EXPORTER_ENDPOINT"
+    AGENT_LOG_LEVEL = "AGENT_LOG_LEVEL"
+    AGENT_DEBUG_ERRORS = "AGENT_DEBUG_ERRORS"
+    ENABLE_APPLICATION_INSIGHTS_LOGGER = "AGENT_APP_INSIGHTS_ENABLED"

azure/ai/agentserver/core/logger.py

@@ -0,0 +1,161 @@
+# pylint: disable=broad-exception-caught,dangerous-default-value
+# ---------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# ---------------------------------------------------------
+import contextvars
+import logging
+import os
+from logging import config
+
+from ._version import VERSION
+from .constants import Constants
+
+default_log_config = {
+    "version": 1,
+    "disable_existing_loggers": False,
+    "loggers": {
+        "azure.ai.agentserver": {
+            "handlers": ["console"],
+            "level": "INFO",
+            "propagate": False,
+        },
+    },
+    "handlers": {
+        "console": {"formatter": "std_out", "class": "logging.StreamHandler", "level": "INFO"},
+    },
+    "formatters": {"std_out": {"format": "%(asctime)s - %(name)s - %(levelname)s - %(message)s"}},
+}
+
+request_context = contextvars.ContextVar("request_context", default=None)
+
+APPINSIGHT_CONNSTR_ENV_NAME = "APPLICATIONINSIGHTS_CONNECTION_STRING"
+
+
+def get_dimensions():
+    env_values = {name: value for name, value in vars(Constants).items() if not name.startswith("_")}
+    res = {"azure.ai.agentserver.version": VERSION}
+    for name, env_name in env_values.items():
+        if isinstance(env_name, str) and not env_name.startswith("_"):
+            runtime_value = os.environ.get(env_name)
+            if runtime_value:
+                res[f"azure.ai.agentserver.{name.lower()}"] = runtime_value
+    return res
+
+
+def get_project_endpoint():
+    project_resource_id = os.environ.get(Constants.AGENT_PROJECT_RESOURCE_ID)
+    if project_resource_id:
+        last_part = project_resource_id.split("/")[-1]
+
+        parts = last_part.split("@")
+        if len(parts) < 2:
+            print(f"invalid project resource id: {project_resource_id}")
+            return None
+        account = parts[0]
+        project = parts[1]
+        return f"https://{account}.services.ai.azure.com/api/projects/{project}"
+    print("environment variable AGENT_PROJECT_RESOURCE_ID not set.")
+    return None
+
+
+def get_application_insights_connstr():
+    try:
+        conn_str = os.environ.get(APPINSIGHT_CONNSTR_ENV_NAME)
+        if not conn_str:
+            print(f"environment variable {APPINSIGHT_CONNSTR_ENV_NAME} not set.")
+            project_endpoint = get_project_endpoint()
+            if project_endpoint:
+                # try to get the project connected application insights
+                from azure.ai.projects import AIProjectClient
+                from azure.identity import DefaultAzureCredential
+
+                project_client = AIProjectClient(credential=DefaultAzureCredential(), endpoint=project_endpoint)
+                conn_str = project_client.telemetry.get_application_insights_connection_string()
+                if not conn_str:
+                    print(f"no connected application insights found for project:{project_endpoint}")
+                else:
+                    os.environ[APPINSIGHT_CONNSTR_ENV_NAME] = conn_str
+        return conn_str
+    except Exception as e:
+        print(f"failed to get application insights with error: {e}")
+        return None
+
+
+class CustomDimensionsFilter(logging.Filter):
+    def filter(self, record):
+        # Add custom dimensions to every log record
+        dimensions = get_dimensions()
+        for key, value in dimensions.items():
+            setattr(record, key, value)
+        cur_request_context = request_context.get()
+        if cur_request_context:
+            for key, value in cur_request_context.items():
+                setattr(record, key, value)
+        return True
+
+
+def configure(log_config: dict = default_log_config):
+    """
+    Configure logging based on the provided configuration dictionary.
+    The dictionary should contain the logging configuration in a format compatible with `logging.config.dictConfig`.
+
+    :param log_config: A dictionary containing logging configuration.
+    :type log_config: dict
+    """
+    try:
+        config.dictConfig(log_config)
+
+        application_insights_connection_string = get_application_insights_connstr()
+        enable_application_insights_logger = (
+            os.environ.get(Constants.ENABLE_APPLICATION_INSIGHTS_LOGGER, "true").lower() == "true"
+        )
+        if application_insights_connection_string and enable_application_insights_logger:
+            from opentelemetry._logs import set_logger_provider
+            from opentelemetry.sdk._logs import (
+                LoggerProvider,
+                LoggingHandler,
+            )
+            from opentelemetry.sdk._logs.export import BatchLogRecordProcessor
+            from opentelemetry.sdk.resources import Resource
+
+            from azure.monitor.opentelemetry.exporter import AzureMonitorLogExporter
+
+            logger_provider = LoggerProvider(resource=Resource.create({"service.name": "azure.ai.agentserver"}))
+            set_logger_provider(logger_provider)
+
+            exporter = AzureMonitorLogExporter(connection_string=application_insights_connection_string)
+
+            logger_provider.add_log_record_processor(BatchLogRecordProcessor(exporter))
+            handler = LoggingHandler(logger_provider=logger_provider)
+            handler.name = "appinsights_handler"
+
+            # Add custom filter to inject dimensions
+            custom_filter = CustomDimensionsFilter()
+            handler.addFilter(custom_filter)
+
+            # Only add to azure.ai.agentserver namespace to avoid infrastructure logs
+            app_logger = logging.getLogger("azure.ai.agentserver")
+            app_logger.setLevel(get_log_level())
+            app_logger.addHandler(handler)
+
+    except Exception as e:
+        print(f"Failed to configure logging: {e}")
+
+
+def get_log_level():
+    log_level = os.getenv(Constants.AGENT_LOG_LEVEL, "INFO").upper()
+    valid_levels = ["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"]
+    if log_level not in valid_levels:
+        print(f"Invalid log level '{log_level}' specified. Defaulting to 'INFO'.")
+        log_level = "INFO"
+    return log_level
+
+
+def get_logger() -> logging.Logger:
+    """
+    If the logger is not already configured, it will be initialized with default settings.
+
+    :return: Configured logger instance.
+    :rtype: logging.Logger
+    """
+    return logging.getLogger("azure.ai.agentserver")

azure/ai/agentserver/core/models/__init__.py

@@ -0,0 +1,7 @@
+# ---------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# ---------------------------------------------------------
+from ._create_response import CreateResponse  # type: ignore
+from .projects import Response, ResponseStreamEvent
+
+__all__ = ["CreateResponse", "Response", "ResponseStreamEvent"]  # type: ignore[var-annotated]

azure/ai/agentserver/core/models/_create_response.py

@@ -0,0 +1,12 @@
+# ---------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# ---------------------------------------------------------
+# pylint: disable=no-name-in-module
+from typing import Optional
+
+from .openai import response_create_params   # type: ignore
+from . import projects as _azure_ai_projects_models
+
+class CreateResponse(response_create_params.ResponseCreateParamsBase, total=False): # type: ignore
+    agent: Optional[_azure_ai_projects_models.AgentReference]
+    stream: Optional[bool]

azure/ai/agentserver/core/models/openai/__init__.py

@@ -0,0 +1,16 @@
+# ---------------------------------------------------------
+# Copyright (c) Microsoft Corporation. All rights reserved.
+# ---------------------------------------------------------
+"""
+Re-exports of OpenAI SDK response types.
+
+This module re-exports types from the OpenAI SDK for convenience.
+These types are fully documented in the OpenAI SDK documentation.
+
+.. note::
+   This module re-exports OpenAI SDK types. For detailed documentation,
+   please refer to the `OpenAI Python SDK documentation <https://github.com/openai/openai-python>`_.
+"""
+from openai.types.responses import *  # pylint: disable=unused-wildcard-import
+
+__all__ = [name for name in globals() if not name.startswith("_")]  # type: ignore[var-annotated]