Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
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
|