edit 3

Author: Harmanpreet Kaur

src/backend/tests/agents/test_agentutils.py

@@ -1,11 +1,14 @@
+import json
 import os
 import sys
-import json
-import pytest
 from unittest.mock import AsyncMock, MagicMock, patch
+import pytest
 from pydantic import ValidationError
-# pylint: disable=E402
 
+from src.backend.agents.agentutils import extract_and_update_transition_states
+from src.backend.models.messages import Step
+# pylint: disable=import-error
+# Environment and module setup
 sys.modules["azure.monitor.events.extension"] = MagicMock()
 
 os.environ["COSMOSDB_ENDPOINT"] = "https://mock-endpoint"
@@ -16,8 +19,7 @@
 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
-from src.backend.agents.agentutils import extract_and_update_transition_states
+# Ensure imports are at the top
 
 
 @pytest.mark.asyncio