Merge branch 'UT-10888-backend' of https://github.com/microsoft/Multi-Agent-Custom-Automation-Engine-Solution-Accelerator into UT-10888-backend

Author: UtkarshMishra-Microsoft

.github/workflows/test.yml

@@ -38,6 +38,7 @@ jobs:
           python -m pip install --upgrade pip
           pip install -r src/backend/requirements.txt
           pip install pytest-cov
+          pip install pytest-asyncio
 
       - name: Check if test files exist
         id: check_tests

src/backend/tests/agents/test_agentutils.py

@@ -1,13 +1,15 @@
-import pytest
-import sys
+# pylint: disable=import-error, wrong-import-position, missing-module-docstring
+import json
 import os
-import json  # Fix for missing import
+import sys
 from unittest.mock import AsyncMock, MagicMock, patch
+import pytest
 from pydantic import ValidationError
 
+
+# Environment and module setup
 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"
 os.environ["COSMOSDB_DATABASE"] = "mock-database"
@@ -16,13 +18,9 @@
 os.environ["AZURE_OPENAI_API_VERSION"] = "2023-01-01"
 os.environ["AZURE_OPENAI_ENDPOINT"] = "https://mock-openai-endpoint"
 
-from autogen_core.components.models import (
-    AssistantMessage,
-    AzureOpenAIChatCompletionClient,
-)
-from src.backend.models.messages import Step
-from src.backend.context.cosmos_memory import CosmosBufferedChatCompletionContext
-from src.backend.agents.agentutils import extract_and_update_transition_states
+# noqa: F401 is to ignore unused import warnings (if any)
+from src.backend.agents.agentutils import extract_and_update_transition_states  # noqa: F401, C0413,E402
+from src.backend.models.messages import Step  # noqa: F401, C0413,E402
 
 
 @pytest.mark.asyncio

src/backend/tests/agents/test_generic.py

@@ -1,11 +1,11 @@
-import pytest
-import sys
 import os
-import json  # Fix for missing import
-from unittest.mock import AsyncMock, MagicMock, patch
-from pydantic import ValidationError
+import unittest
+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
 
-sys.modules["azure.monitor.events.extension"] = MagicMock()
 
 # Set environment variables to mock Config dependencies before any import
 os.environ["COSMOSDB_ENDPOINT"] = "https://mock-endpoint"
@@ -15,16 +15,6 @@
 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"
-import unittest
-from unittest.mock import MagicMock
-from typing import List
-
-from autogen_core.components.models import AzureOpenAIChatCompletionClient
-from autogen_core.components.tools import Tool
-from autogen_core.base import AgentId
-from src.backend.context.cosmos_memory import CosmosBufferedChatCompletionContext
-from src.backend.agents.base_agent import BaseAgent
-from src.backend.agents.generic import get_generic_tools, GenericAgent, dummy_function
 
 
 class TestGenericAgent(unittest.TestCase):