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 01:52:02 +00:00
parent d96ec1b577
commit e782e6c552
1 changed files with 18 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
app/rag_chain.py
View File

@ -1,6 +1,8 @@
# rag_chain.py
import os
import weaviate
from weaviate import Client
from weaviate.connect import ConnectionParams
from weaviate.auth import AuthApiKey
from weaviate.classes.init import Auth
from langchain.vectorstores import Weaviate
@ -10,21 +12,29 @@ from langchain.embeddings import OpenAIEmbeddings
def build_rag_chain():
# 1. Weaviate 클라이언트
auth_config = weaviate.AuthApiKey(api_key="01jryrcctd8c8vxbj4bs2ywrgs")
auth_config = weaviate.AuthApiKey(api_key="01js3q6y7twaxccm5dbh3se9bt")
# client = weaviate.connect_to_weaviate_cloud(cluster_url="http://183.111.96.67:32668",
# auth_credentials=Auth.api_key("01jryrcctd8c8vxbj4bs2ywrgs"),
# auth_credentials=Auth.api_key("01js3q6y7twaxccm5dbh3se9bt"),
# headers={
# "X-OpenAI-Api-Key": "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" # 필요할 경우
# })
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("01js3q6y7twaxccm5dbh3se9bt"), # 필요 없으면 제거
# additional_headers={
# "X-OpenAI-Api-Key": "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" # 필요할 경우
# }
# )
client = Client(
connection_params=ConnectionParams.from_http(
host="183.111.96.67", # 도메인 or IP
port=32668, # 포트
secure=False, # HTTP면 False, HTTPS면 True
auth_credentials=auth_config
)
)
# client = weaviate.connect_to_custom(