northstaranlyticsma24 commited on
Commit
0760685
·
verified ·
1 Parent(s): 73d55da

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +22 -23
app.py CHANGED
@@ -78,29 +78,6 @@ def process_file(file: IO):
78
  ### On Chat Start (Session Start) Section ###
79
  @cl.on_chat_start
80
  async def on_chat_start():
81
-
82
- # Adding cache!
83
- store = LocalFileStore("./cache/")
84
- cached_embedder = CacheBackedEmbeddings.from_bytes_store(
85
- core_embeddings, store, namespace=core_embeddings.model
86
- )
87
-
88
- # Typical QDrant Vector Store Set-up
89
- vectorstore = QdrantVectorStore(
90
- client=client,
91
- collection_name=collection_name,
92
- embedding=cached_embedder)
93
- vectorstore.add_documents(docs)
94
- retriever = vectorstore.as_retriever(search_type="mmr", search_kwargs={"k": 3})
95
-
96
- retrieval_augmented_qa_chain = (
97
- {"context": itemgetter("question") | retriever, "question": itemgetter("question")} ##
98
- | RunnablePassthrough.assign(context=itemgetter("context"))
99
- | chat_prompt | chat_model
100
- )
101
-
102
- cl.user_session.set("midterm_chain", retrieval_augmented_qa_chain)
103
-
104
  files = None
105
 
106
  # Wait for the user to upload a file
@@ -123,6 +100,28 @@ async def on_chat_start():
123
 
124
  # load the file
125
  docs = process_file(file)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
126
 
127
  ### Rename Chains ###
128
  @cl.author_rename
 
78
  ### On Chat Start (Session Start) Section ###
79
  @cl.on_chat_start
80
  async def on_chat_start():
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
81
  files = None
82
 
83
  # Wait for the user to upload a file
 
100
 
101
  # load the file
102
  docs = process_file(file)
103
+
104
+ # Adding cache!
105
+ store = LocalFileStore("./cache/")
106
+ cached_embedder = CacheBackedEmbeddings.from_bytes_store(
107
+ core_embeddings, store, namespace=core_embeddings.model
108
+ )
109
+
110
+ # Typical QDrant Vector Store Set-up
111
+ vectorstore = QdrantVectorStore(
112
+ client=client,
113
+ collection_name=collection_name,
114
+ embedding=cached_embedder)
115
+ vectorstore.add_documents(docs)
116
+ retriever = vectorstore.as_retriever(search_type="mmr", search_kwargs={"k": 3})
117
+
118
+ retrieval_augmented_qa_chain = (
119
+ {"context": itemgetter("question") | retriever, "question": itemgetter("question")} ##
120
+ | RunnablePassthrough.assign(context=itemgetter("context"))
121
+ | chat_prompt | chat_model
122
+ )
123
+
124
+ cl.user_session.set("midterm_chain", retrieval_augmented_qa_chain)
125
 
126
  ### Rename Chains ###
127
  @cl.author_rename