mirror of https://github.com/dapr/dapr-agents.git
Chore: Ruff
Signed-off-by: Casper Guldbech Nielsen <scni@novonordisk.com>
This commit is contained in:
parent
c47e283c40
commit
3fa6945cb5
|
@ -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
|
||||
|
|
|
@ -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(
|
||||
|
|
|
@ -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))
|
||||
|
||||
|
|
|
@ -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)
|
||||
|
|
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue