From a38cbcd864eb3d73f6dc3b41c29a21be6609a64b Mon Sep 17 00:00:00 2001
From: localsoo <want813@local.co.kr>
Date: Tue, 22 Apr 2025 00:07:34 +0000
Subject: [PATCH] Update rag_chain.py

---
 app/rag_chain.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/app/rag_chain.py b/app/rag_chain.py
index 56ab93f..99ca520 100644
--- a/app/rag_chain.py
+++ b/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")