enhanced workflow

Author: Zvi Fried

src/mcp_as_a_judge/coding_task_manager.py

@@ -7,7 +7,7 @@
 
 import json
 import logging
-from datetime import datetime
+import time
 from typing import Optional
 
 from mcp_as_a_judge.db.conversation_history_service import ConversationHistoryService
@@ -19,7 +19,6 @@
 
 async def create_new_coding_task(
     user_request: str,
-    task_name: str,
     task_title: str,
     task_description: str,
     user_requirements: str,
@@ -28,24 +27,22 @@ async def create_new_coding_task(
 ) -> TaskMetadata:
     """
     Create a new coding task with auto-generated task_id.
-    
+
     Args:
         user_request: Original user request
-        task_name: Human-readable task name
         task_title: Display title
         task_description: Detailed description
         user_requirements: Initial requirements
         tags: Task tags
         conversation_service: Conversation service
-        
+
     Returns:
         New TaskMetadata instance
     """
     logger.info(f"📝 Creating new coding task: {task_title}")
 
     # Create new TaskMetadata with auto-generated UUID
     task_metadata = TaskMetadata(
-        name=task_name,
         title=task_title,
         description=task_description,
         user_requirements=user_requirements,
@@ -64,7 +61,6 @@ async def create_new_coding_task(
 async def update_existing_coding_task(
     task_id: str,
     user_request: str,
-    task_name: str,
     task_title: str,
     task_description: str,
     user_requirements: Optional[str],
@@ -74,21 +70,20 @@ async def update_existing_coding_task(
 ) -> TaskMetadata:
     """
     Update an existing coding task.
-    
+
     Args:
         task_id: Immutable task ID
         user_request: Original user request
-        task_name: Updated task name
         task_title: Updated title
         task_description: Updated description
         user_requirements: Updated requirements
-        state: Updated state
+        state: Updated state (None to skip state update)
         tags: Updated tags
         conversation_service: Conversation service
-        
+
     Returns:
         Updated TaskMetadata instance
-        
+
     Raises:
         ValueError: If task not found or invalid state transition
     """
@@ -104,11 +99,10 @@ async def update_existing_coding_task(
         raise ValueError(f"Task not found: {task_id}")
 
     # Update mutable fields
-    existing_metadata.name = task_name
     existing_metadata.title = task_title
     existing_metadata.description = task_description
     existing_metadata.tags = tags
-    existing_metadata.updated_at = datetime.now()
+    existing_metadata.updated_at = int(time.time())
 
     # Update requirements if provided
     if user_requirements is not None:
@@ -183,7 +177,7 @@ async def save_task_metadata_to_history(
             tool_output=json.dumps({
                 "action": action,
                 "current_task_metadata": task_metadata.model_dump(mode='json'),
-                "timestamp": datetime.now().isoformat(),
+                "timestamp": int(time.time()),
             }),
         )
 

src/mcp_as_a_judge/constants.py

@@ -7,6 +7,7 @@
 # LLM Configuration
 MAX_TOKENS = 25000  # Maximum tokens for all LLM requests - increased for comprehensive responses
 DEFAULT_TEMPERATURE = 0.1  # Default temperature for LLM requests
+DEFAULT_REASONING_EFFORT = "low"  # Default reasoning effort level - lowest for speed and efficiency
 
 # Timeout Configuration
 DEFAULT_TIMEOUT = 30  # Default timeout in seconds for operations

src/mcp_as_a_judge/db/cleanup_service.py

@@ -5,7 +5,7 @@
 removing records older than the retention period (default: 1 day).
 """
 
-from datetime import datetime, timedelta
+import time
 
 from sqlalchemy import Engine
 from sqlmodel import Session, select
@@ -35,7 +35,7 @@ def __init__(self, engine: Engine) -> None:
         """
         self.engine = engine
         self.retention_days = RECORD_RETENTION_DAYS
-        self.last_cleanup_time = datetime.utcnow()
+        self.last_cleanup_time = int(time.time())
 
     def cleanup_old_records(self) -> int:
         """
@@ -45,16 +45,17 @@ def cleanup_old_records(self) -> int:
         Returns:
             Number of records deleted
         """
-        # Only run cleanup once per day
-        if (datetime.utcnow() - self.last_cleanup_time).days < 1:
+        # Only run cleanup once per day (86400 seconds)
+        current_time = int(time.time())
+        if (current_time - self.last_cleanup_time) < 86400:
             return 0
 
-        cutoff_date = datetime.utcnow() - timedelta(days=self.retention_days)
+        cutoff_timestamp = current_time - (self.retention_days * 86400)
 
         with Session(self.engine) as session:
             # Count old records
             old_count_stmt = select(ConversationRecord).where(
-                ConversationRecord.timestamp < cutoff_date
+                ConversationRecord.timestamp < cutoff_timestamp
             )
             old_records = session.exec(old_count_stmt).all()
             old_count = len(old_records)
@@ -63,7 +64,7 @@ def cleanup_old_records(self) -> int:
                 logger.info(
                     f"🧹 Daily cleanup: No records older than {self.retention_days} days"
                 )
-                self.last_cleanup_time = datetime.utcnow()
+                self.last_cleanup_time = int(time.time())
                 return 0
 
             # Delete old records
@@ -73,7 +74,7 @@ def cleanup_old_records(self) -> int:
             session.commit()
 
             # Reset cleanup tracking
-            self.last_cleanup_time = datetime.utcnow()
+            self.last_cleanup_time = int(time.time())
 
             logger.info(
                 f"🧹 Daily cleanup: Deleted {old_count} records older than {self.retention_days} days"

src/mcp_as_a_judge/db/conversation_history_service.py

@@ -135,7 +135,7 @@ def format_conversation_history_as_json_array(
                     "source": record.source,
                     "input": record.input,
                     "output": record.output,
-                    "timestamp": record.timestamp.isoformat(),
+                    "timestamp": record.timestamp,  # Already epoch int
                 }
             )
 

src/mcp_as_a_judge/db/interface.py

@@ -6,7 +6,7 @@
 """
 
 from abc import ABC, abstractmethod
-from datetime import datetime
+import time
 
 from sqlmodel import Field, SQLModel
 
@@ -21,9 +21,9 @@ class ConversationRecord(SQLModel, table=True):
     source: str  # tool name
     input: str  # tool input query
     output: str  # tool output string
-    timestamp: datetime = Field(
-        default_factory=datetime.utcnow, index=True
-    )  # when the record was created
+    timestamp: int = Field(
+        default_factory=lambda: int(time.time()), index=True
+    )  # when the record was created (epoch seconds)
 
 
 class ConversationHistoryDB(ABC):

src/mcp_as_a_judge/db/providers/sqlite_provider.py

@@ -6,7 +6,7 @@
 """
 
 import uuid
-from datetime import datetime
+import time
 
 from sqlalchemy import create_engine
 from sqlmodel import Session, SQLModel, asc, desc, select
@@ -141,7 +141,7 @@ async def save_conversation(
     ) -> str:
         """Save a conversation record to SQLite database with LRU cleanup."""
         record_id = str(uuid.uuid4())
-        timestamp = datetime.utcnow()
+        timestamp = int(time.time())
 
         logger.info(
             f"💾 Saving conversation to SQLModel SQLite DB: record {record_id} "

src/mcp_as_a_judge/llm_client.py

@@ -10,6 +10,7 @@
 
 import litellm
 
+from mcp_as_a_judge.constants import DEFAULT_REASONING_EFFORT
 from mcp_as_a_judge.llm_integration import LLMConfig, LLMVendor
 
 # Set global drop_params to handle model-specific parameter restrictions
@@ -156,6 +157,7 @@ async def generate_text(
                 "api_key": self.config.api_key,
                 "max_tokens": max_tokens,
                 "temperature": temperature,
+                "reasoning_effort": DEFAULT_REASONING_EFFORT,  # Set reasoning to lowest level
                 **kwargs,
             }
 

src/mcp_as_a_judge/models.py

@@ -136,13 +136,51 @@ class JudgeCodingPlanUserVars(BaseModel):
     research: str = Field(description="Research findings and analysis")
     research_urls: list[str] = Field(
         default_factory=list,
-        description="URLs from MANDATORY online research - minimum 3 URLs required",
+        description="URLs from online research (conditional based on task needs)",
     )
     conversation_history: list = Field(
         default_factory=list,
         description="Previous conversation history as JSON array with timestamps",
     )
 
+    # Conditional research fields
+    research_required: bool = Field(
+        default=False,
+        description="Whether external research is required for this task"
+    )
+    research_scope: str = Field(
+        default="none",
+        description="Scope of research required (none, light, deep)"
+    )
+    research_rationale: str = Field(
+        default="",
+        description="Explanation of why research is or isn't required"
+    )
+
+    # Conditional internal research fields
+    internal_research_required: bool = Field(
+        default=False,
+        description="Whether internal codebase research is needed"
+    )
+    related_code_snippets: list[str] = Field(
+        default_factory=list,
+        description="Related code snippets from the codebase"
+    )
+
+    # Conditional risk assessment fields
+    risk_assessment_required: bool = Field(
+        default=False,
+        description="Whether risk assessment is needed"
+    )
+    identified_risks: list[str] = Field(
+        default_factory=list,
+        description="Areas that could be harmed by the changes"
+    )
+    risk_mitigation_strategies: list[str] = Field(
+        default_factory=list,
+        description="Strategies to mitigate identified risks"
+    )
+
 
 class JudgeCodeChangeSystemVars(BaseModel):
     """Variables for judge_code_change system prompt."""