[NOTICKET] fixed single source into multiple sources
#12
by sofhiaazzhr - opened
- 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")
|