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:29:55 +00:00
parent 80a9baabb3
commit 4dc0d5d8af
1 changed files with 14 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
app/rag_chain.py
View File

@ -1,6 +1,7 @@
# rag_chain.py # rag_chain.py
import os import os
import weaviate import weaviate
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
from langchain.chains import RetrievalQA from langchain.chains import RetrievalQA
@ -16,12 +17,19 @@ def build_rag_chain():
# headers={ # headers={
# "X-OpenAI-Api-Key": "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" # 필요할 경우 # "X-OpenAI-Api-Key": "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" # 필요할 경우
# }) # })
client = weaviate.Client( # client = 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("01jryrcctd8c8vxbj4bs2ywrgs"), # 필요 없으면 제거 # auth_client_secret=Auth.api_key("01jryrcctd8c8vxbj4bs2ywrgs"), # 필요 없으면 제거
additional_headers={ # additional_headers={
"X-OpenAI-Api-Key": "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" # 필요할 경우 # "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(): if client.is_ready():