leandrocarneiro commited on
Commit
26c225d
·
verified ·
1 Parent(s): 2418314

Update rag.py

Browse files
Files changed (1) hide show
  1. rag.py +1 -1
rag.py CHANGED
@@ -54,7 +54,7 @@ def generate_embeddings_and_vectorstore(path, model):
54
  fc_embeddings = TogetherEmbeddings(model = 'togethercomputer/m2-bert-80M-8k-retrieval', together_api_key = os.environ['TOGETHER_KEY'])
55
  for doc in docs:
56
  vectorstore = Chroma.from_documents(documents=[doc], embedding=fc_embeddings)
57
- time.sleep(1)
58
  print('total de docs no vectorstore=',len(vectorstore.get()['documents']))
59
 
60
  return vectorstore
 
54
  fc_embeddings = TogetherEmbeddings(model = 'togethercomputer/m2-bert-80M-8k-retrieval', together_api_key = os.environ['TOGETHER_KEY'])
55
  for doc in docs:
56
  vectorstore = Chroma.from_documents(documents=[doc], embedding=fc_embeddings)
57
+ time.sleep(1.1)
58
  print('total de docs no vectorstore=',len(vectorstore.get()['documents']))
59
 
60
  return vectorstore