Please pylint

Author: xrmx

instrumentation/opentelemetry-instrumentation-botocore/src/opentelemetry/instrumentation/botocore/extensions/bedrock_utils.py

@@ -75,6 +75,7 @@ def __iter__(self):
             raise
 
     def _process_event(self, event):
+        # pylint: disable=too-many-branches
         if "messageStart" in event:
             # {'messageStart': {'role': 'assistant'}}
             if event["messageStart"].get("role") == "assistant":
@@ -214,6 +215,7 @@ def _process_amazon_titan_chunk(self, chunk):
             self._stream_done_callback(self._response)
 
     def _process_amazon_nova_chunk(self, chunk):
+        # pylint: disable=too-many-branches
         # TODO:  handle tool calls!
         if "messageStart" in chunk:
             # {'messageStart': {'role': 'assistant'}}
@@ -262,7 +264,7 @@ def _process_amazon_nova_chunk(self, chunk):
             return
 
     def _process_anthropic_claude_chunk(self, chunk):
-        # pylint: disable=too-many-return-statements
+        # pylint: disable=too-many-return-statements,too-many-branches
         if not (message_type := chunk.get("type")):
             return
 
@@ -343,10 +345,10 @@ def genai_capture_message_content() -> bool:
 
 def extract_tool_calls(
     message: dict[str, Any], capture_content: bool
-) -> Sequence[Dict[str, Any]]:
+) -> Sequence[Dict[str, Any]] | None:
     content = message.get("content")
     if not content:
-        return
+        return None
 
     tool_uses = [item["toolUse"] for item in content if "toolUse" in item]
     if not tool_uses:
@@ -358,7 +360,7 @@ def extract_tool_calls(
         tool_id_key = "toolUseId"
 
     if not tool_uses:
-        return
+        return None
 
     tool_calls = []
     for tool_use in tool_uses:

instrumentation/opentelemetry-instrumentation-botocore/tests/test_botocore_bedrock.py

@@ -154,6 +154,7 @@ def test_converse_with_content_different_events(
 def converse_tool_call(
     span_exporter, log_exporter, bedrock_runtime_client, expect_content
 ):
+    # pylint:disable=too-many-locals
     messages = [
         {
             "role": "user",
@@ -648,6 +649,7 @@ def _rebuild_stream_message(response):
 def converse_stream_tool_call(
     span_exporter, log_exporter, bedrock_runtime_client, expect_content
 ):
+    # pylint:disable=too-many-locals,too-many-statements
     messages = [
         {
             "role": "user",
@@ -1294,6 +1296,7 @@ def invoke_model_tool_call(
     llm_model_value,
     expect_content,
 ):
+    # pylint:disable=too-many-locals,too-many-statements
     messages = [
         {
             "role": "user",
@@ -1862,6 +1865,7 @@ def invoke_model_with_response_stream_tool_call(
     llm_model_value,
     expect_content,
 ):
+    # pylint:disable=too-many-locals,too-many-statements,too-many-branches
     messages = [
         {
             "role": "user",