agent_course / agent.py
echodrift's picture
feat: test answer
7b83fec
from langgraph.prebuilt import create_react_agent
from langchain_openai import ChatOpenAI
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 = ChatOpenAI(
api_key=os.getenv("API_KEY"),
model=os.getenv("MODEL")
)
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)