Spaces:
Running
on
T4
Running
on
T4
Update auditqa/utils.py
Browse files- auditqa/utils.py +1 -1
auditqa/utils.py
CHANGED
@@ -69,7 +69,7 @@ def get_message_template(type, SYSTEM_PROMPT, USER_PROMPT):
|
|
69 |
SystemMessage(content=SYSTEM_PROMPT),
|
70 |
HumanMessage(content=USER_PROMPT),]
|
71 |
elfi type == 'INF_PROVIDERS':
|
72 |
-
messages =
|
73 |
{"role":"user","content":USER_PROMPT}]
|
74 |
else:
|
75 |
logging.error("No message template found")
|
|
|
69 |
SystemMessage(content=SYSTEM_PROMPT),
|
70 |
HumanMessage(content=USER_PROMPT),]
|
71 |
elfi type == 'INF_PROVIDERS':
|
72 |
+
messages = [{"role": "system", "content": SYSTEM_PROMPT},
|
73 |
{"role":"user","content":USER_PROMPT}]
|
74 |
else:
|
75 |
logging.error("No message template found")
|