Spaces:
Sleeping
Sleeping
Update langgraph_code.py
Browse files- 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 =
|
| 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
|