brendon-ai commited on
Commit
2ff78b9
·
verified ·
1 Parent(s): 6ef0559

Update src/RAGSample.py

Browse files
Files changed (1) hide show
  1. src/RAGSample.py +14 -12
src/RAGSample.py CHANGED
@@ -525,9 +525,9 @@ class RAGApplication:
525
  try:
526
  if not question.strip():
527
  return "Please provide a valid question."
528
-
529
  print(f"\nProcessing question: '{question}'")
530
-
531
  if hasattr(self.retriever, "get_documents_with_confidence"):
532
  docs_with_scores = self.retriever.get_documents_with_confidence(question)
533
  documents = [Document(page_content=d["document"]) for d in docs_with_scores]
@@ -535,25 +535,27 @@ class RAGApplication:
535
  else:
536
  documents = self.retriever.invoke(question)
537
  confidence_info = "Confidence scoring not available."
538
-
539
  print(f"Retrieved {len(documents)} documents")
540
  print(confidence_info)
541
-
542
  doc_texts = "\n\n".join([doc.page_content for doc in documents])
543
  if len(doc_texts) > 500:
544
  doc_texts = doc_texts[:500] + "..."
545
-
546
  answer = self.rag_chain.invoke({"question": question, "documents": doc_texts})
547
-
548
  # Append confidence footer
549
  footer = "\n\n(Note: This answer is based on documents with confidence scores. Review full context if critical.)"
550
  return answer.strip() + footer
551
-
552
- except Exception as e:
553
- print(f"Error in RAG application: {str(e)}")
554
- import traceback
555
- traceback.print_exc()
556
- return f"I apologize, but I encountered an error processing your question: {str(e)}. Please try rephrasing it or ask a different question."
 
 
557
 
558
 
559
  # Main execution block
 
525
  try:
526
  if not question.strip():
527
  return "Please provide a valid question."
528
+
529
  print(f"\nProcessing question: '{question}'")
530
+
531
  if hasattr(self.retriever, "get_documents_with_confidence"):
532
  docs_with_scores = self.retriever.get_documents_with_confidence(question)
533
  documents = [Document(page_content=d["document"]) for d in docs_with_scores]
 
535
  else:
536
  documents = self.retriever.invoke(question)
537
  confidence_info = "Confidence scoring not available."
538
+
539
  print(f"Retrieved {len(documents)} documents")
540
  print(confidence_info)
541
+
542
  doc_texts = "\n\n".join([doc.page_content for doc in documents])
543
  if len(doc_texts) > 500:
544
  doc_texts = doc_texts[:500] + "..."
545
+
546
  answer = self.rag_chain.invoke({"question": question, "documents": doc_texts})
547
+
548
  # Append confidence footer
549
  footer = "\n\n(Note: This answer is based on documents with confidence scores. Review full context if critical.)"
550
  return answer.strip() + footer
551
+
552
+ except Exception as e:
553
+ print(f"Error in RAG application: {str(e)}")
554
+ import traceback
555
+ traceback.print_exc()
556
+ return f"I apologize, but I encountered an error processing your question: {str(e)}. Please try rephrasing it or ask a different question."
557
+
558
+
559
 
560
 
561
  # Main execution block