Spaces:
Running
on
T4
Running
on
T4
Update auditqa/utils.py
Browse files- 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
|