Update app/chains.py

app/chains.py

@@ -45,14 +45,13 @@ standalone_chain = standalone_prompt_formatted | llm
 generation_chain = rag_prompt_formatted | llm
 
 rag_chain = (
-    RunnablePassthrough.assign(new_question=standalone_prompt_formatted | llm)
+    RunnablePassthrough.assign(new_question=standalone_chain)
     | {
-        'context': lambda x: format_context(data_indexer.search(x['new_question'], hybrid_search=x['hybrid_search'])),
-        'standalone_question': lambda x: x['new_question'],
-        'test': lambda x : print(x)
+        'context': lambda x:
+format_context(search(x['new_question'])),
+        'standalone_question': lambda x: x['new_question']
     }
-    | rag_prompt_formatted
-    | llm
+    | generation_chain
 ).with_types(input_type=schemas.RagInput)