From 06066f5f0f2425da9e2887bcfacbd19f74720039 Mon Sep 17 00:00:00 2001 From: David Westgate Date: Sat, 15 Jun 2024 04:11:46 -0700 Subject: [PATCH] code cleanup --- final/app.py | 6 ++---- final/tools.py | 1 - 2 files changed, 2 insertions(+), 5 deletions(-) diff --git a/final/app.py b/final/app.py index 6923d85..aae662f 100644 --- a/final/app.py +++ b/final/app.py @@ -4,7 +4,6 @@ from langchain.agents import AgentExecutor, create_react_agent from langchain.tools import tool from tools import dalle, tcp_dump, get_adapter_interface, ip_loc, pcap_summary from langchain import hub -from langchain_community.tools import ShellTool from langsmith import Client from dotenv import load_dotenv @@ -15,11 +14,10 @@ Main rag tool-chain application. Simply load the tools and prompt the user. """ os.environ["LANGCHAIN_TRACING_V2"] = "true" -os.environ["LANGCHAIN_PROJECT"] = f"LangSmith Introduction" +os.environ["LANGCHAIN_PROJECT"] = f"Final Project" os.environ["LANGCHAIN_ENDPOINT"] = "https://api.smith.langchain.com" - client = Client() -shell_tool = ShellTool() + llm = ChatOpenAI(model_name="gpt-4o", temperature=0) tools = [] tools.extend([tcp_dump, ip_loc, dalle, pcap_summary, get_adapter_interface]) diff --git a/final/tools.py b/final/tools.py index 2d23515..9084129 100644 --- a/final/tools.py +++ b/final/tools.py @@ -6,7 +6,6 @@ from langchain_core.pydantic_v1 import BaseModel, Field from langchain.tools import tool from langchain_community.tools import ShellTool from langchain_community.utilities.dalle_image_generator import DallEAPIWrapper -from scapy import * import requests from scapy.all import rdpcap, wrpcap from scapy.layers.inet import IP, TCP, UDP, Packet, PacketList