Upload folder using huggingface_hub
Browse files- Test_RAG.py +4 -4
Test_RAG.py
CHANGED
@@ -615,13 +615,13 @@ def bot(history, temperature, top_p, top_k, repetition_penalty, hide_full_prompt
|
|
615 |
if do_rag:
|
616 |
# t1 = Thread(target=rag_chain.invoke, args=({"input": history[-1][0]},))
|
617 |
input_text = history[-1][0]
|
618 |
-
response = rag_chain.invoke({"input":
|
619 |
print(response)
|
620 |
else:
|
621 |
-
|
622 |
# t1 = Thread(target=llm.invoke, args=(input_text,))
|
623 |
-
input_text = history[-1][0]
|
624 |
-
response = rag_chain.invoke({"input":
|
625 |
print(response)
|
626 |
# t1.start()
|
627 |
|
|
|
615 |
if do_rag:
|
616 |
# t1 = Thread(target=rag_chain.invoke, args=({"input": history[-1][0]},))
|
617 |
input_text = history[-1][0]
|
618 |
+
response = rag_chain.invoke({"input": input_text })
|
619 |
print(response)
|
620 |
else:
|
621 |
+
input_text = rag_prompt_template.format(input=history[-1][0], context="")
|
622 |
# t1 = Thread(target=llm.invoke, args=(input_text,))
|
623 |
+
# input_text = history[-1][0]
|
624 |
+
response = rag_chain.invoke({"input": input_text })
|
625 |
print(response)
|
626 |
# t1.start()
|
627 |
|