alexkueck commited on
Commit
e3e5859
1 Parent(s): 46dda78

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +16 -1
utils.py CHANGED
@@ -464,6 +464,21 @@ def rag_chain(llm, prompt, retriever):
464
  return result
465
 
466
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
467
  # Funktion zum Erstellen der Liste von Dictionaries
468
  def extract_document_info(documents):
469
  extracted_info = []
@@ -477,7 +492,7 @@ def extract_document_info(documents):
477
  }
478
  extracted_info.append(info)
479
  return extracted_info
480
-
481
 
482
 
483
 
 
464
  return result
465
 
466
 
467
+ def extract_document_info(documents):
468
+ extracted_info = []
469
+ for doc in documents:
470
+ info = {
471
+ 'content': doc.page_content,
472
+ 'metadata': doc.metadata,
473
+ 'titel': doc.metadata.get("title", "Keine Überschrift"),
474
+ 'seite': doc.metadata.get("page", "Unbekannte Seite"),
475
+ 'pfad': doc.metadata.get("path", "Kein Pfad verfügbar")
476
+ }
477
+ extracted_info.append(info)
478
+ return extracted_info
479
+
480
+
481
+ """
482
  # Funktion zum Erstellen der Liste von Dictionaries
483
  def extract_document_info(documents):
484
  extracted_info = []
 
492
  }
493
  extracted_info.append(info)
494
  return extracted_info
495
+ """
496
 
497
 
498