[NOTICKET] fixed single source into multiple sources

#12
by sofhiaazzhr - opened
Files changed (1) hide show
  1. src/api/v1/chat.py +2 -2
src/api/v1/chat.py CHANGED
@@ -63,7 +63,7 @@ def _extract_sources(results: List[Dict[str, Any]]) -> List[Dict[str, Any]]:
63
  for result in results:
64
  if "document_id" in result["metadata"]["data"]:
65
  meta = result["metadata"]
66
- key = (meta.get("document_id"), meta.get("page_label"))
67
  if key not in seen:
68
  seen.add(key)
69
  sources.append({
@@ -73,7 +73,7 @@ def _extract_sources(results: List[Dict[str, Any]]) -> List[Dict[str, Any]]:
73
  })
74
  else:
75
  meta = result["metadata"]
76
- key = (meta.get("table_name"), meta.get("column_name"))
77
  if key not in seen:
78
  seen.add(key)
79
  table_name = meta.get("data", {}).get("table_name")
 
63
  for result in results:
64
  if "document_id" in result["metadata"]["data"]:
65
  meta = result["metadata"]
66
+ key = (meta.get("data", {}).get("document_id"), meta.get("data", {}).get("page_label"))
67
  if key not in seen:
68
  seen.add(key)
69
  sources.append({
 
73
  })
74
  else:
75
  meta = result["metadata"]
76
+ key = (meta.get("data", {}).get("table_name"), meta.get("data", {}).get("column_name"))
77
  if key not in seen:
78
  seen.add(key)
79
  table_name = meta.get("data", {}).get("table_name")