start work on app, readme
This commit is contained in:
parent
329241d9fd
commit
6966d33826
1
hw5/.env.example
Normal file
1
hw5/.env.example
Normal file
@ -0,0 +1 @@
|
||||
OPENAI_API_KEY=""
|
2
hw5/README.md
Normal file
2
hw5/README.md
Normal file
@ -0,0 +1,2 @@
|
||||
###### David Westgate 17 May 2024
|
||||
## HW4 for gensec
|
46
hw5/app.py
46
hw5/app.py
@ -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
|
Reference in New Issue
Block a user