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