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 04:10:23 +00:00
parent 7513149bdb
commit f81a3bd52f
1 changed files with 18 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
app/rag_chain.py
View File

@ -3,7 +3,7 @@ import os
import weaviate import weaviate
from weaviate import Client from weaviate import Client
from weaviate import WeaviateClient from weaviate import WeaviateClient
from weaviate.connect import ConnectionParams from weaviate.connect import ConnectionParams, AuthCredentials
from weaviate.auth import AuthApiKey from weaviate.auth import AuthApiKey
from weaviate.classes.init import Auth from weaviate.classes.init import Auth
from langchain.vectorstores import Weaviate from langchain.vectorstores import Weaviate
@ -107,33 +107,33 @@ def build_rag_chain():
else: else:
print("연결 실패 3. 서버 상태를 확인하세요.") print("연결 실패 3. 서버 상태를 확인하세요.")
client4 = weaviate.Client(
url="http://183.111.96.67:32668", auth_key = AuthCredentials.api_key("01js3q6y7twaxccm5dbh3se9bt") # 없으면 None
auth_client_secret=auth
# 커넥션 설정 (http, grpc 둘 다)
connection_params = ConnectionParams.from_params(
http_host="183.111.96.67",
http_port=32668,
http_secure=False,
grpc_host="183.111.96.67",
grpc_port=32619, # 예시 포트, 실제 grpc 포트 넣어야 함
grpc_secure=False,
auth_credentials=auth_key # 없으면 생략 가능
) )
client4 = WeaviateClient(connection_params=connection_params)
if client4.is_ready(): if client4.is_ready():
print("Weaviate 4 연결 성공!") print("Weaviate 4 연결 성공!")
else: else:
print("연결 실패 4. 서버 상태를 확인하세요.") print("연결 실패 4. 서버 상태를 확인하세요.")
legacy_client = weaviate.Client(connection_params=client4.connection_params) # langchain용
client5 = weaviate.Client(
url="s4x71b3lt32z3md1ya0yw.c0.asia-southeast1.gcp.weaviate.cloud",
auth_client_secret=AuthApiKey("0ojxypfXy2Y1rKw3KQdBUtwcWSrYPWitE7Qu")
)
if client5.is_ready():
print("Weaviate 5 연결 성공!")
else:
print("연결 실패 5. 서버 상태를 확인하세요.")
# 2. 벡터스토어 # 2. 벡터스토어
vectorstore = Weaviate( vectorstore = Weaviate(
client=client4, client=legacy_client,
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)