Spaces:
Sleeping
Sleeping
Update agent.py
Browse files
agent.py
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
from
|
| 2 |
from langchain_core.messages import SystemMessage, HumanMessage
|
| 3 |
from langchain_core.tools import tool
|
| 4 |
from langchain_community.tools.tavily_search import TavilySearchResults
|
|
@@ -91,7 +91,7 @@ sys_msg = SystemMessage(content=system_prompt)
|
|
| 91 |
|
| 92 |
# Build graph function
|
| 93 |
def build_graph():
|
| 94 |
-
llm =
|
| 95 |
llm_with_tools = llm.bind_tools(tools)
|
| 96 |
|
| 97 |
# Node
|
|
|
|
| 1 |
+
from langchain_groq import ChatGroq
|
| 2 |
from langchain_core.messages import SystemMessage, HumanMessage
|
| 3 |
from langchain_core.tools import tool
|
| 4 |
from langchain_community.tools.tavily_search import TavilySearchResults
|
|
|
|
| 91 |
|
| 92 |
# Build graph function
|
| 93 |
def build_graph():
|
| 94 |
+
llm = ChatGroq(model="qwen-qwq-32b", temperature=0)
|
| 95 |
llm_with_tools = llm.bind_tools(tools)
|
| 96 |
|
| 97 |
# Node
|