ppsingh commited on
Commit
a031d5b
·
verified ·
1 Parent(s): 61141b3

Update auditqa/utils.py

Browse files
Files changed (1) hide show
  1. auditqa/utils.py +3 -0
auditqa/utils.py CHANGED
@@ -68,6 +68,9 @@ def get_message_template(type, SYSTEM_PROMPT, USER_PROMPT):
68
  messages = [
69
  SystemMessage(content=SYSTEM_PROMPT),
70
  HumanMessage(content=USER_PROMPT),]
 
 
 
71
  else:
72
  logging.error("No message template found")
73
  raise
 
68
  messages = [
69
  SystemMessage(content=SYSTEM_PROMPT),
70
  HumanMessage(content=USER_PROMPT),]
71
+ elfi type == 'INF_PROVIDERS':
72
+ messages = messages = [{"role": "system", "content": SYSTEM_PROMPT},
73
+ {"role":"user","content":USER_PROMPT}]
74
  else:
75
  logging.error("No message template found")
76
  raise