Chore: Ruff

Signed-off-by: Casper Guldbech Nielsen <scni@novonordisk.com>
This commit is contained in:
Casper Guldbech Nielsen 2025-05-03 17:54:01 +02:00
parent c47e283c40
commit 3fa6945cb5
No known key found for this signature in database
GPG Key ID: B004583B52B9A446
5 changed files with 39 additions and 17 deletions

View File

@ -145,8 +145,10 @@ class DaprAgentsOTel:
return endpoint
def span_decorator(name):
"""Decorator that creates an OpenTelemetry span for a method."""
def decorator(func):
@functools.wraps(func)
def wrapper(self, *args, **kwargs):
@ -158,12 +160,16 @@ def span_decorator(name):
else:
# Fallback if no tracer is available
return func(self, *args, **kwargs)
return wrapper
return decorator
# For async methods
def async_span_decorator(name):
"""Decorator that creates an OpenTelemetry span for an async method."""
def decorator(func):
@functools.wraps(func)
async def wrapper(self, *args, **kwargs):
@ -173,5 +179,7 @@ def async_span_decorator(name):
return await func(self, *args, **kwargs)
else:
return await func(self, *args, **kwargs)
return wrapper
return decorator
return decorator

View File

@ -57,8 +57,8 @@ class FastAPIServerBase(APIServerBase):
"""
Post-initialization to configure core FastAPI app and CORS settings.
"""
#self._logger = _logs.get_logger("fastapi_logger")
# self._logger = _logs.get_logger("fastapi_logger")
# Initialize FastAPI app with title and description
self.app = FastAPI(

View File

@ -34,7 +34,11 @@ from dapr_agents.workflow.messaging.routing import MessageRoutingMixin
from dapr_agents.storage.daprstores.statestore import DaprStateStore
from dapr_agents.workflow import WorkflowApp
from opentelemetry.sdk.trace import Tracer
from dapr_agents.agent.telemetry import DaprAgentsOTel, async_span_decorator, span_decorator
from dapr_agents.agent.telemetry import (
DaprAgentsOTel,
async_span_decorator,
span_decorator,
)
from opentelemetry import trace
from opentelemetry._logs import set_logger_provider
from opentelemetry.instrumentation.asyncio import AsyncioInstrumentor
@ -110,7 +114,7 @@ class AgenticWorkflow(WorkflowApp, DaprPubSub, MessageRoutingMixin):
def model_post_init(self, __context: Any) -> None:
"""Initializes the workflow service, messaging, and metadata storage."""
try:
otel_client = DaprAgentsOTel(
service_name=self.name,
@ -129,9 +133,11 @@ class AgenticWorkflow(WorkflowApp, DaprPubSub, MessageRoutingMixin):
# We can instrument Asyncio automatically
AsyncioInstrumentor().instrument()
except Exception as e:
logger.warning(f"OpenTelemetry initialization failed: {e}. Continuing without telemetry.")
logger.warning(
f"OpenTelemetry initialization failed: {e}. Continuing without telemetry."
)
self._tracer = None
# Set up color formatter for logging and CLI printing
self._text_formatter = ColorTextFormatter()
@ -819,12 +825,8 @@ class AgenticWorkflow(WorkflowApp, DaprPubSub, MessageRoutingMixin):
except Exception:
input_data = await request.json()
logger.info(
f"Starting workflow '{workflow_name}' with input: {input_data}"
)
instance_id = self.run_workflow(
workflow=workflow_name, input=input_data
)
logger.info(f"Starting workflow '{workflow_name}' with input: {input_data}")
instance_id = self.run_workflow(workflow=workflow_name, input=input_data)
asyncio.create_task(self.monitor_workflow_completion(instance_id))

View File

@ -33,7 +33,11 @@ from dapr_agents.workflow.task import WorkflowTask
from dapr_agents.workflow.utils import get_decorated_methods
from pydantic import PrivateAttr
from dapr_agents.agent.telemetry import DaprAgentsOTel, async_span_decorator, span_decorator
from dapr_agents.agent.telemetry import (
DaprAgentsOTel,
async_span_decorator,
span_decorator,
)
from opentelemetry._logs import set_logger_provider
from opentelemetry.trace import Tracer, set_tracer_provider
@ -115,7 +119,9 @@ class WorkflowApp(BaseModel):
# set_logger_provider(otel_logger)
except Exception as e:
logger.warning(f"OpenTelemetry initialization failed: {e}. Continuing without telemetry.")
logger.warning(
f"OpenTelemetry initialization failed: {e}. Continuing without telemetry."
)
self._tracer = None
super().model_post_init(__context)

View File

@ -36,7 +36,11 @@ from dapr_agents.workflow.orchestrators.llm.utils import (
)
from pydantic import PrivateAttr
from dapr_agents.agent.telemetry import DaprAgentsOTel, async_span_decorator, span_decorator
from dapr_agents.agent.telemetry import (
DaprAgentsOTel,
async_span_decorator,
span_decorator,
)
from opentelemetry._logs import set_logger_provider
from opentelemetry.trace import Tracer, set_tracer_provider
@ -75,7 +79,9 @@ class LLMOrchestrator(OrchestratorWorkflowBase):
# set_logger_provider(otel_logger)
except Exception as e:
logger.warning(f"OpenTelemetry initialization failed: {e}. Continuing without telemetry.")
logger.warning(
f"OpenTelemetry initialization failed: {e}. Continuing without telemetry."
)
self._tracer = None
# Initializes local LLM Orchestrator State