start work on app, readme

This commit is contained in:
David Westgate 2024-05-11 10:37:05 -07:00
parent 329241d9fd
commit 6966d33826
3 changed files with 49 additions and 0 deletions

1
hw5/.env.example Normal file
View File

@ -0,0 +1 @@
OPENAI_API_KEY=""

2
hw5/README.md Normal file
View File

@ -0,0 +1,2 @@
###### David Westgate 17 May 2024
## HW4 for gensec

View File

@ -0,0 +1,46 @@
from langchain_community.document_loaders.generic import GenericLoader
from langchain_community.document_loaders.parsers import LanguageParser
from langchain_openai import ChatOpenAI
from langchain_core.runnables import RunnablePassthrough
from langchain_core.prompts import PromptTemplate
from langchain_core.output_parsers import StrOutputParser
from langchain_community.document_loaders import AsyncHtmlLoader
from dotenv import load_dotenv
from validators import url
from time import time
"""
Comments, TODO
"""
load_dotenv()
promt = "You are a malware reverse engineer"
def get_rag_chain():
return (
{"context": retriever | format_docs, "question": RunnablePassthrough()}
| prompt
| llm
| StrOutputParser()
)
llm = ChatOpenAI(model_name="gpt-4-turbo", temperature=0)
print(
"Welcome to HW4"
)
while True:
try:
line: str = input("llm>> ")
if line:
start_time = time()
result: str = get_rag_chain.invoke(line)
end_time = time()
elapsed_time = round(end_time - start_time, 2)
print("\n", result, "\n\nElapsed time: ", elapsed_time, " seconds")
else:
break
except Exception as e:
print(e)
break