resolved pylint errors

Author: Harmanpreet Kaur

src/backend/tests/agents/test_agentutils.py

@@ -1,10 +1,13 @@
-import pytest
-import sys
 import os
-import json  # Fix for missing import
+import sys
+import json
+import pytest
 from unittest.mock import AsyncMock, MagicMock, patch
 from pydantic import ValidationError
+
+# Mocking modules before importing them
 sys.modules["azure.monitor.events.extension"] = MagicMock()
+
 # Set environment variables to mock Config dependencies before any import
 os.environ["COSMOSDB_ENDPOINT"] = "https://mock-endpoint"
 os.environ["COSMOSDB_KEY"] = "mock-key"
@@ -13,8 +16,9 @@
 os.environ["AZURE_OPENAI_DEPLOYMENT_NAME"] = "mock-deployment-name"
 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
+
+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
 
 @pytest.mark.asyncio
 async def test_extract_and_update_transition_states_invalid_response():

src/backend/tests/agents/test_generic.py

@@ -1,19 +1,20 @@
-import sys
 import os
-from unittest.mock import MagicMock
 import unittest
-sys.modules["azure.monitor.events.extension"] = MagicMock()
+from unittest.mock import MagicMock
+from autogen_core.components.models import AzureOpenAIChatCompletionClient
+from autogen_core.base import AgentId
+from src.backend.context.cosmos_memory import CosmosBufferedChatCompletionContext
+from src.backend.agents.generic import get_generic_tools, dummy_function
+
+
+# Set environment variables to mock Config dependencies before any import
 os.environ["COSMOSDB_ENDPOINT"] = "https://mock-endpoint"
 os.environ["COSMOSDB_KEY"] = "mock-key"
 os.environ["COSMOSDB_DATABASE"] = "mock-database"
 os.environ["COSMOSDB_CONTAINER"] = "mock-container"
 os.environ["AZURE_OPENAI_DEPLOYMENT_NAME"] = "mock-deployment-name"
 os.environ["AZURE_OPENAI_API_VERSION"] = "2023-01-01"
 os.environ["AZURE_OPENAI_ENDPOINT"] = "https://mock-openai-endpoint"
-from autogen_core.components.models import AzureOpenAIChatCompletionClient
-from autogen_core.base import AgentId
-from src.backend.context.cosmos_memory import CosmosBufferedChatCompletionContext
-from src.backend.agents.generic import get_generic_tools, GenericAgent, dummy_function
 
 
 class TestGenericAgent(unittest.TestCase):