yashwan2003 commited on
Commit
9c97925
·
verified ·
1 Parent(s): 34ef6db

Update langgraph_code.py

Browse files
Files changed (1) hide show
  1. langgraph_code.py +1 -2
langgraph_code.py CHANGED
@@ -6,7 +6,6 @@ from typing import TypedDict, Annotated, Sequence, Literal
6
  from rapidfuzz import process
7
  from langchain_core.messages import BaseMessage, HumanMessage, AIMessage, SystemMessage
8
  from langchain_openai import ChatOpenAI
9
- from langchain_google_genai import ChatGoogleGenerativeAI
10
  from langgraph.graph import StateGraph, END
11
  from langgraph.prebuilt import ToolNode
12
  from langchain_core.messages import messages_to_dict
@@ -22,7 +21,7 @@ load_dotenv()
22
  DB_PATH = "movie_booking_details.db"
23
 
24
  # Initialize LLM
25
- llm = ChatGoogleGenerativeAI(model="gemini-2.5-flash",temperature=0)
26
 
27
  # ============================================================================
28
  # State Definition
 
6
  from rapidfuzz import process
7
  from langchain_core.messages import BaseMessage, HumanMessage, AIMessage, SystemMessage
8
  from langchain_openai import ChatOpenAI
 
9
  from langgraph.graph import StateGraph, END
10
  from langgraph.prebuilt import ToolNode
11
  from langchain_core.messages import messages_to_dict
 
21
  DB_PATH = "movie_booking_details.db"
22
 
23
  # Initialize LLM
24
+ llm = ChatOpenAI(model="gpt-4o-mini",temperature=0)
25
 
26
  # ============================================================================
27
  # State Definition