mirror of
https://github.com/The-Art-of-Hacking/h4cker
synced 2024-11-21 18:33:03 +00:00
updating log analyzer
This commit is contained in:
parent
0301b452fb
commit
f9a15a3f59
2 changed files with 39 additions and 0 deletions
1
.gitignore
vendored
1
.gitignore
vendored
|
@ -1,2 +1,3 @@
|
||||||
.vscode
|
.vscode
|
||||||
.DS_Store
|
.DS_Store
|
||||||
|
ai_security/AI for Incident Response/.env
|
||||||
|
|
|
@ -0,0 +1,38 @@
|
||||||
|
'''
|
||||||
|
A simple test to interact with the OpenAI API
|
||||||
|
and analyze logs from applications, firewalls, operating systems, and more.
|
||||||
|
Author: Omar Santos, @santosomar
|
||||||
|
'''
|
||||||
|
|
||||||
|
# Import the required libraries
|
||||||
|
# pip3 install openai python-dotenv
|
||||||
|
# Use the line above if you need to install the libraries
|
||||||
|
from dotenv import load_dotenv
|
||||||
|
import openai
|
||||||
|
import os
|
||||||
|
|
||||||
|
# Load the .env file
|
||||||
|
load_dotenv()
|
||||||
|
|
||||||
|
# Get the API key from the environment variable
|
||||||
|
openai.api_key = os.getenv('OPENAI_API_KEY')
|
||||||
|
|
||||||
|
# Read the diff from a file
|
||||||
|
with open('logs.txt', 'r') as file:
|
||||||
|
log_file = file.read()
|
||||||
|
|
||||||
|
# Prepare the prompt
|
||||||
|
prompt = [{"role": "user", "content": f"Explain the following logs:\n\n{log_file}"}]
|
||||||
|
|
||||||
|
# Generate the AI chat completion via the OpenAI API
|
||||||
|
# I am only using GTP 3.5 Turbo for this example.
|
||||||
|
response = openai.ChatCompletion.create(
|
||||||
|
model="gpt-3.5-turbo-16k",
|
||||||
|
messages=prompt,
|
||||||
|
max_tokens=10000
|
||||||
|
)
|
||||||
|
|
||||||
|
# print the response from the OpenAI API
|
||||||
|
print(response.choices[0].message.content)
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue