ashishbangwal commited on
Commit
c926830
·
1 Parent(s): d5d7a80

id error resolved

Browse files
Files changed (2) hide show
  1. utils/PdfUtils.py +3 -3
  2. 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 _ in range(n_parts):
95
- temp.append((pages_content[e][start:end], e + 1))
96
  start = end
97
  end = end + part_size
98
  final_data += temp
99
  else:
100
- final_data.append((pages_content[e], e + 1))
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