From dda737da823c48e7dbca4a34e90f4449f5556fd8 Mon Sep 17 00:00:00 2001 From: localsoo Date: Mon, 21 Apr 2025 01:47:02 +0000 Subject: [PATCH] Update rag_chain.py --- app/rag_chain.py | 38 +++++++++++++++++++++----------------- 1 file changed, 21 insertions(+), 17 deletions(-) diff --git a/app/rag_chain.py b/app/rag_chain.py index de802d6..9d6508b 100644 --- a/app/rag_chain.py +++ b/app/rag_chain.py @@ -17,25 +17,26 @@ def build_rag_chain(): # 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("01jryrcctd8c8vxbj4bs2ywrgs"), # 필요 없으면 제거 - # additional_headers={ - # "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, - grpc_host="183.111.96.67", - http_secure=False, - grpc_port=32619, - grpc_secure=False, - auth_credentials=AuthApiKey("01js3q6y7twaxccm5dbh3se9bt"), # 인증이 필요 없으면 생략 가능 - headers={"X-OpenAI-Api-Key": OPENAI_API_KEY} # 필요시 + 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.connect_to_custom( + # http_host="183.111.96.67", + # http_port=32668, + # grpc_host="183.111.96.67", + # http_secure=False, + # grpc_port=32619, + # grpc_secure=False, + # auth_credentials=AuthApiKey("01js3q6y7twaxccm5dbh3se9bt"), # 인증이 필요 없으면 생략 가능 + # headers={"X-OpenAI-Api-Key": OPENAI_API_KEY} # 필요시 + # ) if client.is_ready(): print("Weaviate 연결 성공!") @@ -58,4 +59,7 @@ def build_rag_chain(): # 4. RetrievalQA chain 구성 - return RetrievalQA.from_chain_type(llm=llm, retriever=retriever) + qa_chain = RetrievalQA.from_chain_type(llm=llm, retriever=retriever) + + client.close() + return qa_chain