gemini support

app.py

@@ -12,11 +12,17 @@ import os, asyncio, trafilatura
 from langchain.docstore.document import Document
 import requests
 from langchain_openai import ChatOpenAI
+from langchain_google_genai import ChatGoogleGenerativeAI
+
+
+
+options = ["openai","gemini"]
+# 创建下拉选项框
+selected_option = st.selectbox("select your llm：", options)
+
+api_key = st.text_input("enter your llm api key:")
 
-load_dotenv()
 
-os.environ["OPENAI_API_KEY"] = os.getenv("OPENAI_API_KEY")
-openai.api_key = os.getenv("OPENAI_API_KEY")
 
 @st.cache_resource
 def get_url_name(url):
@@ -55,7 +61,13 @@ class WebpageQATool(BaseTool):
         raise NotImplementedError
 
 def run_llm(url, query):
-    llm = ChatOpenAI(temperature=0.5)
+    if selected_option == "openai":
+        os.environ["OPENAI_API_KEY"] = api_key
+        llm = ChatOpenAI(temperature=0.5)
+    if selected_option == "gemini":
+        os.environ["GOOGLE_API_KEY"] = api_key
+        llm = ChatGoogleGenerativeAI(temperature=0.5)
+    # llm = ChatOpenAI(temperature=0.5)
     query_website_tool = WebpageQATool(qa_chain=load_qa_with_sources_chain(llm))
     result = query_website_tool._run(url, query)  # Pass the URL and query as arguments
     return result
@@ -64,6 +76,11 @@ st.markdown("<h1 style='text-align: center; color: green;'>Info Retrieval from W
 st.markdown("<h3 style='text-align: center; color: green;'>Developed by <a href='https://github.com/AIAnytime'>AI Anytime with ❤️ </a></h3>", unsafe_allow_html=True)
 st.markdown("<h2 style='text-align: center; color:red;'>Enter the Website URL 👇</h2>", unsafe_allow_html=True)
 
+
+
+input_url = st.dropdown("Please Select you llm")
+
+
 input_url = st.text_input("Enter the URL")
 
 if len(input_url)>0:

requirements.txt

@@ -4,4 +4,5 @@ openai
 trafilatura
 streamlit
 python-dotenv
-langchain_openai
+langchain_openai
+langchain-google-genai