Shekarss's picture
Update agent.py
85fdcdd verified
raw
history blame contribute delete
553 Bytes
from langgraph.prebuilt import create_react_agent
from langchain_openai import ChatOpenAI
from langchain_groq import ChatGroq
from langchain_core.messages import HumanMessage
import os
import dotenv
from tools import get_tools
import json
import pandas as pd
from tqdm import tqdm
dotenv.load_dotenv()
llm = ChatGroq(
model="qwen/qwen3-32b",
api_key=os.getenv("GROQ_API_KEY")
)
tools = get_tools()
with open("system_prompt.txt") as f:
system_prompt = f.read()
agent = create_react_agent(model=llm, tools=tools, prompt=system_prompt)