Spaces:
Running
on
T4
Running
on
T4
Update auditqa/utils.py
Browse files- auditqa/utils.py +6 -1
auditqa/utils.py
CHANGED
@@ -64,8 +64,13 @@ def get_message_template(type, SYSTEM_PROMPT, USER_PROMPT):
|
|
64 |
messages = [
|
65 |
SystemMessage(content=SYSTEM_PROMPT),
|
66 |
HumanMessage(content=USER_PROMPT),]
|
|
|
|
|
|
|
|
|
67 |
else:
|
68 |
-
|
|
|
69 |
|
70 |
return messages
|
71 |
|
|
|
64 |
messages = [
|
65 |
SystemMessage(content=SYSTEM_PROMPT),
|
66 |
HumanMessage(content=USER_PROMPT),]
|
67 |
+
elif type == 'SERVERLESS':
|
68 |
+
messages = [
|
69 |
+
SystemMessage(content=SYSTEM_PROMPT),
|
70 |
+
HumanMessage(content=USER_PROMPT),]
|
71 |
else:
|
72 |
+
logging.error("No message template found")
|
73 |
+
raise
|
74 |
|
75 |
return messages
|
76 |
|