Update auditqa/utils.py

auditqa/utils.py

@@ -64,8 +64,13 @@ def get_message_template(type, SYSTEM_PROMPT, USER_PROMPT):
         messages = [
                  SystemMessage(content=SYSTEM_PROMPT),
                  HumanMessage(content=USER_PROMPT),]
+    elif type == 'SERVERLESS':
+        messages = [
+                 SystemMessage(content=SYSTEM_PROMPT),
+                 HumanMessage(content=USER_PROMPT),]
     else:
-        messages = None
+        logging.error("No message template found")
+        raise
     
     return messages