dapr
/
dapr-agents
mirror of https://github.com/dapr/dapr-agents.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Chore: Ensure model conforms 

Signed-off-by: Casper Guldbech Nielsen <scni@novonordisk.com>
This commit is contained in:
Casper Guldbech Nielsen 2025-05-03 12:34:14 +02:00
parent 0cf7151dbe
commit 0b6d04f330
No known key found for this signature in database
GPG Key ID: B004583B52B9A446
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
dapr_agents/workflow/agentic.py
View File

@ -34,6 +34,7 @@ from dapr_agents.workflow.messaging import DaprPubSub
from dapr_agents.workflow.messaging.routing import MessageRoutingMixin from dapr_agents.workflow.messaging.routing import MessageRoutingMixin
from dapr_agents.storage.daprstores.statestore import DaprStateStore from dapr_agents.storage.daprstores.statestore import DaprStateStore
from dapr_agents.workflow import WorkflowApp from dapr_agents.workflow import WorkflowApp
from opentelemetry.sdk.trace import TracerProvider
if TYPE_CHECKING: if TYPE_CHECKING:
from fastapi import FastAPI from fastapi import FastAPI
@ -103,6 +104,7 @@ class AgenticWorkflow(WorkflowApp, DaprPubSub, MessageRoutingMixin):
Tuple[str, str], Dict[Type[BaseModel], Callable] Tuple[str, str], Dict[Type[BaseModel], Callable]
] = PrivateAttr(default_factory=dict) ] = PrivateAttr(default_factory=dict)
_otel_enabled: Optional[bool] = PrivateAttr(default=True) _otel_enabled: Optional[bool] = PrivateAttr(default=True)
_tracer: Optional[TracerProvider] = PrivateAttr(default=None)
def model_post_init(self, __context: Any) -> None: def model_post_init(self, __context: Any) -> None:
"""Initializes the workflow service, messaging, and metadata storage.""" """Initializes the workflow service, messaging, and metadata storage."""
@ -124,13 +126,13 @@ class AgenticWorkflow(WorkflowApp, DaprPubSub, MessageRoutingMixin):
service_name=self.name, service_name=self.name,
otlp_endpoint=os.getenv("OTEL_EXPORTER_OTLP_ENDPOINT", ""), otlp_endpoint=os.getenv("OTEL_EXPORTER_OTLP_ENDPOINT", ""),
) )
self.tracer = otel_client.create_and_instrument_tracer_provider() self._tracer = otel_client.create_and_instrument_tracer_provider()
trace.set_tracer_provider(self.tracer) trace.set_tracer_provider(self._tracer)
self.otel_logger = otel_client.create_and_instrument_logging_provider( otel_logger = otel_client.create_and_instrument_logging_provider(
logger=logger, logger=logger,
) )
set_logger_provider(self.otel_logger) set_logger_provider(otel_logger)
# We can instrument Asyncio automatically # We can instrument Asyncio automatically
AsyncioInstrumentor().instrument() AsyncioInstrumentor().instrument()
@ -779,7 +781,7 @@ class AgenticWorkflow(WorkflowApp, DaprPubSub, MessageRoutingMixin):
JSONResponse: A 202 Accepted response with the workflow instance ID if successful, JSONResponse: A 202 Accepted response with the workflow instance ID if successful,
or a 400/500 error response if the request fails validation or execution. or a 400/500 error response if the request fails validation or execution.
""" """
with self.tracer.start_as_current_span( with self._tracer.start_as_current_span(
"Workflow Run" "Workflow Run"
) if self.otel_enabled else None as span: ) if self.otel_enabled else None as span:
try: try: