Update app.py

app.py

@@ -14,20 +14,18 @@ from langchain.tools import Tool
 from langgraph.graph import START, StateGraph
 from langgraph.graph.message import add_messages
 from langgraph.prebuilt import ToolNode, tools_condition
-from langchain.chat_models import init_chat_model
-from langchain_mistralai import ChatMistralAI
-from mistralai import Mistral
+from langchain_google_vertexai.gemma import GemmaChatLocalKaggle
 
 
 import getpass
 import os
 
-if "Mistral" not in os.environ:
-    os.environ["Mistral"] = getpass.getpass("Enter your Mistral API key: ")
+if "KAGGLE_API" not in os.environ:
+    os.environ["KAGGLE_API"] = getpass.getpass("Enter your Kaggle API key: ")
 
 print("Loading LLM...")
-chat = ChatMistralAI(
-    model="pixtral-large-2411", mistral_api_key=os.getenv("Mistral")
+chat = GemmaChatLocalKaggle(
+    model="gemma3:4b", kwargs={"api_key": os.getenv("KAGGLE_API")
 )
 print(f"Model {chat.model} downloaded!")