Make check case insensitive

Author: edkaya

genai/controller/generate_controller.py

@@ -90,8 +90,8 @@ def generate():
         messages (List[Dict]): Full conversation history, each with 'role' and 'content'
             Example:
             [
-                {"role": "user", "content": "I have eggs and tomatoes."},
-                {"role": "assistant", "content": "You could make shakshuka."}
+                {"role": "USER", "content": "I have eggs and tomatoes."},
+                {"role": "ASSISTANT", "content": "You could make shakshuka."}
             ]
 
     Returns:

genai/service/rag_service.py

@@ -33,17 +33,18 @@ def prepare_prompt(system_prompt: str,
 
     return prompt
 
+
 def process_raw_messages(raw_messages: List[Dict]) -> List[BaseMessage]:
     """Turns raw messages into BaseMessages, so they can be passed into LLM"""
     processed_messages = []
     for msg in raw_messages:
         role = msg.get("role")
         content = msg.get("content")
 
-        if role == "user":
+        if role.upper() == "USER":
             processed_messages.append(HumanMessage(content=content))
 
-        elif role == "assistant":
+        elif role.upper() == "ASSISTANT":
             processed_messages.append(AIMessage(content=content))
 
     return processed_messages