Merge pull request #307 from CasperGN/fix/specify-loglevel

Fix/specify loglevel

Author: sicoyle

dapr_agents/observability/constants.py

@@ -8,6 +8,7 @@
 try:
     from opentelemetry import context as context_api
     from opentelemetry import trace as trace_api
+    from opentelemetry import _logs as logs_api
     from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
     from opentelemetry.trace import Status, StatusCode
     from opentelemetry.util.types import AttributeValue
@@ -17,6 +18,7 @@
     # OpenTelemetry is not available - instrumentation will be disabled
     # To enable observability features, install: pip install dapr-agents[observability]
     trace_api = None
+    logs_api = None
     BaseInstrumentor = object
     context_api = None
     Status = None
@@ -175,6 +177,7 @@ def safe_json_dumps(obj: Any) -> str:
     "OPENINFERENCE_AVAILABLE",
     # OpenTelemetry core
     "trace_api",
+    "logs_api",
     "context_api",
     "BaseInstrumentor",
     "Status",

dapr_agents/observability/instrumentor.py

@@ -50,6 +50,7 @@
     WRAPT_AVAILABLE,
     BaseInstrumentor,
     trace_api,
+    logs_api,
     context_api,
     wrap_function_wrapper,
 )
@@ -107,6 +108,7 @@ class DaprAgentsInstrumentor(BaseInstrumentor):
 
     # Class-level tracer for global access
     _global_tracer = None
+    _global_logger = None
 
     def __init__(self) -> None:
         """
@@ -116,6 +118,7 @@ def __init__(self) -> None:
         """
         super().__init__()
         self._tracer = None
+        self._logger = None
 
     def instrumentation_dependencies(self) -> Collection[str]:
         """
@@ -164,6 +167,9 @@ def _instrument(self, **kwargs: Any) -> None:
             logger.warning("wrapt not available - instrumentation disabled")
             return
 
+        # Initialize logger for the instrumentor
+        self._initialize_logger(kwargs)
+
         # Initialize OpenTelemetry tracer with provider configuration
         self._initialize_tracer(kwargs)
 
@@ -185,7 +191,20 @@ def _instrument(self, **kwargs: Any) -> None:
 
         logger.info("✅ Dapr Agents OpenTelemetry instrumentation enabled")
 
-    def _initialize_tracer(self, kwargs: dict) -> None:
+    def _initialize_logger(self, kwargs: dict[str, Any]) -> None:
+        """
+        Initialize the logger for the instrumentor.
+
+        Sets up the logger to use the module-level logger defined at the top.
+        """
+        logger_provider = kwargs.get("logger_provider")
+        if not logger_provider:
+            logger_provider = logs_api.get_logger_provider()
+
+        self._logger = logs_api.get_logger(__name__, logger_provider=logger_provider)
+        DaprAgentsInstrumentor._global_logger = self._logger
+
+    def _initialize_tracer(self, kwargs: dict[str, Any]) -> None:
         """
         Initialize OpenTelemetry tracer with provider configuration.
 

dapr_agents/workflow/runners/agent.py

@@ -493,7 +493,12 @@ def serve(
                     "install uvicorn or pass an existing FastAPI app."
                 ) from exc
 
-            uvicorn.run(fastapi_app, host=host, port=port)
+            uvicorn.run(
+                fastapi_app,
+                host=host,
+                port=port,
+                log_level=logging.getLevelName(logger.getEffectiveLevel()).lower(),
+            )
 
         return fastapi_app
 

pyproject.toml

@@ -17,7 +17,7 @@ dependencies = [
     "dapr>=1.16.0rc2",
     "dapr-ext-fastapi>=1.16.0rc2",
     "dapr-ext-workflow>=1.16.0rc2",
-    "durabletask-dapr>=0.2.0a7",
+    "durabletask-dapr>=0.2.0a14",
     "pydantic>=2.11.3,<3.0.0",
     "jinja2>=3.1.0,<4.0.0",
     "pyyaml>=6.0.1,<7.0.0",