diff --git a/.gitignore b/.gitignore index a0f0e53..d636c9c 100644 --- a/.gitignore +++ b/.gitignore @@ -1,2 +1,3 @@ .vscode .DS_Store +ai_security/AI for Incident Response/.env diff --git a/ai_security/AI for Incident Response/analyzing_logs.py b/ai_security/AI for Incident Response/analyzing_logs.py index e69de29..9e8d96f 100644 --- a/ai_security/AI for Incident Response/analyzing_logs.py +++ b/ai_security/AI for Incident Response/analyzing_logs.py @@ -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) + +