加入PaLM支持（未完成）

modules/config.py

@@ -77,6 +77,9 @@ my_api_key = os.environ.get("OPENAI_API_KEY", my_api_key)
 xmchat_api_key = config.get("xmchat_api_key", "")
 os.environ["XMCHAT_API_KEY"] = xmchat_api_key
 
+google_palm_api_key = config.get("google_palm_api_key", "")
+os.environ["GOOGLE_PALM_API_KEY"] = google_palm_api_key
+
 render_latex = config.get("render_latex", True)
 
 if render_latex:

modules/models/PaLM.py

@@ -1,10 +1,11 @@
 from .base_model import BaseLLMModel, CallbackToIterator, ChuanhuCallbackHandler
 from langchain.chat_models import ChatGooglePalm
+import os
 
 class PaLM_Client(BaseLLMModel):
     def __init__(self, model_name, user="") -> None:
         super().__init__(model_name, user)
-        self.llm = ChatGooglePalm(google_api_key="")
+        self.llm = ChatGooglePalm(google_api_key=os.environ["GOOGLE_PALM_API_KEY"])
 
     def get_answer_at_once(self):
         self.llm.generate(self.history)

requirements.txt

@@ -24,3 +24,4 @@ wikipedia
 google.generativeai
 openai
 unstructured
+google-api-python-client