From 4dc0d5d8affba73ee01d27efd15eacba363acf10 Mon Sep 17 00:00:00 2001 From: localsoo Date: Mon, 21 Apr 2025 01:29:55 +0000 Subject: [PATCH] Update rag_chain.py --- app/rag_chain.py | 20 ++++++++++++++------ 1 file changed, 14 insertions(+), 6 deletions(-) diff --git a/app/rag_chain.py b/app/rag_chain.py index fb134c0..dbd7b6b 100644 --- a/app/rag_chain.py +++ b/app/rag_chain.py @@ -1,6 +1,7 @@ # rag_chain.py import os import weaviate +from weaviate.auth import AuthApiKey from weaviate.classes.init import Auth from langchain.vectorstores import Weaviate from langchain.chains import RetrievalQA @@ -16,12 +17,19 @@ def build_rag_chain(): # headers={ # "X-OpenAI-Api-Key": "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" # 필요할 경우 # }) - client = weaviate.Client( - url="http://183.111.96.67:32668", # 예: "http://183.111.96.67:32668" - auth_client_secret=Auth.api_key("01jryrcctd8c8vxbj4bs2ywrgs"), # 필요 없으면 제거 - additional_headers={ - "X-OpenAI-Api-Key": "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" # 필요할 경우 - } + # client = weaviate.Client( + # url="http://183.111.96.67:32668", # 예: "http://183.111.96.67:32668" + # auth_client_secret=Auth.api_key("01jryrcctd8c8vxbj4bs2ywrgs"), # 필요 없으면 제거 + # additional_headers={ + # "X-OpenAI-Api-Key": "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" # 필요할 경우 + # } + # ) + + client = weaviate.connect_to_custom( + http_host="183.111.96.67", + http_port=32668, + auth_credentials=AuthApiKey("01jryrcctd8c8vxbj4bs2ywrgs"), # 인증이 필요 없으면 생략 가능 + headers={"X-OpenAI-Api-Key": "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"} # 필요시 ) if client.is_ready():