Pylint issues fixed

Author: Roopan-Microsoft

src/backend/agents/agentutils.py

@@ -7,7 +7,7 @@
 from pydantic import BaseModel
 
 from src.backend.context.cosmos_memory import CosmosBufferedChatCompletionContext
-from src.backend.models.messages import InputTask, PlanStatus, Step, StepStatus
+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

@@ -15,11 +15,11 @@
 
 from src.backend.context.cosmos_memory import CosmosBufferedChatCompletionContext
 from src.backend.models.messages import (
-    ActionRequest, 
+    ActionRequest,
     ActionResponse,
-    AgentMessage, 
-    Step, 
-    StepStatus
+    AgentMessage,
+    Step,
+    StepStatus,
 )
 from azure.monitor.events.extension import track_event
 

src/backend/agents/planner.py

@@ -15,7 +15,6 @@
 
 from src.backend.context.cosmos_memory import CosmosBufferedChatCompletionContext
 from src.backend.models.messages import (
-    ActionRequest,
     AgentMessage,
     HumanClarification,
     BAgentType,

src/backend/app.py

@@ -6,14 +6,11 @@
 from typing import List, Optional
 from src.backend.middleware.health_check import HealthCheckMiddleware
 from autogen_core.base import AgentId
-from fastapi import Depends, FastAPI, HTTPException, Query, Request
-from fastapi.responses import RedirectResponse
-from fastapi.staticfiles import StaticFiles
+from fastapi import FastAPI, HTTPException, Query, Request
 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 (
-    BaseDataModel,
     HumanFeedback,
     HumanClarification,
     InputTask,
@@ -22,7 +19,11 @@
     AgentMessage,
     PlanWithSteps,
 )
-from src.backend.utils import initialize_runtime_and_context, retrieve_all_agent_tools, rai_success
+from src.backend.utils import (
+    initialize_runtime_and_context,
+    retrieve_all_agent_tools,
+    rai_success,
+)
 import asyncio
 from fastapi.middleware.cors import CORSMiddleware
 from azure.monitor.opentelemetry import configure_azure_monitor

src/backend/models/messages.py

@@ -294,6 +294,7 @@ class RequestToSpeak(BaseModel):
     def to_dict(self):
         return self.model_dump()
 
+
 class GetHumanInputMessage:
     def __init__(self, message):
         self.message = message

src/backend/tests/agents/test_tech_support.py

@@ -1,10 +1,10 @@
 import os
 import sys
 import pytest
-from unittest.mock import AsyncMock, MagicMock
+from unittest.mock import MagicMock
 from autogen_core.components.tools import FunctionTool
 
-sys.modules['azure.monitor.events.extension'] = MagicMock()
+sys.modules["azure.monitor.events.extension"] = MagicMock()
 
 # Set environment variables to mock Config dependencies before any import
 os.environ["COSMOSDB_ENDPOINT"] = "https://mock-endpoint"
@@ -36,11 +36,9 @@
     configure_server,
     grant_database_access,
     provide_tech_training,
-    resolve_technical_issue,
     configure_printer,
     set_up_email_signature,
     configure_mobile_device,
-    manage_software_licenses,
     set_up_remote_desktop,
     troubleshoot_hardware_issue,
     manage_network_security,
@@ -80,25 +78,29 @@ async def test_send_welcome_email():
     assert "John Doe" in result
     assert "[email protected]" in result
 
+
 @pytest.mark.asyncio
 async def test_set_up_office_365_account():
     result = await set_up_office_365_account("Jane Smith", "[email protected]")
     assert "Office 365 Account Setup" in result
     assert "Jane Smith" in result
     assert "[email protected]" in result
 
+
 @pytest.mark.asyncio
 async def test_configure_laptop():
     result = await configure_laptop("John Doe", "Dell XPS 15")
     assert "Laptop Configuration" in result
     assert "Dell XPS 15" in result
 
+
 @pytest.mark.asyncio
 async def test_reset_password():
     result = await reset_password("John Doe")
     assert "Password Reset" in result
     assert "John Doe" in result
 
+
 @pytest.mark.asyncio
 async def test_setup_vpn_access():
     result = await setup_vpn_access("John Doe")
@@ -147,12 +149,14 @@ async def test_assist_procurement_with_tech_equipment():
     assert "Technical Specifications Provided" in result
     assert "Dell Workstation specs" in result
 
+
 @pytest.mark.asyncio
 async def test_provide_tech_support_for_marketing():
     result = await provide_tech_support_for_marketing("Holiday Campaign")
     assert "Tech Support for Marketing Campaign" in result
     assert "Holiday Campaign" in result
 
+
 @pytest.mark.asyncio
 async def test_assist_product_launch():
     result = await assist_product_launch("Smartphone X")
@@ -220,6 +224,7 @@ async def test_configure_mobile_device():
     assert "Emily" in result
     assert "iPhone 13" in result
 
+
 @pytest.mark.asyncio
 async def test_set_up_remote_desktop():
     result = await set_up_remote_desktop("Frank")
@@ -261,6 +266,7 @@ async def test_manage_it_inventory():
     result = await manage_it_inventory()
     assert "IT Inventory Managed" in result
 
+
 @pytest.mark.asyncio
 async def test_configure_firewall_rules():
     result = await configure_firewall_rules("Allow traffic on port 8080")
@@ -336,7 +342,9 @@ async def test_manage_system_updates():
 
 @pytest.mark.asyncio
 async def test_configure_digital_signatures():
-    result = await configure_digital_signatures("John Doe", "Company Approved Signature")
+    result = await configure_digital_signatures(
+        "John Doe", "Company Approved Signature"
+    )
     assert "Digital Signatures Configured" in result
     assert "John Doe" in result
     assert "Company Approved Signature" in result
@@ -373,4 +381,4 @@ def test_get_tech_support_tools():
     tools = get_tech_support_tools()
     assert isinstance(tools, list)
     assert len(tools) > 40  # Ensure all tools are included
-    assert all(isinstance(tool, FunctionTool) for tool in tools)
+    assert all(isinstance(tool, FunctionTool) for tool in tools)

src/backend/tests/auth/test_auth_utils.py

@@ -1,4 +1,3 @@
-import pytest
 from unittest.mock import patch, Mock
 import base64
 import json
@@ -51,4 +50,4 @@ def test_get_tenantid_with_invalid_b64(mock_logger):
     tenant_id = get_tenantid(invalid_b64)
 
     assert tenant_id == ""
-    mock_logger().exception.assert_called_once()
+    mock_logger().exception.assert_called_once()

src/backend/tests/auth/test_sample_user.py

@@ -1,4 +1,3 @@
-import pytest
 from src.backend.auth.sample_user import sample_user  # Adjust path as necessary
 
 
@@ -48,10 +47,13 @@ def test_sample_user_keys():
 
 def test_sample_user_values():
     # Proceed with assertions
-    assert sample_user["Accept"].strip() == '*/*'  # Ensure no hidden characters
+    assert sample_user["Accept"].strip() == "*/*"  # Ensure no hidden characters
     assert sample_user["Content-Type"] == "application/json"
     assert sample_user["Disguised-Host"] == "your_app_service.azurewebsites.net"
-    assert sample_user["X-Ms-Client-Principal-Id"] == "00000000-0000-0000-0000-000000000000"
+    assert (
+        sample_user["X-Ms-Client-Principal-Id"]
+        == "00000000-0000-0000-0000-000000000000"
+    )
     assert sample_user["X-Ms-Client-Principal-Name"] == "[email protected]"
     assert sample_user["X-Forwarded-Proto"] == "https"
 
@@ -79,4 +81,4 @@ def test_sample_user_user_agent():
     user_agent = sample_user["User-Agent"]
     assert "Mozilla/5.0" in user_agent
     assert "Windows NT 10.0" in user_agent
-    assert "Edg/" in user_agent  # Matches Edge's identifier more accurately
+    assert "Edg/" in user_agent  # Matches Edge's identifier more accurately