From 98d8f399afa4c90e3cfbf71403da8e57dc1a6cc3 Mon Sep 17 00:00:00 2001 From: localsoo Date: Fri, 18 Apr 2025 02:18:29 +0000 Subject: [PATCH] Update rag_chain.py --- app/rag_chain.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/app/rag_chain.py b/app/rag_chain.py index 8127482..c9285bf 100644 --- a/app/rag_chain.py +++ b/app/rag_chain.py @@ -14,12 +14,15 @@ def build_rag_chain(): additional_headers={"X-HuggingFace-Api-Key": os.getenv("OPENAI_API_KEY", "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")} ) + # 2. 벡터스토어 vectorstore = Weaviate( client=client, index_name="LangDocs", text_key="text", - embedding=HuggingFaceEmbeddings() # HuggingFaceEmbeddings() 로 교체 가능 + embedding= HuggingFaceEmbeddings( + model_name="sentence-transformers/all-MiniLM-L6-v2" + ) ) # 3. HuggingFace LLM (예: mistralai/Mistral-7B-Instruct-v0.2)