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

Chore: Only instrument fastapi 

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

16
dapr_agents/service/fastapi/base.py
View File

@ -6,6 +6,7 @@ from fastapi import FastAPI
from pydantic import Field, ConfigDict, PrivateAttr
from typing import List, Optional, Any
from dapr_agents.service import APIServerBase
from opentelemetry import _logs, trace
import uvicorn
import asyncio
import signal
@ -63,23 +64,8 @@ class FastAPIServerBase(APIServerBase):
self._otel_enabled = False
if self._otel_enabled:
from dapr_agents.agent import DaprAgentsOTel
from opentelemetry import trace
from opentelemetry._logs import set_logger_provider
from opentelemetry.instrumentation.fastapi import FastAPIInstrumentor
otel_client = DaprAgentsOTel(
service_name="FastAPI Server",
otlp_endpoint=os.getenv("OTEL_EXPORTER_OTLP_ENDPOINT", ""),
)
tracer = otel_client.create_and_instrument_tracer_provider()
trace.set_tracer_provider(tracer)
otel_logger = otel_client.create_and_instrument_logging_provider(
logger=logger,
)
set_logger_provider(otel_logger)
# We can instrument FastAPI automatically
FastAPIInstrumentor.instrument_app(self.app)