edit prints for logs

climateqa/engine/chains/retrieve_documents.py

@@ -104,7 +104,7 @@ async def retrieve_documents(state,config, vectorstore,reranker,llm,rerank_by_qu
     question = current_question["question"]
     index = current_question["index"]
     
-
+    print(f"Retrieve documents for question: {question}")
     await log_event({"question":question,"sources":sources,"index":index},"log_retriever",config)
 
 

climateqa/engine/chains/translation.py

@@ -30,12 +30,11 @@ def make_translation_chain(llm):
 
 
 def make_translation_node(llm):
-    print("---- Translate query ----")
-
-
     translation_chain = make_translation_chain(llm)
 
     def translate_query(state):
+        print("---- Translate query ----")
+
         user_input = state["user_input"]
         translation = translation_chain.invoke({"input":user_input})
         return {"query":translation["translation"]}