openai key 和 智谱的 key 分开

app.py

@@ -6,14 +6,13 @@ from langchain.vectorstores import Zilliz
 from langchain.document_loaders import WebBaseLoader
 from langchain.text_splitter import CharacterTextSplitter
 from langchain.chains import RetrievalQAWithSourcesChain
-from langchain.llms import OpenAI
 import uuid
 from project.llm.zhipuai_llm import ZhipuAILLM
 
 chain: Optional[Callable] = None
 
 
-def web_loader(url_list, openai_key, zilliz_uri, user, password):
+def web_loader(url_list, openai_key, puzhiai_key, zilliz_uri, user, password):
     if not url_list:
         return "please enter url list"
     loader = WebBaseLoader(url_list.split())
@@ -45,7 +44,7 @@ def web_loader(url_list, openai_key, zilliz_uri, user, password):
 
     global chain
     chain = RetrievalQAWithSourcesChain.from_chain_type(
-        ZhipuAILLM(model="glm-3-turbo", temperature=0.1, zhipuai_api_key=openai_key),
+        ZhipuAILLM(model="glm-3-turbo", temperature=0.1, zhipuai_api_key=puzhiai_key),
         chain_type="map_reduce",
         retriever=docsearch.as_retriever(),
     )
@@ -70,7 +69,7 @@ if __name__ == "__main__":
         <h1><center>Langchain And Zilliz Cloud Example</center></h1>
         This is how to use Zilliz Cloud as vector store in LangChain.
         The purpose of this example is to allow you to input multiple URLs (separated by newlines) and then ask questions about the content of the corresponding web pages.
-
+        v.2.26.19.47
         ## 📋 Prerequisite:
 
         1. 🔑 To obtain an OpenAI key, please visit https://platform.openai.com/account/api-keys.
@@ -93,6 +92,7 @@ if __name__ == "__main__":
             placeholder="https://milvus.io/docs/overview.md",
         )
         openai_key_text = gr.Textbox(label="openai api key", type="password", placeholder="sk-******")
+        puzhiai_key_text = gr.Textbox(label="puzhi api key", type="password", placeholder="******")
         with gr.Row():
             zilliz_uri_text = gr.Textbox(
                 label="zilliz cloud uri",
@@ -109,6 +109,7 @@ if __name__ == "__main__":
             inputs=[
                 url_list_text,
                 openai_key_text,
+                puzhiai_key_text,
                 zilliz_uri_text,
                 user_text,
                 password_text,