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 02:13:39 +00:00
parent 5018799361
commit beebdd0760
1 changed files with 10 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
app/rag_chain.py
View File

@ -22,13 +22,10 @@ def build_rag_chain():
# }) # })
OPENAI_API_KEY="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" OPENAI_API_KEY="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"
# client = weaviate.Client( client3 = weaviate.Client(
# url="http://183.111.96.67:32668", # 예: "http://183.111.96.67:32668" url="http://183.111.96.67:32668", # 예: "http://183.111.96.67:32668"
# auth_client_secret=Auth.api_key("01js3q6y7twaxccm5dbh3se9bt"), # 필요 없으면 제거 auth_client_secret=auth
# additional_headers={ )
# "X-OpenAI-Api-Key": "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" # 필요할 경우
# }
# )
# client = Client( # client = Client(
# connection_params=ConnectionParams.from_http( # connection_params=ConnectionParams.from_http(
# host="183.111.96.67", # 도메인 or IP # host="183.111.96.67", # 도메인 or IP
@ -103,10 +100,15 @@ def build_rag_chain():
else: else:
print("연결 실패 2. 서버 상태를 확인하세요.") print("연결 실패 2. 서버 상태를 확인하세요.")
if client3.is_ready():
print("Weaviate 3 연결 성공!")
else:
print("연결 실패 3. 서버 상태를 확인하세요.")
# 2. 벡터스토어 # 2. 벡터스토어
vectorstore = Weaviate( vectorstore = Weaviate(
client=client, client=client3,
index_name="LangDocs", index_name="LangDocs",
text_key="text", text_key="text",
embedding=OpenAIEmbeddings(openai_api_key=OPENAI_API_KEY) embedding=OpenAIEmbeddings(openai_api_key=OPENAI_API_KEY)