Spaces:
Running
Running
Update chatbot_advisor.py
Browse files- chatbot_advisor.py +3 -2
chatbot_advisor.py
CHANGED
|
@@ -2,6 +2,7 @@ from langgraph.graph import StateGraph, START, END
|
|
| 2 |
from typing import TypedDict, Annotated
|
| 3 |
from langchain_core.messages import HumanMessage, BaseMessage, AIMessage, SystemMessage
|
| 4 |
from langchain_openai import ChatOpenAI
|
|
|
|
| 5 |
from langgraph.checkpoint.memory import MemorySaver
|
| 6 |
from langgraph.graph.message import add_messages
|
| 7 |
from langgraph.prebuilt import ToolNode, tools_condition
|
|
@@ -42,8 +43,8 @@ tools = [tavily_search]
|
|
| 42 |
# =====================================================
|
| 43 |
# LLM CONFIGURATION
|
| 44 |
# =====================================================
|
| 45 |
-
llm =
|
| 46 |
-
model="gpt-
|
| 47 |
streaming=True
|
| 48 |
).bind_tools(tools)
|
| 49 |
|
|
|
|
| 2 |
from typing import TypedDict, Annotated
|
| 3 |
from langchain_core.messages import HumanMessage, BaseMessage, AIMessage, SystemMessage
|
| 4 |
from langchain_openai import ChatOpenAI
|
| 5 |
+
from langchain_groq import ChatGroq
|
| 6 |
from langgraph.checkpoint.memory import MemorySaver
|
| 7 |
from langgraph.graph.message import add_messages
|
| 8 |
from langgraph.prebuilt import ToolNode, tools_condition
|
|
|
|
| 43 |
# =====================================================
|
| 44 |
# LLM CONFIGURATION
|
| 45 |
# =====================================================
|
| 46 |
+
llm = ChatGroq(
|
| 47 |
+
model="openai/gpt-oss-120b",
|
| 48 |
streaming=True
|
| 49 |
).bind_tools(tools)
|
| 50 |
|