team-ai / agents /code_execute_agent.py
peichao.dong
update code excute tool
411007e
raw
history blame contribute delete
326 Bytes
from langchain.agents import initialize_agent, AgentType
from models import llm
from agents.tools.shell_tool import shell_tool
from agents.tools.python_code_tool import repl_tool
generate_and_excute_code_agent = initialize_agent([shell_tool, repl_tool], llm(), agent=AgentType.CHAT_ZERO_SHOT_REACT_DESCRIPTION, verbose=True)