Skip to content

Commit 9217303

Browse files
committed
feat: format
1 parent eca1beb commit 9217303

File tree

3 files changed

+4
-0
lines changed

3 files changed

+4
-0
lines changed

veadk/tracing/telemetry/attributes/extractors/llm_attributes_extractors.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -324,9 +324,11 @@ def llm_gen_ai_is_streaming(params: LLMAttributesParams) -> ExtractorResponse:
324324
def llm_gen_ai_operation_name(params: LLMAttributesParams) -> ExtractorResponse:
325325
return ExtractorResponse(content="chat")
326326

327+
327328
def llm_gen_ai_span_kind(params: LLMAttributesParams) -> ExtractorResponse:
328329
return ExtractorResponse(content="llm")
329330

331+
330332
# def llm_gen_ai_system_message(params: LLMAttributesParams) -> ExtractorResponse:
331333
# event_attributes = {
332334
# "content": str(params.llm_request.config.system_instruction),

veadk/tracing/telemetry/attributes/extractors/tool_attributes_extractors.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@
2222
def tool_gen_ai_operation_name(params: ToolAttributesParams) -> ExtractorResponse:
2323
return ExtractorResponse(content="execute_tool")
2424

25+
2526
def tool_gen_ai_span_kind(params: ToolAttributesParams) -> ExtractorResponse:
2627
return ExtractorResponse(content="tool")
2728

veadk/tracing/telemetry/telemetry.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -133,6 +133,7 @@ def _set_agent_output_attribute(span: Span, llm_response: LlmResponse) -> None:
133133
if content and content.parts:
134134
# set gen_ai.output attribute required by APMPlus
135135
span.set_attribute("gen_ai.output", content.model_dump(exclude_none=True))
136+
136137
for idx, part in enumerate(content.parts):
137138
if part.text:
138139
span.add_event(

0 commit comments

Comments
 (0)