From 767104cac86607e058be0b07e62ea9ed944eb154 Mon Sep 17 00:00:00 2001 From: David Westgate Date: Sun, 21 Apr 2024 14:14:40 -0700 Subject: [PATCH] added DALL-E tool and it seems to work --- hw2/app.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/hw2/app.py b/hw2/app.py index fc986c5..ed8139c 100644 --- a/hw2/app.py +++ b/hw2/app.py @@ -2,6 +2,8 @@ from langchain import hub from langchain.agents import AgentExecutor, create_react_agent, load_tools from langchain.tools import tool from langchain_openai import ChatOpenAI +from langchain_community.utilities.dalle_image_generator import DallEAPIWrapper + from dotenv import load_dotenv from tools import lookup_ip, lookup_name from langsmith import Client @@ -20,7 +22,7 @@ load_dotenv() #client = Client() 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]) base_prompt = hub.pull("langchain-ai/react-agent-template")