Shekarss commited on
Commit
aff4080
·
verified ·
1 Parent(s): 2f7f0aa

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +4 -3
agent.py CHANGED
@@ -1,5 +1,6 @@
1
  from langgraph.prebuilt import create_react_agent
2
  from langchain_openai import ChatOpenAI
 
3
  from langchain_core.messages import HumanMessage
4
  import os
5
  import dotenv
@@ -10,9 +11,9 @@ from tqdm import tqdm
10
 
11
  dotenv.load_dotenv()
12
 
13
- llm = ChatOpenAI(
14
- api_key=os.getenv("API_KEY"),
15
- model='gpt-4o-mini'
16
  )
17
 
18
  tools = get_tools()
 
1
  from langgraph.prebuilt import create_react_agent
2
  from langchain_openai import ChatOpenAI
3
+ from langchain.chat_models import ChatGroq
4
  from langchain_core.messages import HumanMessage
5
  import os
6
  import dotenv
 
11
 
12
  dotenv.load_dotenv()
13
 
14
+ llm = ChatGroq(
15
+ model="qwen/qwen3-32b",
16
+ api_key=os.getenv("GROQ_API_KEY")
17
  )
18
 
19
  tools = get_tools()