File size: 516 Bytes
60ba1ca
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
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)