Fix PR check errors: mypy, poetry deprecation, and test failure

Author: Antigravity Agent

pyproject.toml

@@ -2,24 +2,30 @@
 
 # PROJECT
 
-[tool.poetry]
+[project]
 name = "regression_model_template"
 version = "2.0.0"
 description = "Predict the number of regression_model_template available."
-repository = "https://github.com/lgcorzo/mlops-python-package"
-documentation = "https://lgcorzo.github.io/mlops-python-package/"
-authors = []
+authors = [{ name = "lgcorzo" }]
 readme = "README.md"
 license = "MIT"
 keywords = ["mlops", "python", "package"]
-packages = [{ include = "regression_model_template", from = "src" }]
+requires-python = ">=3.12"
+classifiers = [
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.12",
+]
+dynamic = ["dependencies"]
 
-# SCRIPTS
+[project.urls]
+repository = "https://github.com/lgcorzo/mlops-python-package"
+documentation = "https://lgcorzo.github.io/mlops-python-package/"
 
-[tool.poetry.scripts]
-regression_model_template = 'regression_model_template.scripts:main'
+[project.scripts]
+regression_model_template = "regression_model_template.scripts:main"
 
-# DEPENDENCIES
+[tool.poetry]
+packages = [{ include = "regression_model_template", from = "src" }]
 
 [tool.poetry.dependencies]
 python = "^3.12"
@@ -49,7 +55,6 @@ opentelemetry-exporter-otlp = "^1.30.0"
 confluent-kafka = "^2.8.2"
 fastapi = "^0.115.11"
 
-
 [tool.poetry.group.checks.dependencies]
 bandit = "^1.7.9"
 coverage = "^7.5.4"

src/regression_model_template/controller/kafka_app.py

@@ -6,6 +6,7 @@
 import logging
 import time
 import json
+import typing as T
 from typing import Any, Dict, Callable
 
 import uvicorn
@@ -183,7 +184,7 @@ def _process_message(self, msg: Message) -> None:
             prediction_result = self.prediction_callback(input_obj).result
         except Exception as e:
             logger.exception(f"Error during prediction processing: {e}")
-            predictionresponse.result["error"] = "Internal Processing Error"
+            predictionresponse.result["error"] = "An error occurred during prediction processing."
             prediction_result = predictionresponse.result
 
         try:
@@ -220,7 +221,7 @@ def stop(self) -> None:
 
 
 # Global Service Instance
-fastapi_kafka_service: Any = None
+fastapi_kafka_service: "FastAPIKafkaService" = T.cast("FastAPIKafkaService", None)
 
 
 # FastAPI Endpoints
@@ -239,7 +240,7 @@ async def predict(request: PredictionRequest) -> PredictionResponse:  # Use glob
         prediction_result = fastapi_kafka_service.prediction_callback(request)
         logger.info(f"HTTP prediction result: {prediction_result}")
         return prediction_result  # Use the global class
-    except Exception as e:
+    except Exception:
         logger.exception("Error processing HTTP prediction request:")
         raise HTTPException(status_code=500, detail="Internal Server Error")
 
@@ -269,7 +270,7 @@ def predict(self, input_data: PredictionRequest) -> PredictionResponse:
             logger.exception(f"Prediction failed: {e}")
             predictionresponse.result["inference"] = 0
             predictionresponse.result["quality"] = 0
-            predictionresponse.result["error"] = "Internal Processing Error"
+            predictionresponse.result["error"] = "An error occurred during prediction processing."
         return predictionresponse
 
 

tests/controller/test_kafka_app_security.py

@@ -5,7 +5,6 @@
     PredictionRequest,
     PredictionService,
     predict,
-    PredictionResponse,
 )
 import asyncio
 
@@ -25,7 +24,7 @@ def test_prediction_service_sanitization():
     response = service.predict(request)
 
     # Verify
-    assert response.result["error"] == "Internal Processing Error"
+    assert response.result["error"] == "An error occurred during prediction processing."
     assert sensitive_error not in response.result["error"]
     assert response.result["quality"] == 0
     assert response.result["inference"] == 0