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

code cleanup

Browse Source
This commit is contained in:
David Westgate 2024-06-15 04:11:46 -07:00
parent cbf63f0ab5
commit 06066f5f0f
2 changed files with 2 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
final/app.py
View File

@ -4,7 +4,6 @@ from langchain.agents import AgentExecutor, create_react_agent
from langchain.tools import tool from langchain.tools import tool
from tools import dalle, tcp_dump, get_adapter_interface, ip_loc, pcap_summary from tools import dalle, tcp_dump, get_adapter_interface, ip_loc, pcap_summary
from langchain import hub from langchain import hub
from langchain_community.tools import ShellTool
from langsmith import Client from langsmith import Client
from dotenv import load_dotenv 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_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" os.environ["LANGCHAIN_ENDPOINT"] = "https://api.smith.langchain.com"
client = Client() client = Client()
shell_tool = ShellTool()
llm = ChatOpenAI(model_name="gpt-4o", temperature=0) llm = ChatOpenAI(model_name="gpt-4o", temperature=0)
tools = [] tools = []
tools.extend([tcp_dump, ip_loc, dalle, pcap_summary, get_adapter_interface]) tools.extend([tcp_dump, ip_loc, dalle, pcap_summary, get_adapter_interface])

1
final/tools.py
View File

@ -6,7 +6,6 @@ from langchain_core.pydantic_v1 import BaseModel, Field
from langchain.tools import tool from langchain.tools import tool
from langchain_community.tools import ShellTool from langchain_community.tools import ShellTool
from langchain_community.utilities.dalle_image_generator import DallEAPIWrapper from langchain_community.utilities.dalle_image_generator import DallEAPIWrapper
from scapy import *
import requests import requests
from scapy.all import rdpcap, wrpcap from scapy.all import rdpcap, wrpcap
from scapy.layers.inet import IP, TCP, UDP, Packet, PacketList from scapy.layers.inet import IP, TCP, UDP, Packet, PacketList