feat: update code

Author: feng-95

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

@@ -514,7 +514,9 @@ def llm_gen_ai_is_streaming(params: LLMAttributesParams) -> ExtractorResponse:
         ExtractorResponse: Response containing None
     """
     is_streaming = bool(
-        params.invocation_context.run_config and params.invocation_context.run_config.streaming_mode != StreamingMode.NONE)
+        params.invocation_context.run_config
+        and params.invocation_context.run_config.streaming_mode != StreamingMode.NONE
+    )
 
     return ExtractorResponse(content=is_streaming)
 
@@ -807,6 +809,7 @@ def llm_gen_ai_request_functions(params: LLMAttributesParams) -> ExtractorRespon
 
     return ExtractorResponse(content=functions)
 
+
 def llm_server_address(params: LLMAttributesParams) -> ExtractorResponse:
     """Extract the LLM server address (model API base URL).
 
@@ -820,7 +823,10 @@ def llm_server_address(params: LLMAttributesParams) -> ExtractorResponse:
     Returns:
         ExtractorResponse: Response containing the server address or 'unknown'
     """
-    return ExtractorResponse(content=getattr(params.invocation_context.agent, "model_api_base", None) or "unknown")
+    return ExtractorResponse(
+        content=getattr(params.invocation_context.agent, "model_api_base", None)
+        or "unknown"
+    )
 
 
 LLM_ATTRIBUTES = {

veadk/tracing/telemetry/exporters/apmplus_exporter.py

@@ -12,7 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-import os
 import time
 from dataclasses import dataclass
 from typing import Any
@@ -36,7 +35,6 @@
 from typing_extensions import override
 
 from veadk.config import settings
-from veadk.consts import DEFAULT_MODEL_AGENT_API_BASE
 from veadk.tracing.telemetry.exporters.base_exporter import BaseExporter
 from veadk.utils.logger import get_logger
 
@@ -300,15 +298,19 @@ def record_call_llm(
             llm_response: Response object with content and usage metadata
         """
         is_streaming = bool(
-            invocation_context.run_config and invocation_context.run_config.streaming_mode != StreamingMode.NONE)
-        server_address = getattr(invocation_context.agent, "model_api_base", None) or "unknown"
+            invocation_context.run_config
+            and invocation_context.run_config.streaming_mode != StreamingMode.NONE
+        )
+        server_address = (
+            getattr(invocation_context.agent, "model_api_base", None) or "unknown"
+        )
         attributes = {
             "gen_ai_system": "volcengine",
             "gen_ai_response_model": llm_request.model,
             "gen_ai_operation_name": "chat",
             "gen_ai_operation_type": "llm",
             "stream": is_streaming,
-            "server_address":  server_address,
+            "server_address": server_address,
         }  # required by Volcengine APMPlus
 
         if llm_response.usage_metadata: