edit 2

Author: Harmanpreet Kaur

src/backend/tests/agents/test_agentutils.py

@@ -17,8 +17,10 @@
 os.environ["AZURE_OPENAI_API_VERSION"] = "2023-01-01"
 os.environ["AZURE_OPENAI_ENDPOINT"] = "https://mock-openai-endpoint"
 
-from src.backend.models.messages import Step  # Ensure used imports remain
-from src.backend.agents.agentutils import extract_and_update_transition_states  # Ensure used imports remain
+# Ensure used imports remain
+from src.backend.models.messages import Step
+from src.backend.agents.agentutils import extract_and_update_transition_states
+
 
 @pytest.mark.asyncio
 async def test_extract_and_update_transition_states_invalid_response():