Skip to content

Commit 5c26845

Browse files
committed
Updae header
1 parent 81bed13 commit 5c26845

File tree

1 file changed

+4
-4
lines changed
  • code/function/fastapp/api/v1/endpoints

1 file changed

+4
-4
lines changed

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

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
1-
from typing import Any
1+
from typing import Annotated
22

33
import aiohttp
4-
from fastapi import APIRouter, Request
4+
from fastapi import APIRouter, Header
55
from fastapp.models.sample import SampleRequest, SampleResponse
66
from fastapp.utils import setup_logging, setup_tracer
77
from opentelemetry import trace
@@ -13,7 +13,7 @@
1313

1414

1515
@router.post("/sample", response_model=SampleResponse, name="sample")
16-
async def post_predict(data: SampleRequest, request: Request) -> SampleResponse:
16+
async def post_predict(data: SampleRequest, x_forwarded_for: Annotated[str, Header()] = "") -> SampleResponse:
1717
logger.info(f"Received request: {data}")
1818

1919
# Sample request
@@ -23,7 +23,7 @@ async def post_predict(data: SampleRequest, request: Request) -> SampleResponse:
2323
# resp_text = await response.text()
2424
# logger.info(f"Received response status code: {resp_status_code}")
2525

26-
tracer_attributes = {"http.client_ip": request.client.host}
26+
tracer_attributes = {"http.client_ip": x_forwarded_for}
2727
with tracer.start_as_current_span(
2828
"dependency_span", attributes=tracer_attributes
2929
) as span:

0 commit comments

Comments
 (0)