test: Ut 10888 backend

.flake8

@@ -2,4 +2,4 @@
 max-line-length = 88
 extend-ignore = E501
 exclude = .venv, frontend
-ignore = E203, W503, G004, G200
+ignore = E203, W503, G004, G200, E402

.github/workflows/test.yml

@@ -37,8 +37,6 @@ jobs:
         run: |
           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
@@ -50,7 +48,6 @@ jobs:
             echo "Test files found, running tests."
             echo "skip_tests=false" >> $GITHUB_ENV
           fi
-
       - name: Run tests with coverage
         if: env.skip_tests == 'false'
         run: |
@@ -59,4 +56,4 @@ jobs:
       - name: Skip coverage report if no tests
         if: env.skip_tests == 'true'
         run: |
-          echo "Skipping coverage report because no tests were found."
+          echo "Skipping coverage report because no tests were found."

src/backend/agents/agentutils.py

@@ -6,8 +6,8 @@
 )
 from pydantic import BaseModel
 
-from context.cosmos_memory import CosmosBufferedChatCompletionContext
-from models.messages import Step
+from src.backend.context.cosmos_memory import CosmosBufferedChatCompletionContext
+from src.backend.models.messages import Step
 
 common_agent_system_message = "If you do not have the information for the arguments of the function you need to call, do not call the function. Instead, respond back to the user requesting further information. You must not hallucinate or invent any of the information used as arguments in the function. For example, if you need to call a function that requires a delivery address, you must not generate 123 Example St. You must skip calling functions and return a clarification message along the lines of: Sorry, I'm missing some information I need to help you with that. Could you please provide the delivery address so I can do that for you?"
 

src/backend/agents/base_agent.py

@@ -13,15 +13,15 @@
 from autogen_core.components.tool_agent import tool_agent_caller_loop
 from autogen_core.components.tools import Tool
 
