Fix linting and type checking issues

Author: danielbentes

implementations/uim-mock-agent/src/discovery.py

@@ -4,6 +4,7 @@
 This module provides functionality for discovering UIM services and their intents
 by fetching and parsing agents.json files from UIM-compatible web services.
 """
+
 import json
 from typing import Dict
 
@@ -44,7 +45,8 @@ def extract_intent_metadata(agents_data: Dict) -> Dict:
         agents_data: The parsed agents.json data
 
     Returns:
-        Dict: A dictionary containing extracted intents, execute endpoint, and service URL
+        Dict: A dictionary containing extracted intents, execute
+        endpoint, and service URL
     """
     print(f"Debug: agents_data structure: {json.dumps(agents_data, indent=2)}")
     intents = []
@@ -75,7 +77,7 @@ def extract_intent_metadata(agents_data: Dict) -> Dict:
 
 def main():
     """
-    Main function for testing the discovery module functionality.
+    Test the discovery module functionality.
 
     Fetches agents.json, extracts intent metadata, and displays the results.
     """

implementations/uim-mock-agent/src/intent_execution.py

@@ -4,6 +4,7 @@
 This module provides functionality for executing intents against UIM-compatible
 web services, including parameter handling and authentication.
 """
+
 from typing import Any, Dict
 
 import requests

implementations/uim-mock-agent/src/key_management.py

@@ -4,6 +4,7 @@
 This module provides functionality for generating, storing, and retrieving
 cryptographic key pairs used for signing and verifying UIM protocol messages.
 """
+
 import os
 
 from cryptography.hazmat.backends import default_backend

implementations/uim-mock-agent/src/pat_issuance.py

@@ -4,6 +4,7 @@
 This module provides functionality for obtaining Policy Adherence Tokens (PATs)
 from UIM-compatible web services by submitting signed policies.
 """
+
 import base64
 from typing import Dict
 

implementations/uim-mock-agent/src/policy_management.py

@@ -4,6 +4,7 @@
 This module provides functionality for fetching, displaying, and processing
 ODRL policies from UIM-compatible web services.
 """
+
 import requests
 from error_handling import APIError, NetworkError
 from policy_signing import sign_policy, submit_signed_policy_and_get_pat
@@ -80,9 +81,10 @@ def display_policy(policy):
 
 
 def process_policy():
-    """
-    Process the ODRL policy by fetching, displaying, signing, and submitting it
-    to obtain a Policy Adherence Token (PAT).
+    """Process the ODRL policy by fetching, displaying, signing, and submitting it.
+
+    This function handles the complete policy processing workflow to obtain a
+    Policy Adherence Token (PAT).
 
     Returns:
         dict or None: The PAT result if successful, None if an error occurred

implementations/uim-mock-agent/src/policy_signing.py

@@ -4,10 +4,10 @@
 This module provides functionality for signing ODRL policies and submitting them
 to UIM-compatible web services to obtain Policy Adherence Tokens (PATs).
 """
+
 from typing import Dict
 
 import jwt
-from cryptography.hazmat.primitives import serialization
 from error_handling import APIError
 from key_management import get_key_pair
 from pat_issuance import handle_pat_issuance
@@ -48,13 +48,9 @@ def submit_signed_policy_and_get_pat(signed_policy: str, service_url: str) -> Di
     """
     try:
         agent_id = "ai-agent-1"  # This should be a unique identifier for your AI agent
-        _, public_key = get_key_pair(service_url)
-        public_key_pem = public_key.public_bytes(
-            encoding=serialization.Encoding.PEM,
-            format=serialization.PublicFormat.SubjectPublicKeyInfo,
-        ).decode("utf-8")
-
-        result = handle_pat_issuance(signed_policy, agent_id, public_key_pem)
+        # Get the key pair but we only need the agent_id for handle_pat_issuance
+        # The pat_issuance module will handle getting the public key internally
+        result = handle_pat_issuance(signed_policy, agent_id)
 
         if result["success"]:
             return {"pat": result["pat"]}

poetry.lock

@@ -49,6 +49,9 @@ black = "^24.2.0"
 isort = "^5.13.2"
 pylint = "^3.3.4"
 pre-commit = "^3.6.2"
+flake8-docstrings = "^1.7.0"
+flake8-bugbear = "^24.12.12"
+flake8-comprehensions = "^3.16.0"
 
 [tool.poetry.group.docs.dependencies]
 mkdocs = "^1.6.0"