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:57:09 +00:00
parent d3a9bc76b2
commit 2f66a36d10
1 changed files with 20 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
app/rag_chain.py
View File

@ -7,10 +7,12 @@ from weaviate.connect import ConnectionParams
from weaviate.auth import AuthApiKey from weaviate.auth import AuthApiKey
from weaviate.auth import AuthCredentials from weaviate.auth import AuthCredentials
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
from langchain.chat_models import ChatOpenAI from langchain.chat_models import ChatOpenAI
from langchain.embeddings import OpenAIEmbeddings from langchain.embeddings import OpenAIEmbeddings
from langchain.vectorstores import Weaviate as WeaviateLangchain
from langchain.embeddings.openai import OpenAIEmbeddings
def build_rag_chain(): def build_rag_chain():
# 1. Weaviate 클라이언트 # 1. Weaviate 클라이언트
@ -95,14 +97,25 @@ def build_rag_chain():
legacy_client = legacy_weaviate.Client(connection_params=client4.connection_params) # langchain용 legacy_client = legacy_weaviate.Client(connection_params=client4.connection_params) # langchain용
# 2. 벡터스토어 client = weaviate.Client(
vectorstore = Weaviate( url="http://183.111.96.67:32668", # 예: http://localhost:8080
client=legacy_client, auth_client_secret=weaviate.AuthApiKey(api_key="01js3q6y7twaxccm5dbh3se9bt"),
index_name="LangDocs", additional_headers={
text_key="text", "X-OpenAI-Api-Key": OPENAI_API_KEY
embedding=OpenAIEmbeddings(openai_api_key=OPENAI_API_KEY) }
) )
vectorstore = WeaviateLangchain(client=client, index_name="LangDocs", text_key="text", embedding=OpenAIEmbeddings(openai_api_key=OPENAI_API_KEY))
# 2. 벡터스토어
# vectorstore = Weaviate(
# client=client,
# index_name="LangDocs",
# text_key="text",
# embedding=OpenAIEmbeddings(openai_api_key=OPENAI_API_KEY)
# )
# 3. HuggingFace LLM (예: mistralai/Mistral-7B-Instruct-v0.2) # 3. HuggingFace LLM (예: mistralai/Mistral-7B-Instruct-v0.2)
llm = ChatOpenAI(temperature=0, openai_api_key=OPENAI_API_KEY) llm = ChatOpenAI(temperature=0, openai_api_key=OPENAI_API_KEY)