Update tool/rag.py

tool/rag.py

@@ -30,8 +30,8 @@ from langchain_community.vectorstores import FAISS
 from torch import cuda, bfloat16
 device = f'cuda:{cuda.current_device()}' if cuda.is_available() else 'cpu'
 from langchain_openai import OpenAIEmbeddings 
-embeddings = OpenAIEmbeddings(api_key='sk-itPrztYm9F6XZZpsBMJB9O7Vq0pYUABVVBSoThuBxEGTnDik',
-      base_url="https://www.dmxapi.com/v1")
+embeddings = OpenAIEmbeddings(api_key=api_key=os.getenv("OPENAI_API_KEY"),
+             base_url=os.getenv("OPENAI_API_BASE"))
 vectorstore=FAISS.load_local(r"rag", embeddings,allow_dangerous_deserialization =True)   
 
  
@@ -67,8 +67,9 @@ class rag(BaseTool):
     
     def __init__(self, path: str = None):
         super().__init__(  )
-        self.llm = ChatOpenAI(model="gpt-4o-2024-11-20",api_key='sk-itPrztYm9F6XZZpsBMJB9O7Vq0pYUABVVBSoThuBxEGTnDik',
-             base_url="https://www.dmxapi.com/v1")
+        self.llm = ChatOpenAI(model="gpt-4o-2024-11-20",api_key=os.getenv("OPENAI_API_KEY"),
+             base_url=os.getenv("OPENAI_API_BASE")
+                             )
         self.path = path
         # api keys