Spaces:
Sleeping
Sleeping
reply-konhein
commited on
Commit
·
af480d2
1
Parent(s):
f00fb66
Change system prompt to fit audit agent
Browse files- document_qa_engine.py +2 -2
document_qa_engine.py
CHANGED
@@ -118,7 +118,7 @@ class DocumentQAEngine:
|
|
118 |
|
119 |
def inference(self, query, input_messages: List[dict]):
|
120 |
system_message = ChatMessage.from_system(
|
121 |
-
"You are a professional
|
122 |
messages = [system_message]
|
123 |
for message in input_messages:
|
124 |
if message["role"] == "user":
|
@@ -127,7 +127,7 @@ class DocumentQAEngine:
|
|
127 |
messages.append(
|
128 |
ChatMessage.from_user(message["content"]))
|
129 |
messages.append(ChatMessage.from_user("""
|
130 |
-
Relevant information from the uploaded
|
131 |
{% for doc in documents %}
|
132 |
{{ doc.content }}
|
133 |
{% endfor %}
|
|
|
118 |
|
119 |
def inference(self, query, input_messages: List[dict]):
|
120 |
system_message = ChatMessage.from_system(
|
121 |
+
"You are a professional analyzer of git repos, having access to reelvant snippets of the repo content. In 1-3 sentences")
|
122 |
messages = [system_message]
|
123 |
for message in input_messages:
|
124 |
if message["role"] == "user":
|
|
|
127 |
messages.append(
|
128 |
ChatMessage.from_user(message["content"]))
|
129 |
messages.append(ChatMessage.from_user("""
|
130 |
+
Relevant information from the uploaded repo:
|
131 |
{% for doc in documents %}
|
132 |
{{ doc.content }}
|
133 |
{% endfor %}
|