变量改为系统变量

app.py

@@ -22,9 +22,15 @@ from langchain.chains import StuffDocumentsChain
 from langchain_core.prompts import PromptTemplate
 import hashlib
 from project.embeddings.zhipuai_embedding import ZhipuAIEmbeddings
+import os
 
 chain: Optional[Callable] = None
 
+db_host = os.getenv("DB_HOST")
+db_user = os.getenv("DB_USER")
+db_password = os.getenv("DB_PASSWORD")
+zhipuai_api_key = os.getenv("ZHIPU_AI_KEY")
+
 
 def generate_article_id(content):
     # 使用SHA-256哈希算法
@@ -39,7 +45,7 @@ def generate_article_id(content):
     return article_id
 
 
-def web_loader(file, puzhiai_key, zilliz_uri, user, password):
+def web_loader(file):
     if not file:
         return "please upload file"
     loader = TextLoader(file)
@@ -48,7 +54,7 @@ def web_loader(file, puzhiai_key, zilliz_uri, user, password):
     text_splitter = CharacterTextSplitter(chunk_size=512, chunk_overlap=0)
     docs = text_splitter.split_documents(docs)
     #embeddings = OpenAIEmbeddings(model="text-embedding-ada-002", openai_api_key=openai_key)
-    embeddings = ZhipuAIEmbeddings(zhipuai_api_key=puzhiai_key)
+    embeddings = ZhipuAIEmbeddings(zhipuai_api_key=zhipuai_api_key)
 
     if not embeddings:
         return "embeddings not"
@@ -66,9 +72,9 @@ def web_loader(file, puzhiai_key, zilliz_uri, user, password):
         embedding=embeddings,
         ids=article_ids,
         connection_args={
-            "uri": zilliz_uri,
-            "user": user,
-            "password": password,
+            "uri": db_host,
+            "user": db_user,
+            "password": db_password,
             "secure": True,
         },
         collection_name="LangChainCollectionYin"
@@ -149,7 +155,7 @@ if __name__ == "__main__":
             """
         <h1><center>Langchain And Zilliz App</center></h1>
         
-        v.2.28.14.47
+        v.2.28.15.3
         
         """
         )