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