-
Notifications
You must be signed in to change notification settings - Fork 5
/
Copy pathmain.py
29 lines (23 loc) · 1001 Bytes
/
main.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
from tools.retrieval_eventi import get_relevant_document_tool
from tools.aws import ec2_shutdown_tools, ec2_turnon_tools
from tools.tavily import web_search_tool
from tools.utils import get_today_date_tool
# from prompts.react import prompt_react
from langchain import hub
from langchain.agents import AgentExecutor, create_react_agent
from langchain_groq import ChatGroq
import os
from dotenv import load_dotenv
load_dotenv(override=True)
GROQ_API_KEY=os.getenv("GROQ_API_KEY")
tools = [get_relevant_document_tool, get_today_date_tool, web_search_tool]
retrieved_text = ""
# Load ReAct prompt
prompt_react = hub.pull("hwchase17/react")
# Initialize ChatGroq model for language understanding
model = ChatGroq(model_name="llama3-70b-8192", groq_api_key=GROQ_API_KEY, temperature=0)
# Create ReAct agent
react_agent = create_react_agent(model, tools=tools, prompt=prompt_react)
react_agent_executor = AgentExecutor(
agent=react_agent, tools=tools, verbose=True, handle_parsing_errors=True
)