added gemini model

app/services/sql_agent.py

@@ -352,16 +352,19 @@ from langchain_community.agent_toolkits import SQLDatabaseToolkit
 from langchain_groq import ChatGroq
 from langchain_core.messages import HumanMessage, AIMessage
 from langchain_core.prompts import ChatPromptTemplate
-from langchain_core.pydantic_v1 import BaseModel, Field
+# from langchain_core.pydantic_v1 import BaseModel, Field
 from langgraph.graph import StateGraph, END, MessagesState
 from typing import TypedDict, Annotated, List, Literal, Dict, Any
+from langchain_google_genai import ChatGoogleGenerativeAI
+
 from dotenv import load_dotenv
 load_dotenv()
 import os
-os.environ["GROQ_API_KEY"]=os.getenv("GROQ_API_KEY")
+# os.environ["GROQ_API_KEY"]=os.getenv("GROQ_API_KEY")
+os.environ["GEMINI_API_KEY"]=os.getenv("GEMINI_API_KEY")
 
 class SQLAgent:
-    def __init__(self, model="openai/gpt-oss-120b"):
+    def __init__(self, model="gemma2-9b-it"):
         
         # Initialize instance variables
         self.db = None
@@ -373,8 +376,8 @@ class SQLAgent:
         self.app = None
         
         # Setting up LLM
-        self.llm = ChatGroq(model=model,api_key = os.getenv("GROQ_API_KEY"))
-        
+        # self.llm = ChatGroq(model=model,api_key = os.getenv("GROQ_API_KEY"))
+        self.llm = ChatGoogleGenerativeAI(model="gemini-2.5-flash-lite", google_api_key=os.environ["GEMINI_API_KEY"])
         # Register the tool method
         self.query_to_database = self._create_query_tool()
 

pyproject.toml

@@ -5,13 +5,14 @@ description = "Add your description here"
 readme = "README.md"
 requires-python = ">=3.13"
 dependencies = [
-    "bcrypt>=4.3.0",
+    "bcrypt==4.3.0",
     "fastapi>=0.116.1",
     "ipykernel>=6.29.5",
     "ipython>=9.4.0",
     "langchain>=0.3.26",
     "langchain-community>=0.3.27",
     "langchain-core>=0.3.68",
+    "langchain-google-genai>=2.1.9",
     "langchain-groq>=0.3.6",
     "langgraph>=0.5.3",
     "pandas>=2.3.1",

requirements.txt

@@ -15,4 +15,5 @@ ipykernel
 passlib
 python-multipart
 bcrypt==4.3.0
-psycopg2-binary
+psycopg2-binary
+langchain-google-genai==2.1.9