From 52184b1a9345e37ee02019ef0c0a5cd86b54329e Mon Sep 17 00:00:00 2001 From: localsoo Date: Mon, 21 Apr 2025 01:40:31 +0000 Subject: [PATCH] Update rag_chain.py --- app/rag_chain.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/app/rag_chain.py b/app/rag_chain.py index 54d8c6d..de802d6 100644 --- a/app/rag_chain.py +++ b/app/rag_chain.py @@ -24,7 +24,8 @@ def build_rag_chain(): # "X-OpenAI-Api-Key": "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" # 필요할 경우 # } # ) - + OPENAI_API_KEY="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" + client = weaviate.connect_to_custom( http_host="183.111.96.67", http_port=32668, @@ -33,7 +34,7 @@ def build_rag_chain(): grpc_port=32619, grpc_secure=False, auth_credentials=AuthApiKey("01js3q6y7twaxccm5dbh3se9bt"), # 인증이 필요 없으면 생략 가능 - headers={"X-OpenAI-Api-Key": "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"} # 필요시 + headers={"X-OpenAI-Api-Key": OPENAI_API_KEY} # 필요시 ) if client.is_ready(): @@ -47,11 +48,11 @@ def build_rag_chain(): client=client, index_name="LangDocs", text_key="text", - embedding=OpenAIEmbeddings() + embedding=OpenAIEmbeddings(openai_api_key=OPENAI_API_KEY) ) # 3. HuggingFace LLM (예: mistralai/Mistral-7B-Instruct-v0.2) - llm = ChatOpenAI(temperature=0) + llm = ChatOpenAI(temperature=0, openai_api_key=OPENAI_API_KEY) retriever = vectorstore.as_retriever()