Spaces:
Runtime error
Runtime error
Update custom_llm.py
Browse files- custom_llm.py +1 -1
custom_llm.py
CHANGED
|
@@ -51,7 +51,7 @@ def custom_chain_with_history(llm, memory):
|
|
| 51 |
t = ""
|
| 52 |
for x in memory.chat_memory.messages:
|
| 53 |
# for x in memory.messages:
|
| 54 |
-
t += f"<|you|>\n<s>{x.content}</s>\n" if type(x) is AIMessage else f"<|user|>\n{x.content}\n"
|
| 55 |
return "" if len(t) == 0 else t
|
| 56 |
|
| 57 |
def format_docs(docs):
|
|
|
|
| 51 |
t = ""
|
| 52 |
for x in memory.chat_memory.messages:
|
| 53 |
# for x in memory.messages:
|
| 54 |
+
t += f"<|you|>\n<s>{x.content}</s>\n\n" if type(x) is AIMessage else f"<|user|>\n{x.content}\n"
|
| 55 |
return "" if len(t) == 0 else t
|
| 56 |
|
| 57 |
def format_docs(docs):
|