ishaq101 sofhiaazzhr commited on
Commit
c8e3046
·
1 Parent(s): 5db64d1

[NOTICKET] fixed single source into multiple sources (#12)

Browse files

- [NOTICKET] fixed single source into multiple sources (9d826225424dd8fb0044ce6f4a010ab8eac1075b)
- [NOTICKET] remove source_hint line (d1f034edff5a6a78c0a7b9f3402ca2a46baa4f43)


Co-authored-by: Sofhia Az-Zahra <sofhiaazzhr@users.noreply.huggingface.co>

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")