papers_rag.app

Streamlit query chat over the local papers Chroma index.

Attributes

Classes

ChatRole

Roles stored in the Streamlit chat transcript.

ChatMessageKey

Keys on each message dict in st.session_state.

Functions

main(→ None)

Render the Papers RAG chat UI.

run(→ None)

Launch Streamlit for the papers chat (console script papers-rag).

Module Contents

class papers_rag.app.ChatRole[source]

Bases: enum.StrEnum

Roles stored in the Streamlit chat transcript.

USER = 'user'[source]
ASSISTANT = 'assistant'[source]
class papers_rag.app.ChatMessageKey[source]

Bases: enum.StrEnum

Keys on each message dict in st.session_state.

ROLE = 'role'[source]
CONTENT = 'content'[source]
CITATIONS = 'citations'[source]
papers_rag.app.MESSAGES_STATE_KEY: Final = 'messages'[source]
papers_rag.app.STRATEGY_STATE_KEY: Final = 'rag_strategy'[source]
papers_rag.app.main() None[source]

Render the Papers RAG chat UI.

papers_rag.app.run() None[source]

Launch Streamlit for the papers chat (console script papers-rag).

Extra CLI args after papers-rag are forwarded to streamlit run.