BGarida commited on
Commit
3b468ed
·
1 Parent(s): a294dd2

Deploying PDF Version v10

Browse files
Files changed (2) hide show
  1. app.py +8 -16
  2. requirements.txt +5 -5
app.py CHANGED
@@ -13,24 +13,18 @@ logger.debug("Starting application")
13
 
14
  from typing import List
15
 
16
- from langchain.embeddings.openai import OpenAIEmbeddings
17
  from langchain.text_splitter import RecursiveCharacterTextSplitter
18
  from langchain.vectorstores import Chroma
19
- from langchain.chains import (
20
- ConversationalRetrievalChain,
21
- )
22
  from langchain.document_loaders import PyPDFLoader
23
  from langchain.chat_models import ChatOpenAI
24
- from langchain.prompts.chat import (
25
- ChatPromptTemplate,
26
- SystemMessagePromptTemplate,
27
- HumanMessagePromptTemplate,
28
- )
29
- from langchain.docstore.document import Document
30
  from langchain.memory import ChatMessageHistory, ConversationBufferMemory
31
- from chainlit.types import AskFileResponse
32
 
33
  import chainlit as cl
 
34
 
35
  text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=100)
36
 
@@ -134,23 +128,21 @@ async def start():
134
  logger.debug("Defining main function")
135
 
136
  @cl.on_message
137
- async def main(message):
138
  debug_info = f"""
139
  Debug Info:
140
  Message type: {type(message)}
141
  Message: {message}
 
142
  Message dir: {dir(message)}
143
  """
144
  print(debug_info)
145
 
146
  try:
147
- # If message is a string, use it directly
148
- message_content = message if isinstance(message, str) else message.content
149
-
150
  chain = cl.user_session.get("chain") # type: ConversationalRetrievalChain
151
  cb = cl.AsyncLangchainCallbackHandler()
152
 
153
- res = await chain.acall(message_content, callbacks=[cb])
154
  answer = res["answer"]
155
  source_documents = res["source_documents"] # type: List[Document]
156
 
 
13
 
14
  from typing import List
15
 
16
+ from langchain.embeddings import OpenAIEmbeddings
17
  from langchain.text_splitter import RecursiveCharacterTextSplitter
18
  from langchain.vectorstores import Chroma
19
+ from langchain.chains import ConversationalRetrievalChain
 
 
20
  from langchain.document_loaders import PyPDFLoader
21
  from langchain.chat_models import ChatOpenAI
22
+ from langchain.prompts import ChatPromptTemplate, SystemMessagePromptTemplate, HumanMessagePromptTemplate
23
+ from langchain.schema import Document
 
 
 
 
24
  from langchain.memory import ChatMessageHistory, ConversationBufferMemory
 
25
 
26
  import chainlit as cl
27
+ from chainlit.types import AskFileResponse
28
 
29
  text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=100)
30
 
 
128
  logger.debug("Defining main function")
129
 
130
  @cl.on_message
131
+ async def main(message: cl.Message):
132
  debug_info = f"""
133
  Debug Info:
134
  Message type: {type(message)}
135
  Message: {message}
136
+ Message content: {message.content}
137
  Message dir: {dir(message)}
138
  """
139
  print(debug_info)
140
 
141
  try:
 
 
 
142
  chain = cl.user_session.get("chain") # type: ConversationalRetrievalChain
143
  cb = cl.AsyncLangchainCallbackHandler()
144
 
145
+ res = await chain.acall(message.content, callbacks=[cb])
146
  answer = res["answer"]
147
  source_documents = res["source_documents"] # type: List[Document]
148
 
requirements.txt CHANGED
@@ -1,6 +1,6 @@
1
- langchain==0.0.198
2
- chromadb==0.3.22
3
- tiktoken==0.3.3
4
- pypdf==3.8.1
5
  chainlit==0.7.700
6
- openai==0.27.8
 
 
 
 
1
+ llangchain==0.0.252
 
 
 
2
  chainlit==0.7.700
3
+ openai==0.27.8
4
+ chromadb==0.3.26
5
+ tiktoken==0.4.0
6
+ pypdf==3.12.1