HanLee commited on
Commit
2e7a35a
·
1 Parent(s): 477f1cf

feat: foobar

Browse files
Files changed (1) hide show
  1. app/app.py +14 -3
app/app.py CHANGED
@@ -1,15 +1,26 @@
 
 
 
 
 
 
1
  from tempfile import NamedTemporaryFile
2
  from typing import List
3
 
4
  import chainlit as cl
5
  from chainlit.types import AskFileResponse
 
 
 
 
6
  from langchain.chat_models import ChatOpenAI
 
 
7
  from langchain.prompts import ChatPromptTemplate
8
  from langchain.schema import Document, StrOutputParser
9
- from langchain.chains import LLMChain
10
-
11
- from langchain.document_loaders import PDFPlumberLoader
12
  from langchain.text_splitter import RecursiveCharacterTextSplitter
 
 
13
 
14
 
15
  def process_file(*, file: AskFileResponse) -> List[Document]:
 
1
+ # Chroma compatibility issue resolution
2
+ # https://docs.trychroma.com/troubleshooting#sqlite
3
+ __import__('pysqlite3')
4
+ import sys
5
+ sys.modules['sqlite3'] = sys.modules.pop('pysqlite3')
6
+
7
  from tempfile import NamedTemporaryFile
8
  from typing import List
9
 
10
  import chainlit as cl
11
  from chainlit.types import AskFileResponse
12
+
13
+ import chromadb
14
+ from chromadb.config import Settings
15
+ from langchain.chains import LLMChain
16
  from langchain.chat_models import ChatOpenAI
17
+ from langchain.document_loaders import PDFPlumberLoader
18
+ from langchain.embeddings.openai import OpenAIEmbeddings
19
  from langchain.prompts import ChatPromptTemplate
20
  from langchain.schema import Document, StrOutputParser
 
 
 
21
  from langchain.text_splitter import RecursiveCharacterTextSplitter
22
+ from langchain.vectorstores import Chroma
23
+ from langchain.vectorstores.base import VectorStore
24
 
25
 
26
  def process_file(*, file: AskFileResponse) -> List[Document]: