talexm commited on
Commit
45a8fa5
·
1 Parent(s): c641b6b

adding try/catch if non neo4j db supplied

Browse files
Files changed (1) hide show
  1. rag_sec/document_search_system.py +4 -1
rag_sec/document_search_system.py CHANGED
@@ -155,7 +155,10 @@ class DocumentSearchSystem:
155
  blockchain_details = self.data_transformer.secure_transform({"type": "response", "content": response})
156
 
157
  # Log relationships to Neo4j
158
- self.neo4j_handler.log_relationships(query, transformed_query, response, retrieved_docs)
 
 
 
159
 
160
  return {
161
  "status": "success",
 
155
  blockchain_details = self.data_transformer.secure_transform({"type": "response", "content": response})
156
 
157
  # Log relationships to Neo4j
158
+ try:
159
+ self.neo4j_handler.log_relationships(query, transformed_query, response, retrieved_docs)
160
+ except Exception as e:
161
+ print(e)
162
 
163
  return {
164
  "status": "success",