Update rag_chain.py

app/rag_chain.py

@@ -1,6 +1,7 @@
 # rag_chain.py
 import os
 import weaviate
+import mlflow
 from weaviate import Client
 # from weaviate import WeaviateClient
 # from weaviate.connect import ConnectionParams
@@ -15,6 +16,10 @@ from langchain.vectorstores import Weaviate as WeaviateLangchain
 from langchain.embeddings.openai import OpenAIEmbeddings
 
 def build_rag_chain():
+
+    # Enable MLflow tracing
+    mlflow.langchain.autolog()
+
     # 1. Weaviate 클라이언트
     auth = weaviate.AuthApiKey(api_key="01js3q6y7twaxccm5dbh3se9bt")