fix estimated time

Author: neagualexa

src/agent/utils/parse_json_context_to_prompt.py

@@ -97,6 +97,7 @@ def __init__(
         questionTitle: Optional[str] = None,
         questionGuidance: Optional[str] = None,
         questionContent: Optional[str] = None,
+        estimatedTime: Optional[str] = None,
         durationLowerBound: Optional[int] = None,
         durationUpperBound: Optional[int] = None,
         parts: Optional[List[PartDetails]] = [],
@@ -108,6 +109,7 @@ def __init__(
         self.questionTitle = questionTitle
         self.questionGuidance = questionGuidance
         self.questionContent = questionContent
+        self.estimatedTime = estimatedTime
         self.durationLowerBound = durationLowerBound
         self.durationUpperBound = durationUpperBound
         self.parts = [PartDetails(**part) for part in parts] 
@@ -193,6 +195,7 @@ def parse_json_to_structured_prompt(
         'title': question_info.questionTitle,
         'guidance': question_info.questionGuidance,
         'content': question_info.questionContent,
+        'estimated_time': question_info.estimatedTime,
         'duration_lower': question_info.durationLowerBound,
         'duration_upper': question_info.durationUpperBound
     }

src/agent/utils/prompt_context_templates.py

@@ -56,7 +56,9 @@ def format_question_header(
 
         # Duration formatting
         duration_text = "- Expected Duration: "
-        if question_info.get('duration_lower') and question_info.get('duration_upper'):
+        if question_info.get('estimated_time'):
+            duration_text += question_info['estimated_time']
+        elif question_info.get('duration_lower') and question_info.get('duration_upper'):
             duration_text += f"{question_info['duration_lower']}-{question_info['duration_upper']} minutes"
         else:
             duration_text += "Not specified"

src/module.py

@@ -96,6 +96,7 @@ def _build_question_information(context: dict) -> dict:
         "questionTitle": question_data.get("title"),
         "questionGuidance": question_data.get("guidance"),
         "questionContent": question_data.get("content"),
+        "estimatedTime": question_data.get("estimatedTime"),
         "durationLowerBound": None,
         "durationUpperBound": None,
         "parts": [