Skip to content

Commit 93f3bd6

Browse files
Fixed issue for total pages GCS, S3
1 parent 153afcb commit 93f3bd6

File tree

2 files changed

+5
-3
lines changed

2 files changed

+5
-3
lines changed

backend/src/graphDB_dataAccess.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ def create_source_node(self, obj_source_node:sourceNode):
3636
d.processingTime = $pt, d.errorMessage = $e_message, d.nodeCount= $n_count,
3737
d.relationshipCount = $r_count, d.model= $model, d.gcsBucket=$gcs_bucket,
3838
d.gcsBucketFolder= $gcs_bucket_folder, d.language= $language,d.gcsProjectId= $gcs_project_id,
39-
d.is_cancelled=False, d.total_pages=0, d.total_chunks=0, d.processed_chunk=0, d.total_pages=$total_pages,
39+
d.is_cancelled=False, d.total_chunks=0, d.processed_chunk=0, d.total_pages=$total_pages,
4040
d.access_token=$access_token""",
4141
{"fn":obj_source_node.file_name, "fs":obj_source_node.file_size, "ft":obj_source_node.file_type, "st":job_status,
4242
"url":obj_source_node.url,

backend/src/main.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -44,6 +44,7 @@ def create_source_node_graph_url_s3(graph, model, source_url, aws_access_key_id,
4444
obj_source_node.file_type = 'pdf'
4545
obj_source_node.file_size = file_info['file_size_bytes']
4646
obj_source_node.file_source = source_type
47+
obj_source_node.total_pages = 1
4748
obj_source_node.model = model
4849
obj_source_node.url = str(source_url+file_name)
4950
obj_source_node.awsAccessKeyId = aws_access_key_id
@@ -73,6 +74,7 @@ def create_source_node_graph_url_gcs(graph, model, gcs_project_id, gcs_bucket_na
7374
obj_source_node.file_size = file_metadata['fileSize']
7475
obj_source_node.url = file_metadata['url']
7576
obj_source_node.file_source = source_type
77+
obj_source_node.total_pages = 1
7678
obj_source_node.model = model
7779
obj_source_node.file_type = 'pdf'
7880
obj_source_node.gcsBucket = gcs_bucket_name
@@ -442,11 +444,11 @@ def get_labels_and_relationtypes(graph):
442444
RETURN collect {
443445
CALL db.labels() yield label
444446
WHERE NOT label IN ['Chunk','_Bloom_Perspective_']
445-
return label order by label limit 50 } as labels,
447+
return label order by label limit 100 } as labels,
446448
collect {
447449
CALL db.relationshipTypes() yield relationshipType as type
448450
WHERE NOT type IN ['PART_OF', 'NEXT_CHUNK', 'HAS_ENTITY', '_Bloom_Perspective_']
449-
return type order by type LIMIT 50 } as relationshipTypes
451+
return type order by type LIMIT 100 } as relationshipTypes
450452
"""
451453
graphDb_data_Access = graphDBdataAccess(graph)
452454
result = graphDb_data_Access.execute_query(query)

0 commit comments

Comments
 (0)