fix: optional response areas support & gemini

Author: neagualexa

Diff for: index.py

@@ -39,4 +39,7 @@ def handler(event, context):
         "body": chatbot_response
     }
 
+    # Log the response for debugging purposes
+    print("Returning response:", json.dumps(response, indent=2))
+
     return response

Diff for: src/agents/informational_agent/informational_agent.py

@@ -37,7 +37,7 @@ class State(TypedDict):
 
 class InformationalAgent:
     def __init__(self):
-        llm = OpenAILLMs()
+        llm = GoogleAILLMs()
         self.llm = llm.get_llm()
         summarisation_llm = OpenAILLMs()
         self.summarisation_llm = summarisation_llm.get_llm()

Diff for: src/agents/utils/parse_json_to_prompt.py

@@ -142,8 +142,8 @@ def parse_json_to_prompt( questionSubmissionSummary: Optional[List[StudentWorkRe
     questionInformation = QuestionDetails(**questionInformation)
     questionAccessInformation = QuestionAccessInformation(**questionAccessInformation)
 
-    if not questionSubmissionSummary or not questionInformation or not questionAccessInformation:
-        return None
+    if not questionInformation or not questionAccessInformation:
+        return "There must have been an error in fetching the question details. So ask me about the question I am working on such that you can still help me."
 
     def format_response_area_details(responseArea: ResponseAreaDetails, studentSummary: List[StudentWorkResponseArea]) -> str:
         submissionDetails = "\n".join(
@@ -163,7 +163,7 @@ def format_response_area_details(responseArea: ResponseAreaDetails, studentSumma
         return f"""
         ## Response Area: {responseArea.position + 1}
         {f'Area task: What is {responseArea.preResponseText} ?' if responseArea.preResponseText else ''}
-        (Secret) Expected Answer: {responseArea.answer};
+        (Secret - not to be shared) Expected Answer: {responseArea.answer};
         {submissionDetails}"""
 
     def format_part_details(part: PartDetails, currentPart: CurrentPart, summary: List[StudentWorkResponseArea]) -> str:
@@ -188,7 +188,7 @@ def format_part_details(part: PartDetails, currentPart: CurrentPart, summary: Li
     {f"Time spent on this part: {currentPart.timeTakenPart if currentPart.timeTakenPart is not None else 'No recorded duration'}" if currentPart.id == part.publishedPartId else ''}
     Part Content: {part.publishedPartContent.strip() if part.publishedPartContent else 'No content'};
     {responseAreas}
-    {f'Final Part Answer: {part.publishedPartAnswerContent}' if part.publishedPartAnswerContent else 'No direct answer'}
+    {f'Final Part Answer: {part.publishedPartAnswerContent}' if part.publishedPartAnswerContent else 'No direct answer for this part.'}
     {workedSolutions}
 """
 

Diff for: src/module.py

@@ -57,11 +57,16 @@ def chat_module(message: Any, params: Params) -> Result:
         question_submission_summary = question_response_details["questionSubmissionSummary"] if "questionSubmissionSummary" in question_response_details else []
         question_information = question_response_details["questionInformation"] if "questionInformation" in question_response_details else {}
         question_access_information = question_response_details["questionAccessInformation"] if "questionAccessInformation" in question_response_details else {}
-        question_response_details_prompt = parse_json_to_prompt(
-            question_submission_summary,
-            question_information,
-            question_access_information
-        )
+        try:
+            question_response_details_prompt = parse_json_to_prompt(
+                question_submission_summary,
+                question_information,
+                question_access_information
+            )
+            print("INFO:: ", question_response_details_prompt)
+        except Exception as e:
+            print("ERROR:: ", e)
+            raise Exception("Internal Error: The question response details could not be parsed.")
     if "agent_type" in params:
         agent_type = params["agent_type"]
     if "conversation_id" in params: