minor error
Browse files
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=
|
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(
|
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])
|