ppsingh commited on
Commit
f9c30b6
·
verified ·
1 Parent(s): 868541f

Update auditqa/utils.py

Browse files
Files changed (1) hide show
  1. auditqa/utils.py +6 -1
auditqa/utils.py CHANGED
@@ -64,8 +64,13 @@ def get_message_template(type, SYSTEM_PROMPT, USER_PROMPT):
64
  messages = [
65
  SystemMessage(content=SYSTEM_PROMPT),
66
  HumanMessage(content=USER_PROMPT),]
 
 
 
 
67
  else:
68
- messages = None
 
69
 
70
  return messages
71
 
 
64
  messages = [
65
  SystemMessage(content=SYSTEM_PROMPT),
66
  HumanMessage(content=USER_PROMPT),]
67
+ elif type == 'SERVERLESS':
68
+ messages = [
69
+ SystemMessage(content=SYSTEM_PROMPT),
70
+ HumanMessage(content=USER_PROMPT),]
71
  else:
72
+ logging.error("No message template found")
73
+ raise
74
 
75
  return messages
76