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)