david/gensec-westgate-djw2
Archived
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

added DALL-E tool and it seems to work

Browse Source
This commit is contained in:
David Westgate 2024-04-21 14:14:40 -07:00
parent bbd62ce4fc
commit 767104cac8
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
hw2/app.py
View File

@ -2,6 +2,8 @@ from langchain import hub
from langchain.agents import AgentExecutor, create_react_agent, load_tools from langchain.agents import AgentExecutor, create_react_agent, load_tools
from langchain.tools import tool from langchain.tools import tool
from langchain_openai import ChatOpenAI from langchain_openai import ChatOpenAI
from langchain_community.utilities.dalle_image_generator import DallEAPIWrapper
from dotenv import load_dotenv from dotenv import load_dotenv
from tools import lookup_ip, lookup_name from tools import lookup_ip, lookup_name
from langsmith import Client from langsmith import Client
@ -20,7 +22,7 @@ load_dotenv()
#client = Client() #client = Client()
llm = ChatOpenAI(model_name="gpt-4-turbo", temperature=0) llm = ChatOpenAI(model_name="gpt-4-turbo", temperature=0)
tools = load_tools(["serpapi", "terminal"], allow_dangerous_tools=True) tools = load_tools(["serpapi", "terminal", "dalle-image-generator"], allow_dangerous_tools=True)
tools.extend([lookup_name, lookup_ip]) tools.extend([lookup_name, lookup_ip])
base_prompt = hub.pull("langchain-ai/react-agent-template") base_prompt = hub.pull("langchain-ai/react-agent-template")