localsoo/rag-agent-soo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update rag_chain.py

Browse Source
This commit is contained in:
localsoo 2025-04-22 00:07:34 +00:00
parent 363a28d328
commit a38cbcd864
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
app/rag_chain.py
View File

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