changed var names according to linter

Author: haneric00

aws-opentelemetry-distro/tests/amazon/opentelemetry/distro/test-opentelemetry-instrumentation-langchain-v2/mock_langgraph_agent.py

@@ -44,10 +44,10 @@ def calculate(state: State):
                 return {"result": response.content}
 
             # Patch StateGraph to avoid actual execution
-            with patch("langgraph.graph.StateGraph", autospec=True) as MockStateGraph:
+            with patch("langgraph.graph.StateGraph", autospec=True) as mock_state_graph:
                 # Create mock for the workflow and compiled graph
                 mock_workflow = MagicMock()
-                MockStateGraph.return_value = mock_workflow
+                mock_state_graph.return_value = mock_workflow
                 mock_compiled_graph = MagicMock()
                 mock_workflow.compile.return_value = mock_compiled_graph
 
@@ -57,7 +57,7 @@ async def mock_ainvoke(*args, **kwargs):
 
                 mock_compiled_graph.ainvoke = mock_ainvoke
 
-                workflow = MockStateGraph(State)
+                workflow = mock_state_graph(State)
                 workflow.add_node("calculate", calculate)
                 workflow.set_entry_point("calculate")
 
@@ -106,18 +106,18 @@ def mynode_func(state: DummyGraphState) -> DummyGraphState:
         return state
 
     # Patch StateGraph to avoid actual execution
-    with patch("langgraph.graph.StateGraph", autospec=True) as MockStateGraph:
+    with patch("langgraph.graph.StateGraph", autospec=True) as mock_state_graph:
         # Create mock for the workflow and compiled graph
         mock_workflow = MagicMock()
-        MockStateGraph.return_value = mock_workflow
+        mock_state_graph.return_value = mock_workflow
         mock_compiled_graph = MagicMock()
         mock_workflow.compile.return_value = mock_compiled_graph
 
         # Set up response for the invoke method of the compiled graph
         mock_compiled_graph.invoke.return_value = {"result": "init"}
 
         def build_graph():
-            workflow = MockStateGraph(DummyGraphState)
+            workflow = mock_state_graph(DummyGraphState)
             workflow.add_node("mynode", mynode_func)
             workflow.set_entry_point("mynode")
             langgraph = workflow.compile()
@@ -184,10 +184,10 @@ def mynode_func(state: DummyGraphState) -> DummyGraphState:
         return state
 
     # Patch StateGraph to avoid actual execution
-    with patch("langgraph.graph.StateGraph", autospec=True) as MockStateGraph:
+    with patch("langgraph.graph.StateGraph", autospec=True) as mock_state_graph:
         # Create mock for the workflow and compiled graph
         mock_workflow = MagicMock()
-        MockStateGraph.return_value = mock_workflow
+        mock_state_graph.return_value = mock_workflow
         mock_compiled_graph = MagicMock()
         mock_workflow.compile.return_value = mock_compiled_graph
 
@@ -198,7 +198,7 @@ async def mock_ainvoke(*args, **kwargs):
         mock_compiled_graph.ainvoke = mock_ainvoke
 
         def build_graph():
-            workflow = MockStateGraph(DummyGraphState)
+            workflow = mock_state_graph(DummyGraphState)
             workflow.add_node("mynode", mynode_func)
             workflow.set_entry_point("mynode")
             langgraph = workflow.compile()

aws-opentelemetry-distro/tests/amazon/opentelemetry/distro/test-opentelemetry-instrumentation-langchain-v2/test_callback_handler.py

@@ -188,9 +188,9 @@ def mock_create_span(run_id, parent_run_id, name, kind, metadata):
                 self.handler.on_llm_end(response=response, run_id=self.run_id, parent_run_id=self.parent_run_id)
 
                 print("\nAll calls to mock_set_attribute:")
-                for i, call in enumerate(mock_set_attribute.call_args_list):
+                for idx, call in enumerate(mock_set_attribute.call_args_list):
                     args, kwargs = call
-                    print(f"Call {i+1}:", args, kwargs)
+                    print(f"Call {idx+1}:", args, kwargs)
 
                 mock_set_attribute.assert_any_call(self.mock_span, SpanAttributes.GEN_AI_RESPONSE_MODEL, "gpt-4")
                 mock_set_attribute.assert_any_call(self.mock_span, SpanAttributes.GEN_AI_RESPONSE_ID, "response-123")