add gemini

agent.py

@@ -14,6 +14,8 @@ from langchain_core.output_parsers import StrOutputParser
 from langchain_core.tools import Tool
 from langchain_tavily import TavilySearch
 from langchain_tavily import TavilySearch, TavilyExtract
+from langchain_google_genai import ChatGoogleGenerativeAI
+
 
 # from langchain_community.tools.tavily_search import TavilySearchResults
 
@@ -120,15 +122,23 @@ sys_msg = SystemMessage(content=system_prompt)
 
 tools = [
     wiki_search,
-    tavily_search_tool,
-    tavily_extract_tool,
+    # tavily_search_tool,
+    # tavily_extract_tool,
     repl_tool,
     read_excel_file,
     transcribe_audio
 ]
 
 
-llm = ChatGroq(model="qwen-qwq-32b", temperature=0.7)
+# llm = ChatGroq(model="qwen-qwq-32b", temperature=0.7)
+llm = ChatGoogleGenerativeAI(
+    model="gemini-2.0-flash",
+    temperature=0.7,
+    max_tokens=None,
+    timeout=None,
+    max_retries=2,
+    # other params...
+)
 llm_with_tools = llm.bind_tools(tools)
 
 def assistant(state: MessagesState):
@@ -161,6 +171,7 @@ if __name__ == "__main__":
     question = "Hi, I was out sick from my classes on Friday, so I'm trying to figure out what I need to study for my Calculus mid-term next week. My friend from class sent me an audio recording of Professor Willowbrook giving out the recommended reading for the test, but my headphones are broken :(\n\nCould you please listen to the recording for me and tell me the page numbers I'm supposed to go over? I've attached a file called Homework.mp3 that has the recording. Please provide just the page numbers as a comma-delimited list. And please provide the list in ascending order."
     question = "What is the first name of the only Malko Competition recipient from the 20th Century (after 1977) whose nationality on record is a country that no longer exists?"
     question = "Where were the Vietnamese specimens described by Kuznetzov in Nedoshivina's 2010 paper eventually deposited? Just give me the city name without abbreviations."
+    question = "How many studio albums were published by Mercedes Sosa between 2000 and 2009 (included)? You can use the latest 2022 version of english wikipedia."
     # getmessages = [HumanMessage(content=question)]
     # messages = graph.invoke({"messages": messages})
     # for m in messages["messages"]:

requirements.txt

@@ -8,4 +8,5 @@ wikipedia
 duckduckgo-search
 langchain-experimental
 assemblyai
-langchain-tavily
+langchain-tavily
+langchain-google-genai