ci fixes

Author: Zvi Fried

pyproject.toml

@@ -33,6 +33,7 @@ dependencies = [
     "jinja2>=3.1.0",
     "litellm>=1.0.0",
     "sqlmodel>=0.0.24",
+    "tenacity>=9.1.2",
 ]
 
 [project.urls]

src/mcp_as_a_judge/db/providers/sqlite_provider.py

@@ -130,7 +130,9 @@ async def _cleanup_old_messages(self, session_id: str) -> int:
                 oldest_stmt = (
                     select(ConversationRecord)
                     .where(ConversationRecord.session_id == session_id)
-                    .order_by(asc(ConversationRecord.timestamp), asc(ConversationRecord.id))
+                    .order_by(
+                        asc(ConversationRecord.timestamp), asc(ConversationRecord.id)
+                    )
                     .limit(1)
                 )
                 oldest_record = session.exec(oldest_stmt).first()

src/mcp_as_a_judge/llm/llm_client.py

@@ -132,9 +132,7 @@ def _get_model_name(self) -> str:
         wait=wait_exponential(multiplier=2, min=2, max=120),
         reraise=True,
     )
-    async def _generate_text_with_retry(
-        self, completion_params: dict[str, Any]
-    ) -> Any:
+    async def _generate_text_with_retry(self, completion_params: dict[str, Any]) -> Any:
         """Generate text with retry logic for rate limit errors.
 
         This method is decorated with tenacity retry logic to handle
@@ -150,7 +148,9 @@ async def _generate_text_with_retry(
             litellm.RateLimitError: If rate limit is exceeded after all retries
             Exception: For other LiteLLM errors
         """
-        logger.debug(f"Attempting LLM completion with model: {completion_params.get('model')}")
+        logger.debug(
+            f"Attempting LLM completion with model: {completion_params.get('model')}"
+        )
 
         # Run the synchronous LiteLLM call in a thread pool
         response = await asyncio.get_event_loop().run_in_executor(

src/mcp_as_a_judge/messaging/mcp_sampling_provider.py

@@ -12,6 +12,7 @@
 from mcp.server.fastmcp import Context
 from mcp.types import SamplingMessage, TextContent
 
+from mcp_as_a_judge.core.logging_config import get_logger
 from mcp_as_a_judge.messaging.converters import messages_to_mcp_format
 from mcp_as_a_judge.messaging.interface import (
     Message,
@@ -36,6 +37,7 @@ def __init__(self, context: Context):
             context: MCP context with session for sampling
         """
         self.context = context
+        self.logger = get_logger(__name__)
 
     async def _send_message(
         self, messages: list[Message], config: MessagingConfig
@@ -176,8 +178,10 @@ def _normalize_mcp_messages(self, mcp_messages: list[Any]) -> list[Any]:
                             )
                         )
                         continue
-            except Exception:
+            except Exception as e:
                 # Fall through to append original if normalization fails
-                pass
+                self.logger.debug_sync(
+                    f"Failed to normalize message, using original: {e}"
+                )
             normalized.append(msg)
         return normalized

src/mcp_as_a_judge/server.py

@@ -239,7 +239,9 @@ async def set_coding_task(
         return error_result
 
 
-@mcp.tool(description=tool_description_provider.get_description("get_current_coding_task"))  # type: ignore[misc,unused-ignore]
+@mcp.tool(
+    description=tool_description_provider.get_description("get_current_coding_task")
+)  # type: ignore[misc,unused-ignore]
 async def get_current_coding_task(ctx: Context) -> dict:
     """Return the most recently active coding task's task_id and metadata.
 
@@ -284,7 +286,9 @@ async def get_current_coding_task(ctx: Context) -> dict:
                 mode="json", exclude_none=True
             )
         else:
-            response["note"] = "Task metadata not found in history for this session, but a session exists. Use this task_id UUID and proceed; if validation fails, recreate with set_coding_task."
+            response["note"] = (
+                "Task metadata not found in history for this session, but a session exists. Use this task_id UUID and proceed; if validation fails, recreate with set_coding_task."
+            )
 
         return response
     except Exception as e: