alexkueck commited on
Commit
92e67fc
·
verified ·
1 Parent(s): 8c9d8c1

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +5 -1
utils.py CHANGED
@@ -430,10 +430,14 @@ def query(api_llm, payload):
430
  def extract_document_info(documents):
431
  extracted_info = []
432
  for doc in documents:
 
 
 
 
433
  info = {
434
  'content': doc.page_content,
435
  'metadata': doc.metadata,
436
- 'titel': doc.metadata.get("title", "Keine Überschrift"),
437
  'seite': doc.metadata.get("page", "Unbekannte Seite"),
438
  'pfad': doc.metadata.get("path", "Kein Pfad verfügbar")
439
  }
 
430
  def extract_document_info(documents):
431
  extracted_info = []
432
  for doc in documents:
433
+ # Extract the filename from the path to use as the title
434
+ filename = os.path.basename(doc.metadata.get("path", ""))
435
+ title = filename if filename else "Keine Überschrift"
436
+
437
  info = {
438
  'content': doc.page_content,
439
  'metadata': doc.metadata,
440
+ 'titel': title,
441
  'seite': doc.metadata.get("page", "Unbekannte Seite"),
442
  'pfad': doc.metadata.get("path", "Kein Pfad verfügbar")
443
  }