felixflier03 commited on
Commit
3f751a4
·
verified ·
1 Parent(s): de1b94e

Update model_classes.py

Browse files
Files changed (1) hide show
  1. model_classes.py +7 -3
model_classes.py CHANGED
@@ -433,8 +433,12 @@ class EnhancedHybridSearcher:
433
  def _semantic_search(self, query: str, k: int) -> List[Dict[str, Any]]:
434
  """Führt semantische Suche durch"""
435
  try:
436
- # Alternative Methode verwenden
437
- results = self.vectorstore.similarity_search_with_scores(query, k=k) # Diese Zeile ändert sich
 
 
 
 
438
  return [
439
  {
440
  'content': doc.page_content,
@@ -443,7 +447,7 @@ class EnhancedHybridSearcher:
443
  'metadata': doc.metadata if hasattr(doc, 'metadata') else {},
444
  'quality_metrics': self._calculate_quality_metrics(doc.page_content)
445
  }
446
- for doc, score in results
447
  ]
448
  except Exception as e:
449
  print(f"Fehler in semantic search: {str(e)}")
 
433
  def _semantic_search(self, query: str, k: int) -> List[Dict[str, Any]]:
434
  """Führt semantische Suche durch"""
435
  try:
436
+ query_embedding = self.vectorstore.embedding_function.embed_query(query)
437
+ docs_and_scores = self.vectorstore.similarity_search_by_vector_with_scores(
438
+ embedding=query_embedding,
439
+ k=k
440
+ )
441
+
442
  return [
443
  {
444
  'content': doc.page_content,
 
447
  'metadata': doc.metadata if hasattr(doc, 'metadata') else {},
448
  'quality_metrics': self._calculate_quality_metrics(doc.page_content)
449
  }
450
+ for doc, score in docs_and_scores
451
  ]
452
  except Exception as e:
453
  print(f"Fehler in semantic search: {str(e)}")