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