Test with span

Author: marvinbuss

code/function/fastapp/api/v1/endpoints/sample.py

@@ -4,6 +4,7 @@
 from fastapi import APIRouter, Header
 from fastapp.models.sample import SampleRequest, SampleResponse
 from fastapp.utils import setup_logging, setup_tracer
+from opentelemetry.trace import SpanKind
 
 logger = setup_logging(__name__)
 tracer = setup_tracer(__name__)
@@ -15,11 +16,19 @@
 async def post_predict(
     data: SampleRequest, x_forwarded_for: Annotated[str, Header()] = ""
 ) -> SampleResponse:
-    logger.info(f"Received request: '{data}' from ip '{x_forwarded_for}'")
+    logger.info(f"Received request: {data}")
 
     # Sample request
-    async with httpx.AsyncClient() as client:
-        response = await client.get("https://www.bing.com")
-    logger.info(f"Received response status code: {response.status_code}")
+    tracer_attributes = {"http.client_ip": x_forwarded_for}
+    with tracer.start_as_current_span(
+        "dependency_span", attributes=tracer_attributes, kind=SpanKind.CLIENT
+    ) as span:
+        try:
+            async with httpx.AsyncClient() as client:
+                response = await client.get("https://www.bing.com")
+            logger.info(f"Received response status code: {response.status_code}")
+        except Exception as ex:
+            span.set_attribute("status", "exception")
+            span.record_exception(ex)
 
     return SampleResponse(output=f"Hello {data.input}")

code/function/requirements.txt

@@ -6,7 +6,7 @@
 azure-functions~=1.17.0
 fastapi~=0.106.0
 pydantic-settings~=2.1.0
-httpx~=0.25.2
+httpx~=0.26.0
 azure-monitor-opentelemetry~=1.1.1
 opentelemetry-instrumentation-httpx~=0.43b0
 opentelemetry-instrumentation-system-metrics~=0.43b0