PradeepBodhi commited on
Commit
35b5f1a
·
verified ·
1 Parent(s): f7d4abf

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
agent.py CHANGED
@@ -1,4 +1,4 @@
1
- from langchain_openai import AzureChatOpenAI
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 = AzureChatOpenAI(model=os.getenv("AZURE_OPENAI_MODEL"))
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