Spaces:

tall-tree
/

ai-virtual-assistant

Running on CPU Upgrade

yrobel-lima commited on Mar 27

Commit

400265e

•

1 Parent(s): 8770687

Update rag_chain/chain.py

Files changed (1) hide show

rag_chain/chain.py CHANGED Viewed

@@ -175,7 +175,7 @@ def get_rag_chain(
         | practitioners_db_compression_retriever
         | DocumentFormatter("Practitioner #"),
         "tall_tree_db": itemgetter("message")
-        | tall_tree_db_compression_retriever
         | DocumentFormatter("No."),
         "history": RunnableLambda(memory.load_memory_variables) | itemgetter("history"),
         "message": itemgetter("message"),

         | practitioners_db_compression_retriever
         | DocumentFormatter("Practitioner #"),
         "tall_tree_db": itemgetter("message")
+        | tall_tree_db_dense_retriever
         | DocumentFormatter("No."),
         "history": RunnableLambda(memory.load_memory_variables) | itemgetter("history"),
         "message": itemgetter("message"),