-from context.cosmos_memory import CosmosBufferedChatCompletionContext
-from models.messages import (
+from src.backend.context.cosmos_memory import CosmosBufferedChatCompletionContext
+from src.backend.models.messages import (
     ActionRequest,
     ActionResponse,
     AgentMessage,
     Step,
     StepStatus,
 )
-from event_utils import track_event_if_configured
+from src.backend.event_utils import track_event_if_configured
 
 
 class BaseAgent(RoutedAgent):

src/backend/agents/generic.py

@@ -5,8 +5,8 @@
 from autogen_core.components.models import AzureOpenAIChatCompletionClient
 from autogen_core.components.tools import FunctionTool, Tool
 
-from agents.base_agent import BaseAgent
-from context.cosmos_memory import CosmosBufferedChatCompletionContext
+from src.backend.agents.base_agent import BaseAgent
+from src.backend.context.cosmos_memory import CosmosBufferedChatCompletionContext
 
 
 async def dummy_function() -> str:

src/backend/agents/group_chat_manager.py

@@ -9,8 +9,8 @@
 from autogen_core.components import RoutedAgent, default_subscription, message_handler
 from autogen_core.components.models import AzureOpenAIChatCompletionClient
 
-from context.cosmos_memory import CosmosBufferedChatCompletionContext
-from models.messages import (
+from src.backend.context.cosmos_memory import CosmosBufferedChatCompletionContext
+from src.backend.models.messages import (
     ActionRequest,
     AgentMessage,
     BAgentType,
@@ -22,7 +22,7 @@
     StepStatus,
 )
 
-from event_utils import track_event_if_configured
+from src.backend.event_utils import track_event_if_configured
 
 
 @default_subscription

src/backend/agents/hr.py

@@ -6,8 +6,8 @@
 from autogen_core.components.tools import FunctionTool, Tool
 from typing_extensions import Annotated
 
-from agents.base_agent import BaseAgent
-from context.cosmos_memory import CosmosBufferedChatCompletionContext
+from src.backend.agents.base_agent import BaseAgent
+from src.backend.context.cosmos_memory import CosmosBufferedChatCompletionContext
 
 formatting_instructions = "Instructions: returning the output of this function call verbatim to the user in markdown. Then write AGENT SUMMARY: and then include a summary of what you did."
 

src/backend/agents/human.py

@@ -4,15 +4,15 @@
 from autogen_core.base import AgentId, MessageContext
 from autogen_core.components import RoutedAgent, default_subscription, message_handler
 
-from context.cosmos_memory import CosmosBufferedChatCompletionContext
-from models.messages import (
+from src.backend.context.cosmos_memory import CosmosBufferedChatCompletionContext
+from src.backend.models.messages import (
     ApprovalRequest,
     HumanFeedback,
     StepStatus,
     AgentMessage,
     Step,
 )
-from event_utils import track_event_if_configured
+from src.backend.event_utils import track_event_if_configured
 
 
 @default_subscription

src/backend/agents/marketing.py

@@ -5,8 +5,8 @@
 from autogen_core.components.models import AzureOpenAIChatCompletionClient
 from autogen_core.components.tools import FunctionTool, Tool
 
-from agents.base_agent import BaseAgent
-from context.cosmos_memory import CosmosBufferedChatCompletionContext
+from src.backend.agents.base_agent import BaseAgent
+from src.backend.context.cosmos_memory import CosmosBufferedChatCompletionContext
 
 
 # Define new Marketing tools (functions)

src/backend/agents/planner.py

@@ -13,8 +13,8 @@
 )
 from pydantic import BaseModel
 
-from context.cosmos_memory import CosmosBufferedChatCompletionContext
-from models.messages import (
+from src.backend.context.cosmos_memory import CosmosBufferedChatCompletionContext
+from src.backend.models.messages import (
     AgentMessage,
     HumanClarification,
     BAgentType,
@@ -26,7 +26,7 @@
     HumanFeedbackStatus,
 )
 
-from event_utils import track_event_if_configured
+from src.backend.event_utils import track_event_if_configured
 
 
 @default_subscription

src/backend/agents/procurement.py

@@ -6,8 +6,8 @@
 from autogen_core.components.tools import FunctionTool, Tool
 from typing_extensions import Annotated
 
-from agents.base_agent import BaseAgent
-from context.cosmos_memory import CosmosBufferedChatCompletionContext
+from src.backend.agents.base_agent import BaseAgent
+from src.backend.context.cosmos_memory import CosmosBufferedChatCompletionContext
 
 
 # Define new Procurement tools (functions)

src/backend/agents/product.py

@@ -8,8 +8,8 @@
 from autogen_core.components.tools import FunctionTool, Tool
 from typing_extensions import Annotated
 
-from agents.base_agent import BaseAgent
-from context.cosmos_memory import CosmosBufferedChatCompletionContext
+from src.backend.agents.base_agent import BaseAgent
+from src.backend.context.cosmos_memory import CosmosBufferedChatCompletionContext
 
 formatting_instructions = "Instructions: returning the output of this function call verbatim to the user in markdown. Then write AGENT SUMMARY: and then include a summary of what you did."
 

src/backend/agents/tech_support.py

@@ -6,8 +6,8 @@
 from autogen_core.components.tools import FunctionTool, Tool
 from typing_extensions import Annotated
 
-from agents.base_agent import BaseAgent
-from context.cosmos_memory import CosmosBufferedChatCompletionContext
+from src.backend.agents.base_agent import BaseAgent
+from src.backend.context.cosmos_memory import CosmosBufferedChatCompletionContext
 
 formatting_instructions = "Instructions: returning the output of this function call verbatim to the user in markdown. Then write AGENT SUMMARY: and then include a summary of what you did."
 
@@ -523,7 +523,6 @@ async def get_tech_information(
     Document Name: Contoso's IT Policy and Procedure Manual
     Domain: IT Policy
     Description: A comprehensive guide detailing the IT policies and procedures at Contoso, including acceptable use, security protocols, and incident reporting.
-
     At Contoso, we prioritize the security and efficiency of our IT infrastructure. All employees are required to adhere to the following policies:
     - Use strong passwords and change them every 90 days.
     - Report any suspicious emails to the IT department immediately.

src/backend/app.py

@@ -4,13 +4,13 @@
 import os
 import uuid
 from typing import List, Optional
-from middleware.health_check import HealthCheckMiddleware
+from src.backend.middleware.health_check import HealthCheckMiddleware
 from autogen_core.base import AgentId
 from fastapi import FastAPI, HTTPException, Query, Request
-from auth.auth_utils import get_authenticated_user_details
-from config import Config
-from context.cosmos_memory import CosmosBufferedChatCompletionContext
-from models.messages import (
+from src.backend.auth.auth_utils import get_authenticated_user_details
+from src.backend.config import Config
+from src.backend.context.cosmos_memory import CosmosBufferedChatCompletionContext
+from src.backend.models.messages import (
     HumanFeedback,
     HumanClarification,
     InputTask,
@@ -19,8 +19,8 @@
     AgentMessage,
     PlanWithSteps,
 )
-from utils import initialize_runtime_and_context, retrieve_all_agent_tools, rai_success
-from event_utils import track_event_if_configured
+from src.backend.utils import initialize_runtime_and_context, retrieve_all_agent_tools, rai_success
+from src.backend.event_utils import track_event_if_configured
 from fastapi.middleware.cors import CORSMiddleware
 from azure.monitor.opentelemetry import configure_azure_monitor
 from opentelemetry.instrumentation.fastapi import FastAPIInstrumentor

src/backend/context/cosmos_memory.py

@@ -15,8 +15,8 @@
 )
 from azure.cosmos.partition_key import PartitionKey
 
-from config import Config
-from models.messages import BaseDataModel, Plan, Session, Step, AgentMessage
+from src.backend.config import Config
+from src.backend.models.messages import BaseDataModel, Plan, Session, Step, AgentMessage
 
 
 class CosmosBufferedChatCompletionContext(BufferedChatCompletionContext):

src/backend/handlers/runtime_interrupt.py

@@ -3,7 +3,9 @@
 from autogen_core.base import AgentId
 from autogen_core.base.intervention import DefaultInterventionHandler
 
-from models.messages import GetHumanInputMessage, GroupChatMessage
+from src.backend.models.messages import GroupChatMessage
+
+from src.backend.models.messages import GetHumanInputMessage
 
 
 class NeedsUserInputHandler(DefaultInterventionHandler):

src/backend/models/messages.py

@@ -293,3 +293,11 @@ class RequestToSpeak(BaseModel):
 
     def to_dict(self):
         return self.model_dump()
+
+
+class GetHumanInputMessage:
+    def __init__(self, message):
+        self.message = message
+
+    def __str__(self):
+        return f"GetHumanInputMessage: {self.message}"

src/backend/requirements.txt

@@ -14,3 +14,8 @@ opentelemetry-instrumentation-fastapi
 opentelemetry-instrumentation-openai
 opentelemetry-exporter-otlp-proto-http
 opentelemetry-exporter-otlp-proto-grpc
+
+# Testing tools
+pytest>=8.2,<9  # Compatible version for pytest-asyncio
+pytest-asyncio==0.24.0
+pytest-cov==5.0.0

src/backend/tests/agents/test_agentutils.py

@@ -0,0 +1,54 @@
+# pylint: disable=import-error, wrong-import-position, missing-module-docstring
+import os
+import sys
+from unittest.mock import MagicMock
+import pytest
+from pydantic import ValidationError
+
+# Environment and module setup
+sys.modules["azure.monitor.events.extension"] = MagicMock()
+
+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 src.backend.agents.agentutils import extract_and_update_transition_states  # noqa: F401, C0413
+from src.backend.models.messages import Step  # noqa: F401, C0413
+
+
+def test_step_initialization():
+    """Test Step initialization with valid data."""
+    step = Step(
+        data_type="step",
+        plan_id="test_plan",
+        action="test_action",
+        agent="HumanAgent",
+        session_id="test_session",
+        user_id="test_user",
+        agent_reply="test_reply",
+    )
+
+    assert step.data_type == "step"
+    assert step.plan_id == "test_plan"
+    assert step.action == "test_action"
+    assert step.agent == "HumanAgent"
+    assert step.session_id == "test_session"
+    assert step.user_id == "test_user"
+    assert step.agent_reply == "test_reply"
+    assert step.status == "planned"
+    assert step.human_approval_status == "requested"
+
+
+def test_step_missing_required_fields():
+    """Test Step initialization with missing required fields."""
+    with pytest.raises(ValidationError):
+        Step(
+            data_type="step",
+            action="test_action",
+            agent="test_agent",
+            session_id="test_session",
+        )