Update app.py
Browse files
app.py
CHANGED
@@ -79,9 +79,7 @@ def format_prompt(prompt,retrieved_documents,k,history,memory_limit=3):
|
|
79 |
|
80 |
print("checkwohist")
|
81 |
# PROMPT = PROMPT + f"{history[0][0]} [/INST] {history[0][1]} </s>"
|
82 |
-
#
|
83 |
-
# print(PROMPT)
|
84 |
-
# Handle conversation history
|
85 |
for user_message, bot_message in history[0:]:
|
86 |
PROMPT += f"<s>[INST] {user_message} [/INST] {bot_message} </s>"
|
87 |
print("checkwthhist2")
|
@@ -99,13 +97,6 @@ def talk(prompt, history):
|
|
99 |
formatted_prompt = format_prompt(prompt,retrieved_documents,k,history,memory_limit=3) # create a new prompt using the retrieved documents
|
100 |
print("check5")
|
101 |
pd.options.display.max_colwidth = 4000
|
102 |
-
# print(historylog)
|
103 |
-
# print(retrieved_documents['0'])
|
104 |
-
# print(formatted_prompt)
|
105 |
-
# formatted_prompt_with_history = add_history(formatted_prompt, history)
|
106 |
-
|
107 |
-
# formatted_prompt_with_history = formatted_prompt_with_history[:600] # to avoid memory issue
|
108 |
-
# print(formatted_prompt_with_history)
|
109 |
messages = [{"role":"system","content":SYS_PROMPT},{"role":"user","content":formatted_prompt}]
|
110 |
print(messages)
|
111 |
# binding the system context and new prompt for LLM
|
@@ -128,9 +119,9 @@ def talk(prompt, history):
|
|
128 |
|
129 |
try:
|
130 |
# write data to file
|
131 |
-
with open(
|
132 |
data.write(historylog)
|
133 |
-
data = open(
|
134 |
data.seek(0)
|
135 |
# Read the contents of the file to display it.
|
136 |
print("History log printed:")
|
|
|
79 |
|
80 |
print("checkwohist")
|
81 |
# PROMPT = PROMPT + f"{history[0][0]} [/INST] {history[0][1]} </s>"
|
82 |
+
# Handle conversation history
|
|
|
|
|
83 |
for user_message, bot_message in history[0:]:
|
84 |
PROMPT += f"<s>[INST] {user_message} [/INST] {bot_message} </s>"
|
85 |
print("checkwthhist2")
|
|
|
97 |
formatted_prompt = format_prompt(prompt,retrieved_documents,k,history,memory_limit=3) # create a new prompt using the retrieved documents
|
98 |
print("check5")
|
99 |
pd.options.display.max_colwidth = 4000
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
100 |
messages = [{"role":"system","content":SYS_PROMPT},{"role":"user","content":formatted_prompt}]
|
101 |
print(messages)
|
102 |
# binding the system context and new prompt for LLM
|
|
|
119 |
|
120 |
try:
|
121 |
# write data to file
|
122 |
+
with open("file.txt", "a") as data:
|
123 |
data.write(historylog)
|
124 |
+
data = open("file.txt", "r")
|
125 |
data.seek(0)
|
126 |
# Read the contents of the file to display it.
|
127 |
print("History log printed:")
|