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-21 00:30:43 +00:00
parent 4b71798dca
commit 24a02c195d
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
app/rag_chain.py
View File

@ -8,17 +8,16 @@ from langchain.llms import HuggingFaceHub
def build_rag_chain(): def build_rag_chain():
# 1. Weaviate 클라이언트 # 1. Weaviate 클라이언트
client = weaviate.Client( auth_config = weaviate.AuthApiKey(api_key="01jryrcctd8c8vxbj4bs2ywrgs")
url=os.getenv("WEAVIATE_URL", "http://183.111.96.67:30846"),
auth_client_secret=weaviate.AuthApiKey(os.getenv("WEAVIATE_API_KEY", "01jryrcctd8c8vxbj4bs2ywrgs")), client = weaviate.Client("http://183.111.96.67:32668", auth_client_secret=auth_config)
additional_headers={"X-HuggingFace-Api-Key": os.getenv("OPENAI_API_KEY", "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")}
)
# 2. 벡터스토어 # 2. 벡터스토어
vectorstore = Weaviate( vectorstore = Weaviate(
client=client, client=client,
index_name="LangDocs", index_name="Test",
text_key="text", text_key="text",
embedding= HuggingFaceEmbeddings( embedding= HuggingFaceEmbeddings(
model_name="sentence-transformers/all-MiniLM-L6-v2" model_name="sentence-transformers/all-MiniLM-L6-v2"