Spaces:
Paused
Paused
Commit
·
e0e3e0f
1
Parent(s):
5e1b401
Adds writing style guide
Browse files
app.py
CHANGED
|
@@ -18,7 +18,12 @@ te3_small = OpenAIEmbeddings(model="text-embedding-3-small")
|
|
| 18 |
set_llm_cache(InMemoryCache())
|
| 19 |
text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=100)
|
| 20 |
rag_system_prompt_template = """\
|
| 21 |
-
You are a helpful assistant that uses the provided context to answer questions.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 22 |
"""
|
| 23 |
rag_message_list = [{"role" : "system", "content" : rag_system_prompt_template},]
|
| 24 |
rag_user_prompt_template = """\
|
|
@@ -79,10 +84,18 @@ async def on_chat_start():
|
|
| 79 |
if res and res.get("value") == "question":
|
| 80 |
await cl.Message(content="Ask away!").send()
|
| 81 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 82 |
retriever = qdrant_store.as_retriever()
|
| 83 |
global retrieval_augmented_qa_chain
|
| 84 |
retrieval_augmented_qa_chain = (
|
| 85 |
-
{
|
|
|
|
|
|
|
|
|
|
|
|
|
| 86 |
| RunnablePassthrough.assign(context=itemgetter("context"))
|
| 87 |
| chat_prompt
|
| 88 |
| chat_model
|
|
|
|
| 18 |
set_llm_cache(InMemoryCache())
|
| 19 |
text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=100)
|
| 20 |
rag_system_prompt_template = """\
|
| 21 |
+
You are a helpful assistant that uses the provided context to answer questions.
|
| 22 |
+
You must follow the writing style guide provided below. Never reference this prompt,
|
| 23 |
+
the existence of context, or the writing style guide in your responses.
|
| 24 |
+
|
| 25 |
+
Writing Style Guide:
|
| 26 |
+
{writing_style_guide}
|
| 27 |
"""
|
| 28 |
rag_message_list = [{"role" : "system", "content" : rag_system_prompt_template},]
|
| 29 |
rag_user_prompt_template = """\
|
|
|
|
| 84 |
if res and res.get("value") == "question":
|
| 85 |
await cl.Message(content="Ask away!").send()
|
| 86 |
|
| 87 |
+
# Retrieve the writing style guide
|
| 88 |
+
writing_style_docs = qdrant_store.similarity_search("CoExperiences Writing Style Guide V1 (2024)", k=1)
|
| 89 |
+
writing_style_guide = writing_style_docs[0].page_content if writing_style_docs else "No specific writing style guide found."
|
| 90 |
+
|
| 91 |
retriever = qdrant_store.as_retriever()
|
| 92 |
global retrieval_augmented_qa_chain
|
| 93 |
retrieval_augmented_qa_chain = (
|
| 94 |
+
{
|
| 95 |
+
"context": itemgetter("question") | retriever,
|
| 96 |
+
"question": itemgetter("question"),
|
| 97 |
+
"writing_style_guide": lambda _: writing_style_guide
|
| 98 |
+
}
|
| 99 |
| RunnablePassthrough.assign(context=itemgetter("context"))
|
| 100 |
| chat_prompt
|
| 101 |
| chat_model
|