Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Update auditqa/utils.py
Browse files- auditqa/utils.py +1 -1
auditqa/utils.py
CHANGED
|
@@ -68,7 +68,7 @@ def get_message_template(type, SYSTEM_PROMPT, USER_PROMPT):
|
|
| 68 |
messages = [
|
| 69 |
SystemMessage(content=SYSTEM_PROMPT),
|
| 70 |
HumanMessage(content=USER_PROMPT),]
|
| 71 |
-
|
| 72 |
messages = [{"role": "system", "content": SYSTEM_PROMPT},
|
| 73 |
{"role":"user","content":USER_PROMPT}]
|
| 74 |
else:
|
|
|
|
| 68 |
messages = [
|
| 69 |
SystemMessage(content=SYSTEM_PROMPT),
|
| 70 |
HumanMessage(content=USER_PROMPT),]
|
| 71 |
+
elif type == 'INF_PROVIDERS':
|
| 72 |
messages = [{"role": "system", "content": SYSTEM_PROMPT},
|
| 73 |
{"role":"user","content":USER_PROMPT}]
|
| 74 |
else:
|