Update rag_engine.py

rag_engine.py

@@ -6,6 +6,7 @@ from clare_core import (
     get_embedding,
     cosine_similarity,
 )
+from langsmith import traceable
 
 
 def build_rag_chunks_from_file(file, doc_type_val: str) -> List[Dict]:
@@ -45,6 +46,7 @@ def build_rag_chunks_from_file(file, doc_type_val: str) -> List[Dict]:
         return []
 
 
+@traceable(run_type="retriever", name="retrieve_relevant_chunks")
 def retrieve_relevant_chunks(
     question: str,
     rag_chunks: List[Dict],