Update app2.py

app2.py

@@ -1,9 +1,11 @@
 import os
 import streamlit as st
 from langchain_community.utilities import GoogleSearchAPIWrapper
+from langchain_google_genai import GoogleGenerativeAI
 from langchain.llms import GooglePalm
 from langchain.chains import ConversationChain
 from langchain.memory import ConversationBufferMemory
+
 import subprocess
 import git
 import logging
@@ -81,12 +83,13 @@ if st.session_state.GOOGLE_API_KEY:
             return f"An unexpected error occurred while running the command."
 
     def get_agent_response(message: str, system_prompt: str):
-      llm = GooglePalm(google_api_key=st.session_state.GOOGLE_API_KEY)  # Use GooglePalm
-      memory = ConversationBufferMemory()
-      conversation = ConversationChain(llm=llm, memory=memory) 
-      full_prompt = f"{system_prompt}\n{message}"
-      response = conversation.run(full_prompt)
-      return response
+        llm = GoogleGenerativeAI(google_api_key=st.session_state.GOOGLE_API_KEY)  # Use GoogleGenerativeAI
+        memory = ConversationBufferMemory()
+        conversation = ConversationChain(llm=llm, memory=memory) 
+        full_prompt = f"{system_prompt}\n{message}"
+        response = conversation.run(full_prompt)
+            return response
+
 
     def display_agent_info(agent_name: str):
         agent = agents[agent_name]