Update rag_chain.py

2025-04-18 02:18:29 +00:00 · 2025-04-18 02:18:29 +00:00 · 98d8f399af
commit 98d8f399af
parent d437d8c39e
1 changed files with 4 additions and 1 deletions
--- 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)