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")