updating log analyzer

This commit is contained in:
Omar Santos 2023-08-18 22:39:35 -04:00
parent 0301b452fb
commit f9a15a3f59
2 changed files with 39 additions and 0 deletions

1
.gitignore vendored
View File

@ -1,2 +1,3 @@
.vscode .vscode
.DS_Store .DS_Store
ai_security/AI for Incident Response/.env

View File

@ -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)