danicafisher commited on
Commit
0a4803e
1 Parent(s): 3214177

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -16
app.py CHANGED
@@ -31,21 +31,21 @@ chat_prompt = ChatPromptTemplate.from_messages([("system", rag_system_prompt_tem
31
 
32
  @cl.on_chat_start
33
  async def on_chat_start():
34
- qdrant_client = QdrantClient(url=os.environ["QDRANT_ENDPOINT"], api_key=os.environ["QDRANT_API_KEY"])
35
- qdrant_store = Qdrant(
36
- client=qdrant_client,
37
- collection_name="kai_test_docs",
38
- embeddings=te3_small
39
- )
40
- retriever = qdrant_store.as_retriever()
41
 
42
- global retrieval_augmented_qa_chain
43
- retrieval_augmented_qa_chain = (
44
- {"context": itemgetter("question") | retriever, "question": itemgetter("question")}
45
- | RunnablePassthrough.assign(context=itemgetter("context"))
46
- | chat_prompt
47
- | chat_model
48
- )
49
 
50
  await cl.Message(content="YAsk away!").send()
51
 
@@ -55,5 +55,6 @@ def rename(orig_author: str):
55
 
56
  @cl.on_message
57
  async def main(message: cl.Message):
58
- response = retrieval_augmented_qa_chain.invoke({"question": message.content})
59
- await cl.Message(content=response.content).send()
 
 
31
 
32
  @cl.on_chat_start
33
  async def on_chat_start():
34
+ # qdrant_client = QdrantClient(url=os.environ["QDRANT_ENDPOINT"], api_key=os.environ["QDRANT_API_KEY"])
35
+ # qdrant_store = Qdrant(
36
+ # client=qdrant_client,
37
+ # collection_name="kai_test_docs",
38
+ # embeddings=te3_small
39
+ # )
40
+ # retriever = qdrant_store.as_retriever()
41
 
42
+ # global retrieval_augmented_qa_chain
43
+ # retrieval_augmented_qa_chain = (
44
+ # {"context": itemgetter("question") | retriever, "question": itemgetter("question")}
45
+ # | RunnablePassthrough.assign(context=itemgetter("context"))
46
+ # | chat_prompt
47
+ # | chat_model
48
+ # )
49
 
50
  await cl.Message(content="YAsk away!").send()
51
 
 
55
 
56
  @cl.on_message
57
  async def main(message: cl.Message):
58
+ # response = retrieval_augmented_qa_chain.invoke({"question": message.content})
59
+ # await cl.Message(content=response.content).send()
60
+ await cl.Message(content="Response").send()