diff --git a/backend/extensions/trace/rag_wrapper.py b/backend/extensions/trace/rag_wrapper.py index 6d49272f..ee0e22fa 100644 --- a/backend/extensions/trace/rag_wrapper.py +++ b/backend/extensions/trace/rag_wrapper.py @@ -282,6 +282,8 @@ async def wrapper(self, *args, **kwargs): span = get_tracer().start_span(RetrieverSpanNames.RERANKER) span.set_attribute(GEN_AI_SPAN_KIND, RERANKER_SPAN_KIND) span.set_attribute(RERANKER_MODEL_NAME, self.model) + span.set_attribute(GEN_AI_OPERATION_NAME, RERANKER_OPERATION_NAME) + messages = [{ "role": "user", "parts": [