[Mega-Linter] Apply linters fixes

Author: umaannamalai

newrelic/agent.py

@@ -139,6 +139,9 @@ def __asgi_application(*args, **kwargs):
 from newrelic.api.html_insertion import verify_body_exists as __verify_body_exists
 from newrelic.api.lambda_handler import LambdaHandlerWrapper as __LambdaHandlerWrapper
 from newrelic.api.lambda_handler import lambda_handler as __lambda_handler
+from newrelic.api.llm_custom_attributes import (
+    WithLlmCustomAttributes as __WithLlmCustomAttributes,
+)
 from newrelic.api.message_trace import MessageTrace as __MessageTrace
 from newrelic.api.message_trace import MessageTraceWrapper as __MessageTraceWrapper
 from newrelic.api.message_trace import message_trace as __message_trace
@@ -156,9 +159,10 @@ def __asgi_application(*args, **kwargs):
 from newrelic.api.ml_model import (
     record_llm_feedback_event as __record_llm_feedback_event,
 )
-from newrelic.api.ml_model import set_llm_token_count_callback as __set_llm_token_count_callback
+from newrelic.api.ml_model import (
+    set_llm_token_count_callback as __set_llm_token_count_callback,
+)
 from newrelic.api.ml_model import wrap_mlmodel as __wrap_mlmodel
-from newrelic.api.llm_custom_attributes import WithLlmCustomAttributes as __WithLlmCustomAttributes
 from newrelic.api.profile_trace import ProfileTraceWrapper as __ProfileTraceWrapper
 from newrelic.api.profile_trace import profile_trace as __profile_trace
 from newrelic.api.profile_trace import wrap_profile_trace as __wrap_profile_trace

newrelic/api/llm_custom_attributes.py

@@ -13,6 +13,7 @@
 # limitations under the License.
 
 import logging
+
 from newrelic.api.transaction import current_transaction
 
 _logger = logging.getLogger(__name__)
@@ -22,7 +23,9 @@ class WithLlmCustomAttributes(object):
     def __init__(self, custom_attr_dict):
         transaction = current_transaction()
         if not custom_attr_dict or not isinstance(custom_attr_dict, dict):
-            raise TypeError("custom_attr_dict must be a non-empty dictionary. Received type: %s" % type(custom_attr_dict))
+            raise TypeError(
+                "custom_attr_dict must be a non-empty dictionary. Received type: %s" % type(custom_attr_dict)
+            )
 
         # Add "llm." prefix to all keys in attribute dictionary
         context_attrs = {k if k.startswith("llm.") else f"llm.{k}": v for k, v in custom_attr_dict.items()}

newrelic/hooks/mlmodel_langchain.py

@@ -452,9 +452,7 @@ def _record_tool_success(
     try:
         result = str(response)
     except Exception:
-        _logger.debug(
-            f"Failed to convert tool response into a string.\n{traceback.format_exception(*sys.exc_info())}"
-        )
+        _logger.debug(f"Failed to convert tool response into a string.\n{traceback.format_exception(*sys.exc_info())}")
     if settings.ai_monitoring.record_content.enabled:
         full_tool_event_dict.update(
             {

tests/agent_features/test_llm_custom_attributes.py

@@ -1,4 +1,3 @@
-
 # Copyright 2010 New Relic, Inc.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -16,8 +15,8 @@
 import pytest
 
 from newrelic.api.background_task import background_task
-from newrelic.api.transaction import current_transaction
 from newrelic.api.llm_custom_attributes import WithLlmCustomAttributes
+from newrelic.api.transaction import current_transaction
 
 
 @background_task()

tests/external_botocore/test_bedrock_chat_completion.py

@@ -59,8 +59,8 @@
 )
 
 from newrelic.api.background_task import background_task
-from newrelic.api.transaction import add_custom_attribute
 from newrelic.api.llm_custom_attributes import WithLlmCustomAttributes
+from newrelic.api.transaction import add_custom_attribute
 from newrelic.common.object_names import callable_name
 from newrelic.hooks.external_botocore import MODEL_EXTRACTORS
 

tests/external_botocore/test_bedrock_chat_completion_via_langchain.py

@@ -19,7 +19,10 @@
 )
 from conftest import BOTOCORE_VERSION  # pylint: disable=E0611
 from testing_support.fixtures import reset_core_stats_engine, validate_attributes
-from testing_support.ml_testing_utils import set_trace_info, events_with_context_attrs  # noqa: F401
+from testing_support.ml_testing_utils import (  # noqa: F401
+    events_with_context_attrs,
+    set_trace_info,
+)
 from testing_support.validators.validate_custom_event import validate_custom_event_count
 from testing_support.validators.validate_custom_events import validate_custom_events
 from testing_support.validators.validate_transaction_metrics import (

tests/mlmodel_langchain/test_chain.py

@@ -47,8 +47,8 @@
 )
 
 from newrelic.api.background_task import background_task
-from newrelic.api.transaction import add_custom_attribute
 from newrelic.api.llm_custom_attributes import WithLlmCustomAttributes
+from newrelic.api.transaction import add_custom_attribute
 from newrelic.common.object_names import callable_name
 
 _test_openai_chat_completion_messages = (

tests/mlmodel_langchain/test_vectorstore.py

@@ -36,8 +36,8 @@
 )
 
 from newrelic.api.background_task import background_task
-from newrelic.api.transaction import add_custom_attribute
 from newrelic.api.llm_custom_attributes import WithLlmCustomAttributes
+from newrelic.api.transaction import add_custom_attribute
 from newrelic.common.object_names import callable_name
 
 

tests/mlmodel_openai/test_chat_completion.py

@@ -34,10 +34,10 @@
 from testing_support.validators.validate_transaction_metrics import (
     validate_transaction_metrics,
 )
+
 from newrelic.api.background_task import background_task
-from newrelic.api.transaction import add_custom_attribute
 from newrelic.api.llm_custom_attributes import WithLlmCustomAttributes
-
+from newrelic.api.transaction import add_custom_attribute
 
 _test_openai_chat_completion_messages = (
     {"role": "system", "content": "You are a scientist."},

tests/mlmodel_openai/test_chat_completion_error.py

@@ -39,8 +39,8 @@
 )
 
 from newrelic.api.background_task import background_task
-from newrelic.api.transaction import add_custom_attribute
 from newrelic.api.llm_custom_attributes import WithLlmCustomAttributes
+from newrelic.api.transaction import add_custom_attribute
 from newrelic.common.object_names import callable_name
 
 _test_openai_chat_completion_messages = (