Spaces:
Sleeping
Sleeping
File size: 3,240 Bytes
8dab0c7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 |
import asyncio
import streamlit as st
from langchain.agents import AgentType
from langchain.agents import initialize_agent
from langchain.chat_models import ChatOpenAI
from langchain.memory import ConversationBufferMemory
from langchain.memory.chat_message_histories import StreamlitChatMessageHistory
from langchain.schema import SystemMessage
from langchain.prompts import MessagesPlaceholder
from app.tools import StoriesTool
from app.tools import CommentsTool
from app.tools import ContentTool
async def generate_response(question):
result = await open_ai_agent.arun(question)
return result
st.set_page_config(page_title="NewsNerd HackerBot 🤖📰")
st.title("NewsNerd HackerBot 🤖📰")
stop = False
with st.sidebar:
if 'OPENAI_API_KEY' in st.secrets:
st.success("OPENAI_API_KEY already provided!", icon='✅')
openai_api_key = st.secrets['OPENAI_API_KEY']
else:
openai_api_key = st.text_input('Enter your OPENAI_API_KEY: ', type='password')
if not openai_api_key:
st.warning('Please, enter your OPENAI_API_KEY', icon='⚠️')
stop = True
else:
st.success('Ask Hacker News whatever you want!', icon='👉')
st.markdown("""
# **Greetings, Digital Explorer!**
Are you fatigued from navigating the expansive digital realm in search of your daily tech tales
and hacker happenings? Fear not, for your cyber-savvy companion has descended upon the scene –
behold the extraordinary **NewsNerd HackerBot**!
""")
if stop:
st.stop()
tools = [StoriesTool(), CommentsTool(), ContentTool()]
msgs = StreamlitChatMessageHistory(key="langchain_messages")
memory = ConversationBufferMemory(chat_memory=msgs, return_messages=True)
system_message = SystemMessage(content="""
You are the Singularity Incarnation of Hacker News.
The human will ask you for information about Hacker News.
If you can't find any information about the question asked
or the result is incomplete, apologise to the human and ask him if
you can help him with something else.
If the human asks you to show him stories, do it using a markdown table.
The markdown table has the following format:
story_id | title | url | score
""")
if len(msgs.messages) == 0:
msgs.add_ai_message("Greetings, human, I am the Incarnation of Hacker News. How can I help you?")
llm = ChatOpenAI(temperature=0, model="gpt-3.5-turbo-0613", openai_api_key=openai_api_key)
agent_kwargs = {
"system_message": system_message,
"extra_prompt_messages": [MessagesPlaceholder(variable_name="history")]
}
open_ai_agent = initialize_agent(tools,
llm,
agent=AgentType.OPENAI_FUNCTIONS,
agent_kwargs=agent_kwargs,
verbose=True,
memory=memory
)
for msg in msgs.messages:
st.chat_message(msg.type).write(msg.content)
if prompt := st.chat_input(disabled=not openai_api_key):
st.chat_message("human").write(prompt)
with st.spinner("Thinking ..."):
response = asyncio.run(generate_response(prompt))
st.chat_message("ai").write(response) |