Mbonea commited on
Commit
e460efd
·
1 Parent(s): 1fba972

minor error

Browse files
Files changed (1) hide show
  1. App/Embedding/utils/Initialize.py +2 -2
App/Embedding/utils/Initialize.py CHANGED
@@ -53,7 +53,7 @@ def search(query: str, task_id: str):
53
  filtering_conditions = {
54
  "task_id": {"$eq": "task_id"},
55
  }
56
- data =docsearch.similarity_search(query, k=10, filter=filtering_conditions)
57
  return [
58
  {"text": d.page_content, "start": d.metadata["start"], "end": d.metadata["end"]}
59
  for d in data
@@ -62,5 +62,5 @@ def search(query: str, task_id: str):
62
 
63
 
64
  def encode(temp: list[Document]):
65
- docsearch.add_documents([temp])
66
  # return embeddings.embed_documents(texts = [d.page_content for d in temp])
 
53
  filtering_conditions = {
54
  "task_id": {"$eq": "task_id"},
55
  }
56
+ data =docsearch.similarity_search(query, k=4, filter=filtering_conditions)
57
  return [
58
  {"text": d.page_content, "start": d.metadata["start"], "end": d.metadata["end"]}
59
  for d in data
 
62
 
63
 
64
  def encode(temp: list[Document]):
65
+ docsearch.add_documents(temp)
66
  # return embeddings.embed_documents(texts = [d.page_content for d in temp])