Update Components/GooglePalmChat.py

Components/GooglePalmChat.py

@@ -1,6 +1,6 @@
 import os
 from dotenv import load_dotenv
-from langchain.llms import GooglePalm
+from langchain.llms import GoogleGenerativeAI
 from langchain.chains import RetrievalQA
 from langchain.vectorstores import FAISS
 from langchain.prompts import PromptTemplate
@@ -8,7 +8,8 @@ from langchain.prompts import PromptTemplate
 load_dotenv()  # take environment variables from .env (especially openai api key)
 
 # Create Google Palm LLM model
-llm = GooglePalm(google_api_key=os.environ["GOOGLE_PALM_API"], temperature=0.1)
+model_name = "models/text-bison-001" 
+llm = GoogleGenerativeAI(google_api_key=os.environ["GOOGLE_PALM_API"], model_name=model_name,temperature=0.1)
 vectordb_file_path = "faiss_index_V2"
 
 def get_qa_chain(embeddings):