bentobytes commited on
Commit
da554da
·
1 Parent(s): 413baa4

Update indexes.py

Browse files
Files changed (1) hide show
  1. indexes.py +1 -2
indexes.py CHANGED
@@ -13,7 +13,7 @@ from langchain.vectorstores import Chroma
13
  load_dotenv()
14
  openai_api_key=os.getenv('OPENAI_API_KEY')
15
 
16
- def create_indexes(file: tempfile, collection_name: str) -> str:
17
  try:
18
  file_path = file.name
19
  reader = PyPDFLoader(file_path)
@@ -25,7 +25,6 @@ def create_indexes(file: tempfile, collection_name: str) -> str:
25
  persist_directory = './db_metadata'
26
 
27
  vectordb = Chroma.from_documents(
28
- collection_name=collection_name,
29
  documents=documents,
30
  embedding=embeddings,
31
  persist_directory=persist_directory
 
13
  load_dotenv()
14
  openai_api_key=os.getenv('OPENAI_API_KEY')
15
 
16
+ def create_indexes(file: tempfile) -> str:
17
  try:
18
  file_path = file.name
19
  reader = PyPDFLoader(file_path)
 
25
  persist_directory = './db_metadata'
26
 
27
  vectordb = Chroma.from_documents(
 
28
  documents=documents,
29
  embedding=embeddings,
30
  persist_directory=persist_directory