Spaces:
Build error
Build error
Commit ·
c926830
1
Parent(s): d5d7a80
id error resolved
Browse files- utils/PdfUtils.py +3 -3
- utils/VectorDatabase.py +1 -1
utils/PdfUtils.py
CHANGED
|
@@ -91,13 +91,13 @@ def ProcessPdf(pdf_content: bytes) -> List[Tuple[str, int]]:
|
|
| 91 |
part_size = len_content // n_parts
|
| 92 |
start, end = 0, part_size
|
| 93 |
temp = []
|
| 94 |
-
for
|
| 95 |
-
temp.append((pages_content[e][start:end], e +
|
| 96 |
start = end
|
| 97 |
end = end + part_size
|
| 98 |
final_data += temp
|
| 99 |
else:
|
| 100 |
-
final_data.append((pages_content[e], e
|
| 101 |
|
| 102 |
pdf_doc.close()
|
| 103 |
return final_data
|
|
|
|
| 91 |
part_size = len_content // n_parts
|
| 92 |
start, end = 0, part_size
|
| 93 |
temp = []
|
| 94 |
+
for nth_part in range(n_parts):
|
| 95 |
+
temp.append((pages_content[e][start:end], str(e) + "_" + str(nth_part)))
|
| 96 |
start = end
|
| 97 |
end = end + part_size
|
| 98 |
final_data += temp
|
| 99 |
else:
|
| 100 |
+
final_data.append((pages_content[e], str(e)))
|
| 101 |
|
| 102 |
pdf_doc.close()
|
| 103 |
return final_data
|
utils/VectorDatabase.py
CHANGED
|
@@ -27,7 +27,7 @@ class AdvancedClient:
|
|
| 27 |
|
| 28 |
for chunk, _id in file_datas:
|
| 29 |
chunks.append(chunk)
|
| 30 |
-
ids.append(str(_id))
|
| 31 |
|
| 32 |
from .ModelCallingFunctions import generate_embedding
|
| 33 |
|
|
|
|
| 27 |
|
| 28 |
for chunk, _id in file_datas:
|
| 29 |
chunks.append(chunk)
|
| 30 |
+
ids.append(str(_id)) #make sure IDs are string dtpye
|
| 31 |
|
| 32 |
from .ModelCallingFunctions import generate_embedding
|
| 33 |
|