Spaces:
Sleeping
Sleeping
Update chatbot_advisor.py
Browse files- chatbot_advisor.py +9 -1
chatbot_advisor.py
CHANGED
|
@@ -6,6 +6,7 @@ from langgraph.graph.message import add_messages
|
|
| 6 |
from langgraph.prebuilt import ToolNode, tools_condition
|
| 7 |
from langchain_community.tools.tavily_search import TavilySearchResults
|
| 8 |
from langchain_groq import ChatGroq
|
|
|
|
| 9 |
from langchain_core.tools import tool
|
| 10 |
from dotenv import load_dotenv
|
| 11 |
import os
|
|
@@ -42,12 +43,19 @@ tools = [tavily_search]
|
|
| 42 |
# LLM
|
| 43 |
# =====================================================
|
| 44 |
|
| 45 |
-
llm = ChatGroq(
|
| 46 |
model="openai/gpt-oss-20b",
|
| 47 |
streaming=True,
|
| 48 |
temperature=0.3,
|
| 49 |
api_key=os.getenv("GROQ_API_KEY")
|
| 50 |
).bind_tools(tools)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 51 |
|
| 52 |
|
| 53 |
# =====================================================
|
|
|
|
| 6 |
from langgraph.prebuilt import ToolNode, tools_condition
|
| 7 |
from langchain_community.tools.tavily_search import TavilySearchResults
|
| 8 |
from langchain_groq import ChatGroq
|
| 9 |
+
from langchain_openai import ChatOpenAI
|
| 10 |
from langchain_core.tools import tool
|
| 11 |
from dotenv import load_dotenv
|
| 12 |
import os
|
|
|
|
| 43 |
# LLM
|
| 44 |
# =====================================================
|
| 45 |
|
| 46 |
+
"""llm = ChatGroq(
|
| 47 |
model="openai/gpt-oss-20b",
|
| 48 |
streaming=True,
|
| 49 |
temperature=0.3,
|
| 50 |
api_key=os.getenv("GROQ_API_KEY")
|
| 51 |
).bind_tools(tools)
|
| 52 |
+
"""
|
| 53 |
+
llm=ChatOpenAI(
|
| 54 |
+
model="gpt-4.1-nano",
|
| 55 |
+
streaming=True,
|
| 56 |
+
temperature=0.3,
|
| 57 |
+
api_key=os.getenv("OPENAI_API_KEY")
|
| 58 |
+
).bind_tools(tools)
|
| 59 |
|
| 60 |
|
| 61 |
# =====================================================
